▲点击上方“CocoaChina”关注即可免费学习 iOS 开发

原文链接：http://swiftandpainless.com/an-easy-way-to-stub-nsurlsession/
作者：dom
原文日期：2016/01/09

如果你熟悉我这个博客的话，你可能知道我检查问题时，最喜欢的方法是模拟 NSURLSeesion 返回的数据。

那么我们到底要做什么呢，其实是模拟方法的回调数据。而这里的 NSURLSession指的是伪造 web API 的响应。这样做有一些好处，例如：

• 我们不需要一个可用的 web API 来开发我们应用程序的网络请求。

• 能够立马响应，反馈周期更短。

• 测试程序能在没有网络连接的电脑上运行。

一般来说，模拟 NSURLSession 的请求返回数据是通过 NSURLProtocol 来完成的。具体的用例请查看 OHHTTPStubs 和 Mockingjay。使用 NSURLProtocol 的优势在于，当你使用诸如Alamofire 这样的网络请求库时，也能正常模拟数据回调。这种方式很棒，但是对我来说代码太多了。我必须去学习和理解这些代码，以在我的测试中获得预期的效果。

一个简单的解决方案

我将使用 NSURLSession 来做网络请求。下面是如何伪造我的请求返回数据。

为了让它看起来更简单，我已经写了一个 NSURLSession 的替换类和一个协议。整合起来如下所示：

import Foundation

public protocol DHURLSession {

  func dataTaskWithURL(url: NSURL,

    completionHandler: (NSData?, NSURLResponse?, NSError?) -> Void) -> NSURLSessionDataTask

  func dataTaskWithRequest(request: NSURLRequest,

    completionHandler: (NSData?, NSURLResponse?, NSError?) -> Void) -> NSURLSessionDataTask

}

extension NSURLSession: DHURLSession { }

public final class URLSessionMock : DHURLSession {

   

  var url: NSURL?

  var request: NSURLRequest?

  private let dataTaskMock: URLSessionDataTaskMock

   

  public init(data: NSData?, response: NSURLResponse?, error: NSError?) {

    dataTaskMock = URLSessionDataTaskMock()

    dataTaskMock.taskResponse = (data, response, error)

  }

   

  public func dataTaskWithURL(url: NSURL,

    completionHandler: (NSData?, NSURLResponse?, NSError?) -> Void) -> NSURLSessionDataTask {

      self.url = url

      self.dataTaskMock.completionHandler = completionHandler

      return self.dataTaskMock

  }

   

  public func dataTaskWithRequest(request: NSURLRequest,

    completionHandler: (NSData?, NSURLResponse?, NSError?) -> Void) -> NSURLSessionDataTask {

      self.request = request

      self.dataTaskMock.completionHandler = completionHandler

      return self.dataTaskMock

  }

   

  final private class URLSessionDataTaskMock : NSURLSessionDataTask {

     

    typealias CompletionHandler = (NSData!, NSURLResponse!, NSError!) -> Void

    var completionHandler: CompletionHandler?

    var taskResponse: (NSData?, NSURLResponse?, NSError?)?

     

    override func resume() {

      completionHandler?(taskResponse?.0, taskResponse?.1, taskResponse?.2)

    }

  }

}

如上，用来伪造数据的完整帮助代码是 47 行。并且所有代码清晰易懂，既没有 swizzling，也没有复杂的方法。是不是很棒!

使用

为了能够在测试中使用 NSURLSession 替换类，我们需要在代码中注入依赖。一种可能的方式是使用一个懒属性：

lazy var session: DHURLSession = NSURLSession.sharedSession()

然后一个示例测试可能会是这样的：

func testFetchingProfile_ReturnsPopulatedUser() {

  // Arrage

  let responseString = "{\"login\": \"dasdom\", \"id\": 1234567}"

  let responseData = responseString.dataUsingEncoding(NSUTF8StringEncoding)!

  let sessionMock = URLSessionMock(data: responseData, response: nil, error: nil)

  let apiClient = APIClient()

  apiClient.session = sessionMock

   

  // Act

  apiClient.fetchProfileWithName("dasdom")

   

  // Assert

  let user = apiClient.user

  let expectedUser = User(name: "dasdom", id: 1234567)

  XCTAssertEqual(user, expectedUser)

}

我很喜欢这样的解决方案，因为我只要花几分钟时间，通过阅读五十多行代码就能理解替换类。并且没有涉及到 NSURLProtocol 和 swizzling。

这个 NSURLSession 的替换类在 github 上，并且也可以通过CocoaPods 下载。

让我知道你的想法。

微信号：CocoaChinabbs

▲长按二维码“识别”关注即可免费学习 iOS 开发

月薪十万、出任CEO、赢娶白富美、走上人生巅峰不是梦

--------------------------------------

商务合作QQ：645047738

投稿邮箱：support@cocoachina.com