jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-09 14:27:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use climate enums in atag (#70623)

Browse Source
This commit is contained in:
epenet 2022-04-26 09:59:31 +02:00 committed by GitHub
parent 59b7318db4
commit 9f428f268c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/components/atag/climate.py
View File

@ -1,14 +1,13 @@
"""Initialization of ATAG One climate platform.""" """Initialization of ATAG One climate platform."""
from __future__ import annotations from __future__ import annotations
from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature from homeassistant.components.climate import ClimateEntity
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
CURRENT_HVAC_HEAT,
CURRENT_HVAC_IDLE,
HVAC_MODE_AUTO,
HVAC_MODE_HEAT,
PRESET_AWAY, PRESET_AWAY,
PRESET_BOOST, PRESET_BOOST,
ClimateEntityFeature,
HVACAction,
HVACMode,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform from homeassistant.const import ATTR_TEMPERATURE, Platform
@ -25,7 +24,7 @@ PRESET_MAP = {
PRESET_BOOST: "fireplace", PRESET_BOOST: "fireplace",
} }
PRESET_INVERTED = {v: k for k, v in PRESET_MAP.items()} PRESET_INVERTED = {v: k for k, v in PRESET_MAP.items()}
HVAC_MODES = [HVAC_MODE_AUTO, HVAC_MODE_HEAT] HVAC_MODES = [HVACMode.AUTO, HVACMode.HEAT]
async def async_setup_entry( async def async_setup_entry(
@ -61,7 +60,7 @@ class AtagThermostat(AtagEntity, ClimateEntity):
def hvac_action(self) -> str | None: def hvac_action(self) -> str | None:
"""Return the current running hvac operation.""" """Return the current running hvac operation."""
is_active = self.coordinator.data.climate.status is_active = self.coordinator.data.climate.status
return CURRENT_HVAC_HEAT if is_active else CURRENT_HVAC_IDLE return HVACAction.HEATING if is_active else HVACAction.IDLE
@property @property
def current_temperature(self) -> float | None: def current_temperature(self) -> float | None:

11
tests/components/atag/test_climate.py
View File

@ -7,12 +7,11 @@ from homeassistant.components.climate import (
ATTR_HVAC_MODE, ATTR_HVAC_MODE,
ATTR_PRESET_MODE, ATTR_PRESET_MODE,
DOMAIN as CLIMATE_DOMAIN, DOMAIN as CLIMATE_DOMAIN,
HVAC_MODE_HEAT,
SERVICE_SET_HVAC_MODE, SERVICE_SET_HVAC_MODE,
SERVICE_SET_PRESET_MODE, SERVICE_SET_PRESET_MODE,
SERVICE_SET_TEMPERATURE, SERVICE_SET_TEMPERATURE,
) )
from homeassistant.components.climate.const import CURRENT_HVAC_IDLE, PRESET_AWAY from homeassistant.components.climate.const import PRESET_AWAY, HVACAction, HVACMode
from homeassistant.components.homeassistant import DOMAIN as HA_DOMAIN from homeassistant.components.homeassistant import DOMAIN as HA_DOMAIN
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
@ -40,7 +39,7 @@ async def test_climate(
assert entity_registry.async_is_registered(CLIMATE_ID) assert entity_registry.async_is_registered(CLIMATE_ID)
entity = entity_registry.async_get(CLIMATE_ID) entity = entity_registry.async_get(CLIMATE_ID)
assert entity.unique_id == f"{UID}-{Platform.CLIMATE}" assert entity.unique_id == f"{UID}-{Platform.CLIMATE}"
assert hass.states.get(CLIMATE_ID).attributes[ATTR_HVAC_ACTION] == CURRENT_HVAC_IDLE assert hass.states.get(CLIMATE_ID).attributes[ATTR_HVAC_ACTION] == HVACAction.IDLE
async def test_setting_climate( async def test_setting_climate(
@ -72,11 +71,11 @@ async def test_setting_climate(
await hass.services.async_call( await hass.services.async_call(
CLIMATE_DOMAIN, CLIMATE_DOMAIN,
SERVICE_SET_HVAC_MODE, SERVICE_SET_HVAC_MODE,
{ATTR_ENTITY_ID: CLIMATE_ID, ATTR_HVAC_MODE: HVAC_MODE_HEAT}, {ATTR_ENTITY_ID: CLIMATE_ID, ATTR_HVAC_MODE: HVACMode.HEAT},
blocking=True, blocking=True,
) )
await hass.async_block_till_done() await hass.async_block_till_done()
mock_set_hvac.assert_called_once_with(HVAC_MODE_HEAT) mock_set_hvac.assert_called_once_with(HVACMode.HEAT)
async def test_incorrect_modes( async def test_incorrect_modes(
@ -99,7 +98,7 @@ async def test_update_failed(
"""Test data is not destroyed on update failure.""" """Test data is not destroyed on update failure."""
entry = await init_integration(hass, aioclient_mock) entry = await init_integration(hass, aioclient_mock)
await async_setup_component(hass, HA_DOMAIN, {}) await async_setup_component(hass, HA_DOMAIN, {})
assert hass.states.get(CLIMATE_ID).state == HVAC_MODE_HEAT assert hass.states.get(CLIMATE_ID).state == HVACMode.HEAT
coordinator = hass.data[DOMAIN][entry.entry_id] coordinator = hass.data[DOMAIN][entry.entry_id]
with patch("pyatag.AtagOne.update", side_effect=TimeoutError) as updater: with patch("pyatag.AtagOne.update", side_effect=TimeoutError) as updater:
await coordinator.async_refresh() await coordinator.async_refresh()