from __future__ import annotations

from pathlib import Path

import pytest

from ozan_radio.stats import cost_per_track, record_generation, today_stats


def test_cost_per_track_pro():
    costs = {"lyria_pro_usd": 0.08, "lyria_clip_usd": 0.04, "deepseek_per_track_usd": 0.002}
    assert cost_per_track("lyria-3-pro-preview", costs) == pytest.approx(0.082)


def test_cost_per_track_clip():
    costs = {"lyria_pro_usd": 0.08, "lyria_clip_usd": 0.04, "deepseek_per_track_usd": 0.002}
    assert cost_per_track("lyria-3-clip-preview", costs) == 0.04


def test_record_and_today_stats(tmp_path: Path):
    out = tmp_path / "songs"
    out.mkdir()
    record_generation(out, 0.082, "abc", "Test Track")
    stats = today_stats(out)
    assert stats["generated"] == 1
    assert stats["estimated_usd"] == pytest.approx(0.082)
    assert stats["tracks"][0]["title"] == "Test Track"