ohif-viewer/tests/pages/DataOverlayPageObject.ts

import { Page } from '@playwright/test';

export class DataOverlayPageObject {
  readonly page: Page;
  readonly viewportId: string;

  constructor(page: Page, viewportId: string) {
    this.page = page;
    this.viewportId = viewportId;
  }

  get menu() {
    return this.page.getByTestId(`viewport-data-overlay-menu-${this.viewportId}`);
  }

  async toggle() {
    await this.page.getByTestId(`dataOverlayMenu-${this.viewportId}-btn`).click();
  }

  async addSegmentation(segmentation: string) {
    await this.openAddSegmentationDropdown();
    await this.page.getByTestId(segmentation).click();
  }

  async changeSegmentation(from: string, to: string) {
    await this.page.getByTestId(`overlay-ds-select-value-${from.toUpperCase()}`).click();
    await this.page.getByTestId(`${to}-SEG`).click();
  }

  async remove(segmentation: string) {
    await this.page.getByTestId(`overlay-ds-more-button-${segmentation}`).click();
    await this.page.getByTestId(`overlay-ds-remove-button-${segmentation}`).click();
  }

  async openAddSegmentationDropdown() {
    await this.page.getByTestId(`AddSegmentationDataOverlay-${this.viewportId}`).click();
    await this.clickSelectSegmentation();
  }

  async clickSelectSegmentation() {
    await this.page.getByText('SELECT A SEGMENTATION').click();
  }

  async getDropdownOptionLabels(): Promise<string[]> {
    const options = this.page.getByRole('option');
    const labels = (await options.allTextContents()).map(text => text.trim()).filter(Boolean);
    return labels;
  }

  getOverlaySegmentationRow(label: string) {
    return this.menu.getByTestId(`overlay-ds-select-value-${label.toUpperCase()}`);
  }

  get overlaySegmentationRows() {
    return this.menu.locator('[data-cy^="overlay-ds-select-value-"]');
  }

  async changeBackgroundDisplaySet(backgroundLabel: string) {
    await this.page.getByTestId(`overlay-background-ds-select-${this.viewportId}`).click();
    await this.page.getByTestId(backgroundLabel).click();
  }
}