Reset restore dialog state when restore is cancelled

cursoragent · cursoragent · commit 5271dcc692d4 · 2026-03-16T20:21:50.000Z
diff --git a/ui/revenuecatui/src/main/kotlin/com/revenuecat/purchases/ui/revenuecatui/customercenter/viewmodel/CustomerCenterViewModel.kt b/ui/revenuecatui/src/main/kotlin/com/revenuecat/purchases/ui/revenuecatui/customercenter/viewmodel/CustomerCenterViewModel.kt
@@ -510,6 +510,13 @@ internal class CustomerCenterViewModelImpl(
         if (!shouldResumeRestorePurchases(listener, "listener") ||
             !shouldResumeRestorePurchases(purchases.customerCenterListener, "purchases.customerCenterListener")
         ) {
+            _state.update { currentState ->
+                if (currentState is CustomerCenterState.Success) {
+                    currentState.copy(restorePurchasesState = null)
+                } else {
+                    currentState
+                }
+            }
             return
         }
         notifyListenersForRestoreStarted()
diff --git a/ui/revenuecatui/src/test/kotlin/com/revenuecat/purchases/ui/revenuecatui/customercenter/data/CustomerCenterViewModelTests.kt b/ui/revenuecatui/src/test/kotlin/com/revenuecat/purchases/ui/revenuecatui/customercenter/data/CustomerCenterViewModelTests.kt
@@ -963,13 +963,30 @@ class CustomerCenterViewModelTests {
             listener = directListener
         )
 
+        model.state.filterIsInstance<CustomerCenterState.Success>().first()
+
+        val missingPurchasePath = HelpPath(
+            id = "missing_purchase_id",
+            title = "Restore Purchases",
+            type = HelpPath.PathType.MISSING_PURCHASE
+        )
+        model.pathButtonPressed(mockk(relaxed = true), missingPurchasePath, null)
+
+        model.state.first {
+            it is CustomerCenterState.Success &&
+                it.restorePurchasesState == RestorePurchasesState.RESTORING
+        }
+
         model.restorePurchases()
 
+        val updatedState = model.state.value as CustomerCenterState.Success
+
         verify(exactly = 1) { directListener.onRestoreInitiated(any()) }
         verify(exactly = 0) { purchasesListener.onRestoreInitiated(any()) }
         coVerify(exactly = 0) { purchases.awaitRestore() }
         verify(exactly = 0) { directListener.onRestoreStarted() }
         verify(exactly = 0) { purchasesListener.onRestoreStarted() }
+        assertThat(updatedState.restorePurchasesState).isNull()
     }
 
     @Test
