Fix failing tests

Package.swift

@@ -34,7 +34,7 @@ let package = Package(
         ),
         .testTarget(
           name: "MockingjayTests",
-          dependencies: [ "Mockingjay" ],
+          dependencies: [ "Mockingjay", "MockingjayXCTest" ],
           path: "Tests/MockingjayTests"
         )
     ],

Sources/Core/MockingjayProtocol.swift

@@ -29,7 +29,7 @@ public func ==(lhs:Stub, rhs:Stub) -> Bool {
 }
 
 var stubs = [Stub]()
-var registered: Bool = false
+public var registered: Bool = false
 
 public class MockingjayProtocol: URLProtocol {
   // MARK: Stubs

Sources/XCTest/XCTest.swift

@@ -9,6 +9,7 @@
 #if canImport(XCTest)
 import ObjectiveC
 import XCTest
+import Mockingjay
 
 let swizzleTearDown: Void = {
   let tearDown = class_getInstanceMethod(XCTest.self, #selector(XCTest.tearDown))

Tests/MockingjayTests/MockingjayAsyncProtocolTests.swift

@@ -59,6 +59,72 @@ class MockingjayAsyncProtocolTests: XCTestCase, URLSessionDataDelegate  {
     }
     XCTAssertEqual(mutableData as Data, stubData)
   }
+
+  func testDownloadOfAudioFileInChunks() {
+    let request = URLRequest(url: URL(string: "https://fuller.li/")!)
+
+    let thisSourceFile = URL(fileURLWithPath: #file)
+    let thisDirectory = thisSourceFile.deletingLastPathComponent()
+    let resourceURL = thisDirectory.appendingPathComponent("TestAudio.m4a")
+    let data = try! Data(contentsOf: resourceURL)
+
+    let stubResponse = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: "1.1", headerFields: ["Content-Length" : String(data.count)])!
+
+    MockingjayProtocol.addStub(matcher: { (requestedRequest) -> (Bool) in
+      return true
+    }) { (request) -> (Response) in
+      return Response.success(stubResponse, Download.streamContent(data: data, inChunksOf: 2000))
+    }
+    let urlSession = Foundation.URLSession(configuration: configuration, delegate: self, delegateQueue: OperationQueue.current)
+    let dataTask = urlSession.dataTask(with: request)
+    dataTask.resume()
+
+    let mutableData = NSMutableData()
+    while mutableData.length < data.count {
+      let expectation = self.expectation(description: "testProtocolCanReturnedDataInChunks")
+      self.didReceiveDataHandler = { (session: Foundation.URLSession, dataTask: URLSessionDataTask, data: Data) in
+        mutableData.append(data)
+        expectation.fulfill()
+      }
+      waitForExpectations(timeout: 2.0, handler: nil)
+    }
+    XCTAssertEqual(mutableData as Data, data)
+  }
+
+  func testByteRanges() {
+    let length = 100000
+    var request = URLRequest(url: URL(string: "https://fuller.li/")!)
+    request.addValue("bytes=50000-149999", forHTTPHeaderField: "Range")
+
+    let thisSourceFile = URL(fileURLWithPath: #file)
+    let thisDirectory = thisSourceFile.deletingLastPathComponent()
+    let resourceURL = thisDirectory.appendingPathComponent("TestAudio.m4a")
+    let data = try! Data(contentsOf: resourceURL)
+
+    let stubResponse = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: "1.1", headerFields: ["Content-Length" : String(length)])!
+    MockingjayProtocol.addStub(matcher: { (requestedRequest) -> (Bool) in
+      return true
+    }) { (request) -> (Response) in
+      return Response.success(stubResponse, .streamContent(data: data, inChunksOf: 2000))
+    }
+
+    let urlSession = Foundation.URLSession(configuration: configuration, delegate: self, delegateQueue: OperationQueue.current)
+    let dataTask = urlSession.dataTask(with: request)
+    dataTask.resume()
+
+    var mutableData = Data()
+    while mutableData.count < length {
+      let expectation = self.expectation(description: "testProtocolCanReturnedDataInChunks")
+      self.didReceiveDataHandler = { (session: Foundation.URLSession, dataTask: URLSessionDataTask, data: Data) in
+        mutableData.append(data)
+        expectation.fulfill()
+      }
+      waitForExpectations(timeout: 2.0, handler: nil)
+    }
+
+    let correctData = data.subdata(in: 50000 ..< (50000 + length))
+    XCTAssertEqual(mutableData, correctData)
+  }
 
   // MARK: NSURLSessionDataDelegate
   func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {

Tests/MockingjayTests/MockingjayProtocolTests.swift

@@ -13,11 +13,10 @@ import XCTest
 
 class MockingjayProtocolTests : XCTestCase {
 
-  var urlSession:URLSession!
+  lazy var urlSession = URLSession(configuration: URLSessionConfiguration.default)
 
   override func setUp() {
     super.setUp()
-    urlSession = URLSession(configuration: URLSessionConfiguration.default)
   }
 
   override func tearDown() {
@@ -73,6 +72,7 @@ class MockingjayProtocolTests : XCTestCase {
     XCTAssertNil(response)
     XCTAssertNil(data)
     XCTAssertNotNil(error)
+    XCTAssertEqual((error as NSError?)?.domain, "MockingjayTests")
   }
 
   func testProtocolReturnsResponseWithRegisteredStubError() {

Tests/MockingjayTests/MockingjayTests.swift

@@ -20,12 +20,14 @@ class MockingjaySessionTests: XCTestCase {
   }
 
   func testEphemeralSessionConfigurationIncludesProtocol() {
+    ensureMockingjayProtocolRegistration()
     let configuration = URLSessionConfiguration.ephemeral
     let protocolClasses = (configuration.protocolClasses!).map(toString)
     XCTAssertEqual(protocolClasses.first!, "MockingjayProtocol")
   }
 
   func testDefaultSessionConfigurationIncludesProtocol() {
+    ensureMockingjayProtocolRegistration()
     let configuration = URLSessionConfiguration.default
     let protocolClasses = (configuration.protocolClasses!).map(toString)
     XCTAssertEqual(protocolClasses.first!, "MockingjayProtocol")
@@ -52,4 +54,17 @@ class MockingjaySessionTests: XCTestCase {
       XCTAssertNil(error, String(describing: error))
     }
   }
+
+  /// Ensures that `MockingjayProtocol` is added to `URLProtocol` registered classes.
+  ///
+  /// The registration process happens while stubbing, so for test cases, where `stub(_:_:)` is not called
+  /// `MockingjayProtocol` will not be registered. Call this method within those test cases
+  /// if your test relies on `MockingjayProtocol` being registered to pass.
+  func ensureMockingjayProtocolRegistration() {
+    if !registered {
+      URLProtocol.registerClass(MockingjayProtocol.self)
+      URLSessionConfiguration.mockingjaySwizzleDefaultSessionConfiguration()
+      registered = true
+    }
+  }
 }

Tests/MockingjayTests/MockingjayXCTestTests.swift

@@ -7,6 +7,7 @@
 //
 
 import XCTest
+import MockingjayXCTest
 @testable import Mockingjay