homarr/e2e/assistant-management.spec.ts

import { chromium, expect } from "@playwright/test";
import { describe, test } from "vitest";

import { createHomarrContainer } from "./shared/create-homarr-container";
import { createSqliteDbFileAsync } from "./shared/e2e-db";
import { loginAsync } from "./shared/login";
import { seedAdminUserAsync } from "./shared/seed-admin-user";
import * as sqliteSchema from "../packages/db/schema/sqlite";

const adminCredentials = {
  username: "admin",
  password: "Comp(exP4sswOrd",
};

describe("Assistant management", () => {
  test("keeps provider actions clear and responsive", async () => {
    const { db, localMountPath } = await createSqliteDbFileAsync();
    await seedAdminUserAsync(db, adminCredentials);
    await db.insert(sqliteSchema.assistantConfigurations).values({
      id: "default",
      enabled: true,
      provider: "openrouter",
      baseUrl: "https://openrouter.ai/api/v1",
      modelDiscoveryPath: null,
      encryptedApiKey: "encrypted.key",
      modelId: "deepseek/deepseek-v4-flash",
    });

    const homarrContainer = await createHomarrContainer({
      environment: {
        AUTH_PROVIDERS: "credentials",
      },
      mounts: {
        "/appdata": localMountPath,
      },
    }).start();

    const baseUrl = `http://${homarrContainer.getHost()}:${homarrContainer.getMappedPort(7575)}`;
    const browser = await chromium.launch();
    const context = await browser.newContext({ viewport: { width: 1440, height: 1000 } });
    const page = await context.newPage();

    try {
      await loginAsync({ page, baseUrl, credentials: adminCredentials, destination: "/manage/assistant" });
      await expect(page.getByRole("heading", { name: "Assistant" })).toBeVisible();
      await expect(page.getByText("Connection ready")).toBeVisible();
      await expect(page.getByText("Encrypted API key saved")).toBeVisible();
      await expect(page.getByRole("button", { name: "Remove saved API key" })).toBeVisible();
      await page.getByRole("button", { name: "Additional request headers" }).click();
      await expect(page.getByRole("button", { name: "Add header" })).toBeVisible();

      await page.getByRole("button", { name: "Remove saved API key" }).click();
      await expect(page.getByText("Remove the saved API key?")).toBeVisible();
      await expect(page.getByRole("button", { name: "Remove API key" })).toBeVisible();
      await page.getByRole("button", { name: "Keep saved key" }).click();
      await expect(page.getByText("Remove the saved API key?")).toBeHidden();

      await page.setViewportSize({ width: 390, height: 844 });
      await page.reload();
      await expect(page.getByRole("heading", { name: "Assistant" })).toBeVisible();
      await expect(page.getByRole("button", { name: "Remove saved API key" })).toBeVisible();
      await expect(page.getByRole("button", { name: "Save assistant" })).toBeVisible();
      const hasHorizontalOverflow = await page.evaluate(
        () => document.documentElement.scrollWidth > document.documentElement.clientWidth,
      );
      expect(hasHorizontalOverflow).toBe(false);
    } finally {
      await browser.close();
      await homarrContainer.stop();
    }
  }, 120_000);
});