Fix unit tests

mullvad · Jul 31, 2024 · e8e8c4e · e8e8c4e
1 parent b26193f
commit e8e8c4e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/AccountViewModelTest.kt b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/AccountViewModelTest.kt
@@ -2,6 +2,7 @@ package net.mullvad.mullvadvpn.viewmodel
 
 import android.app.Activity
 import app.cash.turbine.test
+import arrow.core.right
 import io.mockk.coEvery
 import io.mockk.coVerify
 import io.mockk.every
@@ -93,6 +94,9 @@ class AccountViewModelTest {
 
     @Test
     fun `onLogoutClick should invoke logout on AccountRepository`() {
+        // Arrange
+        coEvery { mockAccountRepository.logout() } returns Unit.right()
+
         // Act
         viewModel.onLogoutClick()
 

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
@@ -3,13 +3,11 @@ package net.mullvad.mullvadvpn.viewmodel
 import app.cash.turbine.test
 import arrow.core.right
 import io.mockk.MockKAnnotations
-import io.mockk.Runs
 import io.mockk.coEvery
 import io.mockk.coVerify
 import io.mockk.coVerifyOrder
 import io.mockk.every
 import io.mockk.impl.annotations.MockK
-import io.mockk.just
 import io.mockk.mockk
 import io.mockk.unmockkAll
 import kotlinx.coroutines.flow.MutableSharedFlow
@@ -72,7 +70,7 @@ class DeviceRevokedViewModelTest {
     fun `onGoToLoginClicked should invoke logout on AccountRepository`() {
         // Arrange
         coEvery { mockConnectionProxy.disconnect() } returns true.right()
-        coEvery { mockedAccountRepository.logout() } just Runs
+        coEvery { mockedAccountRepository.logout() } returns Unit.right()
 
         // Act
         viewModel.onGoToLoginClicked()
@@ -85,7 +83,7 @@ class DeviceRevokedViewModelTest {
     fun `onGoToLoginClicked should invoke disconnect before logout when connected`() {
         // Arrange
         coEvery { mockConnectionProxy.disconnect() } returns true.right()
-        coEvery { mockedAccountRepository.logout() } just Runs
+        coEvery { mockedAccountRepository.logout() } returns Unit.right()
 
         // Act
         viewModel.onGoToLoginClicked()