twenty/packages/twenty-e2e-testing/lib/pom/helper/uploadImage.ts

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

export class UploadImage {
  private readonly imagePreview: Locator;
  private readonly uploadButton: Locator;
  private readonly removeButton: Locator;

  constructor(public readonly page: Page) {
    this.imagePreview = page.locator('.css-6eut39'); //TODO: add attribute to make it independent of theme
    this.uploadButton = page.getByRole('button', { name: 'Upload' });
    this.removeButton = page.getByRole('button', { name: 'Remove' });
  }

  async clickImagePreview() {
    await this.imagePreview.click();
  }

  async clickUploadButton() {
    await this.uploadButton.click();
  }

  async clickRemoveButton() {
    await this.removeButton.click();
  }
}