Make api an async fixture to grab reference to loop early

zigpy · Oct 26, 2024 · b9ccd07 · b9ccd07
1 parent 69224d1
commit b9ccd07
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -30,13 +30,13 @@ def gateway():
 
 
 @pytest.fixture
-def api(gateway, mock_command_rsp):
+async def api(gateway, mock_command_rsp):
+    loop = asyncio.get_running_loop()
+
     async def mock_connect(config, api):
         transport = MagicMock()
         transport.close = MagicMock(
-            side_effect=lambda: asyncio.get_running_loop().call_soon(
-                gateway.connection_lost, None
-            )
+            side_effect=lambda: loop.call_soon(gateway.connection_lost, None)
         )
 
         gateway._api = api