system_update/server/db/schema.test.ts

// server/db/schema.test.ts
import { describe, it, expect } from "vitest";
import Database from "better-sqlite3";
import { drizzle } from "drizzle-orm/better-sqlite3";
import { migrate } from "drizzle-orm/better-sqlite3/migrator";

function freshMigratedDb() {
  const sqlite = new Database(":memory:");
  const db = drizzle(sqlite);
  migrate(db, { migrationsFolder: "./server/db/migrations" });
  return sqlite;
}

function tableNames(sqlite: Database.Database): string[] {
  return sqlite.prepare("SELECT name FROM sqlite_master WHERE type='table'").all().map((r: any) => r.name);
}
function columnNames(sqlite: Database.Database, table: string): string[] {
  return sqlite.prepare(`PRAGMA table_info(${table})`).all().map((r: any) => r.name);
}

describe("schéma Phase 1", () => {
  it("crée les tables socle", () => {
    const sqlite = freshMigratedDb();
    const tables = tableNames(sqlite);
    for (const t of [
      "machines", "snapshots", "executions",
      "machine_state", "machine_hardware", "machine_metrics_latest",
      "machine_events", "important_messages", "reports", "raw_artifacts",
    ]) {
      expect(tables, `table ${t}`).toContain(t);
    }
  });

  it("ajoute les colonnes étendues sans casser l'existant", () => {
    const sqlite = freshMigratedDb();
    expect(columnNames(sqlite, "machines")).toEqual(
      expect.arrayContaining(["machine_kind", "virtualization", "hardware_profile", "os_version", "updated_at"]),
    );
    expect(columnNames(sqlite, "snapshots")).toEqual(
      expect.arrayContaining(["kind", "schema_version", "important_json"]),
    );
    expect(columnNames(sqlite, "executions")).toEqual(
      expect.arrayContaining(["schema_version", "error_kind", "error_message", "exit_code"]),
    );
    // colonnes jalon 1 conservées
    expect(columnNames(sqlite, "snapshots")).toContain("checked_at");
    expect(columnNames(sqlite, "machines")).toContain("enc_password");
  });
});

describe("schéma Phase 2", () => {
  it("crée les tables de credentials Phase 2", () => {
    const sqlite = freshMigratedDb();
    const tables = tableNames(sqlite);
    expect(tables).toEqual(expect.arrayContaining(["machine_credentials", "machine_host_keys"]));
    // machines conserve ses colonnes secrets legacy (fallback)
    expect(columnNames(sqlite, "machines")).toContain("enc_password");
  });
});

describe("schéma SJ-4 Docker", () => {
  it("crée les tables docker_*", () => {
    const sqlite = freshMigratedDb();
    const tables = tableNames(sqlite);
    for (const t of [
      "docker_settings",
      "docker_compose_roots",
      "docker_compose_stacks",
      "docker_stack_services",
    ]) {
      expect(tables, `table ${t}`).toContain(t);
    }
  });

  it("docker_settings a les colonnes attendues", () => {
    const sqlite = freshMigratedDb();
    expect(columnNames(sqlite, "docker_settings")).toEqual(
      expect.arrayContaining(["machine_id", "enabled", "scan_depth", "prune_mode", "last_scan_at", "updated_at"]),
    );
  });

  it("docker_compose_stacks a les colonnes attendues", () => {
    const sqlite = freshMigratedDb();
    expect(columnNames(sqlite, "docker_compose_stacks")).toEqual(
      expect.arrayContaining(["id", "machine_id", "name", "working_dir", "compose_files_json", "status", "detected_by"]),
    );
  });

  it("docker_stack_services a les colonnes attendues", () => {
    const sqlite = freshMigratedDb();
    expect(columnNames(sqlite, "docker_stack_services")).toEqual(
      expect.arrayContaining(["id", "stack_id", "service_name", "image_ref", "current_image_id", "current_digest"]),
    );
  });
});