import { expect } from "@playwright/test";
import { CellPageObject } from ".";
import BasePage from "../../../Base";

export class RatingCellPageObject extends BasePage {
  readonly cell: CellPageObject;

  constructor(cell: CellPageObject) {
    super(cell.rootPage);
    this.cell = cell;
  }

  get({index, columnHeader}: {index?: number, columnHeader: string}) {
    return this.cell.get({index, columnHeader});
  }

  async verify({index, columnHeader, rating}: {index?: number, columnHeader: string, rating: number}) {
    await expect(await this.get({index, columnHeader}).locator(`div[role="radio"][aria-checked="true"]`)).toHaveCount(rating);
  }
  
}