153 lines
6.3 KiB
Python
153 lines
6.3 KiB
Python
""" Test the normal start of a Switch AC Thermostat """
|
|
from unittest.mock import patch, call
|
|
from datetime import datetime, timedelta
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.components.climate import HVACAction, HVACMode
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
|
|
from homeassistant.helpers.entity_component import EntityComponent
|
|
from homeassistant.components.climate import ClimateEntity, DOMAIN as CLIMATE_DOMAIN
|
|
|
|
from pytest_homeassistant_custom_component.common import MockConfigEntry
|
|
|
|
from custom_components.versatile_thermostat.base_thermostat import BaseThermostat
|
|
from custom_components.versatile_thermostat.thermostat_switch import ThermostatOverSwitch
|
|
|
|
from .commons import * # pylint: disable=wildcard-import, unused-wildcard-import
|
|
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
@pytest.mark.parametrize("expected_lingering_timers", [True])
|
|
async def test_over_switch_ac_full_start(hass: HomeAssistant, skip_hass_states_is_state): # pylint: disable=unused-argument
|
|
"""Test the normal full start of a thermostat in thermostat_over_switch type"""
|
|
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="TheOverSwitchACMockName",
|
|
unique_id="uniqueId",
|
|
data=FULL_SWITCH_AC_CONFIG,
|
|
)
|
|
|
|
tz = get_tz(hass) # pylint: disable=invalid-name
|
|
now: datetime = datetime.now(tz=tz)
|
|
|
|
with patch(
|
|
"custom_components.versatile_thermostat.base_thermostat.BaseThermostat.send_event"
|
|
) as mock_send_event:
|
|
entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
def find_my_entity(entity_id) -> ClimateEntity:
|
|
"""Find my new entity"""
|
|
component: EntityComponent[ClimateEntity] = hass.data[CLIMATE_DOMAIN]
|
|
for entity in component.entities:
|
|
if entity.entity_id == entity_id:
|
|
return entity
|
|
|
|
# The name is in the CONF and not the title of the entry
|
|
entity: BaseThermostat = find_my_entity("climate.theoverswitchmockname")
|
|
|
|
assert entity
|
|
assert isinstance(entity, ThermostatOverSwitch)
|
|
|
|
assert entity.name == "TheOverSwitchMockName"
|
|
assert entity.is_over_climate is False # pylint: disable=protected-access
|
|
assert entity.ac_mode is True
|
|
assert entity.hvac_action is HVACAction.OFF
|
|
assert entity.hvac_mode is HVACMode.OFF
|
|
assert entity.hvac_modes == [HVACMode.HEAT, HVACMode.COOL, HVACMode.OFF]
|
|
assert entity.target_temperature == entity.max_temp
|
|
assert entity.preset_modes == [
|
|
PRESET_NONE,
|
|
PRESET_ECO,
|
|
PRESET_COMFORT,
|
|
PRESET_BOOST,
|
|
PRESET_ACTIVITY,
|
|
]
|
|
assert entity.preset_mode is PRESET_NONE
|
|
assert entity._security_state is False # pylint: disable=protected-access
|
|
assert entity._window_state is None # pylint: disable=protected-access
|
|
assert entity._motion_state is None # pylint: disable=protected-access
|
|
assert entity._presence_state is None # pylint: disable=protected-access
|
|
assert entity._prop_algorithm is not None # pylint: disable=protected-access
|
|
|
|
# should have been called with EventType.PRESET_EVENT and EventType.HVAC_MODE_EVENT
|
|
assert mock_send_event.call_count == 2
|
|
|
|
mock_send_event.assert_has_calls(
|
|
[
|
|
call.send_event(EventType.PRESET_EVENT, {"preset": PRESET_NONE}),
|
|
call.send_event(
|
|
EventType.HVAC_MODE_EVENT,
|
|
{"hvac_mode": HVACMode.OFF},
|
|
),
|
|
]
|
|
)
|
|
|
|
# Select a hvacmode, presence and preset
|
|
await entity.async_set_hvac_mode(HVACMode.COOL)
|
|
assert entity.hvac_mode is HVACMode.COOL
|
|
|
|
event_timestamp = now - timedelta(minutes=4)
|
|
await send_presence_change_event(entity, True, False, event_timestamp)
|
|
assert entity._presence_state == STATE_ON # pylint: disable=protected-access
|
|
|
|
await entity.async_set_hvac_mode(HVACMode.COOL)
|
|
assert entity.hvac_mode is HVACMode.COOL
|
|
|
|
await entity.async_set_preset_mode(PRESET_COMFORT)
|
|
assert entity.preset_mode is PRESET_COMFORT
|
|
assert entity.target_temperature == 23
|
|
|
|
# switch to Eco
|
|
await entity.async_set_preset_mode(PRESET_ECO)
|
|
assert entity.preset_mode is PRESET_ECO
|
|
assert entity.target_temperature == 25
|
|
|
|
# Unset the presence
|
|
event_timestamp = now - timedelta(minutes=3)
|
|
await send_presence_change_event(entity, False, True, event_timestamp)
|
|
assert entity._presence_state == STATE_OFF # pylint: disable=protected-access
|
|
assert entity.target_temperature == 27 # eco_ac_away
|
|
|
|
# Open a window
|
|
with patch(
|
|
"homeassistant.helpers.condition.state", return_value=True
|
|
):
|
|
event_timestamp = now - timedelta(minutes=2)
|
|
try_condition = await send_window_change_event(entity, True, False, event_timestamp)
|
|
|
|
# Confirme the window event
|
|
await try_condition(None)
|
|
|
|
assert entity.hvac_mode is HVACMode.OFF
|
|
assert entity.hvac_action is HVACAction.OFF
|
|
assert entity.target_temperature == 27 # eco_ac_away
|
|
|
|
# Close a window
|
|
with patch(
|
|
"homeassistant.helpers.condition.state", return_value=True
|
|
):
|
|
event_timestamp = now - timedelta(minutes=2)
|
|
try_condition = await send_window_change_event(entity, False, True, event_timestamp)
|
|
|
|
# Confirme the window event
|
|
await try_condition(None)
|
|
|
|
assert entity.hvac_mode is HVACMode.COOL
|
|
assert (entity.hvac_action is HVACAction.OFF or entity.hvac_action is HVACAction.IDLE)
|
|
assert entity.target_temperature == 27 # eco_ac_away
|
|
|
|
await entity.async_set_hvac_mode(HVACMode.HEAT)
|
|
assert entity.hvac_mode is HVACMode.HEAT
|
|
|
|
await entity.async_set_preset_mode(PRESET_COMFORT)
|
|
assert entity.preset_mode is PRESET_COMFORT
|
|
assert entity.target_temperature == 26
|
|
|
|
# switch to Eco
|
|
await entity.async_set_preset_mode(PRESET_ECO)
|
|
assert entity.preset_mode is PRESET_ECO
|
|
assert entity.target_temperature == 27
|