swift - 扩展不同类时出现 "Does not conform to protocol"错误

Question

我试图通过注入(inject)适应 URLSession 和 URLSessionDataTask 协议(protocol)的对象来测试我自己的类。我正在扩展 NSURLSession 和 NSURLSessionDataTask 以采用这些协议(protocol)，以便我可以正常使用现有对象，但在单元测试中使用测试对象。

我有以下代码，错误注释:

typealias SessionHandler = (NSData?, NSURLResponse?, NSError?) -> Void

protocol URLSession {

  func dataTaskWithURL(url: NSURL, completionHandler: SessionHandler) -> URLSessionDataTask

}

protocol URLSessionDataTask {

}

// Type 'NSURLSession' does not conform to protocol 'URLSession'
extension NSURLSession : URLSession {}
extension NSURLSessionDataTask : URLSessionDataTask {}

我理解错误，我的协议(protocol)与 NSURLSession 实现的方法不完全匹配。我该如何解决这个问题？

Answer 1

我最终做的是创建一个协议(protocol)扩展，它创建 NSURLSession 需要的必要方法。

extension NSURLSession : URLSession {
  func dataTaskWithURL(url: NSURL, completionHandler: SessionHandler) -> URLSessionDataTask {
    return dataTaskWithURL(url, completionHandler: completionHandler) as NSURLSessionDataTask
  }
}