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
|
//
// RelaySelectorStub.swift
// PacketTunnelCoreTests
//
// Created by pronebird on 05/09/2023.
// Copyright © 2025 Mullvad VPN AB. All rights reserved.
//
import MullvadREST
import MullvadSettings
import MullvadTypes
import WireGuardKitTypes
/// Relay selector stub that accepts a block that can be used to provide custom implementation.
public final class RelaySelectorStub: RelaySelectorProtocol {
public let relayCache: any RelayCacheProtocol
var selectedRelaysResult: (UInt) throws -> SelectedRelays
var candidatesResult: (() throws -> RelayCandidates)?
init(
relayCache: RelayCacheProtocol = MockRelayCache(),
selectedRelaysResult: @escaping (UInt) throws -> SelectedRelays,
candidatesResult: (() throws -> RelayCandidates)? = nil
) {
self.relayCache = relayCache
self.selectedRelaysResult = selectedRelaysResult
self.candidatesResult = candidatesResult
}
public func selectRelays(
tunnelSettings: LatestTunnelSettings,
connectionAttemptCount: UInt
) throws -> SelectedRelays {
return try selectedRelaysResult(connectionAttemptCount)
}
public func findCandidates(
tunnelSettings: LatestTunnelSettings
) throws -> RelayCandidates {
return try candidatesResult?() ?? RelayCandidates(entryRelays: [], exitRelays: [])
}
}
extension RelaySelectorStub {
/// Returns a relay selector that never fails.
public static func nonFallible() -> RelaySelectorStub {
let publicKey = PrivateKey().publicKey.rawValue
return RelaySelectorStub(
selectedRelaysResult: { _ in
let cityRelay = SelectedRelay(
endpoint: MullvadEndpoint(
ipv4Relay: IPv4Endpoint(ip: .loopback, port: 1300),
ipv4Gateway: .loopback,
ipv6Gateway: .loopback,
publicKey: publicKey
),
hostname: "se-got",
location: Location(
country: "",
countryCode: "se",
city: "",
cityCode: "got",
latitude: 0,
longitude: 0
),
features: nil
)
return SelectedRelays(
entry: cityRelay,
exit: cityRelay,
retryAttempt: 0,
obfuscation: .off
)
}, candidatesResult: nil)
}
/// Returns a relay selector that cannot satisfy constraints .
public static func unsatisfied() -> RelaySelectorStub {
return RelaySelectorStub(
selectedRelaysResult: { _ in
throw NoRelaysSatisfyingConstraintsError(.relayConstraintNotMatching)
},
candidatesResult: {
throw NoRelaysSatisfyingConstraintsError(.relayConstraintNotMatching)
})
}
}
|
