File size: 5,613 Bytes
cd8bd0a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
/**
 * TDD: GET /api/context/analytics/engine?engineId=&days=
 *
 * Auth + isolation pattern mirrors tests/unit/api/compression-preview-engine.test.ts:
 * - makeManagementSessionRequest() for JWT cookie auth.
 * - Temp DATA_DIR, resetDbInstance() before each test, cleanup in test.after().
 */
import test from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs";
import os from "node:os";
import path from "node:path";

import { makeManagementSessionRequest } from "../../helpers/managementSession.ts";

// ─── isolated temp DB ─────────────────────────────────────────────────────────

const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-ctx-analytics-engine-"));
const originalDataDir = process.env.DATA_DIR;
const originalJwtSecret = process.env.JWT_SECRET;

process.env.DATA_DIR = TEST_DATA_DIR;

const core = await import("../../../src/lib/db/core.ts");
const settingsDb = await import("../../../src/lib/db/settings.ts");
const { insertCompressionAnalyticsRow } =
  await import("../../../src/lib/db/compressionAnalytics.ts");
const engineRoute = await import("../../../src/app/api/context/analytics/engine/route.ts");

// ─── helpers ──────────────────────────────────────────────────────────────────

async function setupAuth(): Promise<void> {
  core.resetDbInstance();
  fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
  fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
  await settingsDb.updateSettings({
    requireLogin: true,
    setupComplete: true,
    password: "test-password-hash",
  });
}

// ─── lifecycle ────────────────────────────────────────────────────────────────

test.beforeEach(async () => {
  process.env.DATA_DIR = TEST_DATA_DIR;
  await setupAuth();
});

test.after(() => {
  if (originalDataDir === undefined) delete process.env.DATA_DIR;
  else process.env.DATA_DIR = originalDataDir;
  if (originalJwtSecret === undefined) delete process.env.JWT_SECRET;
  else process.env.JWT_SECRET = originalJwtSecret;
  core.resetDbInstance();
  fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
});

// ─── tests ────────────────────────────────────────────────────────────────────

test("GET /api/context/analytics/engine returns 400 when engineId is missing", async () => {
  const req = await makeManagementSessionRequest("http://localhost/api/context/analytics/engine");
  const res = await engineRoute.GET(req);
  assert.equal(res.status, 400, `Expected 400, got ${res.status}`);
  const body = (await res.json()) as { error: string };
  assert.ok(typeof body.error === "string", "response should have an error string");
});

test("GET /api/context/analytics/engine returns 200 with correct aggregation for headroom", async () => {
  const now = new Date().toISOString();

  insertCompressionAnalyticsRow({
    timestamp: now,
    mode: "stacked",
    engine: "headroom",
    original_tokens: 1000,
    compressed_tokens: 800,
    tokens_saved: 200,
  });
  insertCompressionAnalyticsRow({
    timestamp: now,
    mode: "stacked",
    engine: "headroom",
    original_tokens: 500,
    compressed_tokens: 350,
    tokens_saved: 150,
  });
  // caveman row β€” must NOT appear in headroom result
  insertCompressionAnalyticsRow({
    timestamp: now,
    mode: "stacked",
    engine: "caveman",
    original_tokens: 800,
    compressed_tokens: 600,
    tokens_saved: 200,
  });

  const req = await makeManagementSessionRequest(
    "http://localhost/api/context/analytics/engine?engineId=headroom"
  );
  const res = await engineRoute.GET(req);
  assert.equal(res.status, 200, `Expected 200, got ${res.status}`);

  const body = (await res.json()) as {
    engineId: string;
    runs: number;
    tokensSaved: number;
    avgSavingsPercent: number;
    days: number;
  };

  assert.equal(body.engineId, "headroom");
  assert.equal(body.runs, 2, "should count only the 2 headroom rows");
  assert.equal(body.tokensSaved, 350, "should sum tokens_saved for headroom rows");
  assert.equal(typeof body.avgSavingsPercent, "number", "avgSavingsPercent should be a number");
  assert.equal(body.days, 7, "default days should be 7");
});

test("GET /api/context/analytics/engine respects ?days= parameter", async () => {
  const req = await makeManagementSessionRequest(
    "http://localhost/api/context/analytics/engine?engineId=headroom&days=30"
  );
  const res = await engineRoute.GET(req);
  assert.equal(res.status, 200, `Expected 200, got ${res.status}`);
  const body = (await res.json()) as { days: number };
  assert.equal(body.days, 30, "should pass days=30 through to the result");
});

test("GET /api/context/analytics/engine returns 200 with zero metrics for unknown engine", async () => {
  const req = await makeManagementSessionRequest(
    "http://localhost/api/context/analytics/engine?engineId=nonexistent"
  );
  const res = await engineRoute.GET(req);
  assert.equal(res.status, 200, `Expected 200, got ${res.status}`);
  const body = (await res.json()) as { runs: number; tokensSaved: number };
  assert.equal(body.runs, 0);
  assert.equal(body.tokensSaved, 0);
});