metatext/HTTP/Sources/Stubbing/Stubbing.swift
2020-10-05 15:50:05 -07:00

36 lines
934 B
Swift

// Copyright © 2020 Metabolist. All rights reserved.
import Foundation
public typealias HTTPStub = Result<(HTTPURLResponse, Data), Error>
public protocol Stubbing {
func stub(url: URL) -> HTTPStub?
func data(url: URL) -> Data?
func dataString(url: URL) -> String?
func statusCode(url: URL) -> Int?
}
public 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 }
}