88 lines
2.9 KiB
TypeScript
88 lines
2.9 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import { mkdtempSync, rmSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import test from "node:test";
|
|
import Database from "better-sqlite3";
|
|
import type { ConnectionConfig } from "../src/connections.js";
|
|
import { describeTable, executeQuery, listTables } from "../src/database.js";
|
|
|
|
function sqliteConfig(path: string): ConnectionConfig {
|
|
return {
|
|
id: "sqlite-test",
|
|
name: "local-sqlite",
|
|
db_type: "sqlite",
|
|
host: path,
|
|
port: 0,
|
|
username: "",
|
|
password: "",
|
|
ssh_enabled: false,
|
|
ssl: false,
|
|
};
|
|
}
|
|
|
|
test("queries SQLite connections without the DBX bridge", async () => {
|
|
const dir = mkdtempSync(join(tmpdir(), "dbx-mcp-sqlite-"));
|
|
const path = join(dir, "app.db");
|
|
const db = new Database(path);
|
|
db.exec("create table users (id integer primary key, name text not null); insert into users (name) values ('Ada');");
|
|
db.close();
|
|
|
|
try {
|
|
const result = await executeQuery(sqliteConfig(path), "select id, name from users");
|
|
|
|
assert.deepEqual(result.columns, ["id", "name"]);
|
|
assert.deepEqual(result.rows, [{ id: 1, name: "Ada" }]);
|
|
assert.equal(result.row_count, 1);
|
|
} finally {
|
|
rmSync(dir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("applies query row limits to SQLite connections", async () => {
|
|
const dir = mkdtempSync(join(tmpdir(), "dbx-mcp-sqlite-"));
|
|
const path = join(dir, "app.db");
|
|
const db = new Database(path);
|
|
db.exec("create table users (id integer primary key, name text not null); insert into users (name) values ('Ada'), ('Grace');");
|
|
db.close();
|
|
|
|
try {
|
|
const result = await executeQuery(sqliteConfig(path), "select id, name from users order by id", { maxRows: 1 });
|
|
|
|
assert.deepEqual(result.columns, ["id", "name"]);
|
|
assert.deepEqual(result.rows, [{ id: 1, name: "Ada" }]);
|
|
assert.equal(result.row_count, 1);
|
|
} finally {
|
|
rmSync(dir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("lists and describes SQLite tables without the DBX bridge", async () => {
|
|
const dir = mkdtempSync(join(tmpdir(), "dbx-mcp-sqlite-"));
|
|
const path = join(dir, "app.db");
|
|
const db = new Database(path);
|
|
db.exec("create table users (id integer primary key, name text not null);");
|
|
db.close();
|
|
|
|
try {
|
|
const tables = await listTables(sqliteConfig(path));
|
|
const columns = await describeTable(sqliteConfig(path), "users");
|
|
|
|
assert.deepEqual(tables, [{ name: "users", type: "table" }]);
|
|
assert.deepEqual(
|
|
columns.map((column) => ({
|
|
name: column.name,
|
|
data_type: column.data_type,
|
|
is_nullable: column.is_nullable,
|
|
is_primary_key: column.is_primary_key,
|
|
})),
|
|
[
|
|
{ name: "id", data_type: "INTEGER", is_nullable: true, is_primary_key: true },
|
|
{ name: "name", data_type: "TEXT", is_nullable: false, is_primary_key: false },
|
|
],
|
|
);
|
|
} finally {
|
|
rmSync(dir, { recursive: true, force: true });
|
|
}
|
|
});
|