Fix tests

RevenueCat · Jan 31, 2024 · 0b68b52 · 0b68b52
1 parent b4a44d2
commit 0b68b52
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/Tests/RevenueCatUITests/PaywallFooterTests.swift b/Tests/RevenueCatUITests/PaywallFooterTests.swift
@@ -64,6 +64,8 @@ class PaywallFooterTests: TestCase {
 
         Task {
             _ = try await Self.purchaseHandler.restorePurchases()
+            // Simulates what `RestorePurchasesButton` does after dismissing the alert.
+            Self.purchaseHandler.setRestored(TestData.customerInfo)
         }
 
         expect(customerInfo).toEventually(be(TestData.customerInfo))

diff --git a/Tests/RevenueCatUITests/PresentIfNeededTests.swift b/Tests/RevenueCatUITests/PresentIfNeededTests.swift
@@ -72,6 +72,8 @@ class PresentIfNeededTests: TestCase {
 
         Task {
             _ = try await Self.purchaseHandler.restorePurchases()
+            // Simulates what `RestorePurchasesButton` does after dismissing the alert.
+            Self.purchaseHandler.setRestored(TestData.customerInfo)
         }
 
         expect(customerInfo).toEventually(be(TestData.customerInfo))