Fix unit tests

mullvad · Aug 21, 2024 · 7e83d40 · 7e83d40
1 parent 6d9cc55
commit 7e83d40
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/android/lib/shared/src/main/kotlin/net/mullvad/mullvadvpn/lib/shared/ConnectionProxy.kt b/android/lib/shared/src/main/kotlin/net/mullvad/mullvadvpn/lib/shared/ConnectionProxy.kt
@@ -4,16 +4,14 @@ import arrow.core.Either
 import arrow.core.raise.either
 import arrow.core.raise.ensure
 import kotlinx.coroutines.flow.combine
-import mullvad_daemon.management_interface.location
 import net.mullvad.mullvadvpn.lib.daemon.grpc.ManagementService
 import net.mullvad.mullvadvpn.lib.model.ConnectError
 import net.mullvad.mullvadvpn.lib.model.GeoIpLocation
 import net.mullvad.mullvadvpn.lib.model.TunnelState
-import net.mullvad.mullvadvpn.lib.model.location
 
 class ConnectionProxy(
     private val managementService: ManagementService,
-    private val translationRepository: RelayLocationTranslationRepository,
+    translationRepository: RelayLocationTranslationRepository,
     private val vpnPermissionRepository: VpnPermissionRepository
 ) {
     val tunnelState =

diff --git a/android/lib/shared/src/test/kotlin/net/mullvad/mullvadvpn/lib/shared/ConnectionProxyTest.kt b/android/lib/shared/src/test/kotlin/net/mullvad/mullvadvpn/lib/shared/ConnectionProxyTest.kt
@@ -13,11 +13,14 @@ class ConnectionProxyTest {
 
     private val mockManagementService: ManagementService = mockk(relaxed = true)
     private val mockVpnPermissionRepository: VpnPermissionRepository = mockk()
+    private val mockTranslationRepository: RelayLocationTranslationRepository =
+        mockk(relaxed = true)
 
     private val connectionProxy: ConnectionProxy =
         ConnectionProxy(
             managementService = mockManagementService,
-            vpnPermissionRepository = mockVpnPermissionRepository
+            vpnPermissionRepository = mockVpnPermissionRepository,
+            translationRepository = mockTranslationRepository
         )
 
     @Test