Fix missing device refresh

android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/ConnectViewModel.kt

@@ -118,6 +118,7 @@ class ConnectViewModel(
                 viewModelScope.launch { accountRepository.getAccountData() }
             }
         }
+        viewModelScope.launch { deviceRepository.updateDevice() }
     }
 
     fun onDisconnectClick() {

android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/OutOfTimeViewModel.kt

@@ -59,6 +59,7 @@ class OutOfTimeViewModel(
         }
         verifyPurchases()
         fetchPaymentAvailability()
+        viewModelScope.launch { deviceRepository.updateDevice() }
     }
 
     fun onSitePaymentClick() {

android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/WelcomeViewModel.kt

@@ -59,6 +59,7 @@ class WelcomeViewModel(
         }
         verifyPurchases()
         fetchPaymentAvailability()
+        viewModelScope.launch { deviceRepository.updateDevice() }
     }
 
     private fun hasAddedTimeEffect() =

android/app/src/test/kotlin/net/mullvad/mullvadvpn/viewmodel/ConnectViewModelTest.kt

@@ -3,9 +3,11 @@ package net.mullvad.mullvadvpn.viewmodel
 import androidx.lifecycle.viewModelScope
 import app.cash.turbine.test
 import arrow.core.right
+import io.mockk.Runs
 import io.mockk.coEvery
 import io.mockk.coVerify
 import io.mockk.every
+import io.mockk.just
 import io.mockk.mockk
 import io.mockk.mockkStatic
 import io.mockk.unmockkAll
@@ -102,6 +104,8 @@ class ConnectViewModelTest {
 
         every { mockDeviceRepository.deviceState } returns device
 
+        coEvery { mockDeviceRepository.updateDevice() } just Runs
+
         every { mockInAppNotificationController.notifications } returns notifications
 
         every { mockConnectionProxy.tunnelState } returns tunnelState

android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/ManagementService.kt

@@ -210,6 +210,10 @@ class ManagementService(
             .map { it.toDomain() }
             .mapLeft { GetDeviceStateError.Unknown(it) }
 
+    suspend fun updateDevice() {
+        grpc.updateDevice(Empty.getDefaultInstance())
+    }
+
     suspend fun getDeviceList(token: AccountNumber): Either<GetDeviceListError, List<Device>> =
         Either.catch { grpc.listDevices(StringValue.of(token.value)) }
             .map { it.devicesList.map(ManagementInterface.Device::toDomain) }

android/lib/shared/src/main/kotlin/net/mullvad/mullvadvpn/lib/shared/DeviceRepository.kt

@@ -1,5 +1,6 @@
 package net.mullvad.mullvadvpn.lib.shared
 
+import android.util.Log
 import arrow.core.Either
 import kotlinx.coroutines.CoroutineDispatcher
 import kotlinx.coroutines.CoroutineScope
@@ -33,4 +34,9 @@ class DeviceRepository(
 
     suspend fun deviceList(accountNumber: AccountNumber): Either<GetDeviceListError, List<Device>> =
         managementService.getDeviceList(accountNumber)
+
+    suspend fun updateDevice() {
+        Log.d("mullvad", "Update device")
+        managementService.updateDevice()
+    }
 }