mirror of
https://github.com/musistudio/claude-code-router.git
synced 2026-08-26 17:01:49 +00:00
97d37fad04
Daily usage buckets represent local calendar dates, but standard date parsing interprets their ISO-like form as UTC and shifts them in non-UTC time zones. Parse date-only buckets directly in local time so heatmap totals stay on the reported day.
50 lines
1.6 KiB
JavaScript
50 lines
1.6 KiB
JavaScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
import { buildTokenActivity, activityDateKey } from "@ccr/ui/lib/usage-activity.ts";
|
|
|
|
test("buildTokenActivity summarizes observed token days and streaks", () => {
|
|
withTimezone("America/New_York", () => {
|
|
const summary = buildTokenActivity(
|
|
[
|
|
{ bucket: "2026-06-02", totalTokens: 10 },
|
|
{ bucket: "2026-06-03", totalTokens: 30 },
|
|
{ bucket: "2026-06-04", totalTokens: 60 },
|
|
{ bucket: "2026-06-05", totalTokens: -20 },
|
|
{ bucket: "not-a-date", totalTokens: 999 }
|
|
],
|
|
{ minWeeks: 2 }
|
|
);
|
|
|
|
assert.equal(summary.totalTokens, 100);
|
|
assert.equal(summary.activeDays, 3);
|
|
assert.equal(summary.dayCount, 4);
|
|
assert.equal(summary.longestStreak, 3);
|
|
assert.equal(summary.maxTokens, 60);
|
|
assert.equal(summary.weekCount, 2);
|
|
assert.equal(summary.cells.length, 14);
|
|
|
|
const cellsByDate = new Map(summary.cells.map((cell) => [cell.dateKey, cell]));
|
|
assert.equal(cellsByDate.get("2026-06-02")?.intensity, 1);
|
|
assert.equal(cellsByDate.get("2026-06-03")?.intensity, 3);
|
|
assert.equal(cellsByDate.get("2026-06-04")?.intensity, 4);
|
|
assert.equal(cellsByDate.get("2026-06-05")?.intensity, 0);
|
|
});
|
|
});
|
|
|
|
test("activityDateKey formats local calendar dates", () => {
|
|
assert.equal(activityDateKey(new Date(2026, 0, 5, 14, 30)), "2026-01-05");
|
|
});
|
|
|
|
function withTimezone(timezone, run) {
|
|
const previousTimezone = process.env.TZ;
|
|
process.env.TZ = timezone;
|
|
try {
|
|
run();
|
|
} finally {
|
|
if (previousTimezone === undefined) {
|
|
delete process.env.TZ;
|
|
} else {
|
|
process.env.TZ = previousTimezone;
|
|
}
|
|
}
|
|
}
|