Add e2e tests for conditional payment button text

- Test 'Pay again' button for failed payments
- Test 'Pay now' button for non-failed payable invoices
- Test 'Payment initiated' success message
- Update existing tests with defensive assertions

Co-Authored-By: sahil.lavingia@gmail.com <sahil.lavingia@gmail.com>

Author: devin-ai-integration[bot]

e2e/tests/company/invoices/complete-flow.spec.ts

@@ -106,10 +106,12 @@ test.describe("Invoice submission, approval and rejection", () => {
     await expect(firstRow).toContainText("$23");
     await expect(firstRow).toContainText("Awaiting approval");
     await expect(firstRow.getByRole("button", { name: "Pay now" })).toBeVisible();
+    await expect(firstRow.getByRole("button", { name: "Pay again" })).not.toBeVisible();
     await expect(secondRow).toContainText("Nov 20, 2024");
     await expect(secondRow).toContainText("$623");
     await expect(secondRow).toContainText("Awaiting approval");
     await expect(secondRow.getByRole("button", { name: "Pay now" })).toBeVisible();
+    await expect(secondRow.getByRole("button", { name: "Pay again" })).not.toBeVisible();
     await expect(thirdRow).toContainText("Nov 1, 2024");
     await expect(thirdRow).toContainText("$870");
     await expect(thirdRow).toContainText("Awaiting approval");
@@ -166,6 +168,8 @@ test.describe("Invoice submission, approval and rejection", () => {
     await page.getByRole("cell", { name: workerUserB.legalName ?? "never" }).click();
     await page.getByRole("link", { name: "View invoice" }).click();
     await expect(page.getByRole("heading", { name: "Invoice" })).toBeVisible();
+    await expect(page.locator("header").filter({ hasText: "Invoice" }).getByRole("button", { name: "Pay now" })).toBeVisible();
+    await expect(page.locator("header").filter({ hasText: "Invoice" }).getByRole("button", { name: "Pay again" })).not.toBeVisible();
     await page.locator("header").filter({ hasText: "Invoice" }).getByRole("button", { name: "Pay now" }).click();
 
     await expect(openInvoicesBadge).not.toBeVisible();

e2e/tests/company/invoices/one-off-payments.spec.ts

@@ -378,11 +378,48 @@ test.describe("One-off payments", () => {
       await page.reload();
       await expect(page.getByRole("row", { name: "$123.45" })).toBeVisible();
 
+      await expect(page.getByRole("button", { name: "Pay now" })).toBeVisible();
+      await expect(page.getByRole("button", { name: "Pay again" })).not.toBeVisible();
+      
       await page.getByRole("button", { name: "Pay now" }).click();
       await page.getByRole("button", { name: "Filter" }).click();
       await page.getByRole("menuitem", { name: "Clear all filters" }).click();
 
       await expect(page.getByRole("row", { name: "$123.45 Payment scheduled" })).toBeVisible();
     });
+
+    test("shows 'Pay again' button for failed payments", async ({ page }) => {
+      const { invoice } = await invoicesFactory.create({
+        companyId: company.id,
+        companyContractorId: companyContractor.id,
+        status: "failed",
+        totalAmountInUsdCents: BigInt(50000),
+      });
+      
+      await login(page, adminUser);
+      await page.goto("/invoices");
+      
+      const invoiceRow = page.locator("tbody tr").filter({ hasText: invoice.invoiceNumber });
+      await expect(invoiceRow.getByRole("button", { name: "Pay again" })).toBeVisible();
+      await expect(invoiceRow.getByRole("button", { name: "Pay now" })).not.toBeVisible();
+    });
+
+    test("shows 'Payment initiated' success message when paying", async ({ page }) => {
+      const { invoice } = await invoicesFactory.create({
+        companyId: company.id,
+        companyContractorId: companyContractor.id,
+        status: "approved",
+        totalAmountInUsdCents: BigInt(50000),
+      });
+      
+      await login(page, adminUser);
+      await page.goto("/invoices");
+      
+      const invoiceRow = page.locator("tbody tr").filter({ hasText: invoice.invoiceNumber });
+      await invoiceRow.getByRole("button", { name: "Pay now" }).click();
+      
+      await expect(page.getByText("Payment initiated")).toBeVisible();
+      await expect(page.getByText("Payment sent!")).not.toBeVisible();
+    });
   });
 });