Improve revoked ui state

1 files changed, 4 insertions, 4 deletions

diff --git a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/DeviceRevokedViewModelTest.kt b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/DeviceRevokedViewModelTest.kt
index d8f4fdf119..69941a474d 100644
--- a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/DeviceRevokedViewModelTest.kt
+++ b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/DeviceRevokedViewModelTest.kt
@@ -59,7 +59,7 @@ class DeviceRevokedViewModelTest {
         // Arrange, Act, Assert
         viewModel.uiState.test {
             serviceConnectionState.value = ServiceConnectionState.Disconnected
-            assertEquals(DeviceRevokedUiState(false), awaitItem())
+            assertEquals(DeviceRevokedUiState.UNKNOWN, awaitItem())
         }
     }
 
@@ -68,7 +68,7 @@ class DeviceRevokedViewModelTest {
         // Arrange, Act, Assert
         viewModel.uiState.test {
             serviceConnectionState.value = ServiceConnectionState.ConnectedNotReady(mockk())
-            assertEquals(DeviceRevokedUiState(false), awaitItem())
+            assertEquals(DeviceRevokedUiState.UNKNOWN, awaitItem())
         }
     }
 
@@ -89,9 +89,9 @@ class DeviceRevokedViewModelTest {
 
         // Act, Assert
         viewModel.uiState.test {
-            assertEquals(DeviceRevokedUiState(false), awaitItem())
+            assertEquals(DeviceRevokedUiState.UNKNOWN, awaitItem())
             serviceConnectionState.value = ServiceConnectionState.ConnectedReady(mockedContainer)
-            assertEquals(DeviceRevokedUiState(true), awaitItem())
+            assertEquals(DeviceRevokedUiState.SECURED, awaitItem())
         }
     }