import { describe, expect, test } from "bun:test";
import { getDetailStats } from "./detail-stats";

describe("getDetailStats", () => {
  test("includes completion tokens in detail page stat cards", () => {
    const stats = getDetailStats({
      calls: 12,
      total_tokens: 1000,
      quota: 250000,
      prompt_tokens: 400,
      completion_tokens: 350,
      cache_creation_tokens: 150,
      cache_read_tokens: 100,
    });

    expect(stats.map((stat) => stat.key)).toEqual([
      "calls",
      "total_tokens",
      "quota",
      "prompt_tokens",
      "completion_tokens",
      "cache_creation_tokens",
      "cache_read_tokens",
    ]);
    expect(stats.find((stat) => stat.key === "completion_tokens")).toEqual({
      key: "completion_tokens",
      labelKey: "th.output",
      value: 350,
      format: "tokens",
    });
  });
});