From 31557fe6ad0444da70cfc1e09713d0a255db6787 Mon Sep 17 00:00:00 2001
From: Suejung Shin <suejung.shin@sentry.io>
Date: Thu, 16 Jan 2025 18:58:46 -0800
Subject: [PATCH] fix tests

---
 .../PaymentCard/PaymentCard.test.jsx          | 31 +++++++++++++++++++
 .../PaymentCard/PaymentMethodForm.test.tsx    | 10 +++---
 2 files changed, 36 insertions(+), 5 deletions(-)

diff --git a/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentCard.test.jsx b/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentCard.test.jsx
index d6ccb2f6d3..6173a5af87 100644
--- a/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentCard.test.jsx
+++ b/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentCard.test.jsx
@@ -35,6 +35,7 @@ vi.mock('services/account/useCreateStripeSetupIntent', async () => {
 })
 
 afterEach(() => {
+  queryClient.clear()
   vi.clearAllMocks()
 })
 
@@ -54,6 +55,20 @@ const subscriptionDetail = {
   cancelAtPeriodEnd: false,
 }
 
+const usBankSubscriptionDetail = {
+  defaultPaymentMethod: {
+    usBankAccount: {
+      bankName: 'STRIPE TEST BANK',
+      last4: '6789',
+    },
+  },
+  plan: {
+    value: Plans.USERS_PR_INAPPY,
+  },
+  currentPeriodEnd: 1606851492,
+  cancelAtPeriodEnd: false,
+}
+
 const wrapper = ({ children }) => (
   <QueryClientProvider client={queryClient}>
     <ThemeContextProvider>{children}</ThemeContextProvider>
@@ -200,6 +215,22 @@ describe('PaymentCard', () => {
     })
   })
 
+  describe('when the user has a US bank account', () => {
+    it('renders the bank account details', () => {
+      render(
+        <PaymentCard
+          subscriptionDetail={usBankSubscriptionDetail}
+          provider="gh"
+          owner="codecov"
+        />,
+        { wrapper }
+      )
+
+      expect(screen.getByText(/STRIPE TEST BANK/)).toBeInTheDocument()
+      expect(screen.getByText(/•••• 6789/)).toBeInTheDocument()
+    })
+  })
+
   describe('when the subscription is set to expire', () => {
     it(`doesn't render the next billing`, () => {
       render(
diff --git a/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentMethodForm.test.tsx b/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentMethodForm.test.tsx
index 62c30b5901..20331e0f06 100644
--- a/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentMethodForm.test.tsx
+++ b/src/pages/PlanPage/subRoutes/CurrentOrgPlan/BillingDetails/PaymentCard/PaymentMethodForm.test.tsx
@@ -97,7 +97,7 @@ describe('PaymentMethodForm', () => {
         />,
         { wrapper }
       )
-      await user.click(screen.getByTestId('update-payment-method'))
+      await user.click(screen.getByTestId('save-payment-method'))
 
       expect(screen.queryByText(/Visa/)).not.toBeInTheDocument()
     })
@@ -118,7 +118,7 @@ describe('PaymentMethodForm', () => {
         />,
         { wrapper }
       )
-      await user.click(screen.getByTestId('update-payment-method'))
+      await user.click(screen.getByTestId('save-payment-method'))
 
       expect(screen.getByRole('button', { name: /Save/i })).toBeInTheDocument()
     })
@@ -140,7 +140,7 @@ describe('PaymentMethodForm', () => {
           />,
           { wrapper }
         )
-        await user.click(screen.getByTestId('update-payment-method'))
+        await user.click(screen.getByTestId('save-payment-method'))
         expect(updatePaymentMethod).toHaveBeenCalled()
       })
     })
@@ -163,7 +163,7 @@ describe('PaymentMethodForm', () => {
           { wrapper }
         )
 
-        await user.click(screen.getByTestId('update-payment-method'))
+        await user.click(screen.getByTestId('save-payment-method'))
         await user.click(screen.getByRole('button', { name: /Cancel/ }))
 
         expect(closeForm).toHaveBeenCalled()
@@ -189,7 +189,7 @@ describe('PaymentMethodForm', () => {
         { wrapper }
       )
 
-      await user.click(screen.getByTestId('update-payment-method'))
+      await user.click(screen.getByTestId('save-payment-method'))
 
       expect(screen.getByText(randomError)).toBeInTheDocument()
     })