kilocode/packages/kilo-vscode/tests/unit/model-price-format.test.ts

import { describe, it, expect } from "bun:test"
import { avgPrice, fmtCachedPrice, fmtPrice } from "../../webview-ui/src/components/shared/model-preview-utils"

// Prices arriving at fmtPrice are already in $/M tokens (converted by parseApiPrice).
// This test suite guards against the double-multiplication bug where fmtPrice was
// incorrectly multiplying by 1_000_000 again, turning $3.00/1M into $3,000,000.00/1M.

describe("fmtPrice", () => {
  it("formats Claude Sonnet 4.6 input price correctly ($3/1M)", () => {
    expect(fmtPrice(3)).toBe("$3.00/1M")
  })

  it("formats Claude Sonnet 4.6 output price correctly ($15/1M)", () => {
    expect(fmtPrice(15)).toBe("$15.00/1M")
  })

  it("returns 'Free' for zero price", () => {
    expect(fmtPrice(0)).toBe("Free")
  })

  it("returns dash for negative price", () => {
    expect(fmtPrice(-1)).toBe("—")
    expect(fmtPrice(-1000000)).toBe("—")
  })

  it("uses 4 decimal places for sub-cent prices", () => {
    expect(fmtPrice(0.005)).toBe("$0.0050/1M")
  })

  it("uses 2 decimal places at the $0.01 boundary", () => {
    expect(fmtPrice(0.01)).toBe("$0.01/1M")
  })

  it("formats a typical cheap model price ($0.50/1M)", () => {
    expect(fmtPrice(0.5)).toBe("$0.50/1M")
  })

  it("formats a high-cost model price ($75/1M)", () => {
    expect(fmtPrice(75)).toBe("$75.00/1M")
  })
})

describe("fmtCachedPrice", () => {
  it("formats supported cached prices using the same $/1M formatter", () => {
    expect(fmtCachedPrice({ input: 3, cache: { read: 0.3 } })).toBe("$0.30/1M")
  })

  it("shows free cached pricing when input is free and cache_read is zero", () => {
    expect(fmtCachedPrice({ input: 0, cache: { read: 0 } })).toBe("Free")
  })

  it("returns null when cache_read is zero but input is paid", () => {
    expect(fmtCachedPrice({ input: 3, cache: { read: 0 } })).toBeNull()
  })

  it("returns null when cache_read is missing but input is paid", () => {
    expect(fmtCachedPrice({ input: 3 })).toBeNull()
  })
})

describe("avgPrice", () => {
  it("uses the cache-heavy ratio when cache_read is supported", () => {
    expect(avgPrice({ input: 3, output: 15, cache: { read: 0.3 } })).toBe(2.31)
  })

  it("uses the fallback ratio when cache_read is zero", () => {
    expect(avgPrice({ input: 3, output: 15, cache: { read: 0 } })).toBe(4.2)
  })

  it("uses the fallback ratio when cache_read is missing", () => {
    expect(avgPrice({ input: 3, output: 15 })).toBe(4.2)
  })
})