mirror of https://github.com/home-assistant/core
111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
"""Tests for the initialization of the A. O. Smith integration."""
|
|
|
|
from datetime import timedelta
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
from py_aosmith import AOSmithUnknownException
|
|
import pytest
|
|
|
|
from homeassistant.components.aosmith.const import (
|
|
DOMAIN,
|
|
FAST_INTERVAL,
|
|
REGULAR_INTERVAL,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .conftest import build_device_fixture
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
async def test_config_entry_setup(init_integration: MockConfigEntry) -> None:
|
|
"""Test setup of the config entry."""
|
|
mock_config_entry = init_integration
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
|
|
|
|
async def test_config_entry_not_ready_get_devices_error(
|
|
hass: HomeAssistant, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test the config entry not ready when get_devices fails."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.aosmith.config_flow.AOSmithAPIClient.get_devices",
|
|
side_effect=AOSmithUnknownException("Unknown error"),
|
|
):
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_config_entry_not_ready_get_energy_use_data_error(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test the config entry not ready when get_energy_use_data fails."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
get_devices_fixture = [
|
|
build_device_fixture(
|
|
heat_pump=True,
|
|
mode_pending=False,
|
|
setpoint_pending=False,
|
|
has_vacation_mode=True,
|
|
)
|
|
]
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.aosmith.config_flow.AOSmithAPIClient.get_devices",
|
|
return_value=get_devices_fixture,
|
|
),
|
|
patch(
|
|
"homeassistant.components.aosmith.config_flow.AOSmithAPIClient.get_energy_use_data",
|
|
side_effect=AOSmithUnknownException("Unknown error"),
|
|
),
|
|
):
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
(
|
|
"get_devices_fixture_mode_pending",
|
|
"get_devices_fixture_setpoint_pending",
|
|
"time_to_wait",
|
|
"expected_call_count",
|
|
),
|
|
[
|
|
(False, False, REGULAR_INTERVAL, 1),
|
|
(False, False, FAST_INTERVAL, 0),
|
|
(True, False, FAST_INTERVAL, 1),
|
|
(False, True, FAST_INTERVAL, 1),
|
|
],
|
|
)
|
|
async def test_update(
|
|
freezer: FrozenDateTimeFactory,
|
|
hass: HomeAssistant,
|
|
mock_client: MagicMock,
|
|
init_integration: MockConfigEntry,
|
|
time_to_wait: timedelta,
|
|
expected_call_count: int,
|
|
) -> None:
|
|
"""Test data update with differing intervals depending on device status."""
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert len(entries) == 1
|
|
assert entries[0].state is ConfigEntryState.LOADED
|
|
assert mock_client.get_devices.call_count == 1
|
|
|
|
freezer.tick(time_to_wait)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_client.get_devices.call_count == 1 + expected_call_count
|