原文链接: 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 下载。
让我知道你的想法。