Skip to content

Commit 0d2b2db

Browse files
Chore: Add counterpart for constants for tests
1 parent 6114ad0 commit 0d2b2db

File tree

1 file changed

+15
-6
lines changed

1 file changed

+15
-6
lines changed

DatadogRUM/Tests/Instrumentation/Resources/URLSessionRUMResourcesHandlerTests.swift

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -954,10 +954,10 @@ class URLSessionRUMResourcesHandlerTests: XCTestCase {
954954

955955
// Given
956956
var mockRequest: URLRequest = .mockWith(url: "https://graphql.example.com/api")
957-
mockRequest.setValue("GetUser", forHTTPHeaderField: GraphQLHeaders.operationName)
958-
mockRequest.setValue("query", forHTTPHeaderField: GraphQLHeaders.operationType)
959-
mockRequest.setValue("{\"userId\":\"123\"}", forHTTPHeaderField: GraphQLHeaders.variables)
960-
mockRequest.setValue("query GetUser($userId: ID!) { user(id: $userId) { name } }", forHTTPHeaderField: GraphQLHeaders.payload)
957+
mockRequest.setValue("GetUser", forHTTPHeaderField: ExpectedGraphQLHeaders.operationName)
958+
mockRequest.setValue("query", forHTTPHeaderField: ExpectedGraphQLHeaders.operationType)
959+
mockRequest.setValue("{\"userId\":\"123\"}", forHTTPHeaderField: ExpectedGraphQLHeaders.variables)
960+
mockRequest.setValue("query GetUser($userId: ID!) { user(id: $userId) { name } }", forHTTPHeaderField: ExpectedGraphQLHeaders.payload)
961961

962962
let immutableRequest = ImmutableRequest(request: mockRequest)
963963
let taskInterception = URLSessionTaskInterception(request: immutableRequest, isFirstParty: false)
@@ -989,8 +989,8 @@ class URLSessionRUMResourcesHandlerTests: XCTestCase {
989989

990990
// Given
991991
var mockRequest: URLRequest = .mockWith(url: "https://graphql.example.com/api")
992-
mockRequest.setValue("FailedMutation", forHTTPHeaderField: GraphQLHeaders.operationName)
993-
mockRequest.setValue("mutation", forHTTPHeaderField: GraphQLHeaders.operationType)
992+
mockRequest.setValue("FailedMutation", forHTTPHeaderField: ExpectedGraphQLHeaders.operationName)
993+
mockRequest.setValue("mutation", forHTTPHeaderField: ExpectedGraphQLHeaders.operationType)
994994

995995
let immutableRequest = ImmutableRequest(request: mockRequest)
996996
let taskInterception = URLSessionTaskInterception(request: immutableRequest, isFirstParty: false)
@@ -1028,3 +1028,12 @@ class URLSessionRUMResourcesHandlerTests: XCTestCase {
10281028
return dict
10291029
}
10301030
}
1031+
1032+
// MARK: - Test Helpers
1033+
1034+
struct ExpectedGraphQLHeaders {
1035+
static let operationName: String = "_dd-custom-header-graph-ql-operation-name"
1036+
static let operationType: String = "_dd-custom-header-graph-ql-operation-type"
1037+
static let variables: String = "_dd-custom-header-graph-ql-variables"
1038+
static let payload: String = "_dd-custom-header-graph-ql-payload"
1039+
}

0 commit comments

Comments
 (0)