awesome/test/helpers.js

import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { connect, close } from "../src/db/connection.js";
import { runMigrate } from "../src/db/migrate.js";
import { buildApp } from "../src/app.js";

export async function withApp(fn, { aiIngestKey = "", adminInitPassword = "test123" } = {}) {
  const dir = fs.mkdtempSync(path.join(os.tmpdir(), "awesome-test-"));
  await connect(path.join(dir, "test.duckdb"));
  await runMigrate({ adminInitPassword, aiIngestKey });
  const app = buildApp({
    sessionSecret: "test-secret",
    duckdbPath: path.join(dir, "test.duckdb"),
    baseUrl: "http://test.local",
  });
  const server = app.listen(0);
  await new Promise((r) => server.once("listening", r));
  const base = `http://127.0.0.1:${server.address().port}`;

  async function api(pathname, { method = "GET", body, cookie, headers } = {}) {
    const res = await fetch(base + pathname, {
      method,
      headers: {
        ...(body ? { "content-type": "application/json" } : {}),
        ...(cookie ? { cookie } : {}),
        ...(headers || {}),
      },
      body: body ? JSON.stringify(body) : undefined,
    });
    let json = null;
    try {
      json = await res.json();
    } catch {}
    return { status: res.status, json, headers: res.headers };
  }

  async function loginAdmin() {
    const res = await fetch(base + "/api/admin/login", {
      method: "POST",
      headers: { "content-type": "application/json" },
      body: JSON.stringify({ username: "admin", password: adminInitPassword }),
    });
    const cookies = res.headers.getSetCookie().map((c) => c.split(";")[0]);
    return cookies.join("; ");
  }

  try {
    await fn({ base, api, loginAdmin });
  } finally {
    server.closeAllConnections?.();
    await new Promise((r) => server.close(r));
    await close();
    fs.rmSync(dir, { recursive: true, force: true, maxRetries: 3 });
  }
}