@duetds/date-picker/hydrate/tests/node_modules/expect-puppeteer/lib/matchers/setupPage.js

Duet Date Picker is an open source version of Duet Design System’s accessible date picker.

"use strict";

exports.__esModule = true;
exports.setupPage = void 0;

function waitForFrame(page) {
  let fulfill;
  const promise = new Promise(resolve => {
    fulfill = resolve;
  });

  function checkFrame() {
    const frame = page.frames().find(f => f.parentFrame() !== null);
    if (frame) fulfill(frame);else page.once(`frameattached`, checkFrame);
  }

  checkFrame();
  return promise;
}

const setupPage = (pageType, cb) => {
  let currentPage = page;
  beforeEach(async () => {
    if (pageType === `Page`) {
      cb({
        currentPage
      });
      return;
    }

    await page.goto(`http://localhost:${process.env.TEST_SERVER_PORT}/frame.html`);
    currentPage = await waitForFrame(page);
    cb({
      currentPage
    });
  });
};

exports.setupPage = setupPage;