File size: 4,214 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | // Verify schema migration chain runs correctly across versions.
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
let tempDir;
const originalDataDir = process.env.DATA_DIR;
beforeEach(() => {
tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "9router-mig-"));
process.env.DATA_DIR = tempDir;
// Reset global singleton so each test gets fresh adapter pointed at tempDir
delete global._dbAdapter;
vi.resetModules();
});
afterEach(() => {
// Close adapter to release file handles before rm
try { global._dbAdapter?.instance?.close?.(); } catch {}
delete global._dbAdapter;
if (tempDir) fs.rmSync(tempDir, { recursive: true, force: true });
if (originalDataDir === undefined) delete process.env.DATA_DIR;
else process.env.DATA_DIR = originalDataDir;
});
describe("Schema migrations", () => {
it("fresh DB → applies migrations & stamps schemaVersion", async () => {
const { getAdapter } = await import("@/lib/db/driver.js");
const { latestVersion } = await import("@/lib/db/migrations/index.js");
const db = await getAdapter();
const row = db.get(`SELECT value FROM _meta WHERE key='schemaVersion'`);
expect(parseInt(row.value, 10)).toBe(latestVersion());
const tables = db.all(`SELECT name FROM sqlite_master WHERE type='table'`).map(t => t.name);
expect(tables).toEqual(expect.arrayContaining([
"_meta", "settings", "providerConnections", "providerNodes",
"proxyPools", "apiKeys", "combos", "kv", "usageHistory", "usageDaily", "requestDetails",
]));
});
it("existing DB at older schemaVersion → re-applies pending migrations on restart", async () => {
// 1st boot
const { getAdapter } = await import("@/lib/db/driver.js");
const db = await getAdapter();
db.run(`INSERT INTO settings(id, data) VALUES(1, ?) ON CONFLICT(id) DO UPDATE SET data = excluded.data`, ['{"foo":"bar"}']);
db.run(`UPDATE _meta SET value = '0' WHERE key = 'schemaVersion'`);
db.close?.();
// 2nd boot: full reset to simulate process restart
delete global._dbAdapter;
vi.resetModules();
const { getAdapter: getAdapter2 } = await import("@/lib/db/driver.js");
const { latestVersion } = await import("@/lib/db/migrations/index.js");
const db2 = await getAdapter2();
const row = db2.get(`SELECT value FROM _meta WHERE key='schemaVersion'`);
expect(parseInt(row.value, 10)).toBe(latestVersion());
const settings = db2.get(`SELECT data FROM settings WHERE id=1`);
expect(JSON.parse(settings.data)).toEqual({ foo: "bar" });
});
it("fresh DB + legacy db.json → imports data automatically", async () => {
// Simulate user upgrading: place legacy JSON in DATA_DIR before first boot
const legacy = {
settings: { foo: "legacy-value" },
apiKeys: [{ id: "k1", key: "abc", name: "test", createdAt: new Date().toISOString() }],
modelAliases: { "gpt-4": "gpt-4-turbo" },
};
fs.writeFileSync(path.join(tempDir, "db.json"), JSON.stringify(legacy));
const { getAdapter } = await import("@/lib/db/driver.js");
const db = await getAdapter();
const settings = db.get(`SELECT data FROM settings WHERE id=1`);
expect(JSON.parse(settings.data)).toEqual({ foo: "legacy-value" });
const keys = db.all(`SELECT * FROM apiKeys`);
expect(keys).toHaveLength(1);
expect(keys[0].key).toBe("abc");
const aliases = db.all(`SELECT * FROM kv WHERE scope='modelAliases'`);
expect(aliases).toHaveLength(1);
});
it("auto-sync re-creates missing index when DB lacks it", async () => {
const { getAdapter } = await import("@/lib/db/driver.js");
const db = await getAdapter();
db.exec(`DROP INDEX IF EXISTS idx_pn_type`);
expect(db.all(`PRAGMA index_list(providerNodes)`).map(i => i.name)).not.toContain("idx_pn_type");
db.close?.();
delete global._dbAdapter;
vi.resetModules();
const { getAdapter: getAdapter2 } = await import("@/lib/db/driver.js");
const db2 = await getAdapter2();
const idx = db2.all(`PRAGMA index_list(providerNodes)`).map(i => i.name);
expect(idx).toContain("idx_pn_type");
});
});
|