"""Usage cache reporting for Meta Responses path."""

from types import SimpleNamespace

from agent.usage_pricing import normalize_usage


def test_meta_cached_tokens_flows_to_cache_read():
    usage = SimpleNamespace(
        input_tokens=4000,
        output_tokens=100,
        input_tokens_details=SimpleNamespace(cached_tokens=3920, cache_creation_tokens=0),
        output_tokens_details=None,
        prompt_tokens=None,
        completion_tokens=None,
    )
    cu = normalize_usage(usage, api_mode="codex_responses")
    assert cu.cache_read_tokens == 3920
    assert cu.input_tokens == 80  # 4000 - 3920
    assert cu.prompt_tokens == 4000


def test_meta_cache_write_tokens():
    first = SimpleNamespace(
        input_tokens=4000,
        output_tokens=100,
        input_tokens_details=SimpleNamespace(cached_tokens=0, cache_creation_tokens=4000),
    )
    cu1 = normalize_usage(first, api_mode="codex_responses")
    assert cu1.cache_write_tokens == 4000
    assert cu1.cache_read_tokens == 0

    second = SimpleNamespace(
        input_tokens=4000,
        output_tokens=100,
        input_tokens_details=SimpleNamespace(cached_tokens=3920, cache_creation_tokens=0),
    )
    cu2 = normalize_usage(second, api_mode="codex_responses")
    assert cu2.cache_read_tokens == 3920
    assert cu2.cache_write_tokens == 0