94 lines
7.9 KiB
TypeScript
94 lines
7.9 KiB
TypeScript
import { responsesFixture } from "./support/responses-fixture";
|
|
import { expect, test } from "@playwright/test";
|
|
import { mockAgentApi } from "./support/mock-agent-api";
|
|
import { mockScadaApi } from "./support/mock-scada-api";
|
|
import { mockWorkbenchContext } from "./support/mock-workbench-context";
|
|
|
|
test("unified chart and map commands send success only after rendering, and final answer replaces process text", async ({ page }, testInfo) => {
|
|
test.setTimeout(60000);
|
|
await mockWorkbenchContext(page);
|
|
await page.route("**/runtime-config.js", (route) => route.fulfill({ contentType: "application/javascript", body: 'globalThis.__TJWATER_CONFIG__ = { TJWATER_AUTH_MODE:"disabled", TJWATER_SERVER_API_BASE_URL:"https://tjwater-api.invalid", TJWATER_AGENT_URL:"http://127.0.0.1:8787", TJWATER_MAP_URL:"https://workbench-map.invalid/geoserver", TJWATER_GEOSERVER_WORKSPACE:"tjwater_next", TJWATER_WORKBENCH_RUNTIME_MODE:"auto", TJWATER_ENABLE_MSW:"false", TJWATER_ENABLE_DEV_PANEL:"true" };' }));
|
|
await mockScadaApi(page, [{ device_id: "D1", device_type: "pressure", node_id: "J1", link_id: null, api_query_id: null, measurement_unit: "m", transmission_mode: "test", transmission_frequency: "1min", reliability: 1, x: null, y: null, longitude: 121.48, latitude: 31.1 }]);
|
|
await mockAgentApi(page);
|
|
const receipts: Array<{ actionId: string; status: string; output: unknown }> = [];
|
|
const chart = { title: "监测压力", chart_type: "line", x_data: ["t1", "t2"], series: [{ name: "D1 scada_raw", data: [20, 21] }] };
|
|
await page.route("**/api/v1/frontend-action-registry", (route) => route.fulfill({ headers: { "Access-Control-Allow-Origin": "*" }, json: { schema_version: "frontend-action-registry@2", actions: ["open_chart", "close_chart", "highlight_features", "clear_highlights", "zoom_to_map", "locate_features"].map((id) => ({ id, version: "frontend-action@2" })) } }));
|
|
await page.route("**/api/v1/frontend-actions/*/result", async (route) => {
|
|
if (route.request().method() === "OPTIONS") { await route.fulfill({ status: 204, headers: { "Access-Control-Allow-Origin": "*", "Access-Control-Allow-Headers": "*", "Access-Control-Allow-Methods": "POST, OPTIONS" } }); return; }
|
|
receipts.push(route.request().postDataJSON());
|
|
await route.fulfill({ headers: { "Access-Control-Allow-Origin": "*" }, json: { accepted: true } });
|
|
});
|
|
let responseNumber = 0;
|
|
await page.route("**/api/v1/responses", async (route) => {
|
|
responseNumber += 1;
|
|
const now = Date.now();
|
|
const action = (command: Record<string, unknown>, actionId: string) => ({ version: "frontend-action@2", actionId, toolCallId: actionId, sessionId: "migrated-session", command, issuedAt: now, expiresAt: now + 15000 });
|
|
const commands = responseNumber === 2
|
|
? [{ action: "clear_highlights" }, { action: "close_chart", chart_id: "chat-pressure" }]
|
|
: [{ action: "highlight_features", layer: "scada", feature_ids: ["D1"], effect: "pulse" },
|
|
{ action: "open_chart", chart_id: "chat-pressure", chart, evidence_refs: [] }];
|
|
const chunks = [
|
|
{ type: "start", messageId: "assistant-migrated" },
|
|
{ type: "data-progress", data: { session_id: "migrated-session", id: "business-data", phase: "data", status: "completed", title: "获取业务数据" } },
|
|
{ type: "text-start", id: "text" }, { type: "text-delta", id: "text", delta: "正在整理过程说明" }, { type: "text-end", id: "text" },
|
|
...commands.map((command, index) => ({ type: "data-frontend_action", data: action(command, `command-${responseNumber}-${index}`) })),
|
|
...(responseNumber === 1 ? [{ type: "data-frontend_action", data: action({ action: "locate_features", layer: "scada", feature_ids: ["D1", "missing"] }, "locate-action") }] : []),
|
|
...(responseNumber === 1 ? [{ type: "data-frontend_action", data: action({ action: "zoom_to_map", center: [121.48, 31.1], zoom: 14.5 }, "zoom-action") }] : []),
|
|
{ type: "data-final_answer", data: { session_id: "migrated-session", content: "最终答复：压力曲线已核实。" } },
|
|
{ type: "finish", finishReason: "stop" },
|
|
];
|
|
await route.fulfill({ headers: { "Access-Control-Allow-Origin": "*", "Content-Type": "text/event-stream", "Cache-Control": "no-cache" }, body: responsesFixture(chunks) });
|
|
});
|
|
await page.goto("/", { waitUntil: "domcontentloaded" });
|
|
await page.getByRole("button", { name: /展开 Agent 助手面板/ }).click();
|
|
const prompt = page.getByRole("textbox", { name: "询问供水问题，或描述需要处理的任务" }).filter({ visible: true }).first();
|
|
await page.waitForFunction(() => Boolean(window.__waterNetworkMap?.getLayer("scada-pressure-hit")));
|
|
await prompt.fill("查询历史并展示压力曲线");
|
|
await page.getByRole("button", { name: "发送 Agent 指令" }).filter({ visible: true }).first().click();
|
|
await expect(page.getByText("最终答复：压力曲线已核实。", { exact: true })).toBeVisible();
|
|
await expect(page.getByText("正在整理过程说明", { exact: true })).toBeHidden();
|
|
const display = page.getByLabel("对话展示", { exact: true });
|
|
await expect(display.locator('[data-investigation-chart="chat-pressure"] svg[role="img"]')).toBeVisible();
|
|
await expect.poll(() => receipts.length).toBe(4);
|
|
|
|
expect(receipts.find((receipt) => receipt.actionId === "locate-action")?.output).toEqual({ located_ids: ["D1"], missing_ids: ["missing"], fitted: true });
|
|
expect(receipts.every((receipt) => receipt.status === "succeeded")).toBe(true);
|
|
|
|
const sourceSize = (source: string) => page.evaluate(async (source) => {
|
|
const data = await (window.__waterNetworkMap?.getSource(source) as {
|
|
getData: () => Promise<{ features: unknown[] }>
|
|
} | undefined)?.getData();
|
|
return data?.features.length ?? 0;
|
|
}, source);
|
|
await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(1);
|
|
await page.getByRole("button", { name: "切换地图 Dev Panel" }).click();
|
|
const tools = page.getByTestId("investigation-dev-tools");
|
|
await tools.getByLabel("调查展示 JSONL", { exact: true }).fill([
|
|
{ record_type: "presentation", command: { action: "highlight_features", layer: "scada", feature_ids: ["D1"], effect: "pulse" } },
|
|
{ record_type: "presentation", command: { action: "open_chart", chart_id: "dev-pressure", chart, evidence_refs: [] } },
|
|
].map((record) => JSON.stringify(record)).join("\n"));
|
|
await tools.getByRole("button", { name: "执行本地预览", exact: true }).click();
|
|
await expect(page.getByLabel("本地预览展示", { exact: true }).locator('[data-investigation-chart="dev-pressure"]')).toBeVisible();
|
|
await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
|
|
await page.screenshot({ path: testInfo.outputPath("unified-chat-dev.png") });
|
|
await page.getByRole("button", { name: "切换地图 Dev Panel" }).click();
|
|
const send = async (text: string) => {
|
|
await prompt.fill(text);
|
|
await page.getByRole("button", { name: "发送 Agent 指令" }).filter({ visible: true }).first().click();
|
|
};
|
|
await send("清除本轮对话展示");
|
|
await expect.poll(() => receipts.length).toBe(6);
|
|
await expect(display.locator('[data-investigation-chart="chat-pressure"]')).toHaveCount(0);
|
|
await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(0);
|
|
await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
|
|
await send("重新展示压力曲线");
|
|
await expect.poll(() => receipts.length).toBe(8);
|
|
await expect(display.locator('[data-investigation-chart="chat-pressure"]')).toBeVisible();
|
|
await page.getByRole("button", { name: "新建 Agent 对话", exact: true }).click();
|
|
await expect(display).toContainText("当前没有展示结果");
|
|
await expect(display.getByRole("combobox", { name: "切换展示动作" })).toBeVisible();
|
|
await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(0);
|
|
await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
|
|
await expect(page.locator('[data-investigation-chart="dev-pressure"]')).toBeVisible();
|
|
});
|