Add IPv6 support to the reconnect logic zeroconf listener (#1121)

Author: bdraco

aioesphomeapi/reconnect_logic.py

@@ -8,7 +8,11 @@
 from typing import Callable
 
 import zeroconf
-from zeroconf.const import _TYPE_A as TYPE_A, _TYPE_PTR as TYPE_PTR
+from zeroconf.const import (
+    _TYPE_A as TYPE_A,
+    _TYPE_AAAA as TYPE_AAAA,
+    _TYPE_PTR as TYPE_PTR,
+)
 
 from .client import APIClient
 from .core import (
@@ -23,6 +27,8 @@
 
 _LOGGER = logging.getLogger(__name__)
 
+ADDRESS_RECORD_TYPES = {TYPE_A, TYPE_AAAA}
+
 EXPECTED_DISCONNECT_COOLDOWN = 5.0
 MAXIMUM_BACKOFF_TRIES = 100
 
@@ -398,11 +404,14 @@ def async_update_records(
             return
 
         for record_update in records:
-            # We only consider PTR records and match using the alias name
+            # We only consider A, AAAA, and PTR records and match using the alias name
             new_record = record_update.new
             if not (
                 (new_record.type == TYPE_PTR and new_record.alias == self._ptr_alias)  # type: ignore[attr-defined]
-                or (new_record.type == TYPE_A and new_record.name == self._a_name)
+                or (
+                    new_record.type in ADDRESS_RECORD_TYPES
+                    and new_record.name == self._a_name
+                )
             ):
                 continue
 

tests/test_reconnect_logic.py

@@ -16,7 +16,7 @@
     current_time_millis,
 )
 from zeroconf.asyncio import AsyncZeroconf
-from zeroconf.const import _CLASS_IN, _TYPE_A, _TYPE_PTR
+from zeroconf.const import _CLASS_IN, _TYPE_A, _TYPE_AAAA, _TYPE_PTR
 
 from aioesphomeapi import APIConnectionError, RequiresEncryptionAPIError
 from aioesphomeapi._frame_helper.plain_text import APIPlaintextFrameHelper
@@ -378,6 +378,18 @@ async def on_connect_fail(connect_exception: Exception) -> None:
             ReconnectLogicState.READY,
             "received mDNS record",
         ),
+        (
+            DNSAddress(
+                "mydevice.local.",
+                _TYPE_AAAA,
+                _CLASS_IN,
+                1000,
+                ip_address("::1").packed,
+            ),
+            True,
+            ReconnectLogicState.READY,
+            "received mDNS record",
+        ),
     ),
 )
 @pytest.mark.asyncio