Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-23 20:39:01 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

153
tests/components/homekit_controller/test_climate.py
View File

@@ -1,40 +1,45 @@
"""Basic checks for HomeKitclimate."""
from homeassistant.components.climate.const import (
DOMAIN, SERVICE_SET_HVAC_MODE, SERVICE_SET_TEMPERATURE,
HVAC_MODE_HEAT_COOL, HVAC_MODE_COOL, HVAC_MODE_HEAT, HVAC_MODE_OFF,
SERVICE_SET_HUMIDITY)
from tests.components.homekit_controller.common import (
FakeService, setup_test_component)
DOMAIN,
SERVICE_SET_HVAC_MODE,
SERVICE_SET_TEMPERATURE,
HVAC_MODE_HEAT_COOL,
HVAC_MODE_COOL,
HVAC_MODE_HEAT,
HVAC_MODE_OFF,
SERVICE_SET_HUMIDITY,
)
from tests.components.homekit_controller.common import FakeService, setup_test_component
HEATING_COOLING_TARGET = ('thermostat', 'heating-cooling.target')
HEATING_COOLING_CURRENT = ('thermostat', 'heating-cooling.current')
TEMPERATURE_TARGET = ('thermostat', 'temperature.target')
TEMPERATURE_CURRENT = ('thermostat', 'temperature.current')
HUMIDITY_TARGET = ('thermostat', 'relative-humidity.target')
HUMIDITY_CURRENT = ('thermostat', 'relative-humidity.current')
HEATING_COOLING_TARGET = ("thermostat", "heating-cooling.target")
HEATING_COOLING_CURRENT = ("thermostat", "heating-cooling.current")
TEMPERATURE_TARGET = ("thermostat", "temperature.target")
TEMPERATURE_CURRENT = ("thermostat", "temperature.current")
HUMIDITY_TARGET = ("thermostat", "relative-humidity.target")
HUMIDITY_CURRENT = ("thermostat", "relative-humidity.current")
def create_thermostat_service():
"""Define thermostat characteristics."""
service = FakeService('public.hap.service.thermostat')
service = FakeService("public.hap.service.thermostat")
char = service.add_characteristic('heating-cooling.target')
char = service.add_characteristic("heating-cooling.target")
char.value = 0
char = service.add_characteristic('heating-cooling.current')
char = service.add_characteristic("heating-cooling.current")
char.value = 0
char = service.add_characteristic('temperature.target')
char = service.add_characteristic("temperature.target")
char.value = 0
char = service.add_characteristic('temperature.current')
char = service.add_characteristic("temperature.current")
char.value = 0
char = service.add_characteristic('relative-humidity.target')
char = service.add_characteristic("relative-humidity.target")
char.value = 0
char = service.add_characteristic('relative-humidity.current')
char = service.add_characteristic("relative-humidity.current")
char.value = 0
return service
@@ -42,8 +47,8 @@ def create_thermostat_service():
async def test_climate_respect_supported_op_modes_1(hass, utcnow):
"""Test that climate respects minValue/maxValue hints."""
service = FakeService('public.hap.service.thermostat')
char = service.add_characteristic('heating-cooling.target')
service = FakeService("public.hap.service.thermostat")
char = service.add_characteristic("heating-cooling.target")
char.value = 0
char.minValue = 0
char.maxValue = 1
@@ -51,20 +56,20 @@ async def test_climate_respect_supported_op_modes_1(hass, utcnow):
helper = await setup_test_component(hass, [service])
state = await helper.poll_and_get_state()
assert state.attributes['hvac_modes'] == ['off', 'heat']
assert state.attributes["hvac_modes"] == ["off", "heat"]
async def test_climate_respect_supported_op_modes_2(hass, utcnow):
"""Test that climate respects validValue hints."""
service = FakeService('public.hap.service.thermostat')
char = service.add_characteristic('heating-cooling.target')
service = FakeService("public.hap.service.thermostat")
char = service.add_characteristic("heating-cooling.target")
char.value = 0
char.valid_values = [0, 1, 2]
helper = await setup_test_component(hass, [service])
state = await helper.poll_and_get_state()
assert state.attributes['hvac_modes'] == ['off', 'heat', 'cool']
assert state.attributes["hvac_modes"] == ["off", "heat", "cool"]
async def test_climate_change_thermostat_state(hass, utcnow):
@@ -73,29 +78,37 @@ async def test_climate_change_thermostat_state(hass, utcnow):
helper = await setup_test_component(hass, [ThermostatService()])
await hass.services.async_call(DOMAIN, SERVICE_SET_HVAC_MODE, {
'entity_id': 'climate.testdevice',
'hvac_mode': HVAC_MODE_HEAT,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HVAC_MODE,
{"entity_id": "climate.testdevice", "hvac_mode": HVAC_MODE_HEAT},
blocking=True,
)
assert helper.characteristics[HEATING_COOLING_TARGET].value == 1
await hass.services.async_call(DOMAIN, SERVICE_SET_HVAC_MODE, {
'entity_id': 'climate.testdevice',
'hvac_mode': HVAC_MODE_COOL,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HVAC_MODE,
{"entity_id": "climate.testdevice", "hvac_mode": HVAC_MODE_COOL},
blocking=True,
)
assert helper.characteristics[HEATING_COOLING_TARGET].value == 2
await hass.services.async_call(DOMAIN, SERVICE_SET_HVAC_MODE, {
'entity_id': 'climate.testdevice',
'hvac_mode': HVAC_MODE_HEAT_COOL,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HVAC_MODE,
{"entity_id": "climate.testdevice", "hvac_mode": HVAC_MODE_HEAT_COOL},
blocking=True,
)
assert helper.characteristics[HEATING_COOLING_TARGET].value == 3
await hass.services.async_call(DOMAIN, SERVICE_SET_HVAC_MODE, {
'entity_id': 'climate.testdevice',
'hvac_mode': HVAC_MODE_OFF,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HVAC_MODE,
{"entity_id": "climate.testdevice", "hvac_mode": HVAC_MODE_OFF},
blocking=True,
)
assert helper.characteristics[HEATING_COOLING_TARGET].value == 0
@@ -105,16 +118,20 @@ async def test_climate_change_thermostat_temperature(hass, utcnow):
helper = await setup_test_component(hass, [ThermostatService()])
await hass.services.async_call(DOMAIN, SERVICE_SET_TEMPERATURE, {
'entity_id': 'climate.testdevice',
'temperature': 21,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_TEMPERATURE,
{"entity_id": "climate.testdevice", "temperature": 21},
blocking=True,
)
assert helper.characteristics[TEMPERATURE_TARGET].value == 21
await hass.services.async_call(DOMAIN, SERVICE_SET_TEMPERATURE, {
'entity_id': 'climate.testdevice',
'temperature': 25,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_TEMPERATURE,
{"entity_id": "climate.testdevice", "temperature": 25},
blocking=True,
)
assert helper.characteristics[TEMPERATURE_TARGET].value == 25
@@ -122,16 +139,20 @@ async def test_climate_change_thermostat_humidity(hass, utcnow):
"""Test that we can turn a HomeKit thermostat on and off again."""
helper = await setup_test_component(hass, [create_thermostat_service()])
await hass.services.async_call(DOMAIN, SERVICE_SET_HUMIDITY, {
'entity_id': 'climate.testdevice',
'humidity': 50,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HUMIDITY,
{"entity_id": "climate.testdevice", "humidity": 50},
blocking=True,
)
assert helper.characteristics[HUMIDITY_TARGET].value == 50
await hass.services.async_call(DOMAIN, SERVICE_SET_HUMIDITY, {
'entity_id': 'climate.testdevice',
'humidity': 45,
}, blocking=True)
await hass.services.async_call(
DOMAIN,
SERVICE_SET_HUMIDITY,
{"entity_id": "climate.testdevice", "humidity": 45},
blocking=True,
)
assert helper.characteristics[HUMIDITY_TARGET].value == 45
@@ -149,10 +170,10 @@ async def test_climate_read_thermostat_state(hass, utcnow):
state = await helper.poll_and_get_state()
assert state.state == HVAC_MODE_HEAT
assert state.attributes['current_temperature'] == 19
assert state.attributes['current_humidity'] == 50
assert state.attributes['min_temp'] == 7
assert state.attributes['max_temp'] == 35
assert state.attributes["current_temperature"] == 19
assert state.attributes["current_humidity"] == 50
assert state.attributes["min_temp"] == 7
assert state.attributes["max_temp"] == 35
# Simulate that cooling is on
helper.characteristics[TEMPERATURE_CURRENT].value = 21
@@ -164,8 +185,8 @@ async def test_climate_read_thermostat_state(hass, utcnow):
state = await helper.poll_and_get_state()
assert state.state == HVAC_MODE_COOL
assert state.attributes['current_temperature'] == 21
assert state.attributes['current_humidity'] == 45
assert state.attributes["current_temperature"] == 21
assert state.attributes["current_humidity"] == 45
# Simulate that we are in heat/cool mode
helper.characteristics[TEMPERATURE_CURRENT].value = 21
@@ -191,8 +212,8 @@ async def test_hvac_mode_vs_hvac_action(hass, utcnow):
helper.characteristics[HUMIDITY_TARGET].value = 45
state = await helper.poll_and_get_state()
assert state.state == 'heat'
assert state.attributes['hvac_action'] == 'off'
assert state.state == "heat"
assert state.attributes["hvac_action"] == "off"
# Simulate that current temperature is below target temp
# Heating might be on and hvac_action currently 'heat'
@@ -200,5 +221,5 @@ async def test_hvac_mode_vs_hvac_action(hass, utcnow):
helper.characteristics[HEATING_COOLING_CURRENT].value = 1
state = await helper.poll_and_get_state()
assert state.state == 'heat'
assert state.attributes['hvac_action'] == 'heating'
assert state.state == "heat"
assert state.attributes["hvac_action"] == "heating"