1afed8ae15
* Add support for Fujitsu HVAC devices * Add the entity code to .coveragerc * Only include code that can fail in the try/except block Co-authored-by: Josef Zweck <24647999+zweckj@users.noreply.github.com> * Remove empty keys from manifest * Remove VERSION as it's already the default * Remve the get_devices function and use asyncio.gather to parallelize dev updates * Move initial step to a function * Let KeyError bubble up. If we are passed an invalid mode it's probably worth raising an exception. * Await the gather * Use the async version of the refresh_auth call * Use the serial number as unique id * Use HA constant for precision * Use dev instead of self._dev * Move to property decorated methods * Remove bidict dependency * Setup one config entry for our api credentials instead of per device * Remove bidict from requirements * Signout and remove our api object on unload * Use app credentials from ayla_iot_unofficial * Use entry_id as a key to store our API object * Delete unused code * Create reverse mappings from forward mapping instead of hardcoding them * Clean up the property methods * Only import part of config_entries we are using * Implement suggested changes * Fix tests to use new API consts * Add support for reauth * Use a coordinator instead of doing per-entity refresh * Auto is equivalent to HEAT_COOL not AUTO * Add ON and OFF to list of supported features * Use the mock_setup_entry fixture for the reauth tests * Parametrize testing of config flow exceptions * Only wrap fallable code in try/except * Add tests for coordinator * Use self.coordinator_context instead of self._dev.device_serial_number * Move timeout to ayla_iot_unofficial * Add description for is_europe field * Bump version of ayla-iot-unofficial * Remove turn_on/turn_off warning * Move coordinator creating to __init__ * Add the type of coordinator to the CoordiatorEntity * Update docstring for FujitsuHVACDevice constructor * Fix missed self._dev to dev * Abort instead of showing the form again with an error when usernames are different * Remove useless argument * Fix tests * Implement some suggestions * Use a device property the maps to the coordinator data * Fix api sign out when unloading the entry * Address comments * Fix device lookup * Move API sign in to coordinator setup * Get rid of FujitsuHVACConfigData * Fix async_setup_entry signature * Fix mock_ayla_api * Cleanup common errors * Add test to check that re adding the same account fails * Also patch new_ayla_api in __init__.py * Create a fixture to generate test devices * Add a setup_integration function that does the setup for a mock config entry * Rework unit tests for the coordinator * Fix typos * Use hass session * Rework reauth config flow to only modify password * Update name to be more use-friendly * Fix wrong type for entry in async_unload_entry * Let TimeoutError bubble up as teh base class handles it * Make the mock ayla api return some devices by default * Move test to test_climate.py * Move tests to test_init.py * Remove reauth flow * Remove useless mock setup * Make our mock devices look real * Fix tests * Rename fujitsu_hvac to fujitsu_fglair and rename the integration to FGLair * Add the Fujitsu brand * Add a helper function to generate an entity_id from a device * Use entity_id to remove hardcoded entity ids * Add a test to increase code coverage --------- Co-authored-by: Josef Zweck <24647999+zweckj@users.noreply.github.com> Co-authored-by: Erik Montnemery <erik@montnemery.com> Co-authored-by: Joostlek <joostlek@outlook.com>
128 lines
4.2 KiB
Python
128 lines
4.2 KiB
Python
"""Test the initialization of fujitsu_fglair entities."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from ayla_iot_unofficial import AylaAuthError
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
|
|
from homeassistant.components.fujitsu_fglair.const import API_REFRESH, DOMAIN
|
|
from homeassistant.const import STATE_UNAVAILABLE, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import entity_id, setup_integration
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
async def test_auth_failure(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_devices: list[AsyncMock],
|
|
) -> None:
|
|
"""Test entities become unavailable after auth failure."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
mock_ayla_api.async_get_devices.side_effect = AylaAuthError
|
|
freezer.tick(API_REFRESH)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(entity_id(mock_devices[0])).state == STATE_UNAVAILABLE
|
|
assert hass.states.get(entity_id(mock_devices[1])).state == STATE_UNAVAILABLE
|
|
|
|
|
|
async def test_device_auth_failure(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_devices: list[AsyncMock],
|
|
) -> None:
|
|
"""Test entities become unavailable after auth failure with updating devices."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
for d in mock_ayla_api.async_get_devices.return_value:
|
|
d.async_update.side_effect = AylaAuthError
|
|
|
|
freezer.tick(API_REFRESH)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(entity_id(mock_devices[0])).state == STATE_UNAVAILABLE
|
|
assert hass.states.get(entity_id(mock_devices[1])).state == STATE_UNAVAILABLE
|
|
|
|
|
|
async def test_token_expired(
|
|
hass: HomeAssistant,
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Make sure sign_in is called if the token expired."""
|
|
mock_ayla_api.token_expired = True
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
# Called once during setup and once during update
|
|
assert mock_ayla_api.async_sign_in.call_count == 2
|
|
|
|
|
|
async def test_token_expiring_soon(
|
|
hass: HomeAssistant,
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Make sure sign_in is called if the token expired."""
|
|
mock_ayla_api.token_expiring_soon = True
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
mock_ayla_api.async_refresh_auth.assert_called_once()
|
|
|
|
|
|
@pytest.mark.parametrize("exception", [AylaAuthError, TimeoutError])
|
|
async def test_startup_exception(
|
|
hass: HomeAssistant,
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
) -> None:
|
|
"""Make sure that no devices are added if there was an exception while logging in."""
|
|
mock_ayla_api.async_sign_in.side_effect = exception
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert len(hass.states.async_all()) == 0
|
|
|
|
|
|
async def test_one_device_disabled(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
freezer: FrozenDateTimeFactory,
|
|
mock_devices: list[AsyncMock],
|
|
mock_ayla_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test that coordinator only updates devices that are currently listening."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
for d in mock_devices:
|
|
d.async_update.assert_called_once()
|
|
d.reset_mock()
|
|
|
|
entity = entity_registry.async_get(
|
|
entity_registry.async_get_entity_id(
|
|
Platform.CLIMATE, DOMAIN, mock_devices[0].device_serial_number
|
|
)
|
|
)
|
|
entity_registry.async_update_entity(
|
|
entity.entity_id, disabled_by=er.RegistryEntryDisabler.USER
|
|
)
|
|
await hass.async_block_till_done()
|
|
freezer.tick(API_REFRESH)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all()) == len(mock_devices) - 1
|
|
mock_devices[0].async_update.assert_not_called()
|
|
mock_devices[1].async_update.assert_called_once()
|