allow non-terminal state for PaymentSheet (#4438)

* allow non-terminal state for PaymentSheet * update test name * update error message * update test
stripe · Dec 9, 2021 · 94a1657 · 94a1657
1 parent c75cef9
commit 94a1657
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 12 deletions.
diff --git a/paymentsheet/src/main/java/com/stripe/android/paymentsheet/model/StripeIntentValidator.kt b/paymentsheet/src/main/java/com/stripe/android/paymentsheet/model/StripeIntentValidator.kt
@@ -26,26 +26,25 @@ internal class StripeIntentValidator @Inject constructor() {
             }
             stripeIntent is PaymentIntent &&
                 (
-                    (stripeIntent.status != StripeIntent.Status.RequiresPaymentMethod) &&
-                        (stripeIntent.status != StripeIntent.Status.RequiresAction)
+                    (stripeIntent.status == StripeIntent.Status.Canceled) ||
+                        (stripeIntent.status == StripeIntent.Status.Succeeded) ||
+                        (stripeIntent.status == StripeIntent.Status.RequiresCapture)
                     ) -> {
                 error(
                     """
-                        A PaymentIntent with status='requires_payment_method' or 'requires_action` is required.
-                        The current PaymentIntent has status '${stripeIntent.status}'.
+                        PaymentSheet cannot set up a PaymentIntent in status '${stripeIntent.status}'.
                         See https://stripe.com/docs/api/payment_intents/object#payment_intent_object-status.
                     """.trimIndent()
                 )
             }
             stripeIntent is SetupIntent &&
                 (
-                    (stripeIntent.status != StripeIntent.Status.RequiresPaymentMethod) &&
-                        (stripeIntent.status != StripeIntent.Status.RequiresAction)
+                    (stripeIntent.status == StripeIntent.Status.Canceled) ||
+                        (stripeIntent.status == StripeIntent.Status.Succeeded)
                     ) -> {
                 error(
                     """
-                        A SetupIntent with status='requires_payment_method' or 'requires_action` is required.
-                        The current SetupIntent has status '${stripeIntent.status}'.
+                        PaymentSheet cannot set up a SetupIntent in status '${stripeIntent.status}'.
                         See https://stripe.com/docs/api/setup_intents/object#setup_intent_object-status
                     """.trimIndent()
                 )

diff --git a/paymentsheet/src/test/java/com/stripe/android/paymentsheet/PaymentSheetViewModelTest.kt b/paymentsheet/src/test/java/com/stripe/android/paymentsheet/PaymentSheetViewModelTest.kt
@@ -638,8 +638,7 @@ internal class PaymentSheetViewModelTest {
         viewModel.maybeFetchStripeIntent()
         assertThat((result as? PaymentSheetResult.Failed)?.error?.message)
             .isEqualTo(
-                "A PaymentIntent with status='requires_payment_method' or 'requires_action` is required.\n" +
-                    "The current PaymentIntent has status 'succeeded'.\n" +
+                "PaymentSheet cannot set up a PaymentIntent in status 'succeeded'.\n" +
                     "See https://stripe.com/docs/api/payment_intents/object#payment_intent_object-status."
             )
     }

diff --git a/...entsheet/src/test/java/com/stripe/android/paymentsheet/model/StripeIntentValidatorTest.kt b/...entsheet/src/test/java/com/stripe/android/paymentsheet/model/StripeIntentValidatorTest.kt
@@ -34,11 +34,11 @@ class StripeIntentValidatorTest {
     }
 
     @Test
-    fun `requireValid() processing is not valid`() {
+    fun `requireValid() Succeeded is not valid`() {
         assertFails {
             validator.requireValid(
                 PaymentIntentFixtures.PI_REQUIRES_PAYMENT_METHOD.copy(
-                    status = StripeIntent.Status.Processing
+                    status = StripeIntent.Status.Succeeded
                 )
             )
         }
@@ -71,4 +71,22 @@ class StripeIntentValidatorTest {
             SetupIntentFixtures.SI_NEXT_ACTION_REDIRECT
         )
     }
+
+    @Test
+    fun `PaymentIntent requireValid() allows status = RequiresConfirmation`() {
+        validator.requireValid(
+            PaymentIntentFixtures.PI_REQUIRES_PAYMENT_METHOD.copy(
+                status = StripeIntent.Status.RequiresConfirmation
+            )
+        )
+    }
+
+    @Test
+    fun `PaymentIntent requireValid() allows status = Processing`() {
+        validator.requireValid(
+            PaymentIntentFixtures.PI_REQUIRES_PAYMENT_METHOD.copy(
+                status = StripeIntent.Status.Processing
+            )
+        )
+    }
 }