Loopback tests for URLSession

Author: Pushkar N Kulkarni

TestFoundation/TestNSURLSession.swift

@@ -15,9 +15,12 @@ import XCTest
 import SwiftFoundation
 import SwiftXCTest
 #endif
+import Dispatch
 
 class TestURLSession : XCTestCase {
 
+    var serverPort: Int = -1
+
     static var allTests: [(String, (TestURLSession) -> () throws -> Void)] {
         return [
             ("test_dataTaskWithURL", test_dataTaskWithURL),
@@ -32,8 +35,35 @@ class TestURLSession : XCTestCase {
         ]
     }
 
+    private func runServer(with condition: DispatchSemaphore) throws {
+        let start = 21961
+        for port in start...(start+100) { //we must find at least one port to bind
+            do {
+                serverPort = port
+                let test = try TestURLSessionServer(port: UInt16(port))
+                try test.start(started: condition)
+                try test.readAndRespond()
+                test.stop()
+            } catch let e as ServerError {
+                if e.operation != "bind" { continue }
+                throw e
+            }
+        }
+    }
+
     func test_dataTaskWithURL() {
-        let urlString = "https://restcountries.eu/rest/v1/name/Nepal?fullText=true"
+        let serverReady = DispatchSemaphore(value: 0)
+        let queue = DispatchQueue.global()
+        queue.async {
+            do {
+                try self.runServer(with: serverReady)
+            } catch {
+                XCTAssertTrue(true)
+                return
+            }
+        }
+        serverReady.wait()
+        let urlString = "http://127.0.0.1:\(serverPort)/Nepal"
         let url = URL(string: urlString)!
         let d = DataTask(with: expectation(description: "data task"))                         
         d.run(with: url)
@@ -44,7 +74,18 @@ class TestURLSession : XCTestCase {
     }
 
     func test_dataTaskWithURLCompletionHandler() {
-        let urlString = "https://restcountries.eu/rest/v1/name/USA?fullText=true"
+        let serverReady = DispatchSemaphore(value: 0)
+        let queue = DispatchQueue.global()
+        queue.async {
+            do {
+                try self.runServer(with: serverReady)
+            } catch {
+                XCTAssertTrue(true)
+                return 
+            }
+        }
+        serverReady.wait()
+        let urlString = "http://127.0.0.1:\(serverPort)/USA"
         let url = URL(string: urlString)!
         let config = URLSessionConfiguration.default
         config.timeoutIntervalForRequest = 8
@@ -60,13 +101,7 @@ class TestURLSession : XCTestCase {
 
             let httpResponse = response as! HTTPURLResponse?
             XCTAssertEqual(200, httpResponse!.statusCode, "HTTP response code is not 200") 
-            do {
-                let json = try JSONSerialization.jsonObject(with: data!, options: [])
-                let arr = json as? Array<Any>
-                let first = arr![0]
-                let result = first as? [String : Any]
-                expectedResult = result!["capital"] as! String
-            } catch { }
+            expectedResult = NSString(data: data!, encoding: String.Encoding.utf8.rawValue)!._bridgeToSwift()
             XCTAssertEqual("Washington, D.C.", expectedResult, "Did not receive expected value")
             expect.fulfill()
         }
@@ -75,7 +110,18 @@ class TestURLSession : XCTestCase {
     }
 
     func test_dataTaskWithURLRequest() {
-        let urlString = "https://restcountries.eu/rest/v1/name/Peru?fullText=true"
+        let serverReady = DispatchSemaphore(value: 0)
+        let queue = DispatchQueue.global()
+        queue.async {
+            do {
+                try self.runServer(with: serverReady)
+            } catch {
+                XCTAssertTrue(true)
+                return
+            }
+        }
+        serverReady.wait()
+        let urlString = "http://127.0.0.1:\(serverPort)/Peru"
         let urlRequest = URLRequest(url: URL(string: urlString)!)
         let d = DataTask(with: expectation(description: "data task"))     
         d.run(with: urlRequest)
@@ -86,7 +132,18 @@ class TestURLSession : XCTestCase {
     }
 
     func test_dataTaskWithURLRequestCompletionHandler() {
-        let urlString = "https://restcountries.eu/rest/v1/name/Italy?fullText=true"
+        let serverReady = DispatchSemaphore(value: 0)
+        let queue = DispatchQueue.global()
+        queue.async {
+            do {
+                try self.runServer(with: serverReady)
+            } catch {
+                XCTAssertTrue(true)
+                return
+            }
+        }
+        serverReady.wait()
+        let urlString = "http://127.0.0.1:\(serverPort)/Italy"
         let urlRequest = URLRequest(url: URL(string: urlString)!)
         let config = URLSessionConfiguration.default
         config.timeoutIntervalForRequest = 8
@@ -101,13 +158,7 @@ class TestURLSession : XCTestCase {
             }
             let httpResponse = response as! HTTPURLResponse?
             XCTAssertEqual(200, httpResponse!.statusCode, "HTTP response code is not 200")
-            do {
-                let json = try JSONSerialization.jsonObject(with: data!, options: [])
-                let arr = json as? Array<Any>
-                let first = arr![0]
-                let result = first as? [String : Any]
-                expectedResult = result!["capital"] as! String
-            } catch { }
+            expectedResult = NSString(data: data!, encoding: String.Encoding.utf8.rawValue)!._bridgeToSwift()
             XCTAssertEqual("Rome", expectedResult, "Did not receive expected value")
             expect.fulfill()
         }
@@ -194,14 +245,7 @@ class DataTask: NSObject {
 
 extension DataTask : URLSessionDataDelegate {
      public func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {
-         do {
-            let json = try JSONSerialization.jsonObject(with: data, options: [])
-            let arr = json as? Array<Any>
-            let first = arr![0]
-            let result = first as? [String : Any]
-            capital = result!["capital"] as! String
-         } catch { }
-
+         capital = NSString(data: data, encoding: String.Encoding.utf8.rawValue)!._bridgeToSwift()
          dataTaskExpectation.fulfill()
      }
 }

build.py

@@ -458,8 +458,11 @@
 # TODO: Probably this should be another 'product', but for now it's simply a phase
 foundation_tests = SwiftExecutable('TestFoundation', [
 	'TestFoundation/main.swift',
+        'TestFoundation/HTTPServer.swift',
 ] + glob.glob('./TestFoundation/Test*.swift')) # all TestSomething.swift are considered sources to the test project in the TestFoundation directory
 
+Configuration.current.extra_ld_flags = '-L'+Configuration.current.variables["LIBDISPATCH_BUILD_DIR"]+'/src/.libs'
+
 foundation_tests.add_dependency(foundation_tests_resources)
 foundation.add_phase(foundation_tests_resources)
 foundation.add_phase(foundation_tests)