import { test as base } from "@playwright/experimental-ct-react";

export type TestOptions = {
  mockID: number | undefined;
};

export const test: typeof base = base.extend<TestOptions>({
  browser: async ({ browser }, use) => {
    const oldFunc = browser.newContext;
    browser.newContext = async (options?) => {
      const ret = await oldFunc.call(browser, options);

      await ret.addInitScript({
        content: `window.__TEST_OPTIONS = window.__TEST_OPTIONS || {};`,
      });

      return ret;
    };
    await use(browser);
  },
  context: async ({ context }, use) => {
    await context.addInitScript({
      content: `window.__TEST_OPTIONS = window.__TEST_OPTIONS || {};`,
    });
    await use(context);
  },
});

export { expect } from "@playwright/experimental-ct-react";