diff --git a/canvas/src/components/__tests__/PurchaseSuccessModal.test.tsx b/canvas/src/components/__tests__/PurchaseSuccessModal.test.tsx index 4abdb36c..519942dc 100644 --- a/canvas/src/components/__tests__/PurchaseSuccessModal.test.tsx +++ b/canvas/src/components/__tests__/PurchaseSuccessModal.test.tsx @@ -13,7 +13,7 @@ */ import React from "react"; import { render, screen, fireEvent, cleanup, act } from "@testing-library/react"; -import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; +import { afterEach, afterAll, beforeEach, beforeAll, describe, expect, it, vi } from "vitest"; import { PurchaseSuccessModal } from "../PurchaseSuccessModal"; // ─── URL stub helper ─────────────────────────────────────────────────────────── @@ -35,6 +35,40 @@ async function waitForDialog() { await act(async () => { await new Promise((r) => setTimeout(r, 50)); }); } +// ─── Global mocks ───────────────────────────────────────────────────────────── + +let replaceStateMock: ReturnType; +let pushStateMock: ReturnType; + +beforeAll(() => { + replaceStateMock = vi.spyOn(window.history, "replaceState").mockImplementation( + (_s, _u, url) => { if (url) currentUrl = String(url); } + ); + pushStateMock = vi.spyOn(window.history, "pushState").mockImplementation( + (_s, _u, url) => { if (url) currentUrl = String(url); } + ); + // Mock window.location as a getter so `new URL(window.location.href)` always + // reads the live currentUrl value, not a snapshot made at setup time. + Object.defineProperty(window, "location", { + get() { return new URL(currentUrl); }, + configurable: true, + }); +}); + +afterAll(() => { + replaceStateMock?.mockRestore(); + pushStateMock?.mockRestore(); +}); + +beforeEach(() => { + currentUrl = "http://localhost/"; +}); + +afterEach(() => { + cleanup(); + vi.useRealTimers(); +}); + // ─── Tests ──────────────────────────────────────────────────────────────────── describe("PurchaseSuccessModal — render conditions", () => { @@ -233,4 +267,4 @@ describe("PurchaseSuccessModal — accessibility", () => { // Use getByRole which is more reliable than querySelector expect(screen.getByRole("button", { name: "Close" })).toBeTruthy(); }); -}); +}); \ No newline at end of file