UI Test fixes

primer-io · Dec 13, 2024 · dd6a283 · dd6a283
1 parent ab63aa2
commit dd6a283
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/...rces/View Controllers/MerchantHeadlessCheckoutAvailablePaymentMethodsViewController.swift b/...rces/View Controllers/MerchantHeadlessCheckoutAvailablePaymentMethodsViewController.swift
@@ -312,6 +312,7 @@ extension MerchantHeadlessCheckoutAvailablePaymentMethodsViewController {
         print("\n\nMERCHANT APP\n\(#function)\nerror: \(err)\ncheckoutData: \(String(describing: checkoutData))")
         self.logs.append(#function)
         self.primerError = err
+        self.checkoutData = checkoutData
         self.hideLoadingOverlay()
 
         if let lastViewController = navigationController?.children.last {

diff --git a/.../New UI/StripeAchHeadless/MerchantHeadlessCheckoutStripeAchViewController+StripeAch.swift b/.../New UI/StripeAchHeadless/MerchantHeadlessCheckoutStripeAchViewController+StripeAch.swift
@@ -88,7 +88,7 @@ extension MerchantHeadlessCheckoutStripeAchViewController: PrimerHeadlessUnivers
     func primerHeadlessUniversalCheckoutDidFail(withError err: any Error, checkoutData: PrimerCheckoutData?) {
         print("\n\nMERCHANT APP\n\(#function)\nerror: \(err)\ncheckoutData: \(String(describing: checkoutData))")
         logs.append(#function)
-        presentResultsVC(checkoutData: nil, error: err)
+        presentResultsVC(checkoutData: checkoutData, error: err)
     }
 
     func primerHeadlessUniversalCheckoutDidReceiveAdditionalInfo(_ additionalInfo: PrimerCheckoutAdditionalInfo?) {

diff --git a/Sources/PrimerSDK/Classes/Core/PrimerHeadlessUniversalCheckout/Managers/VaultManager.swift b/Sources/PrimerSDK/Classes/Core/PrimerHeadlessUniversalCheckout/Managers/VaultManager.swift
@@ -25,7 +25,6 @@ extension PrimerHeadlessUniversalCheckout {
         private(set) var resumePaymentId: String?
         private var webViewController: SFSafariViewController?
         private var webViewCompletion: ((_ authorizationToken: String?, _ error: PrimerError?) -> Void)?
-        private var achMandateCompletion: ((Result<Void, PrimerError>) -> Void)?
 
         lazy var createResumePaymentService: CreateResumePaymentServiceProtocol = {
             CreateResumePaymentService(paymentMethodType: paymentMethodType)

diff --git a/...ses/User Interface/TokenizationViewModels/CheckoutWithVaultedPaymentMethodViewModel.swift b/...ses/User Interface/TokenizationViewModels/CheckoutWithVaultedPaymentMethodViewModel.swift
@@ -522,9 +522,14 @@ Make sure you call the decision handler otherwise the SDK will hang.
     }
 
     func handleSuccessfulFlow() {
-        let categories = self.config.paymentMethodManagerCategories
-        PrimerUIManager.dismissOrShowResultScreen(type: .success,
-                                                  paymentMethodManagerCategories: categories ?? [])
+        if let paymentMethodType = config.internalPaymentMethodType,
+              paymentMethodType == .stripeAch {
+            PrimerUIManager.showResultScreen(for: paymentMethodType, error: nil)
+        } else {
+            let categories = self.config.paymentMethodManagerCategories
+            PrimerUIManager.dismissOrShowResultScreen(type: .success,
+                                                      paymentMethodManagerCategories: categories ?? [])
+        }
     }
 
     func handleFailureFlow(errorMessage: String?) {