adjust patch target

bdraco · bdraco · commit 8363e4a090f1 · 2021-09-18T16:19:07.000-10:00
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -18,8 +18,9 @@
 from unittest.mock import patch
 
 import zeroconf as r
-from zeroconf import _core, _protocol, const, Zeroconf, current_time_millis
+from zeroconf import _core, const, Zeroconf, current_time_millis
 from zeroconf.asyncio import AsyncZeroconf
+from zeroconf._protocol import outgoing
 
 from . import has_working_ipv6, _clear_cache, _inject_response, _wait_for_start
 
@@ -670,8 +671,8 @@ def test_guard_against_oversized_packets():
         )
 
     # We are patching to generate an oversized packet
-    with patch.object(_protocol, "_MAX_MSG_ABSOLUTE", 100000), patch.object(
-        _protocol, "_MAX_MSG_TYPICAL", 100000
+    with patch.object(outgoing, "_MAX_MSG_ABSOLUTE", 100000), patch.object(
+        outgoing, "_MAX_MSG_TYPICAL", 100000
     ):
         over_sized_packet = generated.packets()[0]
         assert len(over_sized_packet) > const._MAX_MSG_ABSOLUTE
