f195ecca4b
* Moved climate components with tests into platform dirs. * Updated tests from climate component. * Moved binary_sensor components with tests into platform dirs. * Updated tests from binary_sensor component. * Moved calendar components with tests into platform dirs. * Updated tests from calendar component. * Moved camera components with tests into platform dirs. * Updated tests from camera component. * Moved cover components with tests into platform dirs. * Updated tests from cover component. * Moved device_tracker components with tests into platform dirs. * Updated tests from device_tracker component. * Moved fan components with tests into platform dirs. * Updated tests from fan component. * Moved geo_location components with tests into platform dirs. * Updated tests from geo_location component. * Moved image_processing components with tests into platform dirs. * Updated tests from image_processing component. * Moved light components with tests into platform dirs. * Updated tests from light component. * Moved lock components with tests into platform dirs. * Moved media_player components with tests into platform dirs. * Updated tests from media_player component. * Moved scene components with tests into platform dirs. * Moved sensor components with tests into platform dirs. * Updated tests from sensor component. * Moved switch components with tests into platform dirs. * Updated tests from sensor component. * Moved vacuum components with tests into platform dirs. * Updated tests from vacuum component. * Moved weather components with tests into platform dirs. * Fixed __init__.py files * Fixes for stuff moved as part of this branch. * Fix stuff needed to merge with balloob's branch. * Formatting issues. * Missing __init__.py files. * Fix-ups * Fixup * Regenerated requirements. * Linting errors fixed. * Fixed more broken tests. * Missing init files. * Fix broken tests. * More broken tests * There seems to be a thread race condition. I suspect the logger stuff is running in another thread, which means waiting until the aio loop is done is missing the log messages. Used sleep instead because that allows the logger thread to run. I think the api_streams sensor might not be thread safe. * Disabled tests, will remove sensor in #22147 * Updated coverage and codeowners.
226 lines
8.2 KiB
Python
226 lines
8.2 KiB
Python
"""The test for the NuHeat thermostat module."""
|
|
import unittest
|
|
from unittest.mock import Mock, patch
|
|
from tests.common import get_test_home_assistant
|
|
|
|
from homeassistant.components.climate.const import (
|
|
SUPPORT_HOLD_MODE,
|
|
SUPPORT_OPERATION_MODE,
|
|
SUPPORT_TARGET_TEMPERATURE,
|
|
STATE_HEAT,
|
|
STATE_IDLE)
|
|
import homeassistant.components.nuheat.climate 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."""
|
|
|
|
# pylint: disable=protected-access, no-self-use
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Set up test variables."""
|
|
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)
|
|
|
|
thermostat.get_data = Mock()
|
|
thermostat.resume_schedule = Mock()
|
|
|
|
self.api = Mock()
|
|
self.api.get_thermostat.return_value = thermostat
|
|
|
|
self.hass = get_test_home_assistant()
|
|
self.thermostat = nuheat.NuHeatThermostat(
|
|
self.api, serial_number, temperature_unit)
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop hass."""
|
|
self.hass.stop()
|
|
|
|
@patch("homeassistant.components.nuheat.climate.NuHeatThermostat")
|
|
def test_setup_platform(self, mocked_thermostat):
|
|
"""Test setup_platform."""
|
|
mocked_thermostat.return_value = self.thermostat
|
|
thermostat = mocked_thermostat(self.api, "12345", "F")
|
|
thermostats = [thermostat]
|
|
|
|
self.hass.data[nuheat.NUHEAT_DOMAIN] = (self.api, ["12345"])
|
|
|
|
config = {}
|
|
add_entities = Mock()
|
|
discovery_info = {}
|
|
|
|
nuheat.setup_platform(self.hass, config, add_entities, discovery_info)
|
|
add_entities.assert_called_once_with(thermostats, True)
|
|
|
|
@patch("homeassistant.components.nuheat.climate.NuHeatThermostat")
|
|
def test_resume_program_service(self, mocked_thermostat):
|
|
"""Test resume program service."""
|
|
mocked_thermostat.return_value = self.thermostat
|
|
thermostat = mocked_thermostat(self.api, "12345", "F")
|
|
thermostat.resume_program = Mock()
|
|
thermostat.schedule_update_ha_state = Mock()
|
|
thermostat.entity_id = "climate.master_bathroom"
|
|
|
|
self.hass.data[nuheat.NUHEAT_DOMAIN] = (self.api, ["12345"])
|
|
nuheat.setup_platform(self.hass, {}, Mock(), {})
|
|
|
|
# Explicit entity
|
|
self.hass.services.call(nuheat.DOMAIN, nuheat.SERVICE_RESUME_PROGRAM,
|
|
{"entity_id": "climate.master_bathroom"}, True)
|
|
|
|
thermostat.resume_program.assert_called_with()
|
|
thermostat.schedule_update_ha_state.assert_called_with(True)
|
|
|
|
thermostat.resume_program.reset_mock()
|
|
thermostat.schedule_update_ha_state.reset_mock()
|
|
|
|
# All entities
|
|
self.hass.services.call(
|
|
nuheat.DOMAIN, nuheat.SERVICE_RESUME_PROGRAM, {}, True)
|
|
|
|
thermostat.resume_program.assert_called_with()
|
|
thermostat.schedule_update_ha_state.assert_called_with(True)
|
|
|
|
def test_name(self):
|
|
"""Test name property."""
|
|
assert self.thermostat.name == "Master bathroom"
|
|
|
|
def test_icon(self):
|
|
"""Test name property."""
|
|
assert self.thermostat.icon == "mdi:thermometer"
|
|
|
|
def test_supported_features(self):
|
|
"""Test name property."""
|
|
features = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_HOLD_MODE |
|
|
SUPPORT_OPERATION_MODE)
|
|
assert self.thermostat.supported_features == features
|
|
|
|
def test_temperature_unit(self):
|
|
"""Test temperature unit."""
|
|
assert self.thermostat.temperature_unit == TEMP_FAHRENHEIT
|
|
self.thermostat._temperature_unit = "C"
|
|
assert self.thermostat.temperature_unit == TEMP_CELSIUS
|
|
|
|
def test_current_temperature(self):
|
|
"""Test current temperature."""
|
|
assert self.thermostat.current_temperature == 72
|
|
self.thermostat._temperature_unit = "C"
|
|
assert self.thermostat.current_temperature == 22
|
|
|
|
def test_current_operation(self):
|
|
"""Test current operation."""
|
|
assert self.thermostat.current_operation == STATE_HEAT
|
|
self.thermostat._thermostat.heating = False
|
|
assert self.thermostat.current_operation == STATE_IDLE
|
|
|
|
def test_min_temp(self):
|
|
"""Test min temp."""
|
|
assert self.thermostat.min_temp == 41
|
|
self.thermostat._temperature_unit = "C"
|
|
assert self.thermostat.min_temp == 5
|
|
|
|
def test_max_temp(self):
|
|
"""Test max temp."""
|
|
assert self.thermostat.max_temp == 157
|
|
self.thermostat._temperature_unit = "C"
|
|
assert self.thermostat.max_temp == 69
|
|
|
|
def test_target_temperature(self):
|
|
"""Test target temperature."""
|
|
assert self.thermostat.target_temperature == 72
|
|
self.thermostat._temperature_unit = "C"
|
|
assert self.thermostat.target_temperature == 22
|
|
|
|
def test_current_hold_mode(self):
|
|
"""Test current hold mode."""
|
|
self.thermostat._thermostat.schedule_mode = SCHEDULE_RUN
|
|
assert self.thermostat.current_hold_mode == nuheat.MODE_AUTO
|
|
|
|
self.thermostat._thermostat.schedule_mode = SCHEDULE_HOLD
|
|
assert self.thermostat.current_hold_mode == \
|
|
nuheat.MODE_HOLD_TEMPERATURE
|
|
|
|
self.thermostat._thermostat.schedule_mode = SCHEDULE_TEMPORARY_HOLD
|
|
assert self.thermostat.current_hold_mode == nuheat.MODE_TEMPORARY_HOLD
|
|
|
|
self.thermostat._thermostat.schedule_mode = None
|
|
assert self.thermostat.current_hold_mode == nuheat.MODE_AUTO
|
|
|
|
def test_operation_list(self):
|
|
"""Test the operation list."""
|
|
assert self.thermostat.operation_list == \
|
|
[STATE_HEAT, STATE_IDLE]
|
|
|
|
def test_resume_program(self):
|
|
"""Test resume schedule."""
|
|
self.thermostat.resume_program()
|
|
self.thermostat._thermostat.resume_schedule.assert_called_once_with()
|
|
assert self.thermostat._force_update
|
|
|
|
def test_set_hold_mode(self):
|
|
"""Test set hold mode."""
|
|
self.thermostat.set_hold_mode("temperature")
|
|
assert self.thermostat._thermostat.schedule_mode == SCHEDULE_HOLD
|
|
assert self.thermostat._force_update
|
|
|
|
self.thermostat.set_hold_mode("temporary_temperature")
|
|
assert self.thermostat._thermostat.schedule_mode == \
|
|
SCHEDULE_TEMPORARY_HOLD
|
|
assert self.thermostat._force_update
|
|
|
|
self.thermostat.set_hold_mode("auto")
|
|
assert self.thermostat._thermostat.schedule_mode == SCHEDULE_RUN
|
|
assert self.thermostat._force_update
|
|
|
|
def test_set_temperature(self):
|
|
"""Test set temperature."""
|
|
self.thermostat.set_temperature(temperature=85)
|
|
assert self.thermostat._thermostat.target_fahrenheit == 85
|
|
assert self.thermostat._force_update
|
|
|
|
self.thermostat._temperature_unit = "C"
|
|
self.thermostat.set_temperature(temperature=23)
|
|
assert self.thermostat._thermostat.target_celsius == 23
|
|
assert self.thermostat._force_update
|
|
|
|
@patch.object(nuheat.NuHeatThermostat, "_throttled_update")
|
|
def test_update_without_throttle(self, throttled_update):
|
|
"""Test update without throttle."""
|
|
self.thermostat._force_update = True
|
|
self.thermostat.update()
|
|
throttled_update.assert_called_once_with(no_throttle=True)
|
|
assert not self.thermostat._force_update
|
|
|
|
@patch.object(nuheat.NuHeatThermostat, "_throttled_update")
|
|
def test_update_with_throttle(self, throttled_update):
|
|
"""Test update with throttle."""
|
|
self.thermostat._force_update = False
|
|
self.thermostat.update()
|
|
throttled_update.assert_called_once_with()
|
|
assert not self.thermostat._force_update
|
|
|
|
def test_throttled_update(self):
|
|
"""Test update with throttle."""
|
|
self.thermostat._throttled_update()
|
|
self.thermostat._thermostat.get_data.assert_called_once_with()
|