Fix android tests

2 files changed, 5 insertions, 3 deletions

diff --git a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/SelectLocationViewModelTest.kt b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/SelectLocationViewModelTest.kt
index 59cd9f48b5..3cadfe575f 100644
--- a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/SelectLocationViewModelTest.kt
+++ b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/SelectLocationViewModelTest.kt
@@ -132,7 +132,7 @@ class SelectLocationViewModelTest {
             assertEquals(Unit, awaitItem())
             verify {
                 connectionProxyMock.connect()
-                mockRelayListListener.selectedRelayLocation = mockLocation
+                mockRelayListListener.updateSelectedRelayLocation(mockLocation)
             }
         }
     }
diff --git a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/VpnSettingsViewModelTest.kt b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/VpnSettingsViewModelTest.kt
index 66d301d903..13561737c8 100644
--- a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/VpnSettingsViewModelTest.kt
+++ b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/VpnSettingsViewModelTest.kt
@@ -159,7 +159,9 @@ class VpnSettingsViewModelTest {
         // Arrange
         val wireguardPort: Constraint<Port> = Constraint.Only(Port(99))
         val wireguardConstraints = WireguardConstraints(port = wireguardPort)
-        every { mockRelayListListener.selectedWireguardConstraints = any() } returns Unit
+        every {
+            mockRelayListListener.updateSelectedWireguardConstraints(wireguardConstraints)
+        } returns Unit
 
         // Act
         mockConnectionState.value =
@@ -168,7 +170,7 @@ class VpnSettingsViewModelTest {
 
         // Assert
         verify(exactly = 1) {
-            mockRelayListListener.selectedWireguardConstraints = wireguardConstraints
+            mockRelayListListener.updateSelectedWireguardConstraints(wireguardConstraints)
         }
     }