Try sleep :)

zigpy · Jul 24, 2024 · 2a5afc1 · 2a5afc1
1 parent 57f273a
commit 2a5afc1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_ezsp.py b/tests/test_ezsp.py
@@ -291,6 +291,7 @@ async def test_ezsp_init(conn_mock, reset_mock, version_mock):
     """Test initialize method."""
     zigpy_config = config.CONFIG_SCHEMA({"device": DEVICE_CONFIG})
     await ezsp.EZSP.initialize(zigpy_config)
+    await asyncio.sleep(0.1)  # TODO: figure out why this is necessary
     assert conn_mock.await_count == 1
     assert reset_mock.await_count == 1
     assert version_mock.await_count == 1
@@ -307,6 +308,7 @@ async def test_ezsp_init_failure(conn_mock, close_mock, reset_mock, version_mock
     with pytest.raises(RuntimeError):
         await ezsp.EZSP.initialize(zigpy_config)
 
+    await asyncio.sleep(0.1)  # TODO: figure out why this is necessary
     assert conn_mock.await_count == 1
     assert reset_mock.await_count == 1
     assert version_mock.await_count == 1