From 1afed8ae15934aa1998d9a05ecf6705059296da5 Mon Sep 17 00:00:00 2001
From: Antoine Reversat <a.reversat@gmail.com>
Date: Sun, 18 Aug 2024 09:37:33 -0400
Subject: [PATCH] Add Fujitsu FGLair integration (#109335)
* 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>
---
.strict-typing | 1 +
CODEOWNERS | 2 +
homeassistant/brands/fujitsu.json | 5 +
.../components/fujitsu_fglair/__init__.py | 49 +++++
.../components/fujitsu_fglair/climate.py | 141 +++++++++++++
.../components/fujitsu_fglair/config_flow.py | 73 +++++++
.../components/fujitsu_fglair/const.py | 54 +++++
.../components/fujitsu_fglair/coordinator.py | 63 ++++++
.../components/fujitsu_fglair/manifest.json | 9 +
.../components/fujitsu_fglair/strings.json | 25 +++
homeassistant/generated/config_flows.py | 1 +
homeassistant/generated/integrations.json | 20 +-
mypy.ini | 10 +
requirements_all.txt | 3 +
requirements_test_all.txt | 3 +
tests/components/fujitsu_fglair/__init__.py | 21 ++
tests/components/fujitsu_fglair/conftest.py | 113 +++++++++++
.../snapshots/test_climate.ambr | 189 ++++++++++++++++++
.../components/fujitsu_fglair/test_climate.py | 98 +++++++++
.../fujitsu_fglair/test_config_flow.py | 107 ++++++++++
tests/components/fujitsu_fglair/test_init.py | 128 ++++++++++++
21 files changed, 1111 insertions(+), 4 deletions(-)
create mode 100644 homeassistant/brands/fujitsu.json
create mode 100644 homeassistant/components/fujitsu_fglair/__init__.py
create mode 100644 homeassistant/components/fujitsu_fglair/climate.py
create mode 100644 homeassistant/components/fujitsu_fglair/config_flow.py
create mode 100644 homeassistant/components/fujitsu_fglair/const.py
create mode 100644 homeassistant/components/fujitsu_fglair/coordinator.py
create mode 100644 homeassistant/components/fujitsu_fglair/manifest.json
create mode 100644 homeassistant/components/fujitsu_fglair/strings.json
create mode 100644 tests/components/fujitsu_fglair/__init__.py
create mode 100644 tests/components/fujitsu_fglair/conftest.py
create mode 100644 tests/components/fujitsu_fglair/snapshots/test_climate.ambr
create mode 100644 tests/components/fujitsu_fglair/test_climate.py
create mode 100644 tests/components/fujitsu_fglair/test_config_flow.py
create mode 100644 tests/components/fujitsu_fglair/test_init.py
diff --git a/.strict-typing b/.strict-typing
index 51ca93bb9fa..c8f28c84f8a 100644
--- a/.strict-typing
+++ b/.strict-typing
@@ -196,6 +196,7 @@ homeassistant.components.fritzbox.*
homeassistant.components.fritzbox_callmonitor.*
homeassistant.components.fronius.*
homeassistant.components.frontend.*
+homeassistant.components.fujitsu_fglair.*
homeassistant.components.fully_kiosk.*
homeassistant.components.fyta.*
homeassistant.components.generic_hygrostat.*
diff --git a/CODEOWNERS b/CODEOWNERS
index 6593c02c8a5..367c6eee2bf 100644
--- a/CODEOWNERS
+++ b/CODEOWNERS
@@ -499,6 +499,8 @@ build.json @home-assistant/supervisor
/tests/components/frontend/ @home-assistant/frontend
/homeassistant/components/frontier_silicon/ @wlcrs
/tests/components/frontier_silicon/ @wlcrs
+/homeassistant/components/fujitsu_fglair/ @crevetor
+/tests/components/fujitsu_fglair/ @crevetor
/homeassistant/components/fully_kiosk/ @cgarwood
/tests/components/fully_kiosk/ @cgarwood
/homeassistant/components/fyta/ @dontinelli
diff --git a/homeassistant/brands/fujitsu.json b/homeassistant/brands/fujitsu.json
new file mode 100644
index 00000000000..75d12e33851
--- /dev/null
+++ b/homeassistant/brands/fujitsu.json
@@ -0,0 +1,5 @@
+{
+ "domain": "fujitsu",
+ "name": "Fujitsu",
+ "integrations": ["fujitsu_anywair", "fujitsu_fglair"]
+}
diff --git a/homeassistant/components/fujitsu_fglair/__init__.py b/homeassistant/components/fujitsu_fglair/__init__.py
new file mode 100644
index 00000000000..bd891f05b8d
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/__init__.py
@@ -0,0 +1,49 @@
+"""The Fujitsu HVAC (based on Ayla IOT) integration."""
+
+from __future__ import annotations
+
+from contextlib import suppress
+
+from ayla_iot_unofficial import new_ayla_api
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import aiohttp_client
+
+from .const import API_TIMEOUT, CONF_EUROPE, FGLAIR_APP_ID, FGLAIR_APP_SECRET
+from .coordinator import FGLairCoordinator
+
+PLATFORMS: list[Platform] = [Platform.CLIMATE]
+
+type FGLairConfigEntry = ConfigEntry[FGLairCoordinator]
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: FGLairConfigEntry) -> bool:
+ """Set up Fujitsu HVAC (based on Ayla IOT) from a config entry."""
+ api = new_ayla_api(
+ entry.data[CONF_USERNAME],
+ entry.data[CONF_PASSWORD],
+ FGLAIR_APP_ID,
+ FGLAIR_APP_SECRET,
+ europe=entry.data[CONF_EUROPE],
+ websession=aiohttp_client.async_get_clientsession(hass),
+ timeout=API_TIMEOUT,
+ )
+
+ coordinator = FGLairCoordinator(hass, api)
+ await coordinator.async_config_entry_first_refresh()
+
+ entry.runtime_data = coordinator
+
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: FGLairConfigEntry) -> bool:
+ """Unload a config entry."""
+ unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+ with suppress(TimeoutError):
+ await entry.runtime_data.api.async_sign_out()
+
+ return unload_ok
diff --git a/homeassistant/components/fujitsu_fglair/climate.py b/homeassistant/components/fujitsu_fglair/climate.py
new file mode 100644
index 00000000000..558f4b73a18
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/climate.py
@@ -0,0 +1,141 @@
+"""Support for Fujitsu HVAC devices that use the Ayla Iot platform."""
+
+from typing import Any
+
+from ayla_iot_unofficial.fujitsu_hvac import Capability, FujitsuHVAC
+
+from homeassistant.components.climate import (
+ ClimateEntity,
+ ClimateEntityFeature,
+ HVACMode,
+)
+from homeassistant.const import ATTR_TEMPERATURE, PRECISION_HALVES, UnitOfTemperature
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from . import FGLairConfigEntry
+from .const import (
+ DOMAIN,
+ FUJI_TO_HA_FAN,
+ FUJI_TO_HA_HVAC,
+ FUJI_TO_HA_SWING,
+ HA_TO_FUJI_FAN,
+ HA_TO_FUJI_HVAC,
+ HA_TO_FUJI_SWING,
+)
+from .coordinator import FGLairCoordinator
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: FGLairConfigEntry,
+ async_add_entities: AddEntitiesCallback,
+) -> None:
+ """Set up one Fujitsu HVAC device."""
+ async_add_entities(
+ FGLairDevice(entry.runtime_data, device)
+ for device in entry.runtime_data.data.values()
+ )
+
+
+class FGLairDevice(CoordinatorEntity[FGLairCoordinator], ClimateEntity):
+ """Represent a Fujitsu HVAC device."""
+
+ _attr_temperature_unit = UnitOfTemperature.CELSIUS
+ _attr_precision = PRECISION_HALVES
+ _attr_target_temperature_step = 0.5
+ _attr_has_entity_name = True
+ _attr_name = None
+
+ _enable_turn_on_off_backwards_compatibility: bool = False
+
+ def __init__(self, coordinator: FGLairCoordinator, device: FujitsuHVAC) -> None:
+ """Store the representation of the device and set the static attributes."""
+ super().__init__(coordinator, context=device.device_serial_number)
+
+ self._attr_unique_id = device.device_serial_number
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, device.device_serial_number)},
+ name=device.device_name,
+ manufacturer="Fujitsu",
+ model=device.property_values["model_name"],
+ serial_number=device.device_serial_number,
+ sw_version=device.property_values["mcu_firmware_version"],
+ )
+
+ self._attr_supported_features = (
+ ClimateEntityFeature.TARGET_TEMPERATURE
+ | ClimateEntityFeature.TURN_ON
+ | ClimateEntityFeature.TURN_OFF
+ )
+ if device.has_capability(Capability.OP_FAN):
+ self._attr_supported_features |= ClimateEntityFeature.FAN_MODE
+
+ if device.has_capability(Capability.SWING_HORIZONTAL) or device.has_capability(
+ Capability.SWING_VERTICAL
+ ):
+ self._attr_supported_features |= ClimateEntityFeature.SWING_MODE
+ self._set_attr()
+
+ @property
+ def device(self) -> FujitsuHVAC:
+ """Return the device object from the coordinator data."""
+ return self.coordinator.data[self.coordinator_context]
+
+ @property
+ def available(self) -> bool:
+ """Return if the device is available."""
+ return super().available and self.coordinator_context in self.coordinator.data
+
+ async def async_set_fan_mode(self, fan_mode: str) -> None:
+ """Set Fan mode."""
+ await self.device.async_set_fan_speed(HA_TO_FUJI_FAN[fan_mode])
+ await self.coordinator.async_request_refresh()
+
+ async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
+ """Set HVAC mode."""
+ await self.device.async_set_op_mode(HA_TO_FUJI_HVAC[hvac_mode])
+ await self.coordinator.async_request_refresh()
+
+ async def async_set_swing_mode(self, swing_mode: str) -> None:
+ """Set swing mode."""
+ await self.device.async_set_swing_mode(HA_TO_FUJI_SWING[swing_mode])
+ await self.coordinator.async_request_refresh()
+
+ async def async_set_temperature(self, **kwargs: Any) -> None:
+ """Set target temperature."""
+ if (temperature := kwargs.get(ATTR_TEMPERATURE)) is None:
+ return
+ await self.device.async_set_set_temp(temperature)
+ await self.coordinator.async_request_refresh()
+
+ def _set_attr(self) -> None:
+ if self.coordinator_context in self.coordinator.data:
+ self._attr_fan_mode = FUJI_TO_HA_FAN.get(self.device.fan_speed)
+ self._attr_fan_modes = [
+ FUJI_TO_HA_FAN[mode]
+ for mode in self.device.supported_fan_speeds
+ if mode in FUJI_TO_HA_FAN
+ ]
+ self._attr_hvac_mode = FUJI_TO_HA_HVAC.get(self.device.op_mode)
+ self._attr_hvac_modes = [
+ FUJI_TO_HA_HVAC[mode]
+ for mode in self.device.supported_op_modes
+ if mode in FUJI_TO_HA_HVAC
+ ]
+ self._attr_swing_mode = FUJI_TO_HA_SWING.get(self.device.swing_mode)
+ self._attr_swing_modes = [
+ FUJI_TO_HA_SWING[mode]
+ for mode in self.device.supported_swing_modes
+ if mode in FUJI_TO_HA_SWING
+ ]
+ self._attr_min_temp = self.device.temperature_range[0]
+ self._attr_max_temp = self.device.temperature_range[1]
+ self._attr_current_temperature = self.device.sensed_temp
+ self._attr_target_temperature = self.device.set_temp
+
+ def _handle_coordinator_update(self) -> None:
+ self._set_attr()
+ super()._handle_coordinator_update()
diff --git a/homeassistant/components/fujitsu_fglair/config_flow.py b/homeassistant/components/fujitsu_fglair/config_flow.py
new file mode 100644
index 00000000000..10f703df6d9
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/config_flow.py
@@ -0,0 +1,73 @@
+"""Config flow for Fujitsu HVAC (based on Ayla IOT) integration."""
+
+import logging
+from typing import Any
+
+from ayla_iot_unofficial import AylaAuthError, new_ayla_api
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.helpers import aiohttp_client
+
+from .const import API_TIMEOUT, CONF_EUROPE, DOMAIN, FGLAIR_APP_ID, FGLAIR_APP_SECRET
+
+_LOGGER = logging.getLogger(__name__)
+
+
+STEP_USER_DATA_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_USERNAME): str,
+ vol.Required(CONF_PASSWORD): str,
+ vol.Required(CONF_EUROPE): bool,
+ }
+)
+
+
+class FGLairConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for Fujitsu HVAC (based on Ayla IOT)."""
+
+ async def _async_validate_credentials(
+ self, user_input: dict[str, Any]
+ ) -> dict[str, str]:
+ errors: dict[str, str] = {}
+ api = new_ayla_api(
+ user_input[CONF_USERNAME],
+ user_input[CONF_PASSWORD],
+ FGLAIR_APP_ID,
+ FGLAIR_APP_SECRET,
+ europe=user_input[CONF_EUROPE],
+ websession=aiohttp_client.async_get_clientsession(self.hass),
+ timeout=API_TIMEOUT,
+ )
+ try:
+ await api.async_sign_in()
+ except TimeoutError:
+ errors["base"] = "cannot_connect"
+ except AylaAuthError:
+ errors["base"] = "invalid_auth"
+ except Exception: # pylint: disable=broad-except
+ _LOGGER.exception("Unexpected exception")
+ errors["base"] = "unknown"
+
+ return errors
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle the initial step."""
+ errors: dict[str, str] = {}
+ if user_input:
+ await self.async_set_unique_id(user_input[CONF_USERNAME].lower())
+ self._abort_if_unique_id_configured()
+
+ errors = await self._async_validate_credentials(user_input)
+ if len(errors) == 0:
+ return self.async_create_entry(
+ title=f"FGLair ({user_input[CONF_USERNAME]})",
+ data=user_input,
+ )
+
+ return self.async_show_form(
+ step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
+ )
diff --git a/homeassistant/components/fujitsu_fglair/const.py b/homeassistant/components/fujitsu_fglair/const.py
new file mode 100644
index 00000000000..0e93361f20b
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/const.py
@@ -0,0 +1,54 @@
+"""Constants for the Fujitsu HVAC (based on Ayla IOT) integration."""
+
+from datetime import timedelta
+
+from ayla_iot_unofficial.fujitsu_consts import ( # noqa: F401
+ FGLAIR_APP_ID,
+ FGLAIR_APP_SECRET,
+)
+from ayla_iot_unofficial.fujitsu_hvac import FanSpeed, OpMode, SwingMode
+
+from homeassistant.components.climate import (
+ FAN_AUTO,
+ FAN_HIGH,
+ FAN_LOW,
+ FAN_MEDIUM,
+ SWING_BOTH,
+ SWING_HORIZONTAL,
+ SWING_OFF,
+ SWING_VERTICAL,
+ HVACMode,
+)
+
+API_TIMEOUT = 10
+API_REFRESH = timedelta(minutes=5)
+
+DOMAIN = "fujitsu_fglair"
+
+CONF_EUROPE = "is_europe"
+
+HA_TO_FUJI_FAN = {
+ FAN_LOW: FanSpeed.LOW,
+ FAN_MEDIUM: FanSpeed.MEDIUM,
+ FAN_HIGH: FanSpeed.HIGH,
+ FAN_AUTO: FanSpeed.AUTO,
+}
+FUJI_TO_HA_FAN = {value: key for key, value in HA_TO_FUJI_FAN.items()}
+
+HA_TO_FUJI_HVAC = {
+ HVACMode.OFF: OpMode.OFF,
+ HVACMode.HEAT: OpMode.HEAT,
+ HVACMode.COOL: OpMode.COOL,
+ HVACMode.HEAT_COOL: OpMode.AUTO,
+ HVACMode.DRY: OpMode.DRY,
+ HVACMode.FAN_ONLY: OpMode.FAN,
+}
+FUJI_TO_HA_HVAC = {value: key for key, value in HA_TO_FUJI_HVAC.items()}
+
+HA_TO_FUJI_SWING = {
+ SWING_OFF: SwingMode.OFF,
+ SWING_VERTICAL: SwingMode.SWING_VERTICAL,
+ SWING_HORIZONTAL: SwingMode.SWING_HORIZONTAL,
+ SWING_BOTH: SwingMode.SWING_BOTH,
+}
+FUJI_TO_HA_SWING = {value: key for key, value in HA_TO_FUJI_SWING.items()}
diff --git a/homeassistant/components/fujitsu_fglair/coordinator.py b/homeassistant/components/fujitsu_fglair/coordinator.py
new file mode 100644
index 00000000000..267c0b2c3e5
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/coordinator.py
@@ -0,0 +1,63 @@
+"""Coordinator for Fujitsu HVAC integration."""
+
+import logging
+
+from ayla_iot_unofficial import AylaApi, AylaAuthError
+from ayla_iot_unofficial.fujitsu_hvac import FujitsuHVAC
+
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryError
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+
+from .const import API_REFRESH
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class FGLairCoordinator(DataUpdateCoordinator[dict[str, FujitsuHVAC]]):
+ """Coordinator for Fujitsu HVAC integration."""
+
+ def __init__(self, hass: HomeAssistant, api: AylaApi) -> None:
+ """Initialize coordinator for Fujitsu HVAC integration."""
+ super().__init__(
+ hass,
+ _LOGGER,
+ name="Fujitsu HVAC data",
+ update_interval=API_REFRESH,
+ )
+ self.api = api
+
+ async def _async_setup(self) -> None:
+ try:
+ await self.api.async_sign_in()
+ except AylaAuthError as e:
+ raise ConfigEntryError("Credentials expired for Ayla IoT API") from e
+
+ async def _async_update_data(self) -> dict[str, FujitsuHVAC]:
+ """Fetch data from api endpoint."""
+ listening_entities = set(self.async_contexts())
+ try:
+ if self.api.token_expired:
+ await self.api.async_sign_in()
+
+ if self.api.token_expiring_soon:
+ await self.api.async_refresh_auth()
+
+ devices = await self.api.async_get_devices()
+ except AylaAuthError as e:
+ raise ConfigEntryError("Credentials expired for Ayla IoT API") from e
+
+ if len(listening_entities) == 0:
+ devices = list(filter(lambda x: isinstance(x, FujitsuHVAC), devices))
+ else:
+ devices = list(
+ filter(lambda x: x.device_serial_number in listening_entities, devices)
+ )
+
+ try:
+ for dev in devices:
+ await dev.async_update()
+ except AylaAuthError as e:
+ raise ConfigEntryError("Credentials expired for Ayla IoT API") from e
+
+ return {d.device_serial_number: d for d in devices}
diff --git a/homeassistant/components/fujitsu_fglair/manifest.json b/homeassistant/components/fujitsu_fglair/manifest.json
new file mode 100644
index 00000000000..9286f7c24d9
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/manifest.json
@@ -0,0 +1,9 @@
+{
+ "domain": "fujitsu_fglair",
+ "name": "FGLair",
+ "codeowners": ["@crevetor"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/fujitsu_fglair",
+ "iot_class": "cloud_polling",
+ "requirements": ["ayla-iot-unofficial==1.3.1"]
+}
diff --git a/homeassistant/components/fujitsu_fglair/strings.json b/homeassistant/components/fujitsu_fglair/strings.json
new file mode 100644
index 00000000000..71d8542cd3e
--- /dev/null
+++ b/homeassistant/components/fujitsu_fglair/strings.json
@@ -0,0 +1,25 @@
+{
+ "config": {
+ "step": {
+ "user": {
+ "title": "Enter your FGLair credentials",
+ "data": {
+ "is_europe": "Use european servers",
+ "username": "[%key:common::config_flow::data::username%]",
+ "password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "is_europe": "Allows the user to choose whether to use european servers or not since the API uses different endoint URLs for european vs non-european users"
+ }
+ }
+ },
+ "error": {
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+ "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
+ "unknown": "[%key:common::config_flow::error::unknown%]"
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+ }
+ }
+}
diff --git a/homeassistant/generated/config_flows.py b/homeassistant/generated/config_flows.py
index c3fe4af4a76..b474fbaf54f 100644
--- a/homeassistant/generated/config_flows.py
+++ b/homeassistant/generated/config_flows.py
@@ -200,6 +200,7 @@ FLOWS = {
"fritzbox_callmonitor",
"fronius",
"frontier_silicon",
+ "fujitsu_fglair",
"fully_kiosk",
"fyta",
"garages_amsterdam",
diff --git a/homeassistant/generated/integrations.json b/homeassistant/generated/integrations.json
index a61e58d80d4..9e1250e3b60 100644
--- a/homeassistant/generated/integrations.json
+++ b/homeassistant/generated/integrations.json
@@ -2054,10 +2054,22 @@
"config_flow": true,
"iot_class": "local_polling"
},
- "fujitsu_anywair": {
- "name": "Fujitsu anywAIR",
- "integration_type": "virtual",
- "supported_by": "advantage_air"
+ "fujitsu": {
+ "name": "Fujitsu",
+ "integrations": {
+ "fujitsu_anywair": {
+ "integration_type": "virtual",
+ "config_flow": false,
+ "supported_by": "advantage_air",
+ "name": "Fujitsu anywAIR"
+ },
+ "fujitsu_fglair": {
+ "integration_type": "hub",
+ "config_flow": true,
+ "iot_class": "cloud_polling",
+ "name": "FGLair"
+ }
+ }
},
"fully_kiosk": {
"name": "Fully Kiosk Browser",
diff --git a/mypy.ini b/mypy.ini
index f0a941f20eb..ca10d05af86 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -1716,6 +1716,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.fujitsu_fglair.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.fully_kiosk.*]
check_untyped_defs = true
disallow_incomplete_defs = true
diff --git a/requirements_all.txt b/requirements_all.txt
index bdaa2ecdd91..0988a50a4cc 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -522,6 +522,9 @@ autarco==2.0.0
# homeassistant.components.axis
axis==62
+# homeassistant.components.fujitsu_fglair
+ayla-iot-unofficial==1.3.1
+
# homeassistant.components.azure_event_hub
azure-eventhub==5.11.1
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 8bfd4ec095f..7a27406deae 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -471,6 +471,9 @@ autarco==2.0.0
# homeassistant.components.axis
axis==62
+# homeassistant.components.fujitsu_fglair
+ayla-iot-unofficial==1.3.1
+
# homeassistant.components.azure_event_hub
azure-eventhub==5.11.1
diff --git a/tests/components/fujitsu_fglair/__init__.py b/tests/components/fujitsu_fglair/__init__.py
new file mode 100644
index 00000000000..2ec3fa0fce6
--- /dev/null
+++ b/tests/components/fujitsu_fglair/__init__.py
@@ -0,0 +1,21 @@
+"""Tests for the Fujitsu HVAC (based on Ayla IOT) integration."""
+
+from ayla_iot_unofficial.fujitsu_hvac import FujitsuHVAC
+
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry
+
+
+async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
+ """Fixture for setting up the component."""
+ config_entry.add_to_hass(hass)
+
+ await hass.config_entries.async_setup(config_entry.entry_id)
+ await hass.async_block_till_done()
+
+
+def entity_id(device: FujitsuHVAC) -> str:
+ """Generate the entity id for the given serial."""
+ return f"{Platform.CLIMATE}.{device.device_serial_number}"
diff --git a/tests/components/fujitsu_fglair/conftest.py b/tests/components/fujitsu_fglair/conftest.py
new file mode 100644
index 00000000000..b73007a566b
--- /dev/null
+++ b/tests/components/fujitsu_fglair/conftest.py
@@ -0,0 +1,113 @@
+"""Common fixtures for the Fujitsu HVAC (based on Ayla IOT) tests."""
+
+from collections.abc import Generator
+from unittest.mock import AsyncMock, create_autospec, patch
+
+from ayla_iot_unofficial import AylaApi
+from ayla_iot_unofficial.fujitsu_hvac import FanSpeed, FujitsuHVAC, OpMode, SwingMode
+import pytest
+
+from homeassistant.components.fujitsu_fglair.const import CONF_EUROPE, DOMAIN
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+
+from tests.common import MockConfigEntry
+
+TEST_DEVICE_NAME = "Test device"
+TEST_DEVICE_SERIAL = "testserial"
+TEST_USERNAME = "test-username"
+TEST_PASSWORD = "test-password"
+
+TEST_USERNAME2 = "test-username2"
+TEST_PASSWORD2 = "test-password2"
+
+TEST_SERIAL_NUMBER = "testserial123"
+TEST_SERIAL_NUMBER2 = "testserial345"
+
+TEST_PROPERTY_VALUES = {
+ "model_name": "mock_fujitsu_device",
+ "mcu_firmware_version": "1",
+}
+
+
+@pytest.fixture
+def mock_setup_entry() -> Generator[AsyncMock, None, None]:
+ """Override async_setup_entry."""
+ with patch(
+ "homeassistant.components.fujitsu_fglair.async_setup_entry", return_value=True
+ ) as mock_setup_entry:
+ yield mock_setup_entry
+
+
+@pytest.fixture
+def mock_ayla_api(mock_devices: list[AsyncMock]) -> Generator[AsyncMock]:
+ """Override AylaApi creation."""
+ my_mock = create_autospec(AylaApi)
+
+ with (
+ patch(
+ "homeassistant.components.fujitsu_fglair.new_ayla_api", return_value=my_mock
+ ),
+ patch(
+ "homeassistant.components.fujitsu_fglair.config_flow.new_ayla_api",
+ return_value=my_mock,
+ ),
+ ):
+ my_mock.async_get_devices.return_value = mock_devices
+ yield my_mock
+
+
+@pytest.fixture
+def mock_config_entry() -> MockConfigEntry:
+ """Return a regular config entry."""
+ return MockConfigEntry(
+ domain=DOMAIN,
+ unique_id=TEST_USERNAME,
+ data={
+ CONF_USERNAME: TEST_USERNAME,
+ CONF_PASSWORD: TEST_PASSWORD,
+ CONF_EUROPE: False,
+ },
+ )
+
+
+def _create_device(serial_number: str) -> AsyncMock:
+ dev = AsyncMock(spec=FujitsuHVAC)
+ dev.device_serial_number = serial_number
+ dev.device_name = serial_number
+ dev.property_values = TEST_PROPERTY_VALUES
+ dev.has_capability.return_value = True
+ dev.fan_speed = FanSpeed.AUTO
+ dev.supported_fan_speeds = [
+ FanSpeed.LOW,
+ FanSpeed.MEDIUM,
+ FanSpeed.HIGH,
+ FanSpeed.AUTO,
+ ]
+ dev.op_mode = OpMode.COOL
+ dev.supported_op_modes = [
+ OpMode.OFF,
+ OpMode.ON,
+ OpMode.AUTO,
+ OpMode.COOL,
+ OpMode.DRY,
+ ]
+ dev.swing_mode = SwingMode.SWING_BOTH
+ dev.supported_swing_modes = [
+ SwingMode.OFF,
+ SwingMode.SWING_HORIZONTAL,
+ SwingMode.SWING_VERTICAL,
+ SwingMode.SWING_BOTH,
+ ]
+ dev.temperature_range = [18.0, 26.0]
+ dev.sensed_temp = 22.0
+ dev.set_temp = 21.0
+
+ return dev
+
+
+@pytest.fixture
+def mock_devices() -> list[AsyncMock]:
+ """Generate a list of mock devices that the API can return."""
+ return [
+ _create_device(serial) for serial in (TEST_SERIAL_NUMBER, TEST_SERIAL_NUMBER2)
+ ]
diff --git a/tests/components/fujitsu_fglair/snapshots/test_climate.ambr b/tests/components/fujitsu_fglair/snapshots/test_climate.ambr
new file mode 100644
index 00000000000..31b143c6f95
--- /dev/null
+++ b/tests/components/fujitsu_fglair/snapshots/test_climate.ambr
@@ -0,0 +1,189 @@
+# serializer version: 1
+# name: test_entities[climate.testserial123-entry]
+ EntityRegistryEntrySnapshot({
+ 'aliases': set({
+ }),
+ 'area_id': None,
+ 'capabilities': dict({
+ 'fan_modes': list([
+ 'low',
+ 'medium',
+ 'high',
+ 'auto',
+ ]),
+ 'hvac_modes': list([
+ <HVACMode.OFF: 'off'>,
+ <HVACMode.HEAT_COOL: 'heat_cool'>,
+ <HVACMode.COOL: 'cool'>,
+ <HVACMode.DRY: 'dry'>,
+ ]),
+ 'max_temp': 26.0,
+ 'min_temp': 18.0,
+ 'swing_modes': list([
+ 'off',
+ 'horizontal',
+ 'vertical',
+ 'both',
+ ]),
+ 'target_temp_step': 0.5,
+ }),
+ 'config_entry_id': <ANY>,
+ 'device_class': None,
+ 'device_id': <ANY>,
+ 'disabled_by': None,
+ 'domain': 'climate',
+ 'entity_category': None,
+ 'entity_id': 'climate.testserial123',
+ 'has_entity_name': True,
+ 'hidden_by': None,
+ 'icon': None,
+ 'id': <ANY>,
+ 'labels': set({
+ }),
+ 'name': None,
+ 'options': dict({
+ }),
+ 'original_device_class': None,
+ 'original_icon': None,
+ 'original_name': None,
+ 'platform': 'fujitsu_fglair',
+ 'previous_unique_id': None,
+ 'supported_features': <ClimateEntityFeature: 425>,
+ 'translation_key': None,
+ 'unique_id': 'testserial123',
+ 'unit_of_measurement': None,
+ })
+# ---
+# name: test_entities[climate.testserial123-state]
+ StateSnapshot({
+ 'attributes': ReadOnlyDict({
+ 'current_temperature': 22.0,
+ 'fan_mode': 'auto',
+ 'fan_modes': list([
+ 'low',
+ 'medium',
+ 'high',
+ 'auto',
+ ]),
+ 'friendly_name': 'testserial123',
+ 'hvac_modes': list([
+ <HVACMode.OFF: 'off'>,
+ <HVACMode.HEAT_COOL: 'heat_cool'>,
+ <HVACMode.COOL: 'cool'>,
+ <HVACMode.DRY: 'dry'>,
+ ]),
+ 'max_temp': 26.0,
+ 'min_temp': 18.0,
+ 'supported_features': <ClimateEntityFeature: 425>,
+ 'swing_mode': 'both',
+ 'swing_modes': list([
+ 'off',
+ 'horizontal',
+ 'vertical',
+ 'both',
+ ]),
+ 'target_temp_step': 0.5,
+ 'temperature': 21.0,
+ }),
+ 'context': <ANY>,
+ 'entity_id': 'climate.testserial123',
+ 'last_changed': <ANY>,
+ 'last_reported': <ANY>,
+ 'last_updated': <ANY>,
+ 'state': 'cool',
+ })
+# ---
+# name: test_entities[climate.testserial345-entry]
+ EntityRegistryEntrySnapshot({
+ 'aliases': set({
+ }),
+ 'area_id': None,
+ 'capabilities': dict({
+ 'fan_modes': list([
+ 'low',
+ 'medium',
+ 'high',
+ 'auto',
+ ]),
+ 'hvac_modes': list([
+ <HVACMode.OFF: 'off'>,
+ <HVACMode.HEAT_COOL: 'heat_cool'>,
+ <HVACMode.COOL: 'cool'>,
+ <HVACMode.DRY: 'dry'>,
+ ]),
+ 'max_temp': 26.0,
+ 'min_temp': 18.0,
+ 'swing_modes': list([
+ 'off',
+ 'horizontal',
+ 'vertical',
+ 'both',
+ ]),
+ 'target_temp_step': 0.5,
+ }),
+ 'config_entry_id': <ANY>,
+ 'device_class': None,
+ 'device_id': <ANY>,
+ 'disabled_by': None,
+ 'domain': 'climate',
+ 'entity_category': None,
+ 'entity_id': 'climate.testserial345',
+ 'has_entity_name': True,
+ 'hidden_by': None,
+ 'icon': None,
+ 'id': <ANY>,
+ 'labels': set({
+ }),
+ 'name': None,
+ 'options': dict({
+ }),
+ 'original_device_class': None,
+ 'original_icon': None,
+ 'original_name': None,
+ 'platform': 'fujitsu_fglair',
+ 'previous_unique_id': None,
+ 'supported_features': <ClimateEntityFeature: 425>,
+ 'translation_key': None,
+ 'unique_id': 'testserial345',
+ 'unit_of_measurement': None,
+ })
+# ---
+# name: test_entities[climate.testserial345-state]
+ StateSnapshot({
+ 'attributes': ReadOnlyDict({
+ 'current_temperature': 22.0,
+ 'fan_mode': 'auto',
+ 'fan_modes': list([
+ 'low',
+ 'medium',
+ 'high',
+ 'auto',
+ ]),
+ 'friendly_name': 'testserial345',
+ 'hvac_modes': list([
+ <HVACMode.OFF: 'off'>,
+ <HVACMode.HEAT_COOL: 'heat_cool'>,
+ <HVACMode.COOL: 'cool'>,
+ <HVACMode.DRY: 'dry'>,
+ ]),
+ 'max_temp': 26.0,
+ 'min_temp': 18.0,
+ 'supported_features': <ClimateEntityFeature: 425>,
+ 'swing_mode': 'both',
+ 'swing_modes': list([
+ 'off',
+ 'horizontal',
+ 'vertical',
+ 'both',
+ ]),
+ 'target_temp_step': 0.5,
+ 'temperature': 21.0,
+ }),
+ 'context': <ANY>,
+ 'entity_id': 'climate.testserial345',
+ 'last_changed': <ANY>,
+ 'last_reported': <ANY>,
+ 'last_updated': <ANY>,
+ 'state': 'cool',
+ })
+# ---
diff --git a/tests/components/fujitsu_fglair/test_climate.py b/tests/components/fujitsu_fglair/test_climate.py
new file mode 100644
index 00000000000..fd016e4e226
--- /dev/null
+++ b/tests/components/fujitsu_fglair/test_climate.py
@@ -0,0 +1,98 @@
+"""Test for the climate entities of Fujitsu HVAC."""
+
+from unittest.mock import AsyncMock
+
+from syrupy import SnapshotAssertion
+
+from homeassistant.components.climate import (
+ ATTR_FAN_MODE,
+ ATTR_HVAC_MODE,
+ ATTR_SWING_MODE,
+ ATTR_TEMPERATURE,
+ DOMAIN as CLIMATE_DOMAIN,
+ FAN_AUTO,
+ SERVICE_SET_FAN_MODE,
+ SERVICE_SET_HVAC_MODE,
+ SERVICE_SET_SWING_MODE,
+ SERVICE_SET_TEMPERATURE,
+ SWING_BOTH,
+ HVACMode,
+)
+from homeassistant.components.fujitsu_fglair.const import (
+ HA_TO_FUJI_FAN,
+ HA_TO_FUJI_HVAC,
+ HA_TO_FUJI_SWING,
+)
+from homeassistant.const import ATTR_ENTITY_ID
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import entity_registry as er
+
+from . import entity_id, setup_integration
+
+from tests.common import MockConfigEntry, snapshot_platform
+
+
+async def test_entities(
+ hass: HomeAssistant,
+ entity_registry: er.EntityRegistry,
+ snapshot: SnapshotAssertion,
+ mock_ayla_api: AsyncMock,
+ mock_config_entry: MockConfigEntry,
+) -> None:
+ """Test that coordinator returns the data we expect after the first refresh."""
+ await setup_integration(hass, mock_config_entry)
+ await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
+
+
+async def test_set_attributes(
+ hass: HomeAssistant,
+ entity_registry: er.EntityRegistry,
+ snapshot: SnapshotAssertion,
+ mock_ayla_api: AsyncMock,
+ mock_devices: list[AsyncMock],
+ mock_config_entry: MockConfigEntry,
+) -> None:
+ """Test that setting the attributes calls the correct functions on the device."""
+ await setup_integration(hass, mock_config_entry)
+
+ await hass.services.async_call(
+ CLIMATE_DOMAIN,
+ SERVICE_SET_HVAC_MODE,
+ service_data={ATTR_HVAC_MODE: HVACMode.COOL},
+ target={ATTR_ENTITY_ID: entity_id(mock_devices[0])},
+ blocking=True,
+ )
+ mock_devices[0].async_set_op_mode.assert_called_once_with(
+ HA_TO_FUJI_HVAC[HVACMode.COOL]
+ )
+
+ await hass.services.async_call(
+ CLIMATE_DOMAIN,
+ SERVICE_SET_FAN_MODE,
+ service_data={ATTR_FAN_MODE: FAN_AUTO},
+ target={ATTR_ENTITY_ID: entity_id(mock_devices[0])},
+ blocking=True,
+ )
+ mock_devices[0].async_set_fan_speed.assert_called_once_with(
+ HA_TO_FUJI_FAN[FAN_AUTO]
+ )
+
+ await hass.services.async_call(
+ CLIMATE_DOMAIN,
+ SERVICE_SET_SWING_MODE,
+ service_data={ATTR_SWING_MODE: SWING_BOTH},
+ target={ATTR_ENTITY_ID: entity_id(mock_devices[0])},
+ blocking=True,
+ )
+ mock_devices[0].async_set_swing_mode.assert_called_once_with(
+ HA_TO_FUJI_SWING[SWING_BOTH]
+ )
+
+ await hass.services.async_call(
+ CLIMATE_DOMAIN,
+ SERVICE_SET_TEMPERATURE,
+ service_data={ATTR_TEMPERATURE: 23.0},
+ target={ATTR_ENTITY_ID: entity_id(mock_devices[0])},
+ blocking=True,
+ )
+ mock_devices[0].async_set_set_temp.assert_called_once_with(23.0)
diff --git a/tests/components/fujitsu_fglair/test_config_flow.py b/tests/components/fujitsu_fglair/test_config_flow.py
new file mode 100644
index 00000000000..06e4b2e5bd3
--- /dev/null
+++ b/tests/components/fujitsu_fglair/test_config_flow.py
@@ -0,0 +1,107 @@
+"""Test the Fujitsu HVAC (based on Ayla IOT) config flow."""
+
+from unittest.mock import AsyncMock
+
+from ayla_iot_unofficial import AylaAuthError
+import pytest
+
+from homeassistant.components.fujitsu_fglair.const import CONF_EUROPE, DOMAIN
+from homeassistant.config_entries import SOURCE_USER
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.core import HomeAssistant
+from homeassistant.data_entry_flow import FlowResult, FlowResultType
+
+from .conftest import TEST_PASSWORD, TEST_USERNAME
+
+from tests.common import MockConfigEntry
+
+
+async def _initial_step(hass: HomeAssistant) -> FlowResult:
+ result = await hass.config_entries.flow.async_init(
+ DOMAIN, context={"source": SOURCE_USER}
+ )
+ assert result["type"] is FlowResultType.FORM
+ assert result["errors"] == {}
+
+ return await hass.config_entries.flow.async_configure(
+ result["flow_id"],
+ {
+ CONF_USERNAME: TEST_USERNAME,
+ CONF_PASSWORD: TEST_PASSWORD,
+ CONF_EUROPE: False,
+ },
+ )
+
+
+async def test_full_flow(
+ hass: HomeAssistant, mock_setup_entry: AsyncMock, mock_ayla_api: AsyncMock
+) -> None:
+ """Test full config flow."""
+ result = await _initial_step(hass)
+ mock_ayla_api.async_sign_in.assert_called_once()
+
+ assert result["type"] is FlowResultType.CREATE_ENTRY
+ assert result["title"] == f"FGLair ({TEST_USERNAME})"
+ assert result["data"] == {
+ CONF_USERNAME: TEST_USERNAME,
+ CONF_PASSWORD: TEST_PASSWORD,
+ CONF_EUROPE: False,
+ }
+
+
+async def test_duplicate_entry(
+ hass: HomeAssistant,
+ mock_setup_entry: AsyncMock,
+ mock_ayla_api: AsyncMock,
+ mock_config_entry: MockConfigEntry,
+) -> None:
+ """Test that re-adding the same account fails."""
+ mock_config_entry.add_to_hass(hass)
+ result = await _initial_step(hass)
+ mock_ayla_api.async_sign_in.assert_not_called()
+
+ assert result["type"] is FlowResultType.ABORT
+ assert result["reason"] == "already_configured"
+
+
+@pytest.mark.parametrize(
+ ("exception", "err_msg"),
+ [
+ (AylaAuthError, "invalid_auth"),
+ (TimeoutError, "cannot_connect"),
+ (Exception, "unknown"),
+ ],
+)
+async def test_form_exceptions(
+ hass: HomeAssistant,
+ mock_setup_entry: AsyncMock,
+ mock_ayla_api: AsyncMock,
+ exception: Exception,
+ err_msg: str,
+) -> None:
+ """Test we handle exceptions."""
+
+ mock_ayla_api.async_sign_in.side_effect = exception
+ result = await _initial_step(hass)
+ mock_ayla_api.async_sign_in.assert_called_once()
+
+ assert result["type"] is FlowResultType.FORM
+ assert result["errors"] == {"base": err_msg}
+
+ mock_ayla_api.async_sign_in.side_effect = None
+ result = await hass.config_entries.flow.async_configure(
+ result["flow_id"],
+ {
+ CONF_USERNAME: TEST_USERNAME,
+ CONF_PASSWORD: TEST_PASSWORD,
+ CONF_EUROPE: False,
+ },
+ )
+
+ assert result["type"] is FlowResultType.CREATE_ENTRY
+ assert result["title"] == f"FGLair ({TEST_USERNAME})"
+ assert result["data"] == {
+ CONF_USERNAME: TEST_USERNAME,
+ CONF_PASSWORD: TEST_PASSWORD,
+ CONF_EUROPE: False,
+ }
diff --git a/tests/components/fujitsu_fglair/test_init.py b/tests/components/fujitsu_fglair/test_init.py
new file mode 100644
index 00000000000..fa67ea08661
--- /dev/null
+++ b/tests/components/fujitsu_fglair/test_init.py
@@ -0,0 +1,128 @@
+"""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()
--
GitLab