66 lines
1.5 KiB
Swift
66 lines
1.5 KiB
Swift
|
//
|
||
|
// MockURLProtocol.swift
|
||
|
// APIService
|
||
|
//
|
||
|
// Created by Javier Cicchelli on 04/12/2022.
|
||
|
// Copyright © 2022 Röck+Cöde. All rights reserved.
|
||
|
//
|
||
|
|
||
|
import Foundation
|
||
|
|
||
|
class MockURLProtocol: URLProtocol {
|
||
|
|
||
|
// MARK: Properties
|
||
|
|
||
|
static var mockData: [URL: MockURLResponse] = [:]
|
||
|
|
||
|
// MARK: Functions
|
||
|
|
||
|
override class func canInit(with task: URLSessionTask) -> Bool {
|
||
|
true
|
||
|
}
|
||
|
|
||
|
override class func canInit(with request: URLRequest) -> Bool {
|
||
|
true
|
||
|
}
|
||
|
|
||
|
override class func canonicalRequest(for request: URLRequest) -> URLRequest {
|
||
|
request
|
||
|
}
|
||
|
|
||
|
override func startLoading() {
|
||
|
guard
|
||
|
let url = request.url,
|
||
|
let response = Self.mockData[url]
|
||
|
else {
|
||
|
client?.urlProtocolDidFinishLoading(self)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if let data = response.data {
|
||
|
client?.urlProtocol(self, didLoad: data)
|
||
|
}
|
||
|
|
||
|
if let httpResponse = HTTPURLResponse(
|
||
|
url: url,
|
||
|
statusCode: response.status.rawValue,
|
||
|
httpVersion: nil,
|
||
|
headerFields: response.headers
|
||
|
) {
|
||
|
client?.urlProtocol(self, didReceive: httpResponse, cacheStoragePolicy: .allowedInMemoryOnly)
|
||
|
}
|
||
|
|
||
|
client?.urlProtocolDidFinishLoading(self)
|
||
|
}
|
||
|
|
||
|
override func stopLoading() {}
|
||
|
}
|
||
|
|
||
|
// MARK: - Structs
|
||
|
|
||
|
struct MockURLResponse {
|
||
|
let status: ResponseStatus
|
||
|
let headers: [String: String]
|
||
|
let data: Data?
|
||
|
}
|