swift - 扩展不同类时出现“不符合协议”错误

Question

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

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

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 {}

我理解这个错误，我的协议与 . 所暗示的方法不完全匹配NSURLSession。我该如何解决？

Answer 1

我最终做的是创建一个协议扩展来创建所需的必要方法NSURLSession。

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