import pytest

from app.projectors.live_positions import _classify_device_type


@pytest.mark.parametrize(
    "device_name,expected",
    [
        ("John Mbugua - KDW 573B_cam", "camera"),
        ("John Mbugua - KDW 573B_CAM", "camera"),
        ("John Mbugua - KDW 573B", "tracker"),
        ("Parked - KMGK 596V", "tracker"),
        ("JC400P-92732", "tracker"),
        ("", "tracker"),
        (None, "tracker"),
    ],
)
def test_classify_device_type(device_name: str | None, expected: str) -> None:
    assert _classify_device_type(device_name) == expected