Add header text tests

stripe · May 23, 2024 · d46815b · d46815b
1 parent f921f1b
commit d46815b
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/paymentsheet/src/test/java/com/stripe/android/paymentsheet/ui/HeaderTextFactoryTest.kt b/paymentsheet/src/test/java/com/stripe/android/paymentsheet/ui/HeaderTextFactoryTest.kt
@@ -111,4 +111,26 @@ class HeaderTextFactoryTest {
 
         assertThat(resource).isEqualTo(R.string.stripe_paymentsheet_choose_payment_method)
     }
+
+    @Test
+    fun `Shows correct header for manage saved PMs screen when first opened`() {
+        val resource = getManagedSavedPaymentMethodsHeaderText(isCompleteFlow = true)
+
+        assertThat(resource).isEqualTo(R.string.stripe_paymentsheet_select_payment_method)
+    }
+
+    @Test
+    fun `Shows correct header for manage saved PMs screen when first opened - FlowController`() {
+        val resource = getManagedSavedPaymentMethodsHeaderText(isCompleteFlow = false)
+
+        assertThat(resource).isEqualTo(R.string.stripe_paymentsheet_select_payment_method)
+    }
+
+    private fun getManagedSavedPaymentMethodsHeaderText(isCompleteFlow: Boolean): Int? {
+        return HeaderTextFactory(isCompleteFlow = isCompleteFlow).create(
+            screen = PaymentSheetScreen.ManageSavedPaymentMethods,
+            isWalletEnabled = false,
+            types = emptyList(),
+        )
+    }
 }