Update test case

bdraco · bdraco · commit 034565cfd4f6 · 2021-06-18T21:03:11.000-10:00
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -249,10 +249,14 @@ def mock_split_incoming_msg(service_state_change: r.ServiceStateChange) -> r.DNS
             zeroconf.close()
 
 
-def test_notify_listeners():
+# This test uses asyncio because it needs to verify the listeners
+# run in the event loop
+@pytest.mark.asyncio
+async def test_notify_listeners():
     """Test adding and removing notify listeners."""
     # instantiate a zeroconf instance
-    zc = Zeroconf(interfaces=['127.0.0.1'])
+    aiozc = AsyncZeroconf(interfaces=['127.0.0.1'])
+    zc = aiozc.zeroconf
     notify_called = 0
 
     class TestNotifyListener(r.NotifyListener):
@@ -281,10 +285,11 @@ def on_service_state_change(zeroconf, service_type, state_change, name):
     # start a browser
     browser = ServiceBrowser(zc, "_http._tcp.local.", [on_service_state_change])
     browser.cancel()
+    await asyncio.sleep(0)
 
     assert not notify_called
 
-    zc.close()
+    await aiozc.async_close()
 
 
 def test_generate_service_query_set_qu_bit():
