ewilliams28/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-08-28 05:16:23 -06:00
Code Issues Packages Projects Releases Wiki Activity

chore: stabilize session cost usage cache warmup test (#113460)

Browse Source 
* test: stabilize session cost usage cache warmup

* test(infra): await cost refresh deterministically

---------

Co-authored-by: Peter Steinberger <steipete@gmail.com>
This commit is contained in:
Eugene Harold Krabs
2026-07-25 16:29:18 +02:00
committed by GitHub
parent 8376daad6c
commit 934acf312b
1 changed files with 30 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/infra/session-cost-usage.test.ts
+30 -32
View File
@@ -33,6 +33,7 @@ import {
loadSessionUsageTimeSeries as loadSessionUsageTimeSeriesForAgent,
resolveExistingUsageSessionFile as resolveExistingUsageSessionFileForAgent,
} from "./session-cost-usage.js";
import { testing as sessionCostUsageTestApi } from "./session-cost-usage.test-support.js";
type WithOptionalAgentId<T> = T extends (params: infer P) => unknown
? Omit<P, "agentId"> & { agentId?: string }
@@ -77,6 +78,13 @@ function waitForFast<T>(
return vi.waitFor(callback, { interval: 1, ...options });
}
async function refreshSessionCostUsageForTest(sessionFile: string): Promise<void> {
await sessionCostUsageTestApi.usageCostRefreshRuntime.refreshCostUsageCacheForAgent({
agentId: "main",
sessionFiles: [sessionFile],
});
}
function clearGatewayModelPricingState(): void {
replaceGatewayModelPricingCache(new Map(), 0);
clearGatewayModelPricingFailures();
@@ -931,21 +939,20 @@ describe("session cost usage", () => {
[
JSON.stringify(assistantEntry(undefined, 1_000)),
JSON.stringify(assistantEntry("2026-02-05T12:00:00.000Z", 20)),
"",
].join("\n"),
"utf-8",
);
await withStateDir(root, async () => {
const session = { sessionId: "sess-v8-upgrade", sessionFile };
await loadSessionCostSummariesFromCache({ sessions: [session], agentId: "main" });
await waitForFast(async () => {
const current = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
requestRefresh: false,
});
expect(current.cacheStatus.status).toBe("fresh");
await refreshSessionCostUsageForTest(sessionFile);
const current = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
requestRefresh: false,
});
expect(current.cacheStatus.status).toBe("fresh");
const currentRow = requireValue(
readSessionCostUsageRollupRows("main").find((row) => row.key === sessionFile),
@@ -968,41 +975,32 @@ describe("session cost usage", () => {
).toBe(true);
const rangeEndMs = Date.UTC(2026, 1, 5) + 24 * 60 * 60 * 1000 - 1;
await loadSessionCostSummariesFromCache({
await refreshSessionCostUsageForTest(sessionFile);
const rebuilt = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
startMs: Date.UTC(2026, 1, 5),
endMs: rangeEndMs,
requestRefresh: false,
});
await waitForFast(async () => {
const rebuilt = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
startMs: Date.UTC(2026, 1, 5),
endMs: rangeEndMs,
requestRefresh: false,
});
expect(rebuilt.cacheStatus.status).toBe("fresh");
expect(rebuilt.summaries[0]?.totalTokens).toBe(20);
});
expect(rebuilt.cacheStatus.status).toBe("fresh");
expect(rebuilt.summaries[0]?.totalTokens).toBe(20);
await fs.appendFile(
sessionFile,
`\n${JSON.stringify(assistantEntry("2026-02-05T13:00:00.000Z", 5))}`,
`${JSON.stringify(assistantEntry("2026-02-05T13:00:00.000Z", 5))}\n`,
"utf-8",
);
await loadSessionCostSummariesFromCache({ sessions: [session], agentId: "main" });
await waitForFast(async () => {
const appended = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
startMs: Date.UTC(2026, 1, 5),
endMs: rangeEndMs,
requestRefresh: false,
});
expect(appended.cacheStatus.status).toBe("fresh");
expect(appended.summaries[0]?.totalTokens).toBe(25);
await refreshSessionCostUsageForTest(sessionFile);
const appended = await loadSessionCostSummariesFromCache({
sessions: [session],
agentId: "main",
startMs: Date.UTC(2026, 1, 5),
endMs: rangeEndMs,
requestRefresh: false,
});
expect(appended.cacheStatus.status).toBe("fresh");
expect(appended.summaries[0]?.totalTokens).toBe(25);
const appendedRow = requireValue(
readSessionCostUsageRollupRows("main").find((row) => row.key === sessionFile),