bitwarden/browser
1
0
Fork 0
mirror of https://github.com/bitwarden/browser synced 2026-02-13 23:13:36 +00:00
Code Issues Releases Wiki Activity

Include fetch override in auth fixtures

Browse Source
This commit is contained in:
Matt Gibson
2025-11-10 18:21:51 -08:00
parent 06e6600f6a
commit e0d57086a8
2 changed files with 26 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
libs/playwright-helpers/src/fixtures/auth.fixture.ts
View File

@@ -9,6 +9,8 @@ type BrowserName = "chromium" | "firefox" | "webkit";
import { Play, SingleUserScene, SingleUserSceneTemplate } from "@bitwarden/playwright-helpers";
import { addInitScriptForPlayId } from "./page-extension";
const hostname = new URL(webServerBaseUrl).hostname;
const dataDir = process.env.PLAYWRIGHT_DATA_DIR ?? "playwright-data";
// Ensure data directory exists
@@ -79,6 +81,7 @@ export class AuthFixture {
}
const context = await this._browser.newContext();
this._page = await context.newPage();
await addInitScriptForPlayId(this._page, process.env.PLAY_ID!);
}
return this._page;
}

45
libs/playwright-helpers/src/fixtures/page-extension.ts
View File

@@ -2,28 +2,29 @@ import { Page, TestFixture } from "@playwright/test";
export function pageExtension(): TestFixture<Page, { page: Page; playId: string }> {
return async ({ page, playId }, use) => {
await page.addInitScript(
({ p }) => {
const originalFetch = window.fetch;
window.fetch = async function (
input: string | URL | globalThis.Request,
init?: RequestInit,
) {
// Build a Request that takes into account both the input and any provided init overrides
const baseRequest =
input instanceof globalThis.Request
? init
? new Request(input, init)
: input
: new Request(input, init);
baseRequest.headers.set("x-play-id", p);
return originalFetch(baseRequest);
};
},
{ p: playId },
);
await addInitScriptForPlayId(page, playId);
await use(page);
};
}
export function addInitScriptForPlayId(page: Page, playId: string): Promise<void> {
return page.addInitScript(
({ p }) => {
const originalFetch = window.fetch;
window.fetch = async function (input: string | URL | globalThis.Request, init?: RequestInit) {
// Build a Request that takes into account both the input and any provided init overrides
const baseRequest =
input instanceof globalThis.Request
? init
? new Request(input, init)
: input
: new Request(input, init);
baseRequest.headers.set("x-play-id", p);
return originalFetch(baseRequest);
};
},
{ p: playId },
);
}