jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-12 07:47:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate roku tests to use freezegun (#105418)

Browse Source
This commit is contained in:
Jan-Philipp Benecke 2023-12-09 23:47:19 +01:00 committed by GitHub
parent 64a5271a51
commit a8148cea65
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/components/roku/test_media_player.py
View File

@ -2,6 +2,7 @@
from datetime import timedelta from datetime import timedelta
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from freezegun.api import FrozenDateTimeFactory
import pytest import pytest
from rokuecp import RokuConnectionError, RokuConnectionTimeoutError, RokuError from rokuecp import RokuConnectionError, RokuConnectionTimeoutError, RokuError
@ -153,6 +154,7 @@ async def test_availability(
hass: HomeAssistant, hass: HomeAssistant,
mock_roku: MagicMock, mock_roku: MagicMock,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
freezer: FrozenDateTimeFactory,
error: RokuError, error: RokuError,
) -> None: ) -> None:
"""Test entity availability.""" """Test entity availability."""
@ -160,19 +162,18 @@ async def test_availability(
future = now + timedelta(minutes=1) future = now + timedelta(minutes=1)
mock_config_entry.add_to_hass(hass) mock_config_entry.add_to_hass(hass)
with patch("homeassistant.util.dt.utcnow", return_value=now): freezer.move_to(now)
await hass.config_entries.async_setup(mock_config_entry.entry_id) await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
with patch("homeassistant.util.dt.utcnow", return_value=future): freezer.move_to(future)
mock_roku.update.side_effect = error mock_roku.update.side_effect = error
async_fire_time_changed(hass, future) async_fire_time_changed(hass, future)
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.get(MAIN_ENTITY_ID).state == STATE_UNAVAILABLE assert hass.states.get(MAIN_ENTITY_ID).state == STATE_UNAVAILABLE
future += timedelta(minutes=1) future += timedelta(minutes=1)
freezer.move_to(future)
with patch("homeassistant.util.dt.utcnow", return_value=future):
mock_roku.update.side_effect = None mock_roku.update.side_effect = None
async_fire_time_changed(hass, future) async_fire_time_changed(hass, future)
await hass.async_block_till_done() await hass.async_block_till_done()