class MockSession: URLSession {
    var completionHandler: ((Data?, URLResponse?, Error?) -> Void)?
    var request: URLRequest?
    let stubbedDataTask: MockURLSessionDataTask = MockURLSessionDataTask()

    override func dataTask(with request: URLRequest, completionHandler: @escaping (Data?, URLResponse?, Error?) -> Void) -> URLSessionDataTask {
        self.completionHandler = completionHandler
        self.request = request

        return stubbedDataTask
    }
}

class MockURLSessionDataTask: URLSessionDataTask {
    var resumeWasCalled: Bool = false

    override func resume() {
        resumeWasCalled = true
    }
}