evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use freezegun in lacrosse_view tests (#99036)

Browse source
This commit is contained in:
Erik Montnemery 2023-08-25 16:00:30 +02:00 • committed by GitHub
parent e0af9de877
commit ee073e9e3e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/components/lacrosse_view/test_init.py
View file

@ -1,8 +1,8 @@
"""Test the LaCrosse View initialization.""" """Test the LaCrosse View initialization."""
from datetime import datetime, timedelta from datetime import timedelta
from unittest.mock import patch from unittest.mock import patch
from freezegun import freeze_time from freezegun.api import FrozenDateTimeFactory
from lacrosse_view import HTTPError, LoginError from lacrosse_view import HTTPError, LoginError
from homeassistant.components.lacrosse_view.const import DOMAIN from homeassistant.components.lacrosse_view.const import DOMAIN
@ -74,7 +74,7 @@ async def test_http_error(hass: HomeAssistant) -> None:
assert entries[0].state == ConfigEntryState.SETUP_RETRY assert entries[0].state == ConfigEntryState.SETUP_RETRY
async def test_new_token(hass: HomeAssistant) -> None: async def test_new_token(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> None:
"""Test new token.""" """Test new token."""
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA) config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
@ -92,19 +92,20 @@ async def test_new_token(hass: HomeAssistant) -> None:
assert len(entries) == 1 assert len(entries) == 1
assert entries[0].state == ConfigEntryState.LOADED assert entries[0].state == ConfigEntryState.LOADED
one_hour_after = datetime.now() + timedelta(hours=1)
with patch("lacrosse_view.LaCrosse.login", return_value=True) as login, patch( with patch("lacrosse_view.LaCrosse.login", return_value=True) as login, patch(
"lacrosse_view.LaCrosse.get_sensors", "lacrosse_view.LaCrosse.get_sensors",
return_value=[TEST_SENSOR], return_value=[TEST_SENSOR],
), freeze_time(one_hour_after): ):
async_fire_time_changed(hass, one_hour_after) freezer.tick(timedelta(hours=1))
async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
login.assert_called_once() login.assert_called_once()
async def test_failed_token(hass: HomeAssistant) -> None: async def test_failed_token(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
) -> None:
"""Test if a reauth flow occurs when token refresh fails.""" """Test if a reauth flow occurs when token refresh fails."""
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA) config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
@ -122,12 +123,9 @@ async def test_failed_token(hass: HomeAssistant) -> None:
assert len(entries) == 1 assert len(entries) == 1
assert entries[0].state == ConfigEntryState.LOADED assert entries[0].state == ConfigEntryState.LOADED
one_hour_after = datetime.now() + timedelta(hours=1) with patch("lacrosse_view.LaCrosse.login", side_effect=LoginError("Test")):
freezer.tick(timedelta(hours=1))
with patch( async_fire_time_changed(hass)
"lacrosse_view.LaCrosse.login", side_effect=LoginError("Test")
), freeze_time(one_hour_after):
async_fire_time_changed(hass, one_hour_after)
await hass.async_block_till_done() await hass.async_block_till_done()
entries = hass.config_entries.async_entries(DOMAIN) entries = hass.config_entries.async_entries(DOMAIN)