Fix one payment option method at return

yoomoney · Oct 16, 2018 · 3cb8150 · 3cb8150
1 parent 1d1cbb7
commit 3cb8150
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/YandexCheckoutPayments/Modules/Tokenization/Presenter/TokenizationPresenter.swift b/YandexCheckoutPayments/Modules/Tokenization/Presenter/TokenizationPresenter.swift
@@ -238,6 +238,18 @@ extension TokenizationPresenter: TokenizationStrategyOutput {
                                            moduleOutput: strongSelf)
         }
     }
+
+    func didFinish(on module: TokenizationStrategyInput) {
+        handleOnePaymentOptionMethodAtReturn()
+    }
+
+    func handleOnePaymentOptionMethodAtReturn() {
+        if paymentOptionsCount == 1 {
+            close()
+        } else {
+            presentPaymentMethodsModule()
+        }
+    }
 }
 
 // MARK: - TokenizationViewOutput
@@ -456,7 +468,7 @@ extension TokenizationPresenter: YandexAuthModuleOutput {
     func didFailFetchYamoneyPaymentMethods(on module: YandexAuthModuleInput) {}
 
     func didCancelAuthorizeInYandex(on module: YandexAuthModuleInput) {
-        presentPaymentMethodsModule()
+        handleOnePaymentOptionMethodAtReturn()
     }
 }
 

diff --git a/YandexCheckoutPayments/Modules/Tokenization/Strategy/ApplePayStrategy.swift b/YandexCheckoutPayments/Modules/Tokenization/Strategy/ApplePayStrategy.swift
@@ -53,7 +53,7 @@ extension ApplePayStrategy: TokenizationStrategyInput {
 
     func paymentAuthorizationViewControllerDidFinish(_ controller: PKPaymentAuthorizationViewController) {
         if case .failed = paymentResult {
-            output?.presentPaymentMethodsModule()
+            output?.didFinish(on: self)
         }
     }
 

diff --git a/YandexCheckoutPayments/Modules/Tokenization/Strategy/TokenizationStrategy.swift b/YandexCheckoutPayments/Modules/Tokenization/Strategy/TokenizationStrategy.swift
@@ -75,6 +75,8 @@ protocol TokenizationStrategyOutput: class {
 
     func presentErrorWithMessage(_ message: String)
 
+    func didFinish(on module: TokenizationStrategyInput)
+
     // MARK: - ApplePay
 
     func presentApplePay()