claude-code-router/packages/ui/test/unit/usage-activity.test.mjs

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");
});

test("buildTokenActivity averages sparse data over the selected range window", () => {
  withTimezone("America/New_York", () => {
    const summary = buildTokenActivity(
      [
        { bucket: "2026-06-29", totalTokens: 10 },
        { bucket: "2026-06-30", totalTokens: 30 }
      ],
      { now: "2026-06-30T12:00:00.000Z", range: "30d" }
    );

    assert.equal(summary.totalTokens, 40);
    assert.equal(summary.activeDays, 2);
    assert.equal(summary.dayCount, 30);
    assert.equal(summary.longestStreak, 2);
    assert.equal(Math.round(summary.avgPerDay * 100) / 100, 1.33);
    assert.equal(Math.round(summary.avgPerWeek * 100) / 100, 9.33);
  });
});

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;
    }
  }
}