core/tests/components/thermostat/test_heat_control.py

"""
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'


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_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)