import { test as base } from 'playwright-test-coverage';
import {
  DOMOverlayPageObject,
  MainToolbarPageObject,
  LeftPanelPageObject,
  RightPanelPageObject,
  ViewportPageObject,
  NotFoundStudyPageObject,
} from '../pages';

type PageObjects = {
  DOMOverlayPageObject: DOMOverlayPageObject;
  mainToolbarPageObject: MainToolbarPageObject;
  leftPanelPageObject: LeftPanelPageObject;
  rightPanelPageObject: RightPanelPageObject;
  viewportPageObject: ViewportPageObject;
  notFoundStudyPageObject: NotFoundStudyPageObject;
};

export const test = base.extend<PageObjects>({
  DOMOverlayPageObject: async ({ page }, use) => {
    await use(new DOMOverlayPageObject(page));
  },
  mainToolbarPageObject: async ({ page }, use) => {
    await use(new MainToolbarPageObject(page));
  },
  leftPanelPageObject: async ({ page }, use) => {
    await use(new LeftPanelPageObject(page));
  },
  rightPanelPageObject: async ({ page }, use) => {
    await use(new RightPanelPageObject(page));
  },
  viewportPageObject: async ({ page }, use) => {
    await use(new ViewportPageObject(page));
  },
  notFoundStudyPageObject: async ({ page }, use) => {
    await use(new NotFoundStudyPageObject(page));
  },
});

export { expect } from 'playwright-test-coverage';