venmo#93 Add tests for dataTask with URL

m-herold · Sep 23, 2019 · 70cd45a · 70cd45a
1 parent 9b60014
commit 70cd45a
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/Tests/DVRTests/SessionTests.swift b/Tests/DVRTests/SessionTests.swift
@@ -41,6 +41,32 @@ class SessionTests: XCTestCase {
             XCTFail()
         }
     }
+
+    func testDataTaskWithUrl() {
+        let url = URL(string: "http://example.com")!
+        let dataTask = session.dataTask(with: url)
+
+        XCTAssert(dataTask is SessionDataTask)
+
+        if let dataTask = dataTask as? SessionDataTask, let headers = dataTask.request.allHTTPHeaderFields {
+            XCTAssert(headers["testSessionHeader"] == "testSessionHeaderValue")
+        } else {
+            XCTFail()
+        }
+    }
+
+    func testDataTaskWithUrlAndCompletion() {
+        let url = URL(string: "http://example.com")!
+        let dataTask = session.dataTask(with: url, completionHandler: { _, _, _ in return })
+
+        XCTAssert(dataTask is SessionDataTask)
+
+        if let dataTask = dataTask as? SessionDataTask, let headers = dataTask.request.allHTTPHeaderFields {
+            XCTAssert(headers["testSessionHeader"] == "testSessionHeaderValue")
+        } else {
+            XCTFail()
+        }
+    }
 
     func testPlayback() {
         session.recordingEnabled = false