142 lines
4.9 KiB
Python
142 lines
4.9 KiB
Python
"""
|
|
tests.components.thermostat.test_heat_control
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests heat control thermostat.
|
|
"""
|
|
import unittest
|
|
|
|
from homeassistant.const import (
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_ON,
|
|
STATE_OFF,
|
|
TEMP_CELCIUS,
|
|
)
|
|
import homeassistant.core as ha
|
|
from homeassistant.components import switch, thermostat
|
|
|
|
|
|
entity = 'thermostat.test'
|
|
ent_sensor = 'sensor.test'
|
|
ent_switch = 'switch.test'
|
|
min_temp = 3.0
|
|
max_temp = 65.0
|
|
target_temp = 42.0
|
|
|
|
|
|
class TestThermostatHeatControl(unittest.TestCase):
|
|
""" Test the Heat Control thermostat. """
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
self.hass = ha.HomeAssistant()
|
|
self.hass.config.temperature_unit = TEMP_CELCIUS
|
|
thermostat.setup(self.hass, {'thermostat': {
|
|
'platform': 'heat_control',
|
|
'name': 'test',
|
|
'heater': ent_switch,
|
|
'target_sensor': ent_sensor
|
|
}})
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
def test_setup_defaults_to_unknown(self):
|
|
self.assertEqual('unknown', self.hass.states.get(entity).state)
|
|
|
|
def test_default_setup_params(self):
|
|
state = self.hass.states.get(entity)
|
|
self.assertEqual(7, state.attributes.get('min_temp'))
|
|
self.assertEqual(35, state.attributes.get('max_temp'))
|
|
self.assertEqual(None, state.attributes.get('temperature'))
|
|
|
|
def test_custom_setup_params(self):
|
|
thermostat.setup(self.hass, {'thermostat': {
|
|
'platform': 'heat_control',
|
|
'name': 'test',
|
|
'heater': ent_switch,
|
|
'target_sensor': ent_sensor,
|
|
'min_temp': min_temp,
|
|
'max_temp': max_temp,
|
|
'target_temp': target_temp
|
|
}})
|
|
state = self.hass.states.get(entity)
|
|
self.assertEqual(min_temp, state.attributes.get('min_temp'))
|
|
self.assertEqual(max_temp, state.attributes.get('max_temp'))
|
|
self.assertEqual(target_temp, state.attributes.get('temperature'))
|
|
self.assertEqual(str(target_temp), self.hass.states.get(entity).state)
|
|
|
|
def test_set_target_temp(self):
|
|
thermostat.set_temperature(self.hass, 30)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual('30.0', self.hass.states.get(entity).state)
|
|
|
|
def test_set_target_temp_turns_on_heater(self):
|
|
self._setup_switch(False)
|
|
self._setup_sensor(25)
|
|
self.hass.pool.block_till_done()
|
|
thermostat.set_temperature(self.hass, 30)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(1, len(self.calls))
|
|
call = self.calls[0]
|
|
self.assertEqual('switch', call.domain)
|
|
self.assertEqual(SERVICE_TURN_ON, call.service)
|
|
self.assertEqual(ent_switch, call.data['entity_id'])
|
|
|
|
def test_set_target_temp_turns_off_heater(self):
|
|
self._setup_switch(True)
|
|
self._setup_sensor(30)
|
|
self.hass.pool.block_till_done()
|
|
thermostat.set_temperature(self.hass, 25)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(1, len(self.calls))
|
|
call = self.calls[0]
|
|
self.assertEqual('switch', call.domain)
|
|
self.assertEqual(SERVICE_TURN_OFF, call.service)
|
|
self.assertEqual(ent_switch, call.data['entity_id'])
|
|
|
|
def test_set_temp_change_turns_on_heater(self):
|
|
self._setup_switch(False)
|
|
thermostat.set_temperature(self.hass, 30)
|
|
self.hass.pool.block_till_done()
|
|
self._setup_sensor(25)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(1, len(self.calls))
|
|
call = self.calls[0]
|
|
self.assertEqual('switch', call.domain)
|
|
self.assertEqual(SERVICE_TURN_ON, call.service)
|
|
self.assertEqual(ent_switch, call.data['entity_id'])
|
|
|
|
def test_temp_change_turns_off_heater(self):
|
|
self._setup_switch(True)
|
|
thermostat.set_temperature(self.hass, 25)
|
|
self.hass.pool.block_till_done()
|
|
self._setup_sensor(30)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(1, len(self.calls))
|
|
call = self.calls[0]
|
|
self.assertEqual('switch', call.domain)
|
|
self.assertEqual(SERVICE_TURN_OFF, call.service)
|
|
self.assertEqual(ent_switch, call.data['entity_id'])
|
|
|
|
def _setup_sensor(self, temp, unit=TEMP_CELCIUS):
|
|
""" Setup the test sensor. """
|
|
self.hass.states.set(ent_sensor, temp, {
|
|
ATTR_UNIT_OF_MEASUREMENT: unit
|
|
})
|
|
|
|
def _setup_switch(self, is_on):
|
|
""" Setup the test switch. """
|
|
self.hass.states.set(ent_switch, STATE_ON if is_on else STATE_OFF)
|
|
self.calls = []
|
|
|
|
def log_call(call):
|
|
""" Log service calls. """
|
|
self.calls.append(call)
|
|
|
|
self.hass.services.register('switch', SERVICE_TURN_ON, log_call)
|
|
self.hass.services.register('switch', SERVICE_TURN_OFF, log_call)
|
|
|