"""Tests for rtl_fm modulation token mapping."""

from routes.listening_post import _rtl_fm_demod_mode as listening_post_rtl_mode
from utils.sdr.base import SDRDevice, SDRType
from utils.sdr.rtlsdr import RTLSDRCommandBuilder
from utils.sdr.rtlsdr import _rtl_fm_demod_mode as builder_rtl_mode


def _dummy_rtlsdr_device() -> SDRDevice:
    return SDRDevice(
        sdr_type=SDRType.RTL_SDR,
        index=0,
        name='RTL-SDR',
        serial='00000001',
        driver='rtlsdr',
        capabilities=RTLSDRCommandBuilder.CAPABILITIES,
    )


def test_rtl_fm_modulation_maps_wfm_to_wbfm() -> None:
    assert listening_post_rtl_mode('wfm') == 'wbfm'
    assert builder_rtl_mode('wfm') == 'wbfm'


def test_rtl_fm_modulation_keeps_other_modes() -> None:
    assert listening_post_rtl_mode('fm') == 'fm'
    assert builder_rtl_mode('am') == 'am'


def test_rtlsdr_builder_uses_wbfm_token_for_wfm() -> None:
    builder = RTLSDRCommandBuilder()
    cmd = builder.build_fm_demod_command(
        device=_dummy_rtlsdr_device(),
        frequency_mhz=98.1,
        modulation='wfm',
    )
    mode_index = cmd.index('-M')
    assert cmd[mode_index + 1] == 'wbfm'