fix: wait for logout to be completed when handling user events [WPB-9…

…689] [WPB-10310] (#2914) Co-authored-by: Vitor Hugo Schwaab <[email protected]> Co-authored-by: Mohamad Jaara <[email protected]>
wireapp · Jul 30, 2024 · 8a02f96 · 8a02f96
1 parent e2ac7c3
commit 8a02f96
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/logic/src/commonMain/kotlin/com/wire/kalium/logic/sync/receiver/UserEventReceiver.kt b/logic/src/commonMain/kotlin/com/wire/kalium/logic/sync/receiver/UserEventReceiver.kt
@@ -131,7 +131,7 @@ internal class UserEventReceiverImpl internal constructor(
         return currentClientIdProvider().map { currentClientId ->
             if (currentClientId == event.clientId) {
                 logger.logSuccess("info" to "CURRENT_CLIENT")
-                logout(LogoutReason.REMOVED_CLIENT)
+                logout(LogoutReason.REMOVED_CLIENT, waitUntilCompletes = true)
             } else {
                 logger.logSuccess("info" to "OTHER_CLIENT")
             }
@@ -148,7 +148,7 @@ internal class UserEventReceiverImpl internal constructor(
     private suspend fun handleUserDelete(event: Event.User.UserDelete): Either<CoreFailure, Unit> {
         val logger = kaliumLogger.createEventProcessingLogger(event)
         return if (selfUserId == event.userId) {
-            logout(LogoutReason.DELETED_ACCOUNT)
+            logout(LogoutReason.DELETED_ACCOUNT, waitUntilCompletes = true)
             Either.Right(Unit)
         } else {
             userRepository.markUserAsDeletedAndRemoveFromGroupConversations(event.userId)

diff --git a/logic/src/commonTest/kotlin/com/wire/kalium/logic/sync/receiver/UserEventReceiverTest.kt b/logic/src/commonTest/kotlin/com/wire/kalium/logic/sync/receiver/UserEventReceiverTest.kt
@@ -70,7 +70,7 @@ class UserEventReceiverTest {
         eventReceiver.onEvent(event, TestEvent.liveDeliveryInfo)
 
         coVerify {
-            arrangement.logoutUseCase.invoke(eq(LogoutReason.REMOVED_CLIENT), any())
+            arrangement.logoutUseCase.invoke(eq(LogoutReason.REMOVED_CLIENT), eq(true))
         }.wasInvoked(exactly = once)
     }
 
@@ -99,7 +99,7 @@ class UserEventReceiverTest {
         eventReceiver.onEvent(event, TestEvent.liveDeliveryInfo)
 
         coVerify {
-            arrangement.logoutUseCase.invoke(eq(LogoutReason.DELETED_ACCOUNT), any())
+            arrangement.logoutUseCase.invoke(eq(LogoutReason.DELETED_ACCOUNT), eq(true))
         }.wasInvoked(exactly = once)
     }