Paywall Tester: fix automatic dismissing after purchasing

See #3516 and RevenueCat/revenuecat-docs#579.
RevenueCat · Dec 13, 2023 · 99d605a · 99d605a
1 parent 32c63cd
commit 99d605a
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/Tests/TestingApps/PaywallsTester/PaywallsTester/Views/AppContentView.swift b/Tests/TestingApps/PaywallsTester/PaywallsTester/Views/AppContentView.swift
@@ -159,6 +159,7 @@ struct AppContentView: View {
     #if !os(watchOS)
     private func presentPaywallViewController() {
         let paywall = PaywallViewController(displayCloseButton: true)
+        paywall.delegate = paywallDelegate
         paywall.modalPresentationStyle = .pageSheet
 
         guard let rootController = UIApplication
@@ -176,6 +177,8 @@ struct AppContentView: View {
 
 }
 
+// MARK: -
+
 private struct ProminentButton: View {
 
     var title: String
@@ -224,6 +227,23 @@ extension CustomerInfo {
 
 }
 
+// MARK: - Delegate
+
+private let paywallDelegate = PaywallDelegate()
+
+private final class PaywallDelegate: NSObject, PaywallViewControllerDelegate {
+
+    func paywallViewController(
+        _ controller: PaywallViewController,
+        didFinishPurchasingWith customerInfo: CustomerInfo
+    ) {
+        controller.dismiss(animated: true)
+    }
+
+}
+
+// MARK: -
+
 #if !os(watchOS)
 
 private extension UIApplication {

diff --git a/Tests/TestingApps/PaywallsTester/PaywallsTester/Views/OfferingsList.swift b/Tests/TestingApps/PaywallsTester/PaywallsTester/Views/OfferingsList.swift
@@ -123,6 +123,9 @@ struct OfferingsList: View {
         }
         .sheet(item: self.$presentedPaywall) { paywall in
             PaywallPresenter(offering: paywall.offering, mode: paywall.mode)
+                .onPurchaseCompleted { _ in
+                    self.presentedPaywall = nil
+                }
         }
     }