mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-16 11:52:26 +03:00
37ecab3659
Two shards on release/v3.8.51 went red in one day with the same signature —
"ENOTEMPTY, Directory not empty: /tmp/omniroute-<test>-XXXXXX" — from
combo-same-provider-cascade (Unit Tests fast-path 4/4, on a PR that touches only
.github/) and auth-policy-embeddings-webfetch-7785 (the 20k-test TIA step). Both pass
alone and on re-run: the cleanup races something still writing into the directory
(SQLite WAL/-shm checkpoint, a worker, the backup) and under a loaded hosted runner
the window opens. 1154 test files do their own cleanup with
fs.rmSync(dir, { recursive: true, force: true }); 57 already asked for retries.
One-shot codemod (scripts/ad-hoc/codemod-rm-maxretries.mjs, kept for the record):
every rm / rmSync / rmdirSync option object with `recursive: true` and no
`maxRetries` gains `maxRetries: 5, retryDelay: 100` — Node itself then retries
ENOTEMPTY/EBUSY/EPERM for up to ~0.5 s before giving up. 2243 call sites in 1292
files under tests/, the shared tests/_setup/isolateDataDir.ts exit hook included.
Only the option object changes: no call site, assertion or import is touched.
Validation: prettier and ESLint (with the frozen suppressions) clean on all 1292
files; a random 20-file sample runs green (quota-redis-store hangs identically on
the untouched tree — it needs a Redis on localhost, an environment matter). The
four unit shards on this PR are the full run.
96 lines
2.9 KiB
TypeScript
96 lines
2.9 KiB
TypeScript
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";
|
|
|
|
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-stream-debug-"));
|
|
process.env.DATA_DIR = TEST_DATA_DIR;
|
|
|
|
const core = await import("../../src/lib/db/core.ts");
|
|
const callLogs = await import("../../src/lib/usage/callLogs.ts");
|
|
|
|
async function resetStorage() {
|
|
core.resetDbInstance();
|
|
if (fs.existsSync(TEST_DATA_DIR)) {
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true, maxRetries: 5, retryDelay: 100 });
|
|
}
|
|
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
|
|
}
|
|
|
|
test.beforeEach(async () => {
|
|
await resetStorage();
|
|
});
|
|
|
|
test.after(() => {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true, maxRetries: 5, retryDelay: 100 });
|
|
});
|
|
|
|
test("saveCallLog preserves streamChunks in pipeline payloads", async () => {
|
|
const logId = "req_stream_debug_1";
|
|
|
|
const streamChunks = {
|
|
provider: [
|
|
'data: {"content": "hello"}\n\n',
|
|
'data: {"content": " world"}\n\n',
|
|
"data: [DONE]\n\n",
|
|
],
|
|
openai: [
|
|
'data: {"choices":[{"delta":{"content":"hello"}}]}\n\n',
|
|
'data: {"choices":[{"delta":{"content":" world"}}]}\n\n',
|
|
],
|
|
client: [
|
|
'data: {"choices":[{"delta":{"content":"hello"}}]}\n\n',
|
|
'data: {"choices":[{"delta":{"content":" world"}}]}\n\n',
|
|
],
|
|
};
|
|
|
|
await callLogs.saveCallLog({
|
|
id: logId,
|
|
timestamp: new Date().toISOString(),
|
|
status: 200,
|
|
model: "gemini/gemma-4-26b-a4b-it",
|
|
provider: "gemini",
|
|
pipelinePayloads: {
|
|
clientRawRequest: { body: { stream: true } },
|
|
streamChunks: streamChunks,
|
|
},
|
|
});
|
|
|
|
const detail = await callLogs.getCallLogById(logId);
|
|
|
|
assert.ok(detail, "Log detail should exist");
|
|
assert.ok(detail.pipelinePayloads, "Pipeline payloads should exist");
|
|
assert.ok(detail.pipelinePayloads.streamChunks, "streamChunks should exist in pipeline payloads");
|
|
|
|
assert.deepEqual(detail.pipelinePayloads.streamChunks.provider, streamChunks.provider);
|
|
assert.deepEqual(detail.pipelinePayloads.streamChunks.openai, streamChunks.openai);
|
|
assert.deepEqual(detail.pipelinePayloads.streamChunks.client, streamChunks.client);
|
|
});
|
|
|
|
test("saveCallLog preserves partial streamChunks", async () => {
|
|
const logId = "req_stream_debug_2";
|
|
|
|
const streamChunks = {
|
|
provider: ["raw chunk 1", "raw chunk 2"],
|
|
// other stages missing
|
|
};
|
|
|
|
await callLogs.saveCallLog({
|
|
id: logId,
|
|
status: 200,
|
|
model: "test-model",
|
|
pipelinePayloads: {
|
|
streamChunks: streamChunks,
|
|
},
|
|
});
|
|
|
|
const detail = await callLogs.getCallLogById(logId);
|
|
|
|
assert.ok(detail?.pipelinePayloads?.streamChunks, "streamChunks should exist");
|
|
assert.deepEqual(detail.pipelinePayloads.streamChunks.provider, streamChunks.provider);
|
|
assert.equal(detail.pipelinePayloads.streamChunks.openai, undefined);
|
|
assert.equal(detail.pipelinePayloads.streamChunks.client, undefined);
|
|
});
|