Files
diegosouzapw--omniroute/tests/unit/chatcore-caveman-output-analytics.test.ts
2026-07-13 13:39:12 +08:00

76 lines
2.6 KiB
TypeScript

// Characterization of writeCavemanOutputAnalytics — the caveman-output-only compression analytics
// write extracted from handleChatCore's request-setup compression path (chatCore god-file
// decomposition, #3501). Returns the write promise; uses a real temp DB. Locks: the row written
// (mode=output-caveman, engine=caveman-output, original==compressed==estimatedTokens, tokens_saved
// 0, output_mode) and that the returned promise never rejects (best-effort).
import { test, before, after } from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
const testDataDir = fs.mkdtempSync(path.join(os.tmpdir(), "omni-caveman-test-"));
process.env.DATA_DIR = testDataDir;
const coreDb = await import("../../src/lib/db/core.ts");
const { writeCavemanOutputAnalytics } = await import(
"../../open-sse/handlers/chatCore/cavemanOutputAnalytics.ts"
);
function rowFor(requestId: string): Record<string, unknown> | undefined {
return coreDb
.getDbInstance()
.prepare(
"SELECT mode, engine, original_tokens AS orig, compressed_tokens AS comp, tokens_saved AS saved, output_mode AS om FROM compression_analytics WHERE request_id = ?"
)
.get(requestId) as Record<string, unknown> | undefined;
}
before(async () => {
await coreDb.ensureDbInitialized();
});
after(() => {
coreDb.resetDbInstance();
try {
fs.rmSync(testDataDir, { recursive: true, force: true });
} catch {
// best-effort cleanup
}
});
test("writes a caveman-output analytics row and the promise resolves", async () => {
await writeCavemanOutputAnalytics({
comboName: null,
provider: "openai",
compressionComboId: null,
estimatedTokens: 321,
skillRequestId: "caveman-req-1",
cavemanOutputModeIntensity: "heavy",
});
const row = rowFor("caveman-req-1");
assert.ok(row, "expected a compression_analytics row");
assert.equal(row!.mode, "output-caveman");
assert.equal(row!.engine, "caveman-output");
assert.equal(row!.orig, 321);
assert.equal(row!.comp, 321);
assert.equal(row!.saved, 0);
assert.equal(row!.om, "heavy");
});
test("the returned promise never rejects even on a bad write", async () => {
// resetting the DB instance makes the dynamic insert path fail; the helper must swallow it
coreDb.resetDbInstance();
await assert.doesNotReject(
writeCavemanOutputAnalytics({
comboName: "combo-x",
provider: "openai",
compressionComboId: "cc-1",
estimatedTokens: 10,
skillRequestId: "caveman-req-2",
cavemanOutputModeIntensity: null,
})
);
await coreDb.ensureDbInitialized();
});