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