完善AMESim组件界面与仿真求解稳定性

2026-08-02 00:57:48 +08:00
1 parent e7177ab03e
commit 410ef535e8
@@ -0,0 +1,156 @@
+import { expect, test } from "@playwright/test";
+
+type SymbolCase = {
+  category: string;
+  domain: "medium" | "pneumatic" | "signal" | "mechanical";
+  modelType: string;
+};
+
+const symbolCases: SymbolCase[] = [
+  { category: "media", domain: "medium", modelType: "amesim_ideal_air_medium" },
+  { category: "media", domain: "medium", modelType: "amesim_helium_medium" },
+  { category: "storage", domain: "pneumatic", modelType: "amesim_pnch023" },
+  { category: "storage", domain: "pneumatic", modelType: "amesim_pnch012" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnor001" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnvo001_fixed" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnvo001" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnl00r" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnl0001" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnl0002" },
+  { category: "flow", domain: "pneumatic", modelType: "amesim_pnl0003" },
+  { category: "junctions", domain: "pneumatic", modelType: "amesim_pn3node2" },
+  { category: "junctions", domain: "pneumatic", modelType: "amesim_p4node2" },
+  { category: "boundary", domain: "pneumatic", modelType: "amesim_pnpl01" },
+  { category: "signals", domain: "signal", modelType: "amesim_step0" },
+  { category: "signals", domain: "signal", modelType: "amesim_ud00" },
+  { category: "mechanical", domain: "mechanical", modelType: "amesim_f000" },
+  { category: "mechanical", domain: "mechanical", modelType: "amesim_forc" },
+  { category: "mechanical", domain: "mechanical", modelType: "amesim_mecmas21" },
+  { category: "mechanical", domain: "mechanical", modelType: "amesim_lstp00a" },
+  { category: "mechanical", domain: "mechanical", modelType: "amesim_lmechn1" },
+];
+
+const categoryLabels: Record<string, string> = {
+  boundary: "边界元件",
+  flow: "流动元件",
+  junctions: "连接元件",
+  mechanical: "机械元件",
+  media: "介质物性",
+  signals: "信号元件",
+  storage: "储能元件",
+};
+
+test.beforeEach(async ({ page }) => {
+  await page.addInitScript(() => {
+    const preparedKey = "system-simulation-flow:component-symbols-prepared";
+    if (window.sessionStorage.getItem(preparedKey) !== "1") {
+      window.localStorage.clear();
+      window.sessionStorage.clear();
+      window.sessionStorage.setItem(preparedKey, "1");
+    }
+    window.location.hash = "#/modeling";
+  });
+  await page.route("**/api/components/catalog", async (route) => {
+    await route.fulfill({
+      status: 200,
+      contentType: "application/json",
+      body: JSON.stringify({
+        schemaVersion: 1,
+        libraries: [
+          {
+            id: "amesim",
+            label: "AMESim 组件库",
+            version: "0.2.0",
+            sourcePackage: "tests.e2e.amesim-symbols",
+            temporary: true,
+            order: 1,
+            components: symbolCases.map((item, index) => ({
+              type: item.modelType,
+              label: item.modelType,
+              modelType: item.modelType,
+              modelVersion: "0.1.0",
+              symbol: item.modelType,
+              order: index + 1,
+              category: {
+                id: item.category,
+                label: categoryLabels[item.category],
+                order: Object.keys(categoryLabels).indexOf(item.category) + 1,
+              },
+              ports: [],
+              parameters: [],
+            })),
+          },
+        ],
+      }),
+    });
+  });
+});
+
+test("全部已移植 AMESim 模型使用专属图标及正确领域配色", async ({ page }) => {
+  await page.goto("/");
+
+  const palette = page.locator("aside.palette");
+  await expect(palette.locator('[data-component-symbol^="amesim_"]')).toHaveCount(
+    symbolCases.length,
+  );
+  await expect(palette.locator('[data-component-symbol="generic"]')).toHaveCount(0);
+
+  for (const item of symbolCases) {
+    const symbol = palette.locator(`[data-component-symbol="${item.modelType}"]`);
+    await expect(symbol).toHaveCount(1);
+    await expect(symbol.locator("svg")).toHaveAttribute("data-symbol-domain", item.domain);
+    await expect(symbol).toHaveAttribute(
+      "data-symbol-presentation",
+      item.domain === "medium" ? "bare" : "framed",
+    );
+  }
+});
+
+test("旧工程保存的通用图标会按当前目录定义升级", async ({ page }) => {
+  await page.goto("/");
+  await page.evaluate(() => {
+    window.localStorage.setItem(
+      "system-simulation-flow:autosave",
+      JSON.stringify({
+        name: "legacy-symbol-project",
+        nodes: [
+          {
+            id: "amesim_pnl0001_1",
+            type: "simulationComponent",
+            position: { x: 200, y: 160 },
+            data: {
+              label: "amesim_pnl0001_1",
+              componentType: "amesim_pnl0001",
+              modelType: "amesim_pnl0001",
+              symbol: "pipe",
+              ports: [],
+              parameters: {},
+              parameterUnits: {},
+              rotation: 0,
+              mirrored: false,
+            },
+          },
+        ],
+        edges: [],
+        simulation: {
+          t_start: 0,
+          t_stop: 1,
+          step: 0.1,
+          max_step: 0.05,
+          method: "BDF",
+        },
+      }),
+    );
+  });
+  await page.reload();
+
+  await page.getByRole("button", { name: "恢复", exact: true }).click();
+  const restoredNode = page.locator(
+    '.flow-canvas .react-flow__node[data-id="amesim_pnl0001_1"]',
+  );
+  await expect(restoredNode).toBeVisible();
+  await expect(
+    restoredNode.locator('[data-component-symbol="amesim_pnl0001"]'),
+  ).toBeVisible();
+  await expect(restoredNode.locator('[data-component-symbol="pipe"]')).toHaveCount(0);
+});