wehub-resource-sync
101 lines
2.6 KiB
TypeScript
101 lines
2.6 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { createNodeSqliteAdapterFromDatabase } from "../../../src/lib/db/adapters/nodeSqliteShared.ts";
|
|
|
|
type Row = Record<string, unknown>;
|
|
|
|
class FakeStatement {
|
|
finalized = false;
|
|
|
|
constructor(
|
|
private readonly sql: string,
|
|
private readonly rows: Row[] = []
|
|
) {}
|
|
|
|
run(..._params: unknown[]) {
|
|
return { changes: 1n, lastInsertRowid: 7n };
|
|
}
|
|
|
|
get(..._params: unknown[]) {
|
|
if (this.sql.startsWith("PRAGMA")) {
|
|
return { journal_mode: "wal" };
|
|
}
|
|
return this.rows[0];
|
|
}
|
|
|
|
all(..._params: unknown[]) {
|
|
if (this.sql.startsWith("PRAGMA")) {
|
|
return [{ journal_mode: "wal" }];
|
|
}
|
|
return this.rows;
|
|
}
|
|
|
|
finalize() {
|
|
this.finalized = true;
|
|
}
|
|
}
|
|
|
|
class FakeDb {
|
|
closed = false;
|
|
execCalls: string[] = [];
|
|
statements: FakeStatement[] = [];
|
|
|
|
prepare(sql: string) {
|
|
const statement = new FakeStatement(sql, [{ value: "ok" }]);
|
|
this.statements.push(statement);
|
|
return statement;
|
|
}
|
|
|
|
exec(sql: string) {
|
|
this.execCalls.push(sql);
|
|
}
|
|
|
|
close() {
|
|
this.closed = true;
|
|
}
|
|
}
|
|
|
|
test("createNodeSqliteAdapterFromDatabase wraps node:sqlite statements", () => {
|
|
const db = new FakeDb();
|
|
const adapter = createNodeSqliteAdapterFromDatabase(db, ":memory:");
|
|
|
|
const insert = adapter.prepare("INSERT INTO t VALUES (?)").run("ok");
|
|
assert.deepEqual(insert, { changes: 1, lastInsertRowid: 7 });
|
|
|
|
const row = adapter.prepare("SELECT value FROM t").get() as Row;
|
|
assert.equal(row.value, "ok");
|
|
|
|
assert.equal(adapter.pragma("journal_mode", { simple: true }), "wal");
|
|
assert.deepEqual(adapter.pragma("journal_mode"), [{ journal_mode: "wal" }]);
|
|
});
|
|
|
|
test("createNodeSqliteAdapterFromDatabase uses savepoints for transactions", () => {
|
|
const db = new FakeDb();
|
|
const adapter = createNodeSqliteAdapterFromDatabase(db, ":memory:");
|
|
|
|
const run = adapter.transaction((value: string) => value.toUpperCase());
|
|
assert.equal(run("ok"), "OK");
|
|
assert.equal(db.execCalls[0].startsWith("SAVEPOINT "), true);
|
|
assert.equal(db.execCalls[1].startsWith("RELEASE "), true);
|
|
});
|
|
|
|
test("createNodeSqliteAdapterFromDatabase finalizes cached statements on close", () => {
|
|
const db = new FakeDb();
|
|
let closedHookCalls = 0;
|
|
const adapter = createNodeSqliteAdapterFromDatabase(db, ":memory:", () => {
|
|
closedHookCalls++;
|
|
});
|
|
|
|
adapter.prepare("SELECT 1").get();
|
|
adapter.close();
|
|
|
|
assert.equal(closedHookCalls, 1);
|
|
assert.equal(db.closed, true);
|
|
assert.equal(adapter.open, false);
|
|
assert.equal(
|
|
db.statements.every((statement) => statement.finalized),
|
|
true
|
|
);
|
|
});
|