Dima/OmniRoute
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: fix 4 failing unit tests (copilot-usage, request-log-migration)

Browse Source 
- copilot-usage: use future reset date (2026-12-31) to avoid stale
  quota window causing remainingPercentage to reset to 100%
- request-log-migration: close SQLite DB before cleanup to release
  Windows file locks; remove stale archive dir before second test
This commit is contained in:
William Finger
2026-04-01 13:41:08 +01:00
parent 5bd209aded
commit 7928bede1e
2 changed files with 31 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/copilot-usage.test.mjs
+4 -5
View File
@@ -2,9 +2,8 @@ import test from "node:test";
import assert from "node:assert/strict";
const usageService = await import("../../open-sse/services/usage.ts");
const providerLimitUtils = await import(
"../../src/app/(dashboard)/dashboard/usage/components/ProviderLimits/utils.tsx"
);
const providerLimitUtils =
await import("../../src/app/(dashboard)/dashboard/usage/components/ProviderLimits/utils.tsx");
test("github copilot business seats infer business plan and hide unlimited buckets", async () => {
const originalFetch = globalThis.fetch;
@@ -13,7 +12,7 @@ test("github copilot business seats infer business plan and hide unlimited bucke
new Response(
JSON.stringify({
access_type_sku: "copilot_business_seat",
quota_reset_date: "2026-04-01T00:00:00Z",
quota_reset_date: "2026-12-31T00:00:00Z",
quota_snapshots: {
chat: { unlimited: true },
completions: { unlimited: true },
@@ -61,7 +60,7 @@ test("github copilot individual paid plans no longer normalize as free", async (
new Response(
JSON.stringify({
copilot_plan: "individual",
quota_reset_date: "2026-04-01T00:00:00Z",
quota_reset_date: "2026-12-31T00:00:00Z",
quota_snapshots: {
premium_interactions: {
entitlement: 300,
tests/unit/request-log-migration.test.mjs
+27 -9
View File
@@ -8,17 +8,13 @@ const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-log-migra
process.env.DATA_DIR = TEST_DATA_DIR;
const migrations = await import("../../src/lib/usage/migrations.ts");
const { getDbInstance } = await import("../../src/lib/db/core.ts");
const LEGACY_LOGS_DIR = path.join(TEST_DATA_DIR, "logs");
const LEGACY_CALL_LOGS_DIR = path.join(TEST_DATA_DIR, "call_logs");
const LEGACY_SUMMARY_FILE = path.join(TEST_DATA_DIR, "log.txt");
const MARKER_PATH = path.join(migrations.LOG_ARCHIVES_DIR, "legacy-request-logs.json");
function resetDataDir() {
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
}
function seedLegacyLayout() {
fs.mkdirSync(path.join(LEGACY_LOGS_DIR, "session-a"), { recursive: true });
fs.writeFileSync(
@@ -35,12 +31,28 @@ function seedLegacyLayout() {
fs.writeFileSync(LEGACY_SUMMARY_FILE, "legacy summary\n");
}
test.beforeEach(() => {
resetDataDir();
});
function cleanup() {
// Close the SQLite connection that holds a lock on files inside TEST_DATA_DIR
try {
const db = getDbInstance();
if (db && db.open) db.close();
} catch {
// DB may already be closed
}
// On Windows, rmSync can fail if file handles are still held.
// Retry with a short delay to let the OS release locks.
for (let attempt = 0; attempt < 5; attempt++) {
try {
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
return;
} catch {
/* retry */
}
}
}
test.after(() => {
resetDataDir();
cleanup();
});
test("archives legacy request log layout into a zip and removes old files", async () => {
@@ -60,7 +72,13 @@ test("archives legacy request log layout into a zip and removes old files", asyn
});
test("keeps legacy files in place when zip creation fails", async () => {
// Re-seed legacy layout (first test archived and removed them)
seedLegacyLayout();
// Remove the archive dir created by the first test, then write a file
// at that path so mkdirSync throws EEXIST. This simulates a zip
// creation failure. The migration should leave legacy files intact.
fs.rmSync(migrations.LOG_ARCHIVES_DIR, { recursive: true, force: true });
fs.writeFileSync(migrations.LOG_ARCHIVES_DIR, "not-a-directory");
await assert.rejects(() => migrations.archiveLegacyRequestLogs());