metatext/Development Assets/Stubbing.swift

// Copyright © 2020 Metabolist. All rights reserved.

import Foundation

typealias HTTPStub = Result<(URLResponse, Data), Error>

protocol Stubbing {
    func stub(url: URL) -> HTTPStub?
    func data(url: URL) -> Data?
    func dataString(url: URL) -> String?
    func statusCode(url: URL) -> Int?
}

extension Stubbing {
    func stub(url: URL) -> HTTPStub? {
        if let data = data(url: url),
              let statusCode = statusCode(url: url),
              let response = HTTPURLResponse(
                url: url,
                statusCode: statusCode,
                httpVersion: nil,
                headerFields: nil) {
            return .success((response, data))
        }

        return nil
    }

    func data(url: URL) -> Data? {
        dataString(url: url)?.data(using: .utf8)
    }

    func dataString(url: URL) -> String? { nil }

    func statusCode(url: URL) -> Int? { 200 }
}