testViewControllerSuccess

testViewControllerSuccess

func testViewControllerSuccess() {
    let expect = expectation(description: #function)
    let viewController = ViewController()
    
    let data = jsonString.data(using: .utf8)
    
    URLProtocolMock.requestHandler = { request in
        let response = HTTPURLResponse(url: URL(fileURLWithPath: ""), statusCode: 200, httpVersion: nil, headerFields: nil)!
        return (response, data)
    }
    
    let configuration = URLSessionConfiguration.af.default
    configuration.protocolClasses = [URLProtocolMock.self] + (configuration.protocolClasses ?? [])
    
    let session = Session(configuration: configuration)
    
    let networkManager = NetworkManager(
        session: session,
        router: JSONPlaceHolderAPIAction.getToDo(id: 2))
    
    viewController.downloadData(networkManager, completion: { response in
        let expectedModel = jsonString.data(using: .utf8)
        
        switch response {
        case .failure:
            XCTFail()
        case .success(let data):
            XCTAssertEqual(expectedModel, data)
            expect.fulfill()
        }
    })
    wait(for: [expect], timeout: 1.0)
}