diff --git a/src/browser/pw-session.create-page.navigation-guard.test.ts b/src/browser/pw-session.create-page.navigation-guard.test.ts index 088cbeaa721..fc3f249b952 100644 --- a/src/browser/pw-session.create-page.navigation-guard.test.ts +++ b/src/browser/pw-session.create-page.navigation-guard.test.ts @@ -1,19 +1,7 @@ import { afterEach, describe, expect, it, vi } from "vitest"; import { InvalidBrowserNavigationUrlError } from "./navigation-guard.js"; import { closePlaywrightBrowserConnection, createPageViaPlaywright } from "./pw-session.js"; - -const connectOverCdpMock = vi.fn(); -const getChromeWebSocketUrlMock = vi.fn(); - -vi.mock("playwright-core", () => ({ - chromium: { - connectOverCDP: (...args: unknown[]) => connectOverCdpMock(...args), - }, -})); - -vi.mock("./chrome.js", () => ({ - getChromeWebSocketUrl: (...args: unknown[]) => getChromeWebSocketUrlMock(...args), -})); +import { connectOverCdpMock, getChromeWebSocketUrlMock } from "./pw-session.mock-setup.js"; function installBrowserMocks() { const pageOn = vi.fn(); diff --git a/src/browser/pw-session.get-page-for-targetid.extension-fallback.test.ts b/src/browser/pw-session.get-page-for-targetid.extension-fallback.test.ts index bfb429ba45e..08edc7dd171 100644 --- a/src/browser/pw-session.get-page-for-targetid.extension-fallback.test.ts +++ b/src/browser/pw-session.get-page-for-targetid.extension-fallback.test.ts @@ -1,18 +1,6 @@ import { describe, expect, it, vi } from "vitest"; import { closePlaywrightBrowserConnection, getPageForTargetId } from "./pw-session.js"; - -const connectOverCdpMock = vi.fn(); -const getChromeWebSocketUrlMock = vi.fn(); - -vi.mock("playwright-core", () => ({ - chromium: { - connectOverCDP: (...args: unknown[]) => connectOverCdpMock(...args), - }, -})); - -vi.mock("./chrome.js", () => ({ - getChromeWebSocketUrl: (...args: unknown[]) => getChromeWebSocketUrlMock(...args), -})); +import { connectOverCdpMock, getChromeWebSocketUrlMock } from "./pw-session.mock-setup.js"; describe("pw-session getPageForTargetId", () => { it("falls back to the only page when CDP session attachment is blocked (extension relays)", async () => { diff --git a/src/browser/pw-session.mock-setup.ts b/src/browser/pw-session.mock-setup.ts new file mode 100644 index 00000000000..e62d51c9d14 --- /dev/null +++ b/src/browser/pw-session.mock-setup.ts @@ -0,0 +1,14 @@ +import { vi } from "vitest"; + +export const connectOverCdpMock = vi.fn(); +export const getChromeWebSocketUrlMock = vi.fn(); + +vi.mock("playwright-core", () => ({ + chromium: { + connectOverCDP: (...args: unknown[]) => connectOverCdpMock(...args), + }, +})); + +vi.mock("./chrome.js", () => ({ + getChromeWebSocketUrl: (...args: unknown[]) => getChromeWebSocketUrlMock(...args), +}));