chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
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
|
||||
);
|
||||
});
|
||||
Reference in New Issue
Block a user