hass-core/tests/components/demo/test_climate.py

"""The tests for the demo climate component."""

import pytest
import voluptuous as vol

from homeassistant.components.climate.const import (
    ATTR_AUX_HEAT, ATTR_CURRENT_HUMIDITY, ATTR_HVAC_ACTIONS,
    ATTR_CURRENT_TEMPERATURE, ATTR_FAN_MODE, ATTR_HUMIDITY, ATTR_HVAC_MODES,
    ATTR_MAX_HUMIDITY, ATTR_MAX_TEMP, ATTR_MIN_HUMIDITY, ATTR_MIN_TEMP,
    ATTR_PRESET_MODE, ATTR_SWING_MODE, ATTR_TARGET_TEMP_HIGH,
    ATTR_TARGET_TEMP_LOW, CURRENT_HVAC_COOL, DOMAIN,
    HVAC_MODE_COOL, HVAC_MODE_HEAT, PRESET_AWAY, PRESET_ECO)
from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF, STATE_ON
from homeassistant.setup import async_setup_component
from homeassistant.util.unit_system import METRIC_SYSTEM

from tests.components.climate import common

ENTITY_CLIMATE = 'climate.hvac'
ENTITY_ECOBEE = 'climate.ecobee'
ENTITY_HEATPUMP = 'climate.heatpump'


@pytest.fixture(autouse=True)
async def setup_demo_climate(hass):
    """Initialize setup demo climate."""
    hass.config.units = METRIC_SYSTEM
    assert await async_setup_component(hass, DOMAIN, {
        'climate': {
            'platform': 'demo',
        }
    })


def test_setup_params(hass):
    """Test the initial parameters."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert state.state == HVAC_MODE_COOL
    assert 21 == state.attributes.get(ATTR_TEMPERATURE)
    assert 22 == state.attributes.get(ATTR_CURRENT_TEMPERATURE)
    assert "On High" == state.attributes.get(ATTR_FAN_MODE)
    assert 67 == state.attributes.get(ATTR_HUMIDITY)
    assert 54 == state.attributes.get(ATTR_CURRENT_HUMIDITY)
    assert "Off" == state.attributes.get(ATTR_SWING_MODE)
    assert STATE_OFF == state.attributes.get(ATTR_AUX_HEAT)
    assert state.attributes.get(ATTR_HVAC_MODES) == \
        ['off', 'heat', 'cool', 'auto', 'dry', 'fan_only']


def test_default_setup_params(hass):
    """Test the setup with default parameters."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert 7 == state.attributes.get(ATTR_MIN_TEMP)
    assert 35 == state.attributes.get(ATTR_MAX_TEMP)
    assert 30 == state.attributes.get(ATTR_MIN_HUMIDITY)
    assert 99 == state.attributes.get(ATTR_MAX_HUMIDITY)


async def test_set_only_target_temp_bad_attr(hass):
    """Test setting the target temperature without required attribute."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert 21 == state.attributes.get(ATTR_TEMPERATURE)

    with pytest.raises(vol.Invalid):
        await common.async_set_temperature(hass, None, ENTITY_CLIMATE)

    await hass.async_block_till_done()
    assert 21 == state.attributes.get(ATTR_TEMPERATURE)


async def test_set_only_target_temp(hass):
    """Test the setting of the target temperature."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert 21 == state.attributes.get(ATTR_TEMPERATURE)

    await common.async_set_temperature(hass, 30, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert 30.0 == state.attributes.get(ATTR_TEMPERATURE)


async def test_set_only_target_temp_with_convert(hass):
    """Test the setting of the target temperature."""
    state = hass.states.get(ENTITY_HEATPUMP)
    assert 20 == state.attributes.get(ATTR_TEMPERATURE)

    await common.async_set_temperature(hass, 21, ENTITY_HEATPUMP)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_HEATPUMP)
    assert 21.0 == state.attributes.get(ATTR_TEMPERATURE)


async def test_set_target_temp_range(hass):
    """Test the setting of the target temperature with range."""
    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_TEMPERATURE) is None
    assert 21.0 == state.attributes.get(ATTR_TARGET_TEMP_LOW)
    assert 24.0 == state.attributes.get(ATTR_TARGET_TEMP_HIGH)

    await common.async_set_temperature(
        hass, target_temp_high=25, target_temp_low=20, entity_id=ENTITY_ECOBEE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_TEMPERATURE) is None
    assert 20.0 == state.attributes.get(ATTR_TARGET_TEMP_LOW)
    assert 25.0 == state.attributes.get(ATTR_TARGET_TEMP_HIGH)


async def test_set_target_temp_range_bad_attr(hass):
    """Test setting the target temperature range without attribute."""
    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_TEMPERATURE) is None
    assert 21.0 == state.attributes.get(ATTR_TARGET_TEMP_LOW)
    assert 24.0 == state.attributes.get(ATTR_TARGET_TEMP_HIGH)

    with pytest.raises(vol.Invalid):
        await common.async_set_temperature(
            hass, temperature=None, entity_id=ENTITY_ECOBEE,
            target_temp_low=None, target_temp_high=None)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_TEMPERATURE) is None
    assert 21.0 == state.attributes.get(ATTR_TARGET_TEMP_LOW)
    assert 24.0 == state.attributes.get(ATTR_TARGET_TEMP_HIGH)


async def test_set_target_humidity_bad_attr(hass):
    """Test setting the target humidity without required attribute."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert 67 == state.attributes.get(ATTR_HUMIDITY)

    with pytest.raises(vol.Invalid):
        await common.async_set_humidity(hass, None, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert 67 == state.attributes.get(ATTR_HUMIDITY)


async def test_set_target_humidity(hass):
    """Test the setting of the target humidity."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert 67 == state.attributes.get(ATTR_HUMIDITY)

    await common.async_set_humidity(hass, 64, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert 64.0 == state.attributes.get(ATTR_HUMIDITY)


async def test_set_fan_mode_bad_attr(hass):
    """Test setting fan mode without required attribute."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert "On High" == state.attributes.get(ATTR_FAN_MODE)

    with pytest.raises(vol.Invalid):
        await common.async_set_fan_mode(hass, None, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert "On High" == state.attributes.get(ATTR_FAN_MODE)


async def test_set_fan_mode(hass):
    """Test setting of new fan mode."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert "On High" == state.attributes.get(ATTR_FAN_MODE)

    await common.async_set_fan_mode(hass, "On Low", ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert "On Low" == state.attributes.get(ATTR_FAN_MODE)


async def test_set_swing_mode_bad_attr(hass):
    """Test setting swing mode without required attribute."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert "Off" == state.attributes.get(ATTR_SWING_MODE)

    with pytest.raises(vol.Invalid):
        await common.async_set_swing_mode(hass, None, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert "Off" == state.attributes.get(ATTR_SWING_MODE)


async def test_set_swing(hass):
    """Test setting of new swing mode."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert "Off" == state.attributes.get(ATTR_SWING_MODE)

    await common.async_set_swing_mode(hass, "Auto", ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert "Auto" == state.attributes.get(ATTR_SWING_MODE)


async def test_set_hvac_bad_attr_and_state(hass):
    """Test setting hvac mode without required attribute.

    Also check the state.
    """
    state = hass.states.get(ENTITY_CLIMATE)
    assert state.attributes.get(ATTR_HVAC_ACTIONS) == CURRENT_HVAC_COOL
    assert state.state == HVAC_MODE_COOL

    with pytest.raises(vol.Invalid):
        await common.async_set_hvac_mode(hass, None, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert state.attributes.get(ATTR_HVAC_ACTIONS) == CURRENT_HVAC_COOL
    assert state.state == HVAC_MODE_COOL


async def test_set_hvac(hass):
    """Test setting of new hvac mode."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert state.state == HVAC_MODE_COOL

    await common.async_set_hvac_mode(hass, HVAC_MODE_HEAT, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert state.state == HVAC_MODE_HEAT


async def test_set_hold_mode_away(hass):
    """Test setting the hold mode away."""
    await common.async_set_preset_mode(hass, PRESET_AWAY, ENTITY_ECOBEE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_AWAY


async def test_set_hold_mode_eco(hass):
    """Test setting the hold mode eco."""
    await common.async_set_preset_mode(hass, PRESET_ECO, ENTITY_ECOBEE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_ECOBEE)
    assert state.attributes.get(ATTR_PRESET_MODE) == PRESET_ECO


async def test_set_aux_heat_bad_attr(hass):
    """Test setting the auxiliary heater without required attribute."""
    state = hass.states.get(ENTITY_CLIMATE)
    assert state.attributes.get(ATTR_AUX_HEAT) == STATE_OFF

    with pytest.raises(vol.Invalid):
        await common.async_set_aux_heat(hass, None, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    assert state.attributes.get(ATTR_AUX_HEAT) == STATE_OFF


async def test_set_aux_heat_on(hass):
    """Test setting the axillary heater on/true."""
    await common.async_set_aux_heat(hass, True, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert state.attributes.get(ATTR_AUX_HEAT) == STATE_ON


async def test_set_aux_heat_off(hass):
    """Test setting the auxiliary heater off/false."""
    await common.async_set_aux_heat(hass, False, ENTITY_CLIMATE)
    await hass.async_block_till_done()

    state = hass.states.get(ENTITY_CLIMATE)
    assert state.attributes.get(ATTR_AUX_HEAT) == STATE_OFF