55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
|
"""Helper method tests."""
|
||
|
|
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from homeassistant.components.tado import TadoConnector
|
||
|
|
from homeassistant.components.tado.const import (
|
||
|
|
CONST_OVERLAY_MANUAL,
|
||
|
|
CONST_OVERLAY_TADO_DEFAULT,
|
||
|
|
CONST_OVERLAY_TADO_MODE,
|
||
|
|
CONST_OVERLAY_TIMER,
|
||
|
|
)
|
||
|
|
from homeassistant.components.tado.helper import decide_overlay_mode
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
|
||
|
|
|
||
|
|
def dummy_tado_connector(hass: HomeAssistant, fallback) -> TadoConnector:
|
||
|
|
"""Return dummy tado connector."""
|
||
|
|
return TadoConnector(hass, username="dummy", password="dummy", fallback=fallback)
|
||
|
|
|
||
|
|
|
||
|
|
async def test_overlay_mode_duration_set(hass: HomeAssistant) -> None:
|
||
|
|
"""Test overlay method selection when duration is set."""
|
||
|
|
tado = dummy_tado_connector(hass=hass, fallback=CONST_OVERLAY_TADO_MODE)
|
||
|
|
overlay_mode = decide_overlay_mode(tado=tado, duration="01:00:00", zone_id=1)
|
||
|
|
# Must select TIMER overlay
|
||
|
|
assert overlay_mode == CONST_OVERLAY_TIMER
|
||
|
|
|
||
|
|
|
||
|
|
async def test_overlay_mode_next_time_block_fallback(hass: HomeAssistant) -> None:
|
||
|
|
"""Test overlay method selection when duration is not set."""
|
||
|
|
integration_fallback = CONST_OVERLAY_TADO_MODE
|
||
|
|
tado = dummy_tado_connector(hass=hass, fallback=integration_fallback)
|
||
|
|
overlay_mode = decide_overlay_mode(tado=tado, duration=None, zone_id=1)
|
||
|
|
# Must fallback to integration wide setting
|
||
|
|
assert overlay_mode == integration_fallback
|
||
|
|
|
||
|
|
|
||
|
|
async def test_overlay_mode_tado_default_fallback(hass: HomeAssistant) -> None:
|
||
|
|
"""Test overlay method selection when tado default is selected."""
|
||
|
|
integration_fallback = CONST_OVERLAY_TADO_DEFAULT
|
||
|
|
zone_fallback = CONST_OVERLAY_MANUAL
|
||
|
|
tado = dummy_tado_connector(hass=hass, fallback=integration_fallback)
|
||
|
|
|
||
|
|
class MockZoneData:
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.default_overlay_termination_type = zone_fallback
|
||
|
|
|
||
|
|
zone_id = 1
|
||
|
|
|
||
|
|
zone_data = {"zone": {zone_id: MockZoneData()}}
|
||
|
|
with patch.dict(tado.data, zone_data):
|
||
|
|
overlay_mode = decide_overlay_mode(tado=tado, duration=None, zone_id=zone_id)
|
||
|
|
# Must fallback to zone setting
|
||
|
|
assert overlay_mode == zone_fallback
|