Ensure web redirect presentation gracefully handles testing cases whe…

…re no window is present
primer-io · Apr 19, 2024 · 3616e6b · 3616e6b
1 parent 5c2005d
commit 3616e6b
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/...User Interface/TokenizationViewModels/WebRedirectPaymentMethodTokenizationViewModel.swift b/...User Interface/TokenizationViewModels/WebRedirectPaymentMethodTokenizationViewModel.swift
@@ -198,15 +198,23 @@ class WebRedirectPaymentMethodTokenizationViewModel: PaymentMethodTokenizationVi
                     responseCode: nil
                 )
 
+
                 Analytics.Service.record(events: [presentEvent, networkEvent])
+
+                guard UIApplication.shared.windows.count > 0 else {
+                    self.handleWebViewControllerPresentedCompletion()
+                    seal.fulfill()
+                    return
+                }
+
                 if PrimerUIManager.primerRootViewController == nil {
                     firstly {
                         PrimerUIManager.prepareRootViewController()
                     }
                     .done {
                         PrimerUIManager.primerRootViewController?.present(self.webViewController!, animated: true, completion: {
                             DispatchQueue.main.async {
-                                self.handleWebViewControlllerPresentedCompletion()
+                                self.handleWebViewControllerPresentedCompletion()
                                 seal.fulfill()
                             }
                         })
@@ -215,7 +223,7 @@ class WebRedirectPaymentMethodTokenizationViewModel: PaymentMethodTokenizationVi
                 } else {
                     PrimerUIManager.primerRootViewController?.present(self.webViewController!, animated: true, completion: {
                         DispatchQueue.main.async {
-                            self.handleWebViewControlllerPresentedCompletion()
+                            self.handleWebViewControllerPresentedCompletion()
                             seal.fulfill()
                         }
                     })
@@ -224,7 +232,7 @@ class WebRedirectPaymentMethodTokenizationViewModel: PaymentMethodTokenizationVi
         }
     }
 
-    private func handleWebViewControlllerPresentedCompletion() {
+    private func handleWebViewControllerPresentedCompletion() {
         DispatchQueue.main.async {
             let viewEvent = Analytics.Event.ui(
                 action: .view,