120 lines
4.3 KiB
TypeScript
120 lines
4.3 KiB
TypeScript
import { describe, it, expect } from "vitest";
|
||
import Database from "better-sqlite3";
|
||
import { initSchema } from "./schema";
|
||
import {
|
||
querySymbolRefs,
|
||
querySymbolForwardCalls,
|
||
queryModuleDeps,
|
||
queryModuleReverseDeps,
|
||
queryViolations,
|
||
} from "./query";
|
||
|
||
function setupTestDb(): Database.Database {
|
||
const db = new Database(":memory:");
|
||
initSchema(db);
|
||
// 插入测试数据
|
||
db.prepare(
|
||
"INSERT INTO modules (name, path, layer, created_at) VALUES (?, ?, ?, ?)"
|
||
).run("exams", "src/modules/exams", "modules", "2026-07-07");
|
||
db.prepare(
|
||
"INSERT INTO modules (name, path, layer, created_at) VALUES (?, ?, ?, ?)"
|
||
).run("grades", "src/modules/grades", "modules", "2026-07-07");
|
||
db.prepare(
|
||
"INSERT INTO files (module_id, path, kind, lines) VALUES (?, ?, ?, ?)"
|
||
).run(1, "src/modules/exams/actions.ts", "actions", 100);
|
||
db.prepare(
|
||
"INSERT INTO files (module_id, path, kind, lines) VALUES (?, ?, ?, ?)"
|
||
).run(1, "src/modules/exams/data-access.ts", "data-access", 200);
|
||
db.prepare(
|
||
"INSERT INTO symbols (file_id, name, kind, is_exported, is_async, is_server_action, start_line, end_line) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
|
||
).run(1, "createExamAction", "function", 1, 1, 1, 10, 50);
|
||
db.prepare(
|
||
"INSERT INTO symbols (file_id, name, kind, is_exported, is_async, is_server_action, start_line, end_line) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
|
||
).run(2, "createExam", "function", 1, 1, 0, 20, 80);
|
||
db.prepare(
|
||
"INSERT INTO calls (caller_id, callee_id, call_line, count) VALUES (?, ?, ?, ?)"
|
||
).run(1, 2, 30, 1); // createExamAction 调用 createExam
|
||
db.prepare(
|
||
"INSERT INTO module_deps (source_module_id, target_module_id, dep_type) VALUES (?, ?, ?)"
|
||
).run(1, 2, "data-access-call"); // exams 依赖 grades
|
||
return db;
|
||
}
|
||
|
||
describe("querySymbolRefs", () => {
|
||
it("should find callers of a symbol (reverse)", () => {
|
||
const db = setupTestDb();
|
||
const refs = querySymbolRefs(db, "createExam");
|
||
expect(refs).toHaveLength(1);
|
||
expect(refs[0].caller_name).toBe("createExamAction");
|
||
});
|
||
});
|
||
|
||
describe("querySymbolForwardCalls", () => {
|
||
it("should find callees of a symbol (forward)", () => {
|
||
const db = setupTestDb();
|
||
const calls = querySymbolForwardCalls(db, "createExamAction");
|
||
expect(calls).toHaveLength(1);
|
||
expect(calls[0].callee_name).toBe("createExam");
|
||
});
|
||
});
|
||
|
||
describe("queryModuleDeps", () => {
|
||
it("should find module dependencies (forward)", () => {
|
||
const db = setupTestDb();
|
||
const deps = queryModuleDeps(db, "exams");
|
||
expect(deps).toContainEqual({
|
||
target_module: "grades",
|
||
dep_type: "data-access-call",
|
||
});
|
||
});
|
||
});
|
||
|
||
describe("queryModuleReverseDeps", () => {
|
||
it("should find module reverse dependencies", () => {
|
||
const db = setupTestDb();
|
||
const deps = queryModuleReverseDeps(db, "grades");
|
||
expect(deps).toContainEqual({
|
||
source_module: "exams",
|
||
dep_type: "data-access-call",
|
||
});
|
||
});
|
||
});
|
||
|
||
describe("queryViolations", () => {
|
||
it("should detect files over 800 lines", () => {
|
||
const db = setupTestDb();
|
||
// 插入一个超长文件
|
||
db.prepare(
|
||
"INSERT INTO files (module_id, path, kind, lines) VALUES (?, ?, ?, ?)"
|
||
).run(1, "src/modules/exams/huge.ts", "lib", 900);
|
||
const violations = queryViolations(db);
|
||
expect(violations.long_files).toContainEqual({
|
||
path: "src/modules/exams/huge.ts",
|
||
lines: 900,
|
||
});
|
||
});
|
||
|
||
it("should detect Server Actions without requirePermission", () => {
|
||
const db = setupTestDb();
|
||
const violations = queryViolations(db);
|
||
// createExamAction 是 Server Action 但没有调用 requirePermission
|
||
const missing = violations.server_actions_without_permission.find(
|
||
(v) => v.name === "createExamAction"
|
||
);
|
||
expect(missing).toBeDefined();
|
||
});
|
||
|
||
it("should exclude @public Server Actions from violations", () => {
|
||
const db = setupTestDb();
|
||
// 插入一个带 @public 标记的 Server Action（豁免权限校验）
|
||
db.prepare(
|
||
"INSERT INTO symbols (file_id, name, kind, is_exported, is_async, is_server_action, is_public, start_line, end_line) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)"
|
||
).run(1, "publicLoginAction", "function", 1, 1, 1, 1, 60, 100);
|
||
const violations = queryViolations(db);
|
||
const missing = violations.server_actions_without_permission.find(
|
||
(v) => v.name === "publicLoginAction"
|
||
);
|
||
expect(missing).toBeUndefined();
|
||
});
|
||
});
|