CIAPP-570 - Adjust network instrumentation for dd-sdk usage

Don't convert requests made with URL to URLRequest if we are not injecting headers, since it is not needed
DataDog · Nov 19, 2020 · 4bf3628 · 4bf3628
1 parent 9f46c59
commit 4bf3628
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Sources/DatadogSDKTesting/NetworkInstrumentation/DDNetworkInstrumentation.swift b/Sources/DatadogSDKTesting/NetworkInstrumentation/DDNetworkInstrumentation.swift
@@ -103,7 +103,8 @@ class DDNetworkInstrumentation {
             let sessionTaskId = UUID().uuidString
 
             let block: @convention(block) (URLSession, AnyObject) -> URLSessionTask = { session, argument in
-                if let url = argument as? URL {
+                if let url = argument as? URL,
+                   self.injectHeaders == true {
                     let request = URLRequest(url: url)
                     if selector == #selector(URLSession.dataTask(with:) as (URLSession) -> (URL) -> URLSessionDataTask) {
                         return session.dataTask(with: request)
@@ -179,7 +180,8 @@ class DDNetworkInstrumentation {
             let sessionTaskId = UUID().uuidString
 
             let block: @convention(block) (URLSession, AnyObject, @escaping (Any?, URLResponse?, Error?) -> Void) -> URLSessionTask = { session, argument, completion in
-                if let url = argument as? URL {
+                if let url = argument as? URL,
+                   self.injectHeaders == true {
                     let request = URLRequest(url: url)
 
                     if selector == #selector(URLSession.dataTask(with:completionHandler:) as (URLSession) -> (URL, @escaping (Data?, URLResponse?, Error?) -> Void) -> URLSessionDataTask) {