pySwitchbot/tests/test_utils.py

"""Tests for utils.py functionality."""

from __future__ import annotations

from switchbot.utils import format_mac_upper


def test_format_mac_upper() -> None:
    """Test the format_mac_upper utility function."""
    # Test already formatted with colons (lowercase)
    assert format_mac_upper("aa:bb:cc:dd:ee:ff") == "AA:BB:CC:DD:EE:FF"

    # Test already formatted with colons (uppercase)
    assert format_mac_upper("AA:BB:CC:DD:EE:FF") == "AA:BB:CC:DD:EE:FF"

    # Test with dashes
    assert format_mac_upper("aa-bb-cc-dd-ee-ff") == "AA:BB:CC:DD:EE:FF"
    assert format_mac_upper("AA-BB-CC-DD-EE-FF") == "AA:BB:CC:DD:EE:FF"

    # Test with dots (Cisco format)
    assert format_mac_upper("aabb.ccdd.eeff") == "AA:BB:CC:DD:EE:FF"
    assert format_mac_upper("AABB.CCDD.EEFF") == "AA:BB:CC:DD:EE:FF"

    # Test without separators
    assert format_mac_upper("aabbccddeeff") == "AA:BB:CC:DD:EE:FF"
    assert format_mac_upper("AABBCCDDEEFF") == "AA:BB:CC:DD:EE:FF"

    # Test mixed case without separators
    assert format_mac_upper("AaBbCcDdEeFf") == "AA:BB:CC:DD:EE:FF"

    # Test invalid formats (should return original in uppercase)
    assert format_mac_upper("invalid") == "INVALID"
    assert format_mac_upper("aa:bb:cc") == "AA:BB:CC"  # Too short
    assert (
        format_mac_upper("aa:bb:cc:dd:ee:ff:gg") == "AA:BB:CC:DD:EE:FF:GG"
    )  # Too long

    # Test edge cases
    assert format_mac_upper("") == ""
    assert format_mac_upper("123456789ABC") == "12:34:56:78:9A:BC"
    assert format_mac_upper("12:34:56:78:9a:bc") == "12:34:56:78:9A:BC"