mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-01 09:29:38 +02:00
49 lines
2.4 KiB
TypeScript
49 lines
2.4 KiB
TypeScript
import { expect, expectCanvasNonBlank, expectHeading, gotoWorkbench, test } from "./fixtures";
|
|
|
|
test("chat streams, cancels, regenerates, copies, and clears messages", async ({ page }) => {
|
|
await gotoWorkbench(page, "/chat", { chat: { delayFrames: 6 } });
|
|
await expectHeading(page, "Chat");
|
|
|
|
await page.getByPlaceholder(/Ask using Graph RAG/).fill("cancel this request");
|
|
await page.getByLabel("Send message").click();
|
|
await expect(page.getByLabel("Cancel request")).toBeVisible();
|
|
await page.getByLabel("Cancel request").click();
|
|
await expect(page.getByText("(Cancelled)")).toBeVisible();
|
|
|
|
await page.getByLabel("Clear conversation").click();
|
|
await expect(page.getByText("Start a conversation with TrustGraph.")).toBeVisible();
|
|
|
|
await page.getByPlaceholder(/Ask using Graph RAG/).fill("Who does Alice know?");
|
|
await page.getByLabel("Send message").click();
|
|
await expect(page.getByText(/Mock graph answer from QA knowledge graph/)).toBeVisible();
|
|
await expect(page.getByText("qa-model")).toBeVisible();
|
|
|
|
await page.getByText(/Mock graph answer/).hover();
|
|
await page.getByLabel("Copy message").click();
|
|
await expect.poll(() => page.evaluate(() => (window as Window & { __TRUSTGRAPH_WORKBENCH_CLIPBOARD__?: string }).__TRUSTGRAPH_WORKBENCH_CLIPBOARD__)).toContain("Mock graph answer");
|
|
|
|
await page.getByLabel("Regenerate response").click();
|
|
await expect(page.getByText(/Mock graph answer from QA knowledge graph/).last()).toBeVisible();
|
|
|
|
await page.getByText("Agent").click();
|
|
await page.getByPlaceholder(/Ask using Agent/).fill("Use the agent");
|
|
await page.getByLabel("Send message").click();
|
|
await expect(page.getByText("Thinking")).toBeVisible();
|
|
await expect(page.getByText("Observing")).toBeVisible();
|
|
await expect(page.getByText("Mock agent answer from TrustGraph.")).toBeVisible();
|
|
});
|
|
|
|
test("graph renders seeded triples and responds to controls", async ({ page }) => {
|
|
await gotoWorkbench(page, "/graph");
|
|
await expectHeading(page, "Graph");
|
|
await expect(page.getByText(/nodes/).first()).toBeVisible();
|
|
await expect(page.getByText(/edges/).first()).toBeVisible();
|
|
|
|
await expectCanvasNonBlank(page.locator("canvas").first());
|
|
|
|
await page.getByPlaceholder("Search graph...").fill("Alice");
|
|
await expect(page.getByText(/1 nodes/)).toBeVisible();
|
|
await page.getByRole("button", { name: "Labels" }).click();
|
|
await page.getByLabel("Node limit").selectOption("100");
|
|
await expect(page.getByText(/nodes/).first()).toBeVisible();
|
|
});
|