wehub-resource-sync
77 lines
2.7 KiB
TypeScript
Vendored
77 lines
2.7 KiB
TypeScript
Vendored
import { strict as assert } from "node:assert";
|
|
import { test } from "vitest";
|
|
import { analyzeFieldLineage, identifierInSql, summarizeLineageCounts } from "../../apps/desktop/src/lib/diagram/fieldLineage.ts";
|
|
|
|
test("detects outgoing and incoming foreign key lineage as certain", () => {
|
|
const result = analyzeFieldLineage({
|
|
target: { schema: "public", table: "orders", column: "user_id" },
|
|
tables: [
|
|
{
|
|
schema: "public",
|
|
name: "orders",
|
|
columns: ["id", "user_id"],
|
|
foreignKeys: [{ name: "orders_user_id_fkey", column: "user_id", ref_table: "users", ref_column: "id" }],
|
|
},
|
|
{
|
|
schema: "public",
|
|
name: "order_events",
|
|
columns: ["id", "order_user_id"],
|
|
foreignKeys: [{ name: "events_user_fkey", column: "order_user_id", ref_table: "orders", ref_column: "user_id" }],
|
|
},
|
|
],
|
|
});
|
|
|
|
assert.equal(result.items.length, 2);
|
|
assert.deepEqual(
|
|
result.items.map((item) => [item.kind, item.confidence, item.direction, item.table, item.column]),
|
|
[
|
|
["foreignKey", "certain", "outgoing", "users", "id"],
|
|
["foreignKey", "certain", "incoming", "order_events", "order_user_id"],
|
|
],
|
|
);
|
|
});
|
|
|
|
test("finds view, query history, and same-name column references without scanning table data", () => {
|
|
const result = analyzeFieldLineage({
|
|
target: { schema: "public", table: "users", column: "email" },
|
|
tables: [
|
|
{ schema: "public", name: "users", columns: ["id", "email"], foreignKeys: [] },
|
|
{ schema: "public", name: "newsletter_subscribers", columns: ["id", "email"], foreignKeys: [] },
|
|
],
|
|
views: [
|
|
{
|
|
schema: "public",
|
|
name: "active_user_emails",
|
|
ddl: "CREATE VIEW active_user_emails AS SELECT u.email FROM public.users u WHERE u.active = true",
|
|
},
|
|
],
|
|
histories: [
|
|
{
|
|
id: "h1",
|
|
sql: "select email from users where email like '%@example.com'",
|
|
executed_at: "2026-05-02T00:00:00Z",
|
|
},
|
|
],
|
|
});
|
|
|
|
assert.deepEqual(
|
|
result.items.map((item) => [item.kind, item.confidence, item.table, item.column]),
|
|
[
|
|
["viewReference", "likely", "active_user_emails", undefined],
|
|
["historyReference", "likely", undefined, undefined],
|
|
["sameName", "possible", "newsletter_subscribers", "email"],
|
|
],
|
|
);
|
|
assert.deepEqual(summarizeLineageCounts(result.items), {
|
|
certain: 0,
|
|
likely: 2,
|
|
possible: 1,
|
|
});
|
|
});
|
|
|
|
test("matches quoted and bare identifiers safely", () => {
|
|
assert.equal(identifierInSql('select "User Email" from users', "User Email"), true);
|
|
assert.equal(identifierInSql("select `email` from users", "email"), true);
|
|
assert.equal(identifierInSql("select user_email from users", "email"), false);
|
|
});
|