Fix unit tests

mullvad · Aug 15, 2024 · 5a59129 · 5a59129
1 parent 47b38c9
commit 5a59129
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/...advpn/viewmodel/ChangelogViewModelTest.kt → ...dvpn/viewmodel/MullvadAppViewModelTest.kt b/...advpn/viewmodel/ChangelogViewModelTest.kt → ...dvpn/viewmodel/MullvadAppViewModelTest.kt
@@ -11,18 +11,20 @@ import kotlin.test.assertEquals
 import kotlinx.coroutines.test.runTest
 import net.mullvad.mullvadvpn.lib.common.test.TestCoroutineRule
 import net.mullvad.mullvadvpn.lib.model.BuildVersion
+import net.mullvad.mullvadvpn.lib.shared.ConnectionProxy
 import net.mullvad.mullvadvpn.repository.ChangelogRepository
 import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.extension.ExtendWith
 
 @ExtendWith(TestCoroutineRule::class)
-class ChangelogViewModelTest {
+class MullvadAppViewModelTest {
 
     @MockK private lateinit var mockedChangelogRepository: ChangelogRepository
+    @MockK private lateinit var connectionProxy: ConnectionProxy
 
-    private lateinit var viewModel: ChangelogViewModel
+    private lateinit var viewModel: MullvadAppViewModel
 
     private val buildVersion = BuildVersion("1.0", 10)
 
@@ -43,7 +45,8 @@ class ChangelogViewModelTest {
         // Arrange
         every { mockedChangelogRepository.getVersionCodeOfMostRecentChangelogShowed() } returns
             buildVersion.code
-        viewModel = ChangelogViewModel(mockedChangelogRepository, buildVersion, false)
+        viewModel =
+            MullvadAppViewModel(mockedChangelogRepository, connectionProxy, buildVersion, false)
 
         // If we have the most up to date version code, we should not show the changelog dialog
         viewModel.uiSideEffect.test { expectNoEvents() }
@@ -58,7 +61,8 @@ class ChangelogViewModelTest {
             version
         every { mockedChangelogRepository.getLastVersionChanges() } returns changes
 
-        viewModel = ChangelogViewModel(mockedChangelogRepository, buildVersion, false)
+        viewModel =
+            MullvadAppViewModel(mockedChangelogRepository, connectionProxy, buildVersion, false)
         // Given a new version with a change log we should return it
         viewModel.uiSideEffect.test {
             assertEquals(awaitItem(), Changelog(version = buildVersion.name, changes = changes))
@@ -71,7 +75,8 @@ class ChangelogViewModelTest {
         every { mockedChangelogRepository.getVersionCodeOfMostRecentChangelogShowed() } returns -1
         every { mockedChangelogRepository.getLastVersionChanges() } returns emptyList()
 
-        viewModel = ChangelogViewModel(mockedChangelogRepository, buildVersion, false)
+        viewModel =
+            MullvadAppViewModel(mockedChangelogRepository, connectionProxy, buildVersion, false)
         // Given a new version with a change log we should not return it
         viewModel.uiSideEffect.test { expectNoEvents() }
     }