Question

我们有两个URLSession代理，它们在其正文中使用自定义代码（未显示以使此帖子更清晰）：

   public func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {}

  public func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?) {
     if let error = error {
            //code here
        } else if let response = task.response as? HTTPURLResponse, var dataResponse = self.dataReceived {
            if (400...499).contains(response.statusCode) {
                //handle response here
            }
}

我们有一个mockURLSession类：

protocol URLSessionDataTaskProtocol {
    func resume()
}

protocol URLSessionProtocol {
    typealias DataTaskResult = (Data?, URLResponse?, Error?) -> Void

    func dataTask(with request: Request, completionHandler: @escaping DataTaskResult) -> URLSessionDataTaskProtocol
}

class MockURLSession: URLSessionProtocol {

    typealias DataTaskResult = (Data?, URLResponse?, Error?) -> Void
    var nextDataTask = MockURLSessionDataTask()
    var nextData: Data?
    var nextError: Error?

    private (set) var lastURL: URL?

    func successHttpURLResponse(request: Request) -> URLResponse {
        return HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: "HTTP/1.1", headerFields: nil)!
    }

    func wrongHttpURLResponse(request: Request, statusCode:Int) -> URLResponse {
        return HTTPURLResponse(url: request.url!, statusCode: statusCode, httpVersion: "HTTP/1.1", headerFields: nil)!
    }

    func dataTask(with request: Request, completionHandler: @escaping DataTaskResult) -> URLSessionDataTaskProtocol {
        lastURL = request.url
        nextDataTask.resume()
        completionHandler(nextData, successHttpURLResponse(request: request), nextError)
        return nextDataTask
    }

}

class MockURLSessionDataTask: URLSessionDataTaskProtocol {
    private (set) var resumeWasCalled = false

    func resume() {
        resumeWasCalled = true
    }
}

我们遇到的问题是我必须为模拟服务器的特定响应编写单元测试（例如404）。通常我会写这样的回答：

  let urlReponse = MockURLSession().wrongHttpURLResponse(request: self.urlRequest!, statusCode: 404)

我遇到的问题是因为我们使用的是URLSession委托，所以我不能将mockResponse作为参数注入函数，除非它不会被调用。基于模拟响应测试URLSession委托的最佳方法是什么？

Answer 1

您可以在初始化MockURLSession时注入模拟响应。

将以下内容添加到MockURLSession：

private var mockResponse: HTTPURLResponse?

init() { }
init(mockResponse: HTTPURLResponse) {
    self.mockResponse = mockResponse
}

如果您提供了模拟响应，请修改模拟的dataTask函数以使用模拟响应调用完成处理程序：

if let mockResponse = mockResponse {
    completionHandler(nextData, mockResponse, nextError)
}
else {
    //default case is success
    completionHandler(nextData, successHttpURLResponse(request: request), nextError)
}

现在，当您想要测试错误响应时，请在创建模拟会话时注入它：

let mockResponse = HTTPURLResponse(url: self.urlRequest.url!, statusCode: 404, httpVersion: "HTTP/1.1", headerFields: nil)!
let mockURLSession = MockURLSession(mockResponse: mockResponse)

调用dataTask时，将使用模拟响应：

mockURLSession.dataTask(with: self.urlRequest) { (data, response, error) in
    if let response = response as? HTTPURLResponse {
        print("Status Code: \(response.statusCode)") // "Status Code: 404"
    }
}

测试URLSession使用模拟数据委派

1 个答案: