fix: imports

bdraco · bdraco · commit 9a7f6cdbf1c1 · 2023-12-18T18:33:55.000-10:00
diff --git a/src/zeroconf/_handlers/query_handler.py b/src/zeroconf/_handlers/query_handler.py
@@ -23,7 +23,6 @@
 from typing import TYPE_CHECKING, List, Optional, Set, Tuple, Union, cast
 
 from .._cache import DNSCache, _UniqueRecordsType
-from .._core import _MDNS_PORT
 from .._dns import DNSAddress, DNSPointer, DNSQuestion, DNSRecord, DNSRRSet
 from .._protocol.incoming import DNSIncoming
 from .._services.info import ServiceInfo
@@ -33,6 +32,7 @@
     _ADDRESS_RECORD_TYPES,
     _CLASS_IN,
     _DNS_OTHER_TTL,
+    _MDNS_PORT,
     _ONE_SECOND,
     _SERVICE_TYPE_ENUMERATION_NAME,
     _TYPE_A,
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -8,7 +8,8 @@
 
 import pytest
 
-from zeroconf import _core, const
+from zeroconf import const
+from zeroconf._handlers import query_handler
 
 
 @pytest.fixture(autouse=True)
@@ -23,7 +24,7 @@ def verify_threads_ended():
 @pytest.fixture
 def run_isolated():
     """Change the mDNS port to run the test in isolation."""
-    with patch.object(_core, "_MDNS_PORT", 5454), patch.object(const, "_MDNS_PORT", 5454):
+    with patch.object(query_handler, "_MDNS_PORT", 5454), patch.object(const, "_MDNS_PORT", 5454):
         yield
 
 
