d99334eb07
* Add new LaCrosse View integration * Add new LaCrosse View integration * Add retry logic * Actually use the start time for the retry logic * Get new token after 1 hour * Replace retry logic with more reliable logic * Improve test coverage * Add device info and unique id to config entry * Fix manufacturer name * Improve token refresh and check sensor permission * Improve test cover * Add LaCrosse View to .strict-typing * Remove empty fields in manifest.json * Fix mypy * Add retry logic for get_data * Add missing break statement in retry decorator * Fix requirements * Finish suggestions by Allen Porter * Suggestions by Allen Porter * Fix typing issues with calls to get_locations and get_sensors
102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
"""Test the LaCrosse View initialization."""
|
|
from datetime import datetime, timedelta
|
|
from unittest.mock import patch
|
|
|
|
from freezegun import freeze_time
|
|
from lacrosse_view import HTTPError, LoginError
|
|
|
|
from homeassistant.components.lacrosse_view.const import DOMAIN
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import MOCK_ENTRY_DATA, TEST_SENSOR
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
async def test_unload_entry(hass: HomeAssistant) -> None:
|
|
"""Test the unload entry."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with patch("lacrosse_view.LaCrosse.login", return_value=True), patch(
|
|
"lacrosse_view.LaCrosse.get_sensors",
|
|
return_value=[TEST_SENSOR],
|
|
):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.data[DOMAIN]
|
|
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert entries
|
|
assert len(entries) == 1
|
|
assert entries[0].state == ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_unload(entries[0].entry_id)
|
|
await hass.async_block_till_done()
|
|
assert entries[0].state == ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
async def test_login_error(hass: HomeAssistant) -> None:
|
|
"""Test login error."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with patch("lacrosse_view.LaCrosse.login", side_effect=LoginError("Test")):
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.data[DOMAIN]
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert entries
|
|
assert len(entries) == 1
|
|
assert entries[0].state == ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_http_error(hass: HomeAssistant) -> None:
|
|
"""Test http error."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with patch("lacrosse_view.LaCrosse.login", return_value=True), patch(
|
|
"lacrosse_view.LaCrosse.get_sensors", side_effect=HTTPError
|
|
):
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.data[DOMAIN]
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert entries
|
|
assert len(entries) == 1
|
|
assert entries[0].state == ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_new_token(hass: HomeAssistant) -> None:
|
|
"""Test new token."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_ENTRY_DATA)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with patch("lacrosse_view.LaCrosse.login", return_value=True) as login, patch(
|
|
"lacrosse_view.LaCrosse.get_sensors",
|
|
return_value=[TEST_SENSOR],
|
|
):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
login.assert_called_once()
|
|
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert entries
|
|
assert len(entries) == 1
|
|
assert entries[0].state == ConfigEntryState.LOADED
|
|
|
|
one_hour_after = datetime.utcnow() + timedelta(hours=1)
|
|
|
|
with patch("lacrosse_view.LaCrosse.login", return_value=True) as login, patch(
|
|
"lacrosse_view.LaCrosse.get_sensors",
|
|
return_value=[TEST_SENSOR],
|
|
), freeze_time(one_hour_after):
|
|
async_fire_time_changed(hass, one_hour_after)
|
|
await hass.async_block_till_done()
|
|
|
|
login.assert_called_once()
|