单元测试主要用来检测某个工作单元的结果是否符合预期,以此保证该工作单元的逻辑正确。上次写封装一个 Swift-Style 的网络模块的时候在结尾提了一下单元测试的重要性,评论中有朋友对网络层的单元测试有一些疑惑。我推荐他去看《单元测试的艺术》(这本书让我对单元测试有了新的认识),但由于该书是以 C# 为例写的,可能会对 iOS 开发的朋友造成一定的阅读障碍,所以我还是决定填一下坑,简单介绍一下用 Swift 进行网络层单元测试的方法。不过由于 Swift 的函数式特性,像《单元测试的艺术》中那样单纯地用 OOP 思维编写测试可能会有些麻烦,本文临近结尾部分写了一点自己用过的使用“伪装函数”进行测试的方法,可能大家以前没见过,我自己也是突然想到的,欢迎提出各种意见。
  网络层的单元测试之所以让人感觉难以下手,原因主要有两点:
  · 网络是个不稳定的外部依赖。
  · 网络操作一般会涉及异步过程,而异步过程难以测试。
  要直接测试网络和异步调用,可以使用XCTest提供的expectationWithDescription+waitForExpectationsWithTimeout,举个例子:
  func testFetchDataWithAPI_invalidAPI_failureResult() {
  let expectation = expectationWithDescription("")
  let timeout = 15 as NSTimeInterval
  NetworkManager
  .defaultManager
  .fetchDataWithAPI(.Invalid, responseKey: "") {
  expectation.fulfill()
  XCTAssertTrue($0.isFailure)
  }
  waitForExpectationsWithTimeout(timeout, handler: nil)
  }
  测试方法按 test方法名_测试场景_期望结果 的格式命名。首先在异步回调外面调用expectationWithDescription方法得到一个expectation,这个方法接受一个字符串,用来描述本次测试,我传了个空串,因为我们的测试方法名已经足够清晰了。然后在回调中调用expectation.fulfill()表明满足测试条件,接下来可以进行断言。后别忘了在回调外面加上waitForExpectationsWithTimeout(timeout, handler: nil),如果时间超过timeout回调还没有执行,会测试失败,hander会在超时后调用,可以写一些清空状态和还原现场的操作,以免影响之后的测试,譬如task?.cancel()。但是我这边什么都没做,因为的单元测试之间本来不应该互相有影响。
  上面的测试非常简单吧,但是按《单元测试的艺术》一书中的观点,这样的测试已经不能算是单元测试,而是步入集成测试的范畴了:
  集成测试是对一个工作单元进行的测试,这个测试对被测试的工作单元没有完全的控制,并使用该单元的一个或多个真实的依赖物,例如时间、网络、数据库、线程或随机数产生器等。
  上述这个测试非常不稳定,它依赖于真实的网络状况,我们可能因为网络不佳测试失败,而不是因为我们的代码本身有逻辑错误,而且这个测试有可能非常慢,慢到你不愿意每次一修改代码去跑一遍测试,这样的单元测试有可能形同虚设。
  集成测试当然也非常重要,但一般开发人员也写写单元测试。其实 Alamofire 有采用我上面说的方法进行测试,所以如果你的网络层像我一样是以 Alamofire 为基础构建的,那表示你不太需要再去写这样的测试了,你只要保证跟 Alamofire 无关的那些代码本身逻辑正确,以及正确调用了 Alamofire 即可。
  譬如针对我的这个方法:
/**
Fetch raw object
- parameter api:              API address
- parameter method:           HTTP method, default = POST
- parameter parameters:       Request parameters, default = nil
- parameter responseKey:      Key of target value, use '.' to get nested objects, e.g. "data.vehicle_list"
- parameter jsonArrayHandler: Handle result with raw object
- returns: Optional request object which is cancellable.
*/
func fetchDataWithAPI(api: API,
method: Alamofire.Method = .POST,
parameters: [String: String]? = nil,
responseKey: String,
networkCompletionHandler: NetworkCompletionHandler) -> Cancellable? {
guard let url = api.url else {
printLog("URL Invalid: \(api.rawValue)")
return nil
}
return Alamofire.request(method, url, parameters: parameters).responseJSON {
networkCompletionHandler(self.parseResult($0.result, responseKey: responseKey))
}
}
  我一般会去测试它的返回值是否符合预期:
func testFetchDataWithAPI_invalidURL_returnNil {
let task = NetworkManager
.defaultManager
.fetchDataWithAPI(.InvalidURL, responseKey: "") {}
XCTAssertNil(task)
}
func testFetchDataWithAPI_validAPI_returnNotNil {
let task = NetworkManager
.defaultManager
.fetchDataWithAPI(.ValidURL, responseKey: "") {}
XCTAssertNotNil(task)
}