ai-vector-search.page.ts

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

/**
 * AI Vector Search Page Object
 *
 * Encapsulates locators and interactions for the Vector Search page at /ai/vector-search.
 * Handles both enabled state (search UI with scored results) and disabled state (info banner).
 */
export class AiVectorSearchPage {
  readonly page: Page;
  readonly url = '/ai/vector-search';

  // ── Disabled state ────────────────────────────────────────────────────────
  readonly disabledBanner: Locator;

  // ── Enabled state — search UI ─────────────────────────────────────────────
  readonly searchInput: Locator;
  readonly clearButton: Locator;
  readonly resultsTable: Locator;
  readonly resultRows: Locator;
  readonly scoreBadges: Locator;
  readonly resultCount: Locator;
  readonly loadingRow: Locator;
  readonly errorRow: Locator;
  readonly emptyState: Locator;

  constructor(page: Page) {
    this.page = page;

    this.disabledBanner = page.locator('.ai-disabled-banner, .disabled-card').first();

    this.searchInput = page.locator('input[placeholder*="position"], input[placeholder*="describe"]').first();
    this.clearButton = page.locator('button').filter({ hasText: /clear/i }).first();
    this.resultsTable = page.locator('table.results-table, mat-table').first();
    this.resultRows = page.locator('table.results-table tr:not(thead tr), mat-row');
    this.scoreBadges = page.locator('.score-badge');
    this.resultCount = page.locator('.result-count');
    this.loadingRow = page.locator('.loading-row');
    this.errorRow = page.locator('.error-row');
    this.emptyState = page.locator('.empty-state');
  }

  async goto() {
    await this.page.goto(this.url);
    await this.page.waitForLoadState('networkidle');
  }

  async isDisabled(): Promise<boolean> {
    return await this.disabledBanner.isVisible({ timeout: 3000 }).catch(() => false);
  }

  async search(query: string) {
    await this.searchInput.fill(query);
    // Wait for debounce (600ms) + network
    await this.page.waitForTimeout(1500);
  }

  async clear() {
    await this.clearButton.click();
    await this.page.waitForTimeout(500);
  }

  async getResultCount(): Promise<number> {
    return await this.resultRows.count();
  }

  async getScoreBadgeCount(): Promise<number> {
    return await this.scoreBadges.count();
  }
}