Make sure the Recovery Key option is shown on the IdentityConfirmatio…

…nScreen when available. (#3699) There was a race condition where 2 different states could be processed at the same time. * Fix unit test The view model wasn't stored, so relying the current value publisher didn't have time to fire before deinit.
element-hq · Jan 23, 2025 · a70189d · a70189d
1 parent 88e63b2
commit a70189d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/...s/Screens/Onboarding/IdentityConfirmationScreen/IdentityConfirmationScreenViewModel.swift b/...s/Screens/Onboarding/IdentityConfirmationScreen/IdentityConfirmationScreenViewModel.swift
@@ -25,18 +25,13 @@ class IdentityConfirmationScreenViewModel: IdentityConfirmationScreenViewModelTy
 
         super.init(initialViewState: IdentityConfirmationScreenViewState(learnMoreURL: appSettings.encryptionURL))
 
-        userSession.sessionSecurityStatePublisher
-            .receive(on: DispatchQueue.main)
-            .sink { [weak self] state in
-                Task {
-                    await self?.updateWithSessionSecurityState(state)
-                }
+        Task { [weak self] in
+            for await state in userSession.sessionSecurityStatePublisher.values {
+                // We need to call this inside an AsyncSequence otherwise there's a race condition when the method suspends.
+                await self?.updateWithSessionSecurityState(state)
             }
-            .store(in: &cancellables)
-
-        Task {
-            await updateWithSessionSecurityState(userSession.sessionSecurityStatePublisher.value)
         }
+        .store(in: &cancellables)
     }
 
     // MARK: - Public

diff --git a/...urces/Screens/Onboarding/IdentityConfirmationScreen/View/IdentityConfirmationScreen.swift b/...urces/Screens/Onboarding/IdentityConfirmationScreen/View/IdentityConfirmationScreen.swift
@@ -104,6 +104,8 @@ struct IdentityConfirmationScreen: View {
 // MARK: - Previews
 
 struct IdentityConfirmationScreen_Previews: PreviewProvider, TestablePreview {
+    static var viewModel = makeViewModel()
+
     static var previews: some View {
         NavigationStack {
             IdentityConfirmationScreen(context: viewModel.context)
@@ -113,7 +115,7 @@ struct IdentityConfirmationScreen_Previews: PreviewProvider, TestablePreview {
         })
     }
 
-    private static var viewModel: IdentityConfirmationScreenViewModel {
+    static func makeViewModel() -> IdentityConfirmationScreenViewModel {
         let clientProxy = ClientProxyMock(.init())
         let userSession = UserSessionMock(.init(clientProxy: clientProxy))
         userSession.sessionSecurityStatePublisher = CurrentValuePublisher<SessionSecurityState, Never>(.init(verificationState: .unverified, recoveryState: .enabled))