midscene/packages/web-integration/tests/ai/web/puppeteer/utils.ts

import assert from 'node:assert';
import { PuppeteerWebPage } from '@/puppeteer';
import puppeteer, { type Viewport } from 'puppeteer';

export async function launchPage(
  url: string,
  opt?: {
    viewport?: Viewport;
    headless?: boolean;
  },
) {
  const browser = await puppeteer.launch({
    headless: typeof opt?.headless === 'boolean' ? opt.headless : true,
    args: [
      '--no-sandbox',
      '--disable-setuid-sandbox',
      '--disable-features=PasswordLeakDetection',
      '--disable-save-password-bubble',
    ],
  });
  const originPage = (await browser.pages())[0];
  const viewportConfig = {
    width: opt?.viewport?.width || 1920,
    height: opt?.viewport?.height || 1080,
    deviceScaleFactor: opt?.viewport?.deviceScaleFactor || 1,
  };
  await originPage.setViewport(viewportConfig);
  await originPage.setUserAgent(
    'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36',
  );
  let response;
  try {
    response = await originPage.goto(url);
    await originPage.waitForNetworkIdle({ concurrency: 1 });
  } catch (e) {
    // ignore navigation error
    if (response?.status) {
      assert(
        response.status() <= 399,
        `Page load failed: ${response.status()}`,
      );
    }
  }
  const page = new PuppeteerWebPage(originPage);

  return {
    page,
    originPage,
    reset: async () => {
      const pages = await browser.pages();
      await Promise.all(pages.map((page) => page.close()));
      await browser.close();
      await new Promise((resolve) => setTimeout(resolve, 1000));
    },
  };
}