Files
llm-wiki/tests/graph-workbench-interactions.regression-1.sh
T
2026-07-12 21:26:08 +08:00

196 lines
6.7 KiB
Bash
Executable File

#!/bin/bash
# Regression: workbench uses the shared graph interaction stack for zoom, drag, hover, drawer, minimap, and reset.
set -euo pipefail
REPO_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
# shellcheck disable=SC1091
source "$REPO_ROOT/tests/lib/graph-html-engine-helpers.sh"
tmp_dir="$(mktemp -d)"
server_pid=""
web_pid=""
server_port="${GRAPH_WORKBENCH_SERVER_PORT:-18787}"
web_port="${GRAPH_WORKBENCH_WEB_PORT:-15180}"
cleanup() {
if [ -n "$server_pid" ]; then kill "$server_pid" 2>/dev/null || true; fi
if [ -n "$web_pid" ]; then kill "$web_pid" 2>/dev/null || true; fi
rm -rf "$tmp_dir"
}
trap cleanup EXIT
if lsof -i TCP:"$server_port" -sTCP:LISTEN >/dev/null 2>&1; then
fail "port $server_port is already in use"
fi
if lsof -i TCP:"$web_port" -sTCP:LISTEN >/dev/null 2>&1; then
fail "port $web_port is already in use"
fi
npm run build -w @llm-wiki/graph-engine > /dev/null 2>&1 \
|| fail "graph-engine build should succeed before workbench browser regression"
workbench_kb="$tmp_dir/home/llm-wiki/phase-6-workbench"
mkdir -p "$workbench_kb/wiki/entities" "$tmp_dir/home/.llm-wiki-agent"
cp "$REPO_ROOT/tests/fixtures/graph-interactive-basic/wiki/graph-data.json" "$workbench_kb/wiki/graph-data.json"
cat > "$workbench_kb/.wiki-schema.md" <<'EOF'
# Test schema
EOF
cat > "$workbench_kb/purpose.md" <<'EOF'
# Phase 6 Workbench Test
EOF
cat > "$workbench_kb/wiki/entities/A.md" <<'EOF'
# 节点A
这是节点A的正文。参见 [[wiki/entities/B.md]]。
EOF
cat > "$workbench_kb/wiki/entities/B.md" <<'EOF'
# 节点B
这是节点B的正文。
EOF
cat > "$workbench_kb/wiki/entities/C.md" <<'EOF'
# 节点C
这是节点C的正文。
EOF
node - "$workbench_kb/wiki/graph-data.json" <<'NODE'
const fs = require("fs");
const file = process.argv[2];
const data = JSON.parse(fs.readFileSync(file, "utf8"));
const byId = new Map(data.nodes.map((node) => [node.id, node]));
const base = byId.get("A") || data.nodes[0];
for (const id of ["D", "E", "F", "G"]) {
if (byId.has(id)) continue;
data.nodes.push({
...base,
id,
label: `节点${id}`,
type: "entity",
community: "t1",
source_path: `wiki/entities/${id}.md`,
x: (base.x || 500) + (id.charCodeAt(0) - 67) * 24,
y: (base.y || 300) + (id.charCodeAt(0) - 67) * 18
});
}
const existingEdges = new Set(data.edges.map((edge) => edge.id));
for (const id of ["D", "E", "F", "G"]) {
const edgeId = `A-${id}`;
if (existingEdges.has(edgeId)) continue;
data.edges.push({ id: edgeId, from: "A", to: id, type: "EXTRACTED", relation_type: "同社区", weight: 0.5 });
}
for (const node of data.nodes) {
node.source_path = `wiki/entities/${node.id}.md`;
node.content = `# ${node.label}\n\n这是${node.label}的内容。\n`;
if (node.id === "C") node.type = "source";
}
const t1Members = data.nodes.filter((node) => node.community === "t1").map((node) => node.id);
if (!data.learning) {
data.learning = {
version: 1,
entry: {
recommended_start_node_id: "A",
recommended_start_reason: "测试入口",
default_mode: "global"
},
views: {
path: { enabled: false, start_node_id: null, node_ids: [], degraded: false },
community: { enabled: true, community_id: "t1", label: "t1", node_ids: t1Members, is_weak: false, degraded: false },
global: { enabled: true, node_ids: data.nodes.map((node) => node.id), degraded: false }
},
communities: []
};
}
let community = data.learning.communities.find((item) => item.id === "t1");
if (!community) {
community = { id: "t1", label: "t1", node_count: t1Members.length, members: t1Members };
data.learning.communities.push(community);
}
if (community) {
community.members = t1Members;
community.label = community.label || "t1";
community.is_primary = true;
community.recommended_start_node_id = "A";
community.color_index = 0;
community.node_count = community.members.length;
}
if (data.learning.views?.community) {
data.learning.views.community.enabled = true;
data.learning.views.community.community_id = "t1";
data.learning.views.community.label = "t1";
data.learning.views.community.node_ids = t1Members;
data.learning.views.community.is_weak = false;
data.learning.views.community.degraded = false;
}
if (data.learning.views?.global) {
data.learning.views.global.enabled = true;
data.learning.views.global.node_ids = data.nodes.map((node) => node.id);
data.learning.views.global.degraded = false;
}
if (data.learning.entry) {
data.learning.entry.recommended_start_node_id = "A";
data.learning.entry.recommended_start_reason = "测试入口";
data.learning.entry.default_mode = "global";
}
community.summary = "t1 测试社区";
if (data.meta) {
data.meta.total_nodes = data.nodes.length;
data.meta.total_edges = data.edges.length;
}
fs.writeFileSync(file, `${JSON.stringify(data, null, 2)}\n`);
NODE
for id in D E F G; do
cat > "$workbench_kb/wiki/entities/$id.md" <<EOF
# 节点$id
这是节点$id 的正文。
EOF
done
cat > "$tmp_dir/home/.llm-wiki-agent/config.json" <<JSON
{
"version": 1,
"externalKnowledgeBases": [],
"lastUsedKbPath": "$workbench_kb"
}
JSON
HOME="$tmp_dir/home" HOST=127.0.0.1 PORT="$server_port" npm run dev -w @llm-wiki-agent/server > "$tmp_dir/server.log" 2>&1 &
server_pid="$!"
HOME="$tmp_dir/home" LLM_WIKI_AGENT_API_ORIGIN="http://127.0.0.1:$server_port" npm run dev -w @llm-wiki-agent/web -- --host 127.0.0.1 --port "$web_port" --force > "$tmp_dir/web.log" 2>&1 &
web_pid="$!"
for _ in $(seq 1 120); do
if curl -fsS "http://127.0.0.1:$server_port/api/knowledge-bases" >/dev/null 2>&1 \
&& curl -fsS "http://127.0.0.1:$web_port" >/dev/null 2>&1; then
break
fi
sleep 0.25
done
curl -fsS "http://127.0.0.1:$server_port/api/knowledge-bases" >/dev/null 2>&1 \
|| fail "workbench server did not start; see $tmp_dir/server.log"
curl -fsS "http://127.0.0.1:$web_port" >/dev/null 2>&1 \
|| fail "workbench web did not start; see $tmp_dir/web.log"
artifact_dir="$tmp_dir/artifacts"
mkdir -p "$artifact_dir"
playwright_node_path="$(
npx --yes -p playwright -c 'node -e "const path=require(\"path\"); console.log(path.dirname(process.env.PATH.split(\":\")[0]))"'
)"
chrome_executable="${GRAPH_WORKBENCH_CHROME_EXECUTABLE:-}"
if [ -z "$chrome_executable" ] && [ -x "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome" ]; then
chrome_executable="/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
fi
GRAPH_WORKBENCH_URL="http://127.0.0.1:$web_port" \
GRAPH_WORKBENCH_ARTIFACT_DIR="$artifact_dir" \
GRAPH_WORKBENCH_CHROME_EXECUTABLE="$chrome_executable" \
NODE_PATH="$playwright_node_path" \
node "$REPO_ROOT/tests/browser/graph-workbench-interactions.mjs" \
|| fail "workbench graph interaction browser regression should pass"
echo "Artifacts: $artifact_dir"
echo "PASS: graph workbench interaction regression"