"""Tests for core.quality.source_map — each download source's tier/string
mapped into a unified AudioQuality.

These mappers express each source's *claimed* capability tier. The values
are later verified post-download by the quality guard, so over-claiming is
the failure mode to avoid: an unknown tier must NOT pretend to be lossless.
"""

import pytest

from core.quality.source_map import (
    quality_from_tidal_tier,
    quality_from_qobuz,
    quality_from_deezer,
    quality_from_amazon,
)


# ── Tidal / HiFi (Tidal-backed) ────────────────────────────────────────────

@pytest.mark.parametrize("tier", ["HI_RES_LOSSLESS", "HI_RES", "hi_res", "hires", "HIRES"])
def test_tidal_hires_is_flac_24_96(tier):
    aq = quality_from_tidal_tier(tier)
    assert aq.format == "flac"
    assert aq.bit_depth == 24
    assert aq.sample_rate == 96000


@pytest.mark.parametrize("tier", ["LOSSLESS", "lossless"])
def test_tidal_lossless_is_flac_16_44(tier):
    aq = quality_from_tidal_tier(tier)
    assert aq.format == "flac"
    assert aq.bit_depth == 16
    assert aq.sample_rate == 44100


def test_tidal_high_is_lossy_aac_320():
    aq = quality_from_tidal_tier("HIGH")
    assert aq.format == "aac"
    assert aq.bitrate == 320
    assert aq.bit_depth is None  # lossy: no bit depth


def test_tidal_low_is_lossy_aac_96():
    aq = quality_from_tidal_tier("LOW")
    assert aq.format == "aac"
    assert aq.bitrate == 96


def test_tidal_unknown_tier_does_not_overclaim():
    # An unrecognised tier must not masquerade as lossless.
    aq = quality_from_tidal_tier("SOMETHING_NEW")
    assert aq.format == "unknown"
    assert aq.bit_depth is None
    assert aq.sample_rate is None


# ── Qobuz (real API values) ────────────────────────────────────────────────

def test_qobuz_hires_khz_to_hz():
    aq = quality_from_qobuz(96.0, 24)
    assert aq.format == "flac"
    assert aq.sample_rate == 96000
    assert aq.bit_depth == 24


def test_qobuz_cd_quality_fractional_khz():
    aq = quality_from_qobuz(44.1, 16)
    assert aq.format == "flac"
    assert aq.sample_rate == 44100
    assert aq.bit_depth == 16


def test_qobuz_192k():
    aq = quality_from_qobuz(192.0, 24)
    assert aq.sample_rate == 192000
    assert aq.bit_depth == 24


# ── Deezer (config code) ───────────────────────────────────────────────────

def test_deezer_flac_is_16_44():
    aq = quality_from_deezer("flac")
    assert aq.format == "flac"
    assert aq.bit_depth == 16
    assert aq.sample_rate == 44100


def test_deezer_mp3_320():
    aq = quality_from_deezer("mp3_320")
    assert aq.format == "mp3"
    assert aq.bitrate == 320
    assert aq.bit_depth is None


def test_deezer_mp3_128():
    aq = quality_from_deezer("mp3_128")
    assert aq.format == "mp3"
    assert aq.bitrate == 128


# ── Amazon (real sampleRate preferred, tier fallback) ──────────────────────

def test_amazon_prefers_real_sample_rate():
    aq = quality_from_amazon("HD", sample_rate=88200, bit_depth=24)
    assert aq.format == "flac"
    assert aq.sample_rate == 88200
    assert aq.bit_depth == 24


def test_amazon_hd_tier_fallback():
    aq = quality_from_amazon("HD")
    assert aq.format == "flac"
    assert aq.sample_rate == 44100
    assert aq.bit_depth == 16


def test_amazon_uhd_tier_fallback():
    aq = quality_from_amazon("UHD")
    assert aq.format == "flac"
    assert aq.bit_depth == 24
    assert aq.sample_rate == 96000