mirror of
https://github.com/Termix-SSH/Termix.git
synced 2026-10-28 14:15:24 +00:00
111 lines
3.6 KiB
TypeScript
111 lines
3.6 KiB
TypeScript
import { afterEach, describe, expect, it } from "vitest";
|
|
import { createTestDb, type TestDb } from "@termix/plugin-sdk/testing";
|
|
import { pluginDir } from "./helpers";
|
|
|
|
// command_history as core's SQLite bootstrap created it before 2.9.0, plus the
|
|
// index performance-indexes.ts added.
|
|
const LEGACY_DDL = `
|
|
CREATE TABLE command_history (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id TEXT NOT NULL,
|
|
host_id INTEGER NOT NULL,
|
|
command TEXT NOT NULL,
|
|
executed_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (user_id) REFERENCES users (id) ON DELETE CASCADE,
|
|
FOREIGN KEY (host_id) REFERENCES ssh_data (id) ON DELETE CASCADE
|
|
);
|
|
CREATE INDEX idx_command_history_user_host ON command_history (user_id, host_id);
|
|
`;
|
|
|
|
let db: TestDb | null = null;
|
|
|
|
afterEach(() => {
|
|
db?.close();
|
|
db = null;
|
|
});
|
|
|
|
const tableExists = (name: string) =>
|
|
!!db!.sqlite
|
|
.prepare("SELECT name FROM sqlite_master WHERE type = 'table' AND name = ?")
|
|
.get(name);
|
|
|
|
const indexesOn = (table: string) =>
|
|
(
|
|
db!.sqlite
|
|
.prepare(
|
|
"SELECT name FROM sqlite_master WHERE type = 'index' AND tbl_name = ? AND name NOT LIKE 'sqlite_%'",
|
|
)
|
|
.all(table) as { name: string }[]
|
|
).map((row) => row.name);
|
|
|
|
describe("adopting command_history", () => {
|
|
it("keeps every existing row and its index when the legacy table is there", async () => {
|
|
db = await createTestDb(pluginDir, {
|
|
before: (sqlite) => {
|
|
sqlite.exec("INSERT INTO users (id) VALUES ('user-1'), ('user-2')");
|
|
sqlite.exec("INSERT INTO ssh_data (id) VALUES (1), (2)");
|
|
sqlite.exec(LEGACY_DDL);
|
|
sqlite.exec(`
|
|
INSERT INTO command_history (user_id, host_id, command, executed_at)
|
|
VALUES ('user-1', 1, 'ls -la', '2026-01-01T00:00:00.000Z'),
|
|
('user-2', 2, 'uptime', '2026-01-02T00:00:00.000Z');
|
|
`);
|
|
},
|
|
});
|
|
|
|
expect(db.applied).toEqual(["0001_adopt_command_history"]);
|
|
expect(tableExists("command_history")).toBe(false);
|
|
expect(
|
|
db.sqlite
|
|
.prepare(
|
|
"SELECT user_id, host_id, command, executed_at FROM p_ssh_terminal_command_history ORDER BY id",
|
|
)
|
|
.all(),
|
|
).toEqual([
|
|
{
|
|
user_id: "user-1",
|
|
host_id: 1,
|
|
command: "ls -la",
|
|
executed_at: "2026-01-01T00:00:00.000Z",
|
|
},
|
|
{
|
|
user_id: "user-2",
|
|
host_id: 2,
|
|
command: "uptime",
|
|
executed_at: "2026-01-02T00:00:00.000Z",
|
|
},
|
|
]);
|
|
expect(indexesOn("p_ssh_terminal_command_history")).toEqual([
|
|
"idx_command_history_user_host",
|
|
]);
|
|
});
|
|
|
|
it("creates a fresh table when there is no legacy one", async () => {
|
|
db = await createTestDb(pluginDir);
|
|
expect(tableExists("p_ssh_terminal_command_history")).toBe(true);
|
|
expect(indexesOn("p_ssh_terminal_command_history")).toEqual([
|
|
"idx_command_history_user_host",
|
|
]);
|
|
});
|
|
|
|
it("cascades when the user or the host is deleted", async () => {
|
|
db = await createTestDb(pluginDir, {
|
|
before: (sqlite) => {
|
|
sqlite.exec("INSERT INTO users (id) VALUES ('user-1'), ('user-2')");
|
|
sqlite.exec("INSERT INTO ssh_data (id) VALUES (1), (2)");
|
|
},
|
|
});
|
|
db.sqlite.exec(`
|
|
INSERT INTO p_ssh_terminal_command_history (user_id, host_id, command)
|
|
VALUES ('user-1', 1, 'a'), ('user-2', 1, 'b'), ('user-2', 2, 'c');
|
|
`);
|
|
db.sqlite.exec("DELETE FROM users WHERE id = 'user-1'");
|
|
db.sqlite.exec("DELETE FROM ssh_data WHERE id = 2");
|
|
expect(
|
|
db.sqlite
|
|
.prepare("SELECT command FROM p_ssh_terminal_command_history")
|
|
.all(),
|
|
).toEqual([{ command: "b" }]);
|
|
});
|
|
});
|