Attempt to fix ghost error in CI.

Author: jmattsson

requirements.test.txt

@@ -1,7 +1,7 @@
-asyncmock
-mypy
+asyncmock ~= 0.4
+mypy ~= 1.19
 powersensor_local
-pytest
-pytest-asyncio
+pytest ~= 9.0
+pytest-asyncio ~= 1.3
 pytest-homeassistant-custom-component
 zeroconf

tests/test_discovery_service.py

@@ -24,9 +24,6 @@
 MAC = "a4cf1218f158"
 
 
-logging.getLogger().setLevel(logging.CRITICAL)
-
-
 @pytest.fixture
 def mock_service_info():
     """Create a mock service info."""
@@ -108,7 +105,6 @@ async def test_discovery_add_and_remove(
 
     service.remove_service(mock_zc, zc_info.type, zc_info.name)
 
-    # mock_zc.get_service_info.assert_called_once_with(zc_info.type, zc_info.name)
     mock_zc.get_service_info.assert_called_once_with(zc_info.type, zc_info.name)
     mock_send.assert_not_called()
     await asyncio.sleep(service._debounce_seconds + 1)
@@ -178,20 +174,30 @@ async def test_discovery_remove_cancel(
     # reset mock_send
     mock_send = Mock()
     monkeypatch.setattr(PowersensorServiceListener, "dispatch", mock_send)
-    # cache plug data for checking
+
+    # ensure we start from a known state
+    assert len(service._pending_removals) == 0
 
     service.remove_service(mock_zc, zc_info.type, zc_info.name)
 
-    # mock_zc.get_service_info.assert_called_once_with(zc_info.type, zc_info.name)
     mock_zc.get_service_info.assert_called_once_with(zc_info.type, zc_info.name)
     mock_send.assert_not_called()
+    assert len(service._pending_removals) == 1
 
+    # give the remove a head start before triggering the cancellation
+    await asyncio.sleep(0.5)
+
+    # re-add the service, which should cancel the pending remove
     service.add_service(mock_zc, zc_info.type, zc_info.name)
     assert mock_zc.get_service_info.call_count == 2
     mock_zc.get_service_info.assert_has_calls(
         [call(zc_info.type, zc_info.name), call(zc_info.type, zc_info.name)]
     )
 
+    # let the removal task do its cancellation stuff
+    await asyncio.sleep(0.5)
+    assert len(service._pending_removals) == 0
+
 
 @pytest.mark.asyncio
 async def test_discovery_add_and_two_remove_calls(