64 lines
2.1 KiB
TypeScript
64 lines
2.1 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { buildSqlServerDatabaseTreeNodes } from "../../apps/desktop/src/lib/sqlServerTree.ts";
|
|
import type { ObjectInfo } from "../../apps/desktop/src/types/database.ts";
|
|
|
|
function obj(name: string, objectType = "TABLE"): ObjectInfo {
|
|
return {
|
|
name,
|
|
object_type: objectType,
|
|
};
|
|
}
|
|
|
|
test("SQL Server database tree groups dbo objects and keeps non-default schemas", () => {
|
|
const nodes = buildSqlServerDatabaseTreeNodes("conn", "app", ["dbo", "sales"], [
|
|
obj("customers"),
|
|
obj("customer_view", "VIEW"),
|
|
obj("get_total", "FUNCTION"),
|
|
]);
|
|
|
|
const topLevel = nodes.map((n) => ({ id: n.id, label: n.label, type: n.type }));
|
|
assert.deepEqual(topLevel, [
|
|
{ id: "conn:app:__tables", label: "tree.tables", type: "group-tables" },
|
|
{ id: "conn:app:__views", label: "tree.views", type: "group-views" },
|
|
{ id: "conn:app:__functions", label: "tree.functions", type: "group-functions" },
|
|
{ id: "conn:app:sales", label: "sales", type: "schema" },
|
|
]);
|
|
|
|
const tableGroup = nodes.find((n) => n.type === "group-tables");
|
|
assert.deepEqual(
|
|
tableGroup?.children?.map((c) => c.label),
|
|
["customers"],
|
|
);
|
|
|
|
const functionGroup = nodes.find((n) => n.type === "group-functions");
|
|
assert.deepEqual(
|
|
functionGroup?.children?.map((c) => c.label),
|
|
["get_total"],
|
|
);
|
|
});
|
|
|
|
test("SQL Server database tree shows only schemas when dbo has no objects", () => {
|
|
const nodes = buildSqlServerDatabaseTreeNodes("conn", "app", ["dbo", "archive"], []);
|
|
|
|
assert.deepEqual(
|
|
nodes.map((node) => ({ id: node.id, label: node.label, type: node.type })),
|
|
[{ id: "conn:app:archive", label: "archive", type: "schema" }],
|
|
);
|
|
});
|
|
|
|
test("SQL Server database tree groups procedures alongside tables", () => {
|
|
const nodes = buildSqlServerDatabaseTreeNodes("conn", "app", ["dbo"], [
|
|
obj("orders"),
|
|
obj("sp_refresh", "PROCEDURE"),
|
|
]);
|
|
|
|
assert.deepEqual(
|
|
nodes.map((n) => ({ label: n.label, type: n.type })),
|
|
[
|
|
{ label: "tree.tables", type: "group-tables" },
|
|
{ label: "tree.procedures", type: "group-procedures" },
|
|
],
|
|
);
|
|
});
|