Question

我正在看一个用Swift 4编写的iOS应用程序。它具有使用URLSession的相当简单的网络层，但是该应用程序没有单元测试，在我开始重构之前，我渴望通过引入一些解决方案来解决这个问题。测试。

在执行此操作之前，我必须能够模拟URLSession，以便在测试期间不要创建真实的网络请求。我无法在当前的实现中看到如何实现这一目标？在我的测试中将URLSession注入的入口点在哪里。

我已经提取了网络代码，并使用相同的逻辑创建了一个简单的应用，如下所示：

Endpoint.swift

import Foundation

protocol Endpoint {
    var baseURL: String { get }
}

extension Endpoint {
    var urlComponent: URLComponents {
        let component = URLComponents(string: baseURL)
        return component!
    }

    var request: URLRequest {
        return URLRequest(url: urlComponent.url!)
    }
}

struct RandomUserEndpoint: Endpoint {
    var baseURL: String {
        return RandomUserClient.baseURL
    }
}

APIClient.swift

import Foundation

enum Either<T> {
    case success(T), error(Error)
}

enum APIError: Error {
    case unknown, badResponse, jsonDecoder
}

enum HTTPMethod: String {
    case get = "GET"
    case put = "PUT"
    case post = "POST"
    case patch = "PATCH"
    case delete = "DELETE"
    case head = "HEAD"
    case options = "OPTIONS"
}

protocol APIClient {
    var session: URLSession { get }
    func get<T: Codable>(with request: URLRequest, completion: @escaping (Either<T>) -> Void)
}

extension APIClient {
    var session: URLSession {
        return URLSession.shared
    }

   func get<T: Codable>(with request: URLRequest, completion: @escaping (Either<T>) -> Void) {
        let task = session.dataTask(with: request) { (data, response, error) in
            guard error == nil else { return completion(.error(error!)) }
            guard let response = response as? HTTPURLResponse, 200..<300 ~= response.statusCode else { completion(.error(APIError.badResponse)); return }

            guard let data = data else { return }

            guard let value = try? JSONDecoder().decode(T.self, from: data) else { completion(.error(APIError.jsonDecoder)); return }

            DispatchQueue.main.async {
                completion(.success(value))
            }
        }
        task.resume()
    }

}

RandomUserClient.swift

import Foundation

class RandomUserClient: APIClient {
    static let baseURL = "https://randomuser.me/api/"

    func fetchRandomUser(with endpoint: RandomUserEndpoint, method: HTTPMethod, completion: @escaping (Either<RandomUserResponse>)-> Void) {
        var request = endpoint.request
        request.httpMethod = method.rawValue
        get(with: request, completion: completion)
    }

}

RandomUserModel.swift

import Foundation

typealias RandomUser = Result

struct RandomUserResponse: Codable {
    let results: [Result]?
}

struct Result: Codable {
    let name: Name
}

struct Name: Codable {
    let title: String
    let first: String
    let last: String
}

使用此代码的非常简单的应用程序可能类似于

class ViewController: UIViewController {

    let fetchUserButton: UIButton = {
        let button = UIButton(type: .system)
        button.setTitle("FETCH", for: .normal)
        button.setTitleColor(.black, for: .normal)
        button.titleLabel?.font = UIFont.boldSystemFont(ofSize: 36)
        button.addTarget(self, action: #selector(fetchRandomUser), for: .touchUpInside)
        button.translatesAutoresizingMaskIntoConstraints = false
        button.isEnabled = true
        return button
    }()

    override func viewDidLoad() {
        super.viewDidLoad()

        view.backgroundColor = .white

        view.addSubview(fetchUserButton)
        NSLayoutConstraint.activate([
            fetchUserButton.centerXAnchor.constraint(equalTo: view.centerXAnchor),
            fetchUserButton.centerYAnchor.constraint(equalTo: view.centerYAnchor),
        ])
    }

    @objc func fetchRandomUser() {
        let client = RandomUserClient()
        fetchUserButton.isEnabled = false
        client.fetchRandomUser(with: RandomUserEndpoint(), method: .get) { [unowned self] (either) in
            switch either {
            case .success(let user):
                guard let name = user.results?.first?.name else { return }
                let message = "Your new name is... \n\(name.first.uppercased()) \(name.last.uppercased())"
                self.showAlertView(title: "", message: message)
                self.fetchUserButton.isEnabled = true
            case .error(let error):
                print(error.localizedDescription)
            }
        }
    }

    func showAlertView(title: String, message: String) {
        let alert = UIAlertController(title: title, message: message, preferredStyle: UIAlertController.Style.alert)
        alert.addAction(UIAlertAction(title: "Close", style: UIAlertAction.Style.default, handler: nil))
        self.present(alert, animated: true, completion: nil)
    }
}

理想情况下，我希望有一种模拟URLSession的方法，以便可以正确测试，但是不确定如何使用当前代码实现。

Answer 1

在这种情况下，如果您断定RandomUserClient，那么实际上可能更有意义。

您扩展了RandomUserClient，并让它接受URLSession的实例，该实例本身被注入到您的APIClient中。

class RandomUserClient: APIClient {
    var session: URLSession
    static let baseURL = "https://randomuser.me/api/"

    init(session: URLSession) {
        self.session = session
    }

    func fetchRandomUser(with endpoint: RandomUserEndpoint, method: HTTPMethod, completion: @escaping (Either<RandomUserResponse>)-> Void) {
        var request = endpoint.request
        request.httpMethod = method.rawValue

        get(with: request, session: session, completion: completion)
    }

}

您的视图控制器将需要更新，以便对RandomUserClient进行初始化，例如lazy var client = RandomUserClient(session: URLSession.shared)

您还需要对您的APIClient协议和扩展进行重构，以接受URLSession新注入的依赖项

protocol APIClient {
    func get<T: Codable>(with request: URLRequest, session: URLSession, completion: @escaping (Either<T>) -> Void)
}

extension APIClient {
    func get<T: Codable>(with request: URLRequest, session: URLSession, completion: @escaping (Either<T>) -> Void) {
        let task = session.dataTask(with: request) { (data, response, error) in
            guard error == nil else { return completion(.error(error!)) }
            guard let response = response as? HTTPURLResponse, 200..<300 ~= response.statusCode else { completion(.error(APIError.badResponse)); return }

            guard let data = data else { return }

            guard let value = try? JSONDecoder().decode(T.self, from: data) else { completion(.error(APIError.jsonDecoder)); return }

            DispatchQueue.main.async {
                completion(.success(value))
            }
        }
        task.resume()
    }

}

请注意增加了session: URLSession。

如何模拟URLSession？

1 个答案: