Add some tests.

stripe · Dec 18, 2024 · e44078b · e44078b
1 parent c82c03e
commit e44078b
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/payments-core/src/test/java/com/stripe/android/networking/StripeApiRepositoryTest.kt b/payments-core/src/test/java/com/stripe/android/networking/StripeApiRepositoryTest.kt
@@ -197,6 +197,18 @@ internal class StripeApiRepositoryTest {
         assertThat(detachUrl).isEqualTo(expectedUrl)
     }
 
+    @Test
+    fun testGetElementsDetachPaymentMethodUrl() {
+        val paymentMethodId = "pm_1ETDEa2eZvKYlo2CN5828c52"
+        val detachUrl = stripeApiRepository.getElementsDetachPaymentMethodUrl(paymentMethodId)
+        val expectedUrl = arrayOf(
+            "https://api.stripe.com/v1/elements/payment_methods/",
+            paymentMethodId,
+            "/detach"
+        ).joinToString("")
+        assertThat(detachUrl).isEqualTo(expectedUrl)
+    }
+
     @Test
     fun testGetPaymentMethodsUrl() {
         assertThat(StripeApiRepository.paymentMethodsUrl)

diff --git a/...heet/src/test/java/com/stripe/android/paymentsheet/repositories/CustomerRepositoryTest.kt b/...heet/src/test/java/com/stripe/android/paymentsheet/repositories/CustomerRepositoryTest.kt
@@ -398,6 +398,28 @@ internal class CustomerRepositoryTest {
             assertThat(result.isFailure).isTrue()
         }
 
+    @Test
+    fun `detachPaymentMethod() should call elements endpoint when customerSessionClientSecret exists`() =
+        runTest {
+            givenElementsDetachPaymentMethodReturns(
+                Result.success(
+                    PaymentMethodFixtures.CARD_PAYMENT_METHOD
+                )
+            )
+
+            val result = repository.detachPaymentMethod(
+                customerInfo = CustomerRepository.CustomerInfo(
+                    id = "customer_id",
+                    ephemeralKeySecret = "ephemeral_key",
+                    customerSessionClientSecret = "cuss_123",
+                ),
+                paymentMethodId = "payment_method_id",
+                canRemoveDuplicates = false,
+            )
+
+            assertThat(result.getOrNull()).isEqualTo(PaymentMethodFixtures.CARD_PAYMENT_METHOD)
+        }
+
     @Test
     fun `detachPaymentMethod() with 'canRemoveDuplicates' as 'true' should remove duplicate payment methods and remove provided payment method ID last`() =
         runTest {
@@ -643,6 +665,21 @@ internal class CustomerRepositoryTest {
         }
     }
 
+    private fun givenElementsDetachPaymentMethodReturns(
+        result: Result<PaymentMethod>
+    ) {
+        stripeRepository.stub {
+            onBlocking {
+                detachPaymentMethod(
+                    customerSessionClientSecret = any(),
+                    productUsageTokens = any(),
+                    paymentMethodId = anyString(),
+                    requestOptions = any(),
+                )
+            }.doReturn(result)
+        }
+    }
+
     private fun givenAttachPaymentMethodReturns(
         result: Result<PaymentMethod>
     ) {