core/tests/components/airzone/test_select.py

"""The select tests for the Airzone platform."""

from unittest.mock import patch

from aioairzone.common import OperationMode
from aioairzone.const import (
    API_COLD_ANGLE,
    API_DATA,
    API_HEAT_ANGLE,
    API_MODE,
    API_SLEEP,
    API_SYSTEM_ID,
    API_ZONE_ID,
)
import pytest

from homeassistant.components.select import ATTR_OPTIONS, DOMAIN as SELECT_DOMAIN
from homeassistant.const import ATTR_ENTITY_ID, ATTR_OPTION, SERVICE_SELECT_OPTION
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError

from .util import async_init_integration


async def test_airzone_create_selects(hass: HomeAssistant) -> None:
    """Test creation of selects."""

    await async_init_integration(hass)

    state = hass.states.get("select.despacho_cold_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.despacho_heat_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.despacho_mode")
    assert state is None

    state = hass.states.get("select.despacho_sleep")
    assert state.state == "off"

    state = hass.states.get("select.dorm_1_cold_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.dorm_1_heat_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.dorm_1_mode")
    assert state is None

    state = hass.states.get("select.dorm_1_sleep")
    assert state.state == "off"

    state = hass.states.get("select.dorm_2_cold_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.dorm_2_heat_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.dorm_2_mode")
    assert state is None

    state = hass.states.get("select.dorm_2_sleep")
    assert state.state == "off"

    state = hass.states.get("select.dorm_ppal_cold_angle")
    assert state.state == "45deg"

    state = hass.states.get("select.dorm_ppal_heat_angle")
    assert state.state == "50deg"

    state = hass.states.get("select.dorm_ppal_mode")
    assert state is None

    state = hass.states.get("select.dorm_ppal_sleep")
    assert state.state == "30m"

    state = hass.states.get("select.salon_cold_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.salon_heat_angle")
    assert state.state == "90deg"

    state = hass.states.get("select.salon_mode")
    assert state.state == "heat"
    assert state.attributes.get(ATTR_OPTIONS) == [
        "cool",
        "dry",
        "fan",
        "heat",
        "stop",
    ]

    state = hass.states.get("select.salon_sleep")
    assert state.state == "off"


async def test_airzone_select_sleep(hass: HomeAssistant) -> None:
    """Test select sleep."""

    await async_init_integration(hass)

    put_hvac_sleep = {
        API_DATA: [
            {
                API_SYSTEM_ID: 1,
                API_ZONE_ID: 3,
                API_SLEEP: 30,
            }
        ]
    }

    with pytest.raises(ServiceValidationError):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.dorm_1_sleep",
                ATTR_OPTION: "Invalid",
            },
            blocking=True,
        )

    with patch(
        "homeassistant.components.airzone.AirzoneLocalApi.put_hvac",
        return_value=put_hvac_sleep,
    ):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.dorm_1_sleep",
                ATTR_OPTION: "30m",
            },
            blocking=True,
        )

    state = hass.states.get("select.dorm_1_sleep")
    assert state.state == "30m"


async def test_airzone_select_mode(hass: HomeAssistant) -> None:
    """Test select HVAC mode."""

    await async_init_integration(hass)

    put_hvac_mode = {
        API_DATA: [
            {
                API_SYSTEM_ID: 1,
                API_ZONE_ID: 1,
                API_MODE: OperationMode.COOLING,
            }
        ]
    }

    with pytest.raises(ServiceValidationError):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.salon_mode",
                ATTR_OPTION: "Invalid",
            },
            blocking=True,
        )

    with patch(
        "homeassistant.components.airzone.AirzoneLocalApi.put_hvac",
        return_value=put_hvac_mode,
    ):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.salon_mode",
                ATTR_OPTION: "cool",
            },
            blocking=True,
        )

    state = hass.states.get("select.salon_mode")
    assert state.state == "cool"


async def test_airzone_select_grille_angle(hass: HomeAssistant) -> None:
    """Test select sleep."""

    await async_init_integration(hass)

    # Cold Angle

    put_hvac_cold_angle = {
        API_DATA: [
            {
                API_SYSTEM_ID: 1,
                API_ZONE_ID: 3,
                API_COLD_ANGLE: 1,
            }
        ]
    }

    with patch(
        "homeassistant.components.airzone.AirzoneLocalApi.put_hvac",
        return_value=put_hvac_cold_angle,
    ):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.dorm_1_cold_angle",
                ATTR_OPTION: "50deg",
            },
            blocking=True,
        )

    state = hass.states.get("select.dorm_1_cold_angle")
    assert state.state == "50deg"

    # Heat Angle

    put_hvac_heat_angle = {
        API_DATA: [
            {
                API_SYSTEM_ID: 1,
                API_ZONE_ID: 3,
                API_HEAT_ANGLE: 2,
            }
        ]
    }
    with patch(
        "homeassistant.components.airzone.AirzoneLocalApi.put_hvac",
        return_value=put_hvac_heat_angle,
    ):
        await hass.services.async_call(
            SELECT_DOMAIN,
            SERVICE_SELECT_OPTION,
            {
                ATTR_ENTITY_ID: "select.dorm_1_heat_angle",
                ATTR_OPTION: "45deg",
            },
            blocking=True,
        )

    state = hass.states.get("select.dorm_1_heat_angle")
    assert state.state == "45deg"