From c738f24efdc18f6c3f2bdadcf1d2ab41bfae77b3 Mon Sep 17 00:00:00 2001
From: Mat Schmid <mats@stripe.com>
Date: Mon, 25 Nov 2024 11:20:08 -0500
Subject: [PATCH] Fallback to third position

---
 .../Source/PaymentSheet/PaymentMethodType.swift          | 9 ++++++---
 .../PaymentSheet/PaymentSheetPaymentMethodTypeTest.swift | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/PaymentMethodType.swift b/StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/PaymentMethodType.swift
index b62fbe87313..8d3b3c46287 100644
--- a/StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/PaymentMethodType.swift
+++ b/StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/PaymentMethodType.swift
@@ -191,10 +191,13 @@ extension PaymentSheet {
                 configuration.isEligibleForBankTab
             }
 
+            // Places the payment method type at either the index for `link`, or in the third / last position.
             func insertAtBankTabPosition(paymentMethodType: PaymentMethodType) {
-                let bankTabPosition = elementsSession.orderedPaymentMethodTypes.firstIndex(of: .link)
-                if let bankTabPosition, recommendedPaymentMethodTypes.indices.contains(bankTabPosition) {
-                    recommendedPaymentMethodTypes.insert(paymentMethodType, at: bankTabPosition)
+                let linkIndex = elementsSession.orderedPaymentMethodTypes.firstIndex(of: .link)
+                let fallbackIndex = 2
+                let bankTabIndex = linkIndex ?? fallbackIndex
+                if recommendedPaymentMethodTypes.indices.contains(bankTabIndex) {
+                    recommendedPaymentMethodTypes.insert(paymentMethodType, at: bankTabIndex)
                 } else {
                     recommendedPaymentMethodTypes.append(paymentMethodType)
                 }
diff --git a/StripePaymentSheet/StripePaymentSheetTests/PaymentSheet/PaymentSheetPaymentMethodTypeTest.swift b/StripePaymentSheet/StripePaymentSheetTests/PaymentSheet/PaymentSheetPaymentMethodTypeTest.swift
index c32ed218c03..925adeb450a 100644
--- a/StripePaymentSheet/StripePaymentSheetTests/PaymentSheet/PaymentSheetPaymentMethodTypeTest.swift
+++ b/StripePaymentSheet/StripePaymentSheetTests/PaymentSheet/PaymentSheetPaymentMethodTypeTest.swift
@@ -467,7 +467,7 @@ class PaymentSheetPaymentMethodTypeTest: XCTestCase {
             ),
             configuration: configuration
         )
-        XCTAssertEqual(types, [.stripe(.cashApp), .stripe(.amazonPay), .stripe(.card), .stripe(.klarna), .linkCardBrand])
+        XCTAssertEqual(types, [.stripe(.cashApp), .stripe(.amazonPay), .linkCardBrand, .stripe(.card), .stripe(.klarna)])
     }
 
     func testPaymentMethodTypesLinkCardBrand_noDefaults() {