TestURLProtocol.swift

// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//

classTestURLProtocol:LoopbackServerTest{
func test_interceptResponse(){
leturlString="http://127.0.0.1:\(TestURLProtocol.serverPort)/USA"
leturl=URL(string: urlString)!
letconfig=URLSessionConfiguration.default
 config.protocolClasses =[CustomProtocol.self]
 config.timeoutIntervalForRequest =8
letsession=URLSession(configuration: config, delegate:nil, delegateQueue:nil)
letexpect=expectation(description:"GET \(urlString): with a custom protocol")
lettask= session.dataTask(with: url){ data, response, error in
defer{ expect.fulfill()}
iflet e = error as?URLError{
XCTAssertEqual(e.code,.timedOut,"Unexpected error code")
return
}
lethttpResponse= response as!HTTPURLResponse?
XCTAssertEqual(429, httpResponse!.statusCode,"HTTP response code is not 429")
}
 task.resume()
waitForExpectations(timeout:12)
}

func test_interceptRequest(){
leturlString="ssh://127.0.0.1:\(TestURLProtocol.serverPort)/USA"
leturl=URL(string: urlString)!
letconfig=URLSessionConfiguration.default
 config.protocolClasses =[InterceptableRequest.self]
 config.timeoutIntervalForRequest =8
letsession=URLSession(configuration: config, delegate:nil, delegateQueue:nil)
letexpect=expectation(description:"GET \(urlString): with a custom protocol")
lettask= session.dataTask(with: url){ data, response, error in
defer{ expect.fulfill()}
iflet e = error as?URLError{
XCTAssertEqual(e.code,.timedOut,"Unexpected error code")
return
}
lethttpResponse= response as!HTTPURLResponse?
letresponseURL=URL(string:"http://google.com")
XCTAssertEqual(responseURL, httpResponse?.url,"Unexpected url")
XCTAssertEqual(200, httpResponse!.statusCode,"HTTP response code is not 200")
}
 task.resume()
waitForExpectations(timeout:12)
}

func test_multipleCustomProtocols(){
leturlString="http://127.0.0.1:\(TestURLProtocol.serverPort)/Nepal"
leturl=URL(string: urlString)!
letconfig=URLSessionConfiguration.default
 config.protocolClasses =[InterceptableRequest.self,CustomProtocol.self]
letexpect=expectation(description:"GET \(urlString): with a custom protocol")
letsession=URLSession(configuration: config)
lettask= session.dataTask(with: url){ data, response, error in
defer{ expect.fulfill()}
iflet e = error as?URLError{
XCTAssertEqual(e.code,.timedOut,"Unexpected error code")
return
}
lethttpResponse= response as!HTTPURLResponse
print(httpResponse.statusCode)
XCTAssertEqual(429, httpResponse.statusCode,"Status code is not 429")
}
 task.resume()
waitForExpectations(timeout:12)
}

func test_customProtocolResponseWithDelegate(){
leturlString="http://127.0.0.1:\(TestURLSession.serverPort)/Peru"
leturl=URL(string: urlString)!
letd=DataTask(with:expectation(description:"GET \(urlString): with a custom protocol and delegate"), protocolClasses:[CustomProtocol.self])
 d.responseReceivedExpectation =expectation(description:"GET \(urlString): response received")
 d.run(with: url)
waitForExpectations(timeout:12)
}

func test_customProtocolSetDataInResponseWithDelegate(){
leturlString="http://127.0.0.1:\(TestURLSession.serverPort)/Nepal"
leturl=URL(string: urlString)!
letd=DataTask(with:expectation(description:"GET \(urlString): with a custom protocol and delegate"), protocolClasses:[CustomProtocol.self])
 d.run(with: url)
waitForExpectations(timeout:12)
if !d.error {
XCTAssertEqual(d.capital,"Kathmandu","test_dataTaskWithURLRequest returned an unexpected result")
}
}

func test_finishLoadingWithNoResponse()throws{
leturl=tryXCTUnwrap(URL(string:"https://test/url"))
letconfiguration=URLSessionConfiguration.default
 configuration.protocolClasses =[TestURLServer.self]
letsession=URLSession(configuration: configuration)

letexpect=expectation(description:"GET \(url.absoluteString)")
lettask= session.dataTask(with: url){ data, response, error in
defer{ expect.fulfill()}
XCTAssertNil(data)
XCTAssertNil(response)
XCTAssertNotNil(error)
}

 task.resume()
waitForExpectations(timeout:2)
}
}

classInterceptableRequest:URLProtocol{

overrideclassfunc canInit(with request:URLRequest)->Bool{
return request.url?.scheme =="ssh"
}

overrideclassfunc canonicalRequest(for request:URLRequest)->URLRequest{
return request
}

overridefunc startLoading(){
leturlString="http://google.com"
leturl=URL(string: urlString)!
letresponse=HTTPURLResponse(url: url, statusCode:200, httpVersion:"HTTP/1.1", headerFields:[:])
self.client?.urlProtocol(self, didReceive: response!, cacheStoragePolicy:.notAllowed)
self.client?.urlProtocolDidFinishLoading(self)

}

overridefunc stopLoading(){
return
}
}

classCustomProtocol:URLProtocol{

overrideclassfunc canInit(with request:URLRequest)->Bool{
returntrue
}

func sendResponse(statusCode:Int, headers:[String:String]=[:], data:Data){
letresponse=HTTPURLResponse(url:self.request.url!, statusCode: statusCode, httpVersion:"HTTP/1.1", headerFields: headers)
letcapital="Kathmandu"
letdata= capital.data(using:.utf8)
self.client?.urlProtocol(self, didReceive: response!, cacheStoragePolicy:.notAllowed)
self.client?.urlProtocol(self, didLoad: data!)
self.client?.urlProtocolDidFinishLoading(self)
}

overrideclassfunc canonicalRequest(for request:URLRequest)->URLRequest{
return request
}

overridefunc startLoading(){
sendResponse(statusCode:429, data:Data())
}

overridefunc stopLoading(){
return
}
}


publicclassTestURLServer:URLProtocol{
publicoverrideclassfunc canInit(with request:URLRequest)->Bool{
returntrue
}

publicoverrideclassfunc canonicalRequest(for request:URLRequest)->URLRequest{
return request
}

publicoverridefunc startLoading(){
varinfo:[String:Any]=[:]
iflet url = request.url {
info[NSURLErrorFailingURLStringErrorKey]= url.absoluteString
info[NSURLErrorFailingURLErrorKey]= url
}
leterror=URLError(.networkConnectionLost, userInfo: info)

 client?.urlProtocol(self, didFailWithError: error)
 client?.urlProtocolDidFinishLoading(self)
}

publicoverridefunc stopLoading(){
}
}