wehub-resource-sync
105 lines
5.1 KiB
TypeScript
Vendored
105 lines
5.1 KiB
TypeScript
Vendored
import assert from "node:assert/strict";
|
|
import { test } from "vitest";
|
|
import { appendVisibleDatabaseSelection, buildDraftVisibleDatabasesConnectionId, connectionCanChooseVisibleDatabases, visibleDatabaseSelectionIsStale, initialVisibleDatabaseSelection } from "../../apps/desktop/src/lib/connection/connectionVisibleDatabases.ts";
|
|
import { connectionUsesVisibleSchemaFilter, filterDatabaseNamesForConnection, filterDatabaseNamesForVisiblePicker } from "../../apps/desktop/src/lib/database/visibleDatabases.ts";
|
|
import type { ConnectionConfig } from "../../apps/desktop/src/types/database.ts";
|
|
|
|
function config(overrides: Partial<ConnectionConfig> = {}): ConnectionConfig {
|
|
return {
|
|
id: "conn",
|
|
name: "Local",
|
|
db_type: "mysql",
|
|
driver_profile: "mysql",
|
|
host: "127.0.0.1",
|
|
port: 3306,
|
|
username: "root",
|
|
password: "",
|
|
database: undefined,
|
|
visible_databases: undefined,
|
|
transport_layers: [],
|
|
connect_timeout_secs: 5,
|
|
query_timeout_secs: 30,
|
|
idle_timeout_secs: 60,
|
|
ssl: false,
|
|
ca_cert_path: "",
|
|
client_cert_path: "",
|
|
client_key_path: "",
|
|
sysdba: false,
|
|
jdbc_driver_paths: [],
|
|
redis_sentinel_master: "",
|
|
redis_sentinel_nodes: "",
|
|
redis_sentinel_username: "",
|
|
redis_sentinel_password: "",
|
|
redis_sentinel_tls: false,
|
|
redis_cluster_nodes: "",
|
|
etcd_endpoints: "",
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
test("draft visible database connection ids are namespaced", () => {
|
|
assert.equal(buildDraftVisibleDatabasesConnectionId("abc"), "__visible_draft_abc");
|
|
});
|
|
|
|
test("initial selection uses configured visible databases when available", () => {
|
|
assert.deepEqual(initialVisibleDatabaseSelection(["app", "analytics", "billing"], ["billing", "missing"]), ["billing"]);
|
|
});
|
|
|
|
test("initial selection uses default visible database names when no filter is configured", () => {
|
|
assert.deepEqual(initialVisibleDatabaseSelection(["app", "mysql", "sys"], undefined, config()), ["app"]);
|
|
});
|
|
|
|
test("visible database picker ignores saved filters while keeping default system database hiding", () => {
|
|
const databaseNames = ["app", "analytics", "mysql", "sys"];
|
|
const connection = config({ visible_databases: ["app"] });
|
|
assert.deepEqual(filterDatabaseNamesForVisiblePicker(databaseNames, connection), ["app", "analytics"]);
|
|
assert.deepEqual(initialVisibleDatabaseSelection(databaseNames, connection.visible_databases, connection), ["app"]);
|
|
});
|
|
|
|
test("Redis visible database picker keeps every database and initial selection uses saved filters", () => {
|
|
const databaseNames = ["0", "1", "2"];
|
|
const connection = config({ db_type: "redis", driver_profile: "redis", visible_databases: ["0"] });
|
|
assert.deepEqual(filterDatabaseNamesForVisiblePicker(databaseNames, connection), ["0", "1", "2"]);
|
|
assert.deepEqual(initialVisibleDatabaseSelection(databaseNames, connection.visible_databases, connection), ["0"]);
|
|
});
|
|
|
|
test("connection database filtering still applies saved visible database filters for sidebar display", () => {
|
|
assert.deepEqual(filterDatabaseNamesForConnection(["app", "analytics", "mysql", "sys"], config({ visible_databases: ["app"] })), ["app"]);
|
|
});
|
|
|
|
test("append visible database selection only when filter is enabled", () => {
|
|
assert.deepEqual(appendVisibleDatabaseSelection(["app"], "analytics"), ["app", "analytics"]);
|
|
assert.deepEqual(appendVisibleDatabaseSelection(["app", "analytics"], "analytics"), ["app", "analytics"]);
|
|
assert.equal(appendVisibleDatabaseSelection(undefined, "analytics"), undefined);
|
|
});
|
|
|
|
test("append visible database selection trims new database names and ignores empty names", () => {
|
|
assert.deepEqual(appendVisibleDatabaseSelection(["app"], " analytics "), ["app", "analytics"]);
|
|
assert.deepEqual(appendVisibleDatabaseSelection(["app"], " "), ["app"]);
|
|
});
|
|
|
|
test("ZooKeeper connections do not offer visible database selection", () => {
|
|
assert.equal(connectionCanChooseVisibleDatabases(config({ db_type: "zookeeper" })), false);
|
|
});
|
|
|
|
test("Cloudflare D1 does not offer a visible database filter for its fixed main namespace", () => {
|
|
assert.equal(connectionCanChooseVisibleDatabases(config({ db_type: "cloudflare-d1" })), false);
|
|
});
|
|
|
|
test("OceanBase Oracle uses schema filtering for visible object selection", () => {
|
|
assert.equal(connectionUsesVisibleSchemaFilter(config({ db_type: "oceanbase-oracle" })), true);
|
|
assert.equal(connectionUsesVisibleSchemaFilter(config({ db_type: "mysql", driver_profile: "oceanbase" })), false);
|
|
});
|
|
|
|
test("Dameng default SYSDBA user remains selectable", () => {
|
|
assert.deepEqual(filterDatabaseNamesForConnection(["SYS", "SYSDBA", "SYSAUDITOR"], config({ db_type: "dameng" })), ["SYSDBA"]);
|
|
});
|
|
|
|
test("visible database selection is stale when connection target changes", () => {
|
|
const previous = config({ host: "db.internal", visible_databases: ["app"] });
|
|
assert.equal(visibleDatabaseSelectionIsStale(previous, config({ host: "db.internal" })), false);
|
|
assert.equal(visibleDatabaseSelectionIsStale(previous, config({ host: "db2.internal" })), true);
|
|
assert.equal(visibleDatabaseSelectionIsStale(previous, config({ username: "readonly" })), true);
|
|
assert.equal(visibleDatabaseSelectionIsStale(previous, config({ database: "admin" })), true);
|
|
});
|