Files
fa-agent/tests/e2e/formula-export.spec.ts
T
2026-09-26 01:12:10 +08:00

134 lines
5.7 KiB
TypeScript

import { expect, test } from "@playwright/test";
import fs from "node:fs";
const password =
process.env.E2E_PASSWORD ||
fs.readFileSync("var/local-access.txt", "utf8").match(/密码:(.+)/)![1];
const csv = (customerCode: string) =>
`期间,科室,学科,医生等级,医生,产品一级分类,实际收入,成本,毛利贡献,客户编码\n2025-08,测试儿保,普儿,高级,医生A,医疗服务,10000,4000,6000,${customerCode}\n2026-08,测试儿保,普儿,高级,医生A,医疗服务,5000,4000,3000,${customerCode}\n`;
test("新增列:补充信息触发追问并重新确认后生成公式工作簿", async ({ page }) => {
await page.setViewportSize({ width: 1280, height: 900 });
await page.goto("/");
await page.getByLabel("用户名", { exact: true }).fill("admin");
await page.getByLabel("密码", { exact: true }).fill(password);
await page.getByRole("button", { name: "登录", exact: true }).click();
await expect(page.getByRole("button", { name: "导入新表格" })).toBeVisible();
const uploaded = page.waitForResponse(
(response) =>
response.url().endsWith("/api/datasets") &&
response.request().method() === "POST",
);
await page.getByLabel("上传财务表格").setInputFiles({
name: `formula-e2e-${Date.now()}.csv`,
mimeType: "text/csv",
buffer: Buffer.from(csv(`E2E-${Date.now()}`)),
});
await uploaded;
await page
.getByRole("dialog", { name: "确认导入字段" })
.getByRole("button", { name: "确认并导入" })
.click();
await expect(
page.getByRole("heading", { name: "集团经营总览", exact: true }),
).toBeVisible({ timeout: 60000 });
let designCalls = 0;
let sentMessages: { role: string; content: string }[] = [];
await page.route("**/formula-design", async (route) => {
sentMessages = JSON.parse(route.request().postData()!).messages;
designCalls++;
await route.fulfill({
json:
designCalls === 1 || designCalls === 3
? {
complete: false,
questions: [
designCalls === 1
? "缺失值和零分母如何处理?"
: "请说明新增边界条件的计算口径。",
],
column: null,
}
: {
complete: true,
questions: [],
column: {
name: "收入成本差",
meaning: "实际收入减成本",
scope: "所选分组与期间",
missing: "结果不可用(#N/A);零分母不可用",
expression: {
operator: "subtract",
left: { metric: "revenue" },
right: { metric: "cost" },
},
},
},
});
});
await page.getByRole("button", { name: "生成公式工作簿" }).click();
const dialog = page.getByRole("dialog", { name: "生成公式工作簿" });
await dialog.getByRole("checkbox").check();
await dialog
.getByLabel("列计算描述")
.fill("新增收入成本差;请询问缺失值和零分母处理方式");
await dialog.getByRole("button", { name: "发送描述" }).click();
await expect(dialog).toContainText("缺失值和零分母如何处理?");
await dialog.getByLabel("列计算描述").fill("都显示不可用");
await dialog.getByRole("button", { name: "发送描述" }).click();
await expect(dialog.getByText("收入成本差", { exact: true })).toBeVisible();
await expect(dialog).toContainText("计算表达式:(实际收入 − 成本)");
await expect(dialog).toContainText("本轮描述 2/6");
await dialog.getByLabel("列计算描述").fill("增加一项特殊边界条件");
await dialog.getByRole("button", { name: "发送描述" }).click();
await expect(dialog).toContainText("请说明新增边界条件的计算口径。");
await expect(
dialog.getByRole("button", { name: "确认并生成公式工作簿" }),
).toHaveCount(0);
await dialog.getByLabel("列计算描述").fill("边界条件按收入减成本计算");
await dialog.getByRole("button", { name: "发送描述" }).click();
await expect(
dialog.getByRole("button", { name: "确认并生成公式工作簿" }),
).toBeVisible();
await expect(dialog).toContainText("本轮描述 4/6");
expect(sentMessages.map((message) => message.role)).toEqual([
"user",
"assistant",
"user",
"assistant",
"user",
"assistant",
"user",
]);
expect(sentMessages.at(-1)?.content).toContain("边界条件按收入减成本");
await dialog.getByRole("button", { name: "保存到定义库" }).click();
await expect(dialog.getByLabel("已保存定义")).toHaveValue(/.+/);
await dialog.getByRole("button", { name: "确认并生成公式工作簿" }).click();
await expect(dialog.getByRole("status")).toContainText("生成完成", {
timeout: 30000,
});
const download = page.waitForEvent("download");
await dialog.getByRole("button", { name: "下载公式工作簿" }).click();
expect((await download).suggestedFilename()).toBe(
"traceiq-formula-analysis.xlsx",
);
await dialog.getByLabel("关闭", { exact: true }).click();
await page.getByRole("button", { name: "生成公式工作簿" }).click();
const reopened = page.getByRole("dialog", { name: "生成公式工作簿" });
await reopened.getByLabel("已保存定义").selectOption({ label: "收入成本差" });
await expect(reopened).toContainText("计算表达式:(实际收入 − 成本)");
await reopened
.getByRole("button", { name: "确认并生成公式工作簿" })
.click();
await expect(reopened.getByRole("status")).toContainText("生成完成", {
timeout: 30000,
});
expect(designCalls).toBe(4);
});