1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
import Foundation
import Testing
@testable import Hutch
// MARK: - URLProtocol stub
/// Captures outgoing URLRequests and returns a minimal 401 so callers fail fast
/// without touching the real network.
private final class CapturingURLProtocol: URLProtocol, @unchecked Sendable {
nonisolated(unsafe) static var capturedRequests: [URLRequest] = []
override class func canInit(with request: URLRequest) -> Bool { true }
override class func canonicalRequest(for request: URLRequest) -> URLRequest { request }
override func startLoading() {
CapturingURLProtocol.capturedRequests.append(request)
let response = HTTPURLResponse(
url: request.url!,
statusCode: 401,
httpVersion: nil,
headerFields: nil
)!
client?.urlProtocol(self, didReceive: response, cacheStoragePolicy: .notAllowed)
client?.urlProtocol(self, didLoad: Data())
client?.urlProtocolDidFinishLoading(self)
}
override func stopLoading() {}
static func makeSession() -> URLSession {
let config = URLSessionConfiguration.ephemeral
config.protocolClasses = [CapturingURLProtocol.self]
return URLSession(configuration: config)
}
}
// MARK: - Tests
/// Tests are serialized because CapturingURLProtocol uses shared static state.
@Suite(.serialized)
struct BundleUserAgentTests {
// MARK: Bundle extension
@Test
func hutchUserAgentHasNameSlashVersion() {
let ua = Bundle.main.hutchUserAgent
let parts = ua.split(separator: "/", maxSplits: 1)
#expect(parts.count == 2)
#expect(parts[0] == "Hutch")
#expect(!parts[1].isEmpty)
}
@Test
func hutchUserAgentContainsNoParenthesizedContext() {
// The old SystemStatusService user-agent appended "(System Status)".
// The shared agent should be plain "Hutch/<version>".
#expect(!Bundle.main.hutchUserAgent.contains("("))
}
// MARK: SRHTClient
@Test
func sRHTClientSetsUserAgentOnExecute() async {
CapturingURLProtocol.capturedRequests = []
let client = SRHTClient(session: CapturingURLProtocol.makeSession(), token: "test-token")
_ = try? await client.execute(
service: .builds,
query: "{ jobs { results { id } } }",
responseType: [String: String].self
)
guard let captured = CapturingURLProtocol.capturedRequests.first else {
Issue.record("No request was captured by SRHTClient.execute.")
return
}
#expect(captured.value(forHTTPHeaderField: "User-Agent") == Bundle.main.hutchUserAgent)
}
@Test
func sRHTClientSetsUserAgentOnFetchText() async {
CapturingURLProtocol.capturedRequests = []
let client = SRHTClient(session: CapturingURLProtocol.makeSession(), token: "test-token")
let url = try! #require(URL(string: "https://builds.sr.ht/~test/job/1/log"))
_ = try? await client.fetchText(url: url)
guard let captured = CapturingURLProtocol.capturedRequests.first else {
Issue.record("No request was captured by SRHTClient.fetchText.")
return
}
#expect(captured.value(forHTTPHeaderField: "User-Agent") == Bundle.main.hutchUserAgent)
}
// MARK: SystemStatusService
@Test
func systemStatusServiceSetsUserAgent() async {
CapturingURLProtocol.capturedRequests = []
let service = SystemStatusService(session: CapturingURLProtocol.makeSession())
_ = try? await service.fetchSnapshotHTML()
guard let captured = CapturingURLProtocol.capturedRequests.first else {
Issue.record("No request was captured by SystemStatusService.")
return
}
#expect(captured.value(forHTTPHeaderField: "User-Agent") == Bundle.main.hutchUserAgent)
}
// MARK: HutchStatsService
@Test
func hutchStatsServiceSetsUserAgent() async {
CapturingURLProtocol.capturedRequests = []
let service = HutchStatsService(
session: CapturingURLProtocol.makeSession(),
configuration: AppConfiguration(environment: [:])
)
_ = try? await service.fetchContributionCalendar(actor: "testuser", endingOn: .now)
guard let captured = CapturingURLProtocol.capturedRequests.first else {
Issue.record("No request was captured by HutchStatsService.")
return
}
#expect(captured.value(forHTTPHeaderField: "User-Agent") == Bundle.main.hutchUserAgent)
}
}
|
