Merge Pull Request #882

# Conflicts:
#	tests/unit/copilot-usage.test.mjs
This commit is contained in:
diegosouzapw
2026-04-01 18:04:52 -03:00

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());