core/tests/components/climate/test_nuheat.py

"""The test for the NuHeat thermostat module."""
import unittest
from unittest.mock import PropertyMock, Mock, patch

from homeassistant.components.climate import STATE_HEAT, STATE_IDLE
import homeassistant.components.climate.nuheat as nuheat
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT

SCHEDULE_HOLD = 3
SCHEDULE_RUN = 1
SCHEDULE_TEMPORARY_HOLD = 2


class TestNuHeat(unittest.TestCase):
    """Tests for NuHeat climate."""

    def setUp(self):

        serial_number = "12345"
        temperature_unit = "F"

        thermostat = Mock(
            serial_number=serial_number,
            room="Master bathroom",
            online=True,
            heating=True,
            temperature=2222,
            celsius=22,
            fahrenheit=72,
            max_celsius=69,
            max_fahrenheit=157,
            min_celsius=5,
            min_fahrenheit=41,
            schedule_mode=SCHEDULE_RUN,
            target_celsius=22,
            target_fahrenheit=72)

        api = Mock()
        api.get_thermostat.return_value = thermostat

        self.thermostat = nuheat.NuHeatThermostat(
            api, serial_number, temperature_unit)

    def test_name(self):
        """Test name property."""
        self.assertEqual(self.thermostat.name, "Master bathroom")

    def test_temperature_unit(self):
        """Test temperature unit."""
        self.assertEqual(self.thermostat.temperature_unit, TEMP_FAHRENHEIT)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.temperature_unit, TEMP_CELSIUS)

    def test_current_temperature(self):
        """Test current temperature."""
        self.assertEqual(self.thermostat.current_temperature, 72)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.current_temperature, 22)

    def test_current_operation(self):
        """Test current operation."""
        self.assertEqual(self.thermostat.current_operation, STATE_HEAT)

        self.thermostat._thermostat.heating = False
        self.assertEqual(self.thermostat.current_operation, STATE_IDLE)

    def test_min_temp(self):
        """Test min temp."""
        self.assertEqual(self.thermostat.min_temp, 41)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.min_temp, 5)

    def test_max_temp(self):
        """Test max temp."""
        self.assertEqual(self.thermostat.max_temp, 157)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.max_temp, 69)

    def test_target_temperature(self):
        """Test target temperature."""
        self.assertEqual(self.thermostat.target_temperature, 72)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.target_temperature, 22)


    def test_target_temperature_low(self):
        """Test low target temperature."""
        self.assertEqual(self.thermostat.target_temperature_low, 72)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.target_temperature_low, 22)

    def test_target_temperature_high(self):
        """Test high target temperature."""
        self.assertEqual(self.thermostat.target_temperature_high, 72)

        self.thermostat._temperature_unit = "C"
        self.assertEqual(self.thermostat.target_temperature_high, 22)

    @patch.object(
        nuheat.NuHeatThermostat, "is_away_mode_on", new_callable=PropertyMock)
    def test_current_hold_mode_away(self, is_away_mode_on):
        """Test current hold mode while away."""
        is_away_mode_on.return_value = True
        self.assertEqual(self.thermostat.current_hold_mode, nuheat.MODE_AWAY)

    @patch.object(
        nuheat.NuHeatThermostat, "is_away_mode_on", new_callable=PropertyMock)
    def test_current_hold_mode(self, is_away_mode_on):
        """Test current hold mode."""
        is_away_mode_on.return_value = False

        self.thermostat._thermostat.schedule_mode = SCHEDULE_RUN
        self.assertEqual(self.thermostat.current_hold_mode, nuheat.MODE_AUTO)

        self.thermostat._thermostat.schedule_mode = SCHEDULE_HOLD
        self.assertEqual(
            self.thermostat.current_hold_mode, nuheat.MODE_HOLD_TEMPERATURE)

        self.thermostat._thermostat.schedule_mode = SCHEDULE_TEMPORARY_HOLD
        self.assertEqual(
            self.thermostat.current_hold_mode, nuheat.MODE_TEMPORARY_HOLD)

    def test_is_away_mode_on(self):
        """Test is away mode on."""
        _thermostat = self.thermostat._thermostat
        _thermostat.target_celsius = _thermostat.min_celsius
        _thermostat.schedule_mode = SCHEDULE_HOLD
        self.assertTrue(self.thermostat.is_away_mode_on)

        _thermostat.target_celsius = _thermostat.min_celsius + 1
        self.assertFalse(self.thermostat.is_away_mode_on)

        _thermostat.target_celsius = _thermostat.min_celsius
        _thermostat.schedule_mode = SCHEDULE_RUN
        self.assertFalse(self.thermostat.is_away_mode_on)

    @patch.object(
        nuheat.NuHeatThermostat, "is_away_mode_on", new_callable=PropertyMock)
    @patch.object(nuheat.NuHeatThermostat, "set_temperature")
    def test_turn_away_mode_on_while_home(self, set_temp, is_away_mode_on):
        """Test turn away mode on when not away."""
        is_away_mode_on.return_value = False
        self.thermostat.turn_away_mode_on()
        set_temp.assert_called_once_with(temperature=self.thermostat.min_temp)
        self.assertTrue(self.thermostat._force_update)

    @patch.object(
        nuheat.NuHeatThermostat, "is_away_mode_on", new_callable=PropertyMock)
    @patch.object(nuheat.NuHeatThermostat, "set_temperature")
    def test_turn_away_mode_on_while_away(self, set_temp, is_away_mode_on):
        """Test turn away mode on when away."""
        is_away_mode_on.return_value = True
        self.thermostat.turn_away_mode_on()
        set_temp.assert_not_called()

    def test_set_temperature(self):
        """Test set temperature."""
        self.thermostat.set_temperature(temperature=85)
        self.assertEqual(self.thermostat._thermostat.target_fahrenheit, 85)
        self.assertTrue(self.thermostat._force_update)

        self.thermostat._temperature_unit = "C"
        self.thermostat.set_temperature(temperature=23)
        self.assertEqual(self.thermostat._thermostat.target_celsius, 23)
        self.assertTrue(self.thermostat._force_update)

    @patch.object(nuheat.NuHeatThermostat, "_throttled_update")
    def test_forced_update(self, throttled_update):
        """Test update without throttle."""
        self.thermostat._force_update = True
        self.thermostat.update()
        throttled_update.assert_called_once_with(no_throttle=True)
        self.assertFalse(self.thermostat._force_update)

    @patch.object(nuheat.NuHeatThermostat, "_throttled_update")
    def test_throttled_update(self, throttled_update):
        """Test update with throttle."""
        self.thermostat._force_update = False
        self.thermostat.update()
        throttled_update.assert_called_once_with()
        self.assertFalse(self.thermostat._force_update)