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
|
//
// AppMessageHandlerTests.swift
// PacketTunnelCoreTests
//
// Created by Jon Petersson on 2023-09-28.
// Copyright © 2023 Mullvad VPN AB. All rights reserved.
//
import Combine
@testable import MullvadMockData
@testable import MullvadREST
import MullvadTypes
import PacketTunnelCore
import XCTest
final class AppMessageHandlerTests: XCTestCase {
func testHandleAppMessageForSendURLRequest() async throws {
let sendRequestExpectation = expectation(description: "Expect sending request")
let urlRequestProxy = URLRequestProxyStub(sendRequestExpectation: sendRequestExpectation)
let appMessageHandler = createAppMessageHandler(urlRequestProxy: urlRequestProxy)
let url = URL(string: "localhost")!
let urlRequest = ProxyURLRequest(
id: UUID(),
urlRequest: URLRequest(url: url)
)!
_ = try? await appMessageHandler.handleAppMessage(
TunnelProviderMessage.sendURLRequest(urlRequest).encode()
)
await fulfillment(of: [sendRequestExpectation], timeout: .UnitTest.timeout)
}
func testHandleAppMessageForCancelURLRequest() async throws {
let cancelRequestExpectation = expectation(description: "Expect cancelling request")
let urlRequestProxy = URLRequestProxyStub(cancelRequestExpectation: cancelRequestExpectation)
let appMessageHandler = createAppMessageHandler(urlRequestProxy: urlRequestProxy)
_ = try? await appMessageHandler.handleAppMessage(
TunnelProviderMessage.cancelURLRequest(UUID()).encode()
)
await fulfillment(of: [cancelRequestExpectation], timeout: .UnitTest.timeout)
}
func testHandleAppMessageForTunnelStatus() async throws {
let stateExpectation = expectation(description: "Expect getting state")
let actor = PacketTunnelActorStub(stateExpectation: stateExpectation)
let appMessageHandler = createAppMessageHandler(actor: actor)
_ = try? await appMessageHandler.handleAppMessage(
TunnelProviderMessage.getTunnelStatus.encode()
)
await fulfillment(of: [stateExpectation], timeout: .UnitTest.timeout)
}
func testHandleAppMessageForKeyRotation() async throws {
let keyRotationExpectation = expectation(description: "Expect key rotation")
let actor = PacketTunnelActorStub(keyRotationExpectation: keyRotationExpectation)
let appMessageHandler = createAppMessageHandler(actor: actor)
_ = try? await appMessageHandler.handleAppMessage(
TunnelProviderMessage.privateKeyRotation.encode()
)
await fulfillment(of: [keyRotationExpectation], timeout: .UnitTest.timeout)
}
func testHandleAppMessageForReconnectTunnel() async throws {
let reconnectExpectation = expectation(description: "Expect reconnecting state")
let actor = PacketTunnelActorStub(reconnectExpectation: reconnectExpectation)
let appMessageHandler = createAppMessageHandler(actor: actor)
let relayConstraints = RelayConstraints(
exitLocations: .only(UserSelectedRelays(locations: [.hostname("se", "sto", "se6-wireguard")]))
)
let candidates = try RelaySelector.WireGuard.findCandidates(
by: relayConstraints.exitLocations,
in: ServerRelaysResponseStubs.sampleRelays,
filterConstraint: relayConstraints.filter,
daitaEnabled: false
)
let match = try RelaySelector.WireGuard.pickCandidate(
from: candidates,
wireguard: ServerRelaysResponseStubs.sampleRelays.wireguard,
portConstraint: relayConstraints.port,
numberOfFailedAttempts: 0
)
let selectedRelays = SelectedRelays(
entry: nil,
exit: SelectedRelay(
endpoint: match.endpoint,
hostname: match.relay.hostname,
location: match.location
),
retryAttempt: 0
)
_ = try? await appMessageHandler.handleAppMessage(
TunnelProviderMessage.reconnectTunnel(.preSelected(selectedRelays)).encode()
)
await fulfillment(of: [reconnectExpectation], timeout: .UnitTest.timeout)
}
}
extension AppMessageHandlerTests {
func createAppMessageHandler(
actor: PacketTunnelActorProtocol = PacketTunnelActorStub(),
urlRequestProxy: URLRequestProxyProtocol = URLRequestProxyStub()
) -> AppMessageHandler {
return AppMessageHandler(
packetTunnelActor: actor,
urlRequestProxy: urlRequestProxy
)
}
}
|
