authentik/web/e2e/fixtures/FormFixture.ts

import { PageFixture } from "#e2e/fixtures/PageFixture";
import type { LocatorContext } from "#e2e/selectors/types";

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

export class FormFixture extends PageFixture {
    static fixtureName = "Form";

    //#region Selector Methods

    //#endregion

    //#region Field Methods

    /**
     * Set the value of a text input.
     *
     * @param fieldName The name of the form element.
     * @param value the value to set.
     */
    public findTextualInput = async (
        fieldName: string | RegExp,
        context: LocatorContext = this.page,
    ) => {
        const textbox = context.getByRole("textbox", { name: fieldName });
        const searchbox = context.getByRole("searchbox", { name: fieldName });
        const spinbutton = context.getByRole("spinbutton", { name: fieldName });
        // Comboboxes (e.g. the Query Language input) wrap an inner textbox.
        const comboboxTextbox = context
            .getByRole("combobox", { name: fieldName })
            .getByRole("textbox");

        const control = textbox.or(searchbox).or(spinbutton).or(comboboxTextbox);

        await expect(control, `Field (${fieldName}) should be visible`).toBeVisible();

        return control;
    };

    /**
     * Set the value of a text input.
     *
     * @param target The name of the form element.
     * @param value the value to set.
     */
    public fill = async (
        target: string | RegExp | Locator,
        value: string,
        context: LocatorContext = this.page,
    ): Promise<void> => {
        let control: Locator;

        if (typeof target === "string" || target instanceof RegExp) {
            control = await this.findTextualInput(target, context);
        } else {
            control = target;
        }

        await control.fill(value);
    };

    /**
     * Search for a row containing the given text.
     *
     * @returns A locator for the row entry matching the query.
     */
    public search = async (
        query: string,
        context: LocatorContext = this.page,
    ): Promise<Locator> => {
        const searchInput = await this.findTextualInput(/search/i, context);
        // We have to wait for the user to appear in the table,
        // but several UI elements will be rendered asynchronously.
        // We attempt several times to find the user to avoid flakiness.

        const tries = 10;
        let found = false;

        for (let i = 0; i < tries; i++) {
            await this.fill(searchInput, query);
            await searchInput.press("Enter");

            const $rowEntry = context.getByRole("row", {
                name: query,
            });

            this.logger.info(`${i + 1}/${tries} Waiting for "${query}" to appear in the table`);

            found = await $rowEntry
                .waitFor({
                    timeout: 1500,
                })
                .then(() => true)
                .catch(() => false);

            if (found) {
                this.logger.info(`"${query}" found in the table`);
                return $rowEntry;
            }
        }

        throw new Error(`"${query}" not found in the table`);
    };

    /**
     * Set the value of a radio or checkbox input.
     *
     * @param fieldName The name of the form element.
     * @param value the value to set.
     */
    public setInputCheck = async (
        fieldName: string,
        value: boolean = true,
        parent: LocatorContext = this.page,
    ): Promise<void> => {
        const control = parent.locator("ak-switch-input", {
            hasText: fieldName,
        });

        await control.scrollIntoViewIfNeeded();

        await expect(control, `Field (${fieldName}) should be visible`).toBeVisible();

        const currentChecked = await control
            .getAttribute("checked")
            .then((value) => value !== null);

        if (currentChecked === value) {
            return;
        }

        await control.click();
    };

    /**
     * Set the value of a radio or checkbox input.
     *
     * @param fieldName The name of the form element.
     * @param pattern the value to set.
     */
    public setRadio = async (
        groupName: string,
        fieldName: string,
        parent: LocatorContext = this.page,
    ): Promise<void> => {
        const group = parent.getByRole("radiogroup", { name: groupName });

        await expect(group, `Field "${groupName}" should be visible`).toBeVisible();
        const control = parent.getByRole("radio", { name: fieldName });

        await control.setChecked(true);
    };

    /**
     * Set the value of a search select input.
     *
     * @param fieldLabel The name of the search select element.
     * @param pattern The text to match against the search select entry.
     */
    public selectSearchValue = async (
        fieldLabel: string,
        pattern: string | RegExp,
        parent: LocatorContext = this.page,
    ): Promise<void> => {
        const control = parent.getByRole("textbox", { name: fieldLabel });

        await expect(
            control,
            `Search select control (${fieldLabel}) should be visible`,
        ).toBeVisible();

        const fieldName = await control.getAttribute("name");

        if (!fieldName) {
            throw new Error(`Unable to find name attribute on search select (${fieldLabel})`);
        }

        // Find the search select input control and activate it.
        await control.click();

        if (typeof pattern === "string") {
            this.fill(control, pattern, parent);
        }

        const button = this.page
            // ---
            .locator(`div[data-managed-for*="${fieldName}"] button`, {
                hasText: pattern,
            });

        await expect(button, `Search select entry (${pattern}) should be visible`).toBeVisible();

        await button.click();
        await this.page.keyboard.press("Tab");
        await control.blur();
    };

    public setFormGroup = async (
        pattern: string | RegExp,
        value: boolean = true,
        parent: LocatorContext = this.page,
    ) => {
        const control = parent
            .locator("ak-form-group", {
                hasText: pattern,
            })
            .first();

        const currentOpen = await control.getAttribute("open").then((value) => value !== null);

        if (currentOpen === value) {
            this.logger.debug(`Form group ${pattern} is already ${value ? "open" : "closed"}`);
            return;
        }

        this.logger.debug(`Toggling form group ${pattern} to ${value ? "open" : "closed"}`);

        await control.click();

        if (value) {
            await expect(control).toHaveAttribute("open");
        } else {
            await expect(control).not.toHaveAttribute("open");
        }
    };

    //#endregion

    //#region Lifecycle

    constructor(page: Page, testName: string) {
        super({ page, testName });
    }

    //#endregion
}