Make the actor reconnect reason explicit

mullvad · Jun 4, 2024 · 476afec · 476afec
1 parent 66336ae
commit 476afec
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/ios/PacketTunnelCore/Actor/PacketTunnelActor+Public.swift b/ios/PacketTunnelCore/Actor/PacketTunnelActor+Public.swift
@@ -38,7 +38,7 @@ extension PacketTunnelActor {
 
      - Parameter nextRelay: next relay to connect to.
      */
-    public nonisolated func reconnect(to nextRelay: NextRelay, reconnectReason: ActorReconnectReason = .userInitiated) {
+    public nonisolated func reconnect(to nextRelay: NextRelay, reconnectReason: ActorReconnectReason) {
         eventChannel.send(.reconnect(nextRelay, reason: reconnectReason))
     }
 

diff --git a/ios/PacketTunnelCoreTests/Mocks/SettingsReaderStub.swift b/ios/PacketTunnelCoreTests/Mocks/SettingsReaderStub.swift
@@ -46,7 +46,8 @@ extension SettingsReaderStub {
             relayConstraints: RelayConstraints(),
             dnsServers: .gateway,
             obfuscation: WireGuardObfuscationSettings(state: .off, port: .automatic),
-            quantumResistance: .on
+            quantumResistance: .on,
+            multihopState: .off
         )
         return SettingsReaderStub {
             return staticSettings

diff --git a/ios/PacketTunnelCoreTests/PacketTunnelActorTests.swift b/ios/PacketTunnelCoreTests/PacketTunnelActorTests.swift
@@ -383,7 +383,7 @@ final class PacketTunnelActorTests: XCTestCase {
             reconnectingStateExpectation.fulfill()
         }
 
-        actor.reconnect(to: .random)
+        actor.reconnect(to: .random, reconnectReason: .userInitiated)
 
         await fulfillment(
             of: [reconnectingStateExpectation],
@@ -414,7 +414,7 @@ final class PacketTunnelActorTests: XCTestCase {
         let reconnectingState: (ObservedState) -> Bool = { if case .reconnecting = $0 { true } else { false } }
         await expect(reconnectingState, on: actor) { reconnectingStateExpectation.fulfill() }
 
-        actor.reconnect(to: .random)
+        actor.reconnect(to: .random, reconnectReason: .userInitiated)
         await fulfillment(
             of: [reconnectingStateExpectation],
             timeout: Duration.milliseconds(100).timeInterval
@@ -445,7 +445,7 @@ final class PacketTunnelActorTests: XCTestCase {
         // Cancel the state sink to avoid overfulfilling the connected expectation
         stateSink?.cancel()
 
-        actor.reconnect(to: .random)
+        actor.reconnect(to: .random, reconnectReason: .userInitiated)
         await fulfillment(of: [stopMonitorExpectation], timeout: 1)
     }
 }