File size: 4,481 Bytes
a76eef5 893b0be a76eef5 893b0be a76eef5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import { expect, test } from "@playwright/test";
import { TEST_IDS } from "../src/lib/constants.js";
const HF_TOKEN = "hf_aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
const HF_TOKEN_STORAGE_KEY = "hf_token";
const STORAGE_STATE_FILE = "e2e/home_test_storage_state.json";
test("home page has expected token model", async ({ page }) => {
await page.goto("/");
await expect(page.getByText("Add a Hugging Face Token")).toBeVisible();
});
// Group tests that depend on sequential execution and shared state
test.describe.serial("Token Handling and Subsequent Tests", () => {
// Test that sets the token and saves state
test("filling up token input, closes modal, and saves state", async ({ page }) => {
await page.goto("/");
await expect(page.getByText("Add a Hugging Face Token")).toBeVisible();
const input = page.getByPlaceholder("Enter HF Token");
await expect(input).toBeVisible();
await input.fill(HF_TOKEN);
await input.blur();
await page.getByText("Submit").click();
await expect(page.getByText("Add a Hugging Face Token")).not.toBeVisible();
// Save storage state
await page.context().storageState({ path: STORAGE_STATE_FILE });
});
// Nested describe for tests that use the saved state
test.describe("Tests requiring persisted token", () => {
test.use({ storageState: STORAGE_STATE_FILE });
test("can create a conversation with persisted token", async ({ page }) => {
await page.goto("/");
// Expect modal NOT to be visible due to persisted token
await expect(page.getByText("Add a Hugging Face Token")).not.toBeVisible();
// Verify token is in localStorage
const storedToken = await page.evaluate(
key => JSON.parse(window.localStorage.getItem(key) ?? ""),
HF_TOKEN_STORAGE_KEY
);
expect(storedToken).toBe(HF_TOKEN);
const userInput = page.getByRole("textbox", { name: "Enter user message" });
await expect(userInput).toBeVisible();
await userInput.fill("Hello Hugging Face!");
await userInput.blur();
expect(await userInput.inputValue()).toBe("Hello Hugging Face!");
// Reload the page
await page.reload();
// Re-select the input field and check its value
const userInputAfterReload = page.getByRole("textbox", { name: "Enter user message" });
await expect(userInputAfterReload).toBeVisible();
expect(await userInputAfterReload.inputValue()).toBe("Hello Hugging Face!");
});
test("checkpoints, resetting, and restoring", async ({ page }) => {
await page.goto("/");
const userMsg = "user message: hi";
const assistantMsg = "assistant message: hey";
// Fill user message
await page.getByRole("textbox", { name: "Enter user message" }).click();
await page.getByRole("textbox", { name: "Enter user message" }).fill(userMsg);
// Blur
await page.locator(".relative > div:nth-child(2) > div").first().click();
// Fill assistant message
await page.getByRole("button", { name: "Add message" }).click();
await page.getByRole("textbox", { name: "Enter assistant message" }).fill(assistantMsg);
// Blur
await page.locator(".relative > div:nth-child(2) > div").first().click();
// Create Checkpoint
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_trigger}"]`).click();
await page.getByRole("button", { name: "Create new" }).click();
// Check that there are checkpoints
await expect(page.locator(`[data-test-id="${TEST_IDS.checkpoint}"] `)).toBeVisible();
// Get out of menu
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_menu}"] `).press("Escape");
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_menu}"] `).press("Escape");
// Reset
await page.locator(`[data-test-id="${TEST_IDS.reset}"]`).click();
// Check that messages are gone now
await expect(page.getByRole("textbox", { name: "Enter user message" })).toHaveValue("");
// Call in a checkpoint
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_trigger}"]`).click();
await page.locator(`[data-test-id="${TEST_IDS.checkpoint}"] `).click();
// Get out of menu
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_menu}"] `).press("Escape");
await page.locator(`[data-test-id="${TEST_IDS.checkpoints_menu}"] `).press("Escape");
// Check that the messages are back
await expect(page.getByRole("textbox", { name: "Enter user message" })).toHaveValue(userMsg);
await expect(page.getByRole("textbox", { name: "Enter assistant message" })).toHaveValue(assistantMsg);
});
});
});
|