Add unit test for disconnect on welcome viewmodel

1 files changed, 13 insertions, 0 deletions

diff --git a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/WelcomeViewModelTest.kt b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/WelcomeViewModelTest.kt
index 4aaa000047..1c7a7d0e3b 100644
--- a/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/WelcomeViewModelTest.kt
+++ b/android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/WelcomeViewModelTest.kt
@@ -3,6 +3,7 @@ package net.mullvad.mullvadvpn.viewmodel
 import androidx.lifecycle.viewModelScope
 import app.cash.turbine.test
 import io.mockk.coEvery
+import io.mockk.coVerify
 import io.mockk.every
 import io.mockk.mockk
 import io.mockk.mockkStatic
@@ -213,6 +214,18 @@ class WelcomeViewModelTest {
             }
         }
 
+    @Test
+    fun `when on disconnect click is called should call connection proxy disconnect`() = runTest {
+        // Arrange
+        coEvery { mockConnectionProxy.disconnect() } returns true
+
+        // Act
+        viewModel.onDisconnectClick()
+
+        // Assert
+        coVerify { mockConnectionProxy.disconnect() }
+    }
+
     companion object {
         private const val PURCHASE_RESULT_EXTENSIONS_CLASS =
             "net.mullvad.mullvadvpn.util.PurchaseResultExtensionsKt"