Add Tests For LVA (#312)

* Add Tests

tests/unit/test_zeroconf.py

@@ -0,0 +1,178 @@
+"""Unit tests for HomeAssistantZeroconf."""
+
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+# ---------------------------------------------------------------------------
+# Helpers
+# ---------------------------------------------------------------------------
+
+
+def make_zeroconf(**kwargs):
+    defaults = dict(
+        port=6053,
+        mac_address="aa:bb:cc:dd:ee:ff",
+        host_ip_address="192.168.1.100",
+        name="lva-test",
+    )
+    defaults.update(kwargs)
+
+    with patch("linux_voice_assistant.zeroconf.AsyncZeroconf") as mock_zc_cls:
+        mock_zc = MagicMock()
+        mock_zc_cls.return_value = mock_zc
+        from linux_voice_assistant.zeroconf import HomeAssistantZeroconf
+
+        instance = HomeAssistantZeroconf(**defaults)
+        instance._mock_zc = mock_zc
+        instance._mock_zc_cls = mock_zc_cls
+        return instance
+
+
+# ---------------------------------------------------------------------------
+# __init__
+# ---------------------------------------------------------------------------
+
+
+class TestInit:
+    def test_name_stored(self):
+        zc = make_zeroconf(name="my-lva")
+        assert zc.name == "my-lva"
+
+    def test_port_stored(self):
+        zc = make_zeroconf(port=1234)
+        assert zc.port == 1234
+
+    def test_mac_address_stored(self):
+        zc = make_zeroconf(mac_address="11:22:33:44:55:66")
+        assert zc.mac_address == "11:22:33:44:55:66"
+
+    def test_host_ip_stored(self):
+        zc = make_zeroconf(host_ip_address="10.0.0.1")
+        assert zc.host_ip_address == "10.0.0.1"
+
+    def test_name_defaults_to_mac_when_not_provided(self):
+        with patch("linux_voice_assistant.zeroconf.AsyncZeroconf"):
+            from linux_voice_assistant.zeroconf import HomeAssistantZeroconf
+
+            zc = HomeAssistantZeroconf(
+                port=6053,
+                mac_address="aa:bb:cc:dd:ee:ff",
+                host_ip_address="192.168.1.1",
+                name=None,
+            )
+        assert zc.name == "aa:bb:cc:dd:ee:ff"
+
+    def test_async_zeroconf_instantiated(self):
+        with patch("linux_voice_assistant.zeroconf.AsyncZeroconf") as mock_cls:
+            mock_cls.return_value = MagicMock()
+            from linux_voice_assistant.zeroconf import HomeAssistantZeroconf
+
+            HomeAssistantZeroconf(
+                port=6053,
+                mac_address="aa:bb:cc:dd:ee:ff",
+                host_ip_address="192.168.1.1",
+            )
+            mock_cls.assert_called_once()
+
+
+# ---------------------------------------------------------------------------
+# register_server()
+# ---------------------------------------------------------------------------
+
+
+class TestRegisterServer:
+    @pytest.mark.asyncio
+    async def test_register_service_called(self):
+        zc = make_zeroconf()
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo"):
+            await zc.register_server()
+
+        zc._mock_zc.async_register_service.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_service_name_contains_device_name(self):
+        zc = make_zeroconf(name="lva-aabbccddee")
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        # First positional arg is service type, second is full service name
+        assert "lva-aabbccddee" in captured["args"][1]
+
+    @pytest.mark.asyncio
+    async def test_service_type_is_esphomelib(self):
+        zc = make_zeroconf()
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        assert captured["args"][0] == "_esphomelib._tcp.local."
+
+    @pytest.mark.asyncio
+    async def test_service_properties_contain_mac(self):
+        zc = make_zeroconf(mac_address="aa:bb:cc:dd:ee:ff")
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        props = captured["kwargs"].get("properties", {})
+        assert "mac" in props
+        assert props["mac"] == "aa:bb:cc:dd:ee:ff"
+
+    @pytest.mark.asyncio
+    async def test_service_properties_contain_version(self):
+        zc = make_zeroconf()
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        props = captured["kwargs"].get("properties", {})
+        assert "version" in props
+
+    @pytest.mark.asyncio
+    async def test_service_port_matches(self):
+        zc = make_zeroconf(port=9999)
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        assert captured["kwargs"].get("port") == 9999
+
+    @pytest.mark.asyncio
+    async def test_service_address_matches_host_ip(self):
+        import socket
+
+        zc = make_zeroconf(host_ip_address="10.0.0.5")
+        zc._mock_zc.async_register_service = AsyncMock()
+
+        captured = {}
+
+        with patch("linux_voice_assistant.zeroconf.AsyncServiceInfo") as mock_info_cls:
+            mock_info_cls.side_effect = lambda *a, **kw: captured.update({"args": a, "kwargs": kw}) or MagicMock()
+            await zc.register_server()
+
+        addresses = captured["kwargs"].get("addresses", [])
+        assert socket.inet_aton("10.0.0.5") in addresses