"""Tests for HackRF command builder."""

from utils.sdr.base import SDRDevice, SDRType
from utils.sdr.hackrf import HackRFCommandBuilder


def _make_device(serial: str = "abc123") -> SDRDevice:
    return SDRDevice(
        sdr_type=SDRType.HACKRF,
        index=0,
        name="HackRF One",
        serial=serial,
        driver="hackrf",
        capabilities=HackRFCommandBuilder.CAPABILITIES,
    )


class TestHackRFCapabilities:
    def test_gain_max_reflects_combined_lna_vga(self):
        """gain_max should be LNA(40) + VGA(62) = 102."""
        assert HackRFCommandBuilder.CAPABILITIES.gain_max == 102.0

    def test_frequency_range(self):
        caps = HackRFCommandBuilder.CAPABILITIES
        assert caps.freq_min_mhz == 1.0
        assert caps.freq_max_mhz == 6000.0

    def test_tx_capable(self):
        assert HackRFCommandBuilder.CAPABILITIES.tx_capable is True


class TestSplitGain:
    def test_low_gain_all_to_lna(self):
        builder = HackRFCommandBuilder()
        lna, vga = builder._split_gain(30)
        assert lna == 30
        assert vga == 0

    def test_gain_at_lna_max(self):
        builder = HackRFCommandBuilder()
        lna, vga = builder._split_gain(40)
        assert lna == 40
        assert vga == 0

    def test_high_gain_splits_across_stages(self):
        builder = HackRFCommandBuilder()
        lna, vga = builder._split_gain(80)
        assert lna == 40
        assert vga == 40

    def test_max_combined_gain(self):
        builder = HackRFCommandBuilder()
        lna, vga = builder._split_gain(102)
        assert lna == 40
        assert vga == 62


class TestBuildAdsbCommand:
    def test_contains_soapysdr_device_type(self):
        builder = HackRFCommandBuilder()
        cmd = builder.build_adsb_command(_make_device(), gain=40)
        assert "--device-type" in cmd
        assert "soapysdr" in cmd

    def test_includes_serial_in_device_string(self):
        builder = HackRFCommandBuilder()
        cmd = builder.build_adsb_command(_make_device(serial="deadbeef"), gain=40)
        device_idx = cmd.index("--device")
        assert "deadbeef" in cmd[device_idx + 1]


class TestBuildIQCaptureCommand:
    def test_outputs_cu8_to_stdout(self):
        builder = HackRFCommandBuilder()
        cmd = builder.build_iq_capture_command(_make_device(), frequency_mhz=100.0, sample_rate=2048000, gain=40)
        assert "-F" in cmd
        assert "CU8" in cmd
        assert cmd[-1] == "-"

    def test_gain_split_in_command(self):
        builder = HackRFCommandBuilder()
        cmd = builder.build_iq_capture_command(_make_device(), frequency_mhz=100.0, gain=80)
        gain_idx = cmd.index("-g")
        assert cmd[gain_idx + 1] == "LNA=40,VGA=40"