fix update tests

home-assistant · Oct 3, 2024 · ac6ee34 · ac6ee34
1 parent 25f4f52
commit ac6ee34
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tests/components/smlight/conftest.py b/tests/components/smlight/conftest.py
@@ -71,6 +71,16 @@ def mock_setup_entry() -> Generator[AsyncMock]:
         yield mock_setup_entry
 
 
+@pytest.fixture
+def mock_update_sleep() -> Generator[AsyncMock]:
+    """Fixture to mock asyncio.sleep."""
+    with patch(
+        "homeassistant.components.smlight.update.asyncio.sleep",
+        return_value=AsyncMock(),
+    ) as mock_sleep:
+        yield mock_sleep
+
+
 @pytest.fixture
 def mock_smlight_client(request: pytest.FixtureRequest) -> Generator[MagicMock]:
     """Mock the SMLIGHT API client."""

diff --git a/tests/components/smlight/test_update.py b/tests/components/smlight/test_update.py
@@ -87,6 +87,7 @@ async def test_update_setup(
     await hass.config_entries.async_unload(entry.entry_id)
 
 
+@pytest.mark.usefixtures("mock_update_sleep")
 async def test_update_firmware(
     hass: HomeAssistant,
     freezer: FrozenDateTimeFactory,
@@ -134,6 +135,7 @@ async def test_update_firmware(
     assert state.attributes[ATTR_LATEST_VERSION] == "v2.5.2"
 
 
+@pytest.mark.usefixtures("mock_update_sleep")
 async def test_update_legacy_firmware_v2(
     hass: HomeAssistant,
     freezer: FrozenDateTimeFactory,
@@ -213,6 +215,7 @@ async def _call_event_function(event: MessageEvent):
     assert state.attributes[ATTR_IN_PROGRESS] is False
 
 
+@pytest.mark.usefixtures("mock_update_sleep")
 async def test_update_release_notes(
     hass: HomeAssistant,
     freezer: FrozenDateTimeFactory,