vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-12 09:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: advance session cost retry timers

Browse Source
This commit is contained in:
Shakker
2026-05-09 00:34:17 +01:00
parent a1d288cbb7
commit e5975ef859
1 changed files with 44 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
src/infra/session-cost-usage.test.ts
View File

@@ -1054,43 +1054,54 @@ describe("session cost usage", () => {
"utf-8",
);
await withStateDir(root, async () => {
await refreshCostUsageCache();
const cachePath = path.join(sessionsDir, ".usage-cost-cache.json");
const lockPath = `${cachePath}.lock`;
await fs.writeFile(
lockPath,
`${JSON.stringify({ pid: process.pid, startedAt: Date.now() })}\n`,
"utf-8",
);
vi.useFakeTimers({ toFake: ["setTimeout", "clearTimeout"] });
try {
await withStateDir(root, async () => {
await refreshCostUsageCache();
const cachePath = path.join(sessionsDir, ".usage-cost-cache.json");
const lockPath = `${cachePath}.lock`;
await fs.writeFile(
lockPath,
`${JSON.stringify({ pid: process.pid, startedAt: Date.now() })}\n`,
"utf-8",
);
try {
const cold = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
});
expect(cold.summary).toBeNull();
try {
const cold = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
});
expect(cold.summary).toBeNull();
await new Promise((resolve) => setTimeout(resolve, 75));
const stillMissing = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
requestRefresh: false,
});
expect(stillMissing.summary).toBeNull();
} finally {
await fs.rm(lockPath, { force: true });
}
await vi.advanceTimersByTimeAsync(75);
const stillMissing = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
requestRefresh: false,
});
expect(stillMissing.summary).toBeNull();
} finally {
await fs.rm(lockPath, { force: true });
}
await waitFor(async () => {
const warm = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
requestRefresh: false,
});
return warm.summary?.totalTokens === 10;
await vi.waitFor(
async () => {
const warm = await loadSessionCostSummaryFromCache({
sessionId: "sess-cache-lock-busy",
sessionFile,
requestRefresh: false,
});
expect(warm.summary?.totalTokens).toBe(10);
},
{
interval: 1,
timeout: 200,
},
);
});
});
} finally {
vi.useRealTimers();
}
});
it("summarizes a single session file", async () => {