fix tests

stripe · Dec 24, 2024 · a2fce78 · a2fce78
1 parent 991a2e4
commit a2fce78
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/Stripe/StripeiOSTests/CustomerAdapterTests.swift b/Stripe/StripeiOSTests/CustomerAdapterTests.swift
@@ -7,7 +7,6 @@ import Foundation
 import OHHTTPStubs
 import OHHTTPStubsSwift
 @_spi(STP) @testable import StripeCore
-@_spi(STP) @testable import StripeCore
 @testable import StripeCoreTestUtils
 @_spi(STP) @testable import StripePayments
 @_spi(STP) @_spi(CustomerSessionBetaAccess) @testable import StripePaymentSheet
@@ -160,10 +159,10 @@ class CustomerAdapterTests: APIStubbedTestCase {
     }
 
     func testFetchPM_CardAndUSBankAccount() async throws {
-        let expectedPaymentMethods_card = [STPFixtures.paymentMethod()]
+        let expectedPaymentMethods_card = STPFixtures.paymentMethod()
         let expectedPaymentMethods_cardJSON = [STPFixtures.paymentMethodJSON()]
 
-        let expectedPaymentMethods_usbank = [STPFixtures.bankAccountPaymentMethod()]
+        let expectedPaymentMethods_usbank = STPFixtures.bankAccountPaymentMethod()
         let expectedPaymentMethods_usbankJSON = [STPFixtures.bankAccountPaymentMethodJSON()]
 
         let apiClient = stubbedAPIClient()
@@ -177,8 +176,8 @@ class CustomerAdapterTests: APIStubbedTestCase {
                                         apiClient: apiClient)
         let pms = try await sut.fetchPaymentMethods()
         XCTAssertEqual(pms.count, 2)
-        XCTAssertEqual(pms[0].stripeId, expectedPaymentMethods_card[0].stripeId)
-        XCTAssertEqual(pms[1].stripeId, expectedPaymentMethods_usbank[0].stripeId)
+        XCTAssertEqual(pms[0].stripeId, expectedPaymentMethods_usbank.stripeId)
+        XCTAssertEqual(pms[1].stripeId, expectedPaymentMethods_card.stripeId)
     }
 
     func testAttachPM() async throws {