Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use fixtures in deCONZ climate tests (#121242)

pull/121283/head
Robert Svensson 2024-07-05 09:01:45 +02:00 committed by GitHub
parent 5a24ee0bc0
commit dcef25c0fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 128 additions and 138 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

262
tests/components/deconz/test_climate.py
View File

@ -1,6 +1,6 @@
"""deCONZ climate platform tests.""" """deCONZ climate platform tests."""
from unittest.mock import patch from collections.abc import Callable
import pytest import pytest
@ -35,6 +35,7 @@ from homeassistant.components.deconz.climate import (
DECONZ_PRESET_MANUAL, DECONZ_PRESET_MANUAL,
) )
from homeassistant.components.deconz.const import CONF_ALLOW_CLIP_SENSOR from homeassistant.components.deconz.const import CONF_ALLOW_CLIP_SENSOR
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
ATTR_TEMPERATURE, ATTR_TEMPERATURE,
@ -44,32 +45,13 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError from homeassistant.exceptions import ServiceValidationError
from .test_gateway import (
DECONZ_WEB_REQUEST,
mock_deconz_put_request,
setup_deconz_integration,
)
from tests.test_util.aiohttp import AiohttpClientMocker from tests.test_util.aiohttp import AiohttpClientMocker
async def test_no_sensors( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker "sensor_payload",
) -> None: [
"""Test that no sensors in deconz results in no climate entities.""" {
await setup_deconz_integration(hass, aioclient_mock)
assert len(hass.states.async_all()) == 0
async def test_simple_climate_device(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket
) -> None:
"""Test successful creation of climate entities.
This is a simple water heater that only supports setting temperature and on and off.
"""
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 59, "battery": 59,
@ -101,10 +83,18 @@ async def test_simple_climate_device(
"uniqueid": "14:b4:57:ff:fe:d5:4e:77-01-0201", "uniqueid": "14:b4:57:ff:fe:d5:4e:77-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_simple_climate_device(
hass: HomeAssistant,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
mock_deconz_websocket,
) -> None:
"""Test successful creation of climate entities.
This is a simple water heater that only supports setting temperature and on and off.
"""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_thermostat = hass.states.get("climate.thermostat") climate_thermostat = hass.states.get("climate.thermostat")
assert climate_thermostat.state == HVACMode.HEAT assert climate_thermostat.state == HVACMode.HEAT
@ -156,7 +146,7 @@ async def test_simple_climate_device(
# Verify service calls # Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config") aioclient_mock = mock_put_request("/sensors/0/config")
# Service turn on thermostat # Service turn on thermostat
@ -189,12 +179,10 @@ async def test_simple_climate_device(
) )
async def test_climate_device_without_cooling_support( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test successful creation of sensor entities.""" {
data = {
"sensors": {
"1": { "1": {
"name": "Thermostat", "name": "Thermostat",
"type": "ZHAThermostat", "type": "ZHAThermostat",
@ -209,10 +197,15 @@ async def test_climate_device_without_cooling_support(
"uniqueid": "00:00:00:00:00:00:00:00-00", "uniqueid": "00:00:00:00:00:00:00:00-00",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) async def test_climate_device_without_cooling_support(
hass: HomeAssistant,
config_entry_setup: ConfigEntry,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
mock_deconz_websocket,
) -> None:
"""Test successful creation of sensor entities."""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_thermostat = hass.states.get("climate.thermostat") climate_thermostat = hass.states.get("climate.thermostat")
assert climate_thermostat.state == HVACMode.AUTO assert climate_thermostat.state == HVACMode.AUTO
@ -289,7 +282,7 @@ async def test_climate_device_without_cooling_support(
# Verify service calls # Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/1/config") aioclient_mock = mock_put_request("/sensors/1/config")
# Service set HVAC mode to auto # Service set HVAC mode to auto
@ -355,24 +348,22 @@ async def test_climate_device_without_cooling_support(
blocking=True, blocking=True,
) )
await hass.config_entries.async_unload(config_entry.entry_id) await hass.config_entries.async_unload(config_entry_setup.entry_id)
states = hass.states.async_all() states = hass.states.async_all()
assert len(states) == 2 assert len(states) == 2
for state in states: for state in states:
assert state.state == STATE_UNAVAILABLE assert state.state == STATE_UNAVAILABLE
await hass.config_entries.async_remove(config_entry.entry_id) await hass.config_entries.async_remove(config_entry_setup.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_climate_device_with_cooling_support( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test successful creation of sensor entities.""" {
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 25, "battery": 25,
@ -399,10 +390,15 @@ async def test_climate_device_with_cooling_support(
"uniqueid": "00:24:46:00:00:11:6f:56-01-0201", "uniqueid": "00:24:46:00:00:11:6f:56-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_climate_device_with_cooling_support(
hass: HomeAssistant,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
mock_deconz_websocket,
) -> None:
"""Test successful creation of sensor entities."""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_thermostat = hass.states.get("climate.zen_01") climate_thermostat = hass.states.get("climate.zen_01")
assert climate_thermostat.state == HVACMode.HEAT assert climate_thermostat.state == HVACMode.HEAT
@ -458,7 +454,7 @@ async def test_climate_device_with_cooling_support(
# Verify service calls # Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config") aioclient_mock = mock_put_request("/sensors/0/config")
# Service set temperature to 20 # Service set temperature to 20
@ -471,12 +467,10 @@ async def test_climate_device_with_cooling_support(
assert aioclient_mock.mock_calls[1][2] == {"coolsetpoint": 2000.0} assert aioclient_mock.mock_calls[1][2] == {"coolsetpoint": 2000.0}
async def test_climate_device_with_fan_support( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test successful creation of sensor entities.""" {
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 25, "battery": 25,
@ -503,10 +497,15 @@ async def test_climate_device_with_fan_support(
"uniqueid": "00:24:46:00:00:11:6f:56-01-0201", "uniqueid": "00:24:46:00:00:11:6f:56-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_climate_device_with_fan_support(
hass: HomeAssistant,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
mock_deconz_websocket,
) -> None:
"""Test successful creation of sensor entities."""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_thermostat = hass.states.get("climate.zen_01") climate_thermostat = hass.states.get("climate.zen_01")
assert climate_thermostat.state == HVACMode.HEAT assert climate_thermostat.state == HVACMode.HEAT
@ -580,7 +579,7 @@ async def test_climate_device_with_fan_support(
# Verify service calls # Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config") aioclient_mock = mock_put_request("/sensors/0/config")
# Service set fan mode to off # Service set fan mode to off
@ -613,12 +612,10 @@ async def test_climate_device_with_fan_support(
) )
async def test_climate_device_with_preset( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test successful creation of sensor entities.""" {
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 25, "battery": 25,
@ -646,10 +643,15 @@ async def test_climate_device_with_preset(
"uniqueid": "00:24:46:00:00:11:6f:56-01-0201", "uniqueid": "00:24:46:00:00:11:6f:56-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_climate_device_with_preset(
hass: HomeAssistant,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
mock_deconz_websocket,
) -> None:
"""Test successful creation of sensor entities."""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_zen_01 = hass.states.get("climate.zen_01") climate_zen_01 = hass.states.get("climate.zen_01")
@ -703,7 +705,7 @@ async def test_climate_device_with_preset(
# Verify service calls # Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config") aioclient_mock = mock_put_request("/sensors/0/config")
# Service set preset to HASS preset # Service set preset to HASS preset
@ -736,12 +738,10 @@ async def test_climate_device_with_preset(
) )
async def test_clip_climate_device( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker "sensor_payload",
) -> None: [
"""Test successful creation of sensor entities.""" {
data = {
"sensors": {
"1": { "1": {
"name": "Thermostat", "name": "Thermostat",
"type": "ZHAThermostat", "type": "ZHAThermostat",
@ -763,12 +763,13 @@ async def test_clip_climate_device(
"uniqueid": "00:00:00:00:00:00:00:02-00", "uniqueid": "00:00:00:00:00:00:00:02-00",
}, },
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data):
config_entry = await setup_deconz_integration(
hass, aioclient_mock, options={CONF_ALLOW_CLIP_SENSOR: True}
) )
@pytest.mark.parametrize("config_entry_options", [{CONF_ALLOW_CLIP_SENSOR: True}])
async def test_clip_climate_device(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test successful creation of sensor entities."""
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 3
assert hass.states.get("climate.clip_thermostat").state == HVACMode.HEAT assert hass.states.get("climate.clip_thermostat").state == HVACMode.HEAT
assert ( assert (
@ -779,7 +780,7 @@ async def test_clip_climate_device(
# Disallow clip sensors # Disallow clip sensors
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
config_entry, options={CONF_ALLOW_CLIP_SENSOR: False} config_entry_setup, options={CONF_ALLOW_CLIP_SENSOR: False}
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -789,7 +790,7 @@ async def test_clip_climate_device(
# Allow clip sensors # Allow clip sensors
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
config_entry, options={CONF_ALLOW_CLIP_SENSOR: True} config_entry_setup, options={CONF_ALLOW_CLIP_SENSOR: True}
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -801,12 +802,10 @@ async def test_clip_climate_device(
) )
async def test_verify_state_update( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test that state update properly.""" {
data = {
"sensors": {
"1": { "1": {
"name": "Thermostat", "name": "Thermostat",
"type": "ZHAThermostat", "type": "ZHAThermostat",
@ -821,10 +820,11 @@ async def test_verify_state_update(
"uniqueid": "00:00:00:00:00:00:00:00-00", "uniqueid": "00:00:00:00:00:00:00:00-00",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_verify_state_update(hass: HomeAssistant, mock_deconz_websocket) -> None:
"""Test that state update properly."""
assert hass.states.get("climate.thermostat").state == HVACMode.AUTO assert hass.states.get("climate.thermostat").state == HVACMode.AUTO
assert ( assert (
hass.states.get("climate.thermostat").attributes["hvac_action"] hass.states.get("climate.thermostat").attributes["hvac_action"]
@ -848,8 +848,9 @@ async def test_verify_state_update(
) )
@pytest.mark.usefixtures("config_entry_setup")
async def test_add_new_climate_device( async def test_add_new_climate_device(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket hass: HomeAssistant, mock_deconz_websocket
) -> None: ) -> None:
"""Test that adding a new climate device works.""" """Test that adding a new climate device works."""
event_added_sensor = { event_added_sensor = {
@ -873,7 +874,6 @@ async def test_add_new_climate_device(
}, },
} }
await setup_deconz_integration(hass, aioclient_mock)
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
await mock_deconz_websocket(data=event_added_sensor) await mock_deconz_websocket(data=event_added_sensor)
@ -888,12 +888,10 @@ async def test_add_new_climate_device(
) )
async def test_not_allow_clip_thermostat( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker "sensor_payload",
) -> None: [
"""Test that CLIP thermostats are not allowed.""" {
data = {
"sensors": {
"1": { "1": {
"name": "CLIP thermostat sensor", "name": "CLIP thermostat sensor",
"type": "CLIPThermostat", "type": "CLIPThermostat",
@ -902,22 +900,19 @@ async def test_not_allow_clip_thermostat(
"uniqueid": "00:00:00:00:00:00:00:00-00", "uniqueid": "00:00:00:00:00:00:00:00-00",
}, },
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data):
await setup_deconz_integration(
hass, aioclient_mock, options={CONF_ALLOW_CLIP_SENSOR: False}
) )
@pytest.mark.parametrize("config_entry_options", [{CONF_ALLOW_CLIP_SENSOR: False}])
@pytest.mark.usefixtures("config_entry_setup")
async def test_not_allow_clip_thermostat(hass: HomeAssistant) -> None:
"""Test that CLIP thermostats are not allowed."""
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_no_mode_no_state( @pytest.mark.parametrize(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket "sensor_payload",
) -> None: [
"""Test that a climate device without mode and state works.""" {
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 25, "battery": 25,
@ -939,10 +934,11 @@ async def test_no_mode_no_state(
"uniqueid": "00:24:46:00:00:11:6f:56-01-0201", "uniqueid": "00:24:46:00:00:11:6f:56-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_no_mode_no_state(hass: HomeAssistant) -> None:
"""Test that a climate device without mode and state works."""
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 2
climate_thermostat = hass.states.get("climate.zen_01") climate_thermostat = hass.states.get("climate.zen_01")
@ -951,16 +947,11 @@ async def test_no_mode_no_state(
assert climate_thermostat.attributes["preset_mode"] is DECONZ_PRESET_AUTO assert climate_thermostat.attributes["preset_mode"] is DECONZ_PRESET_AUTO
assert climate_thermostat.attributes["hvac_action"] is HVACAction.IDLE assert climate_thermostat.attributes["hvac_action"] is HVACAction.IDLE
# Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config")
@pytest.mark.parametrize(
async def test_boost_mode( "sensor_payload",
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_deconz_websocket [
) -> None: {
"""Test that a climate device with boost mode and different state works."""
data = {
"sensors": {
"0": { "0": {
"config": { "config": {
"battery": 58, "battery": 58,
@ -994,9 +985,11 @@ async def test_boost_mode(
"uniqueid": "84:fd:27:ff:fe:8a:eb:89-01-0201", "uniqueid": "84:fd:27:ff:fe:8a:eb:89-01-0201",
} }
} }
} ],
with patch.dict(DECONZ_WEB_REQUEST, data): )
config_entry = await setup_deconz_integration(hass, aioclient_mock) @pytest.mark.usefixtures("config_entry_setup")
async def test_boost_mode(hass: HomeAssistant, mock_deconz_websocket) -> None:
"""Test that a climate device with boost mode and different state works."""
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 3
@ -1023,6 +1016,3 @@ async def test_boost_mode(
climate_thermostat = hass.states.get("climate.thermostat") climate_thermostat = hass.states.get("climate.thermostat")
assert climate_thermostat.attributes["preset_mode"] is PRESET_BOOST assert climate_thermostat.attributes["preset_mode"] is PRESET_BOOST
assert climate_thermostat.attributes["hvac_action"] is HVACAction.HEATING assert climate_thermostat.attributes["hvac_action"] is HVACAction.HEATING
# Verify service calls
mock_deconz_put_request(aioclient_mock, config_entry.data, "/sensors/0/config")