"""The tests for the cover platform.""" from typing import Any import pytest from homeassistant.components.cover import ( ATTR_CURRENT_POSITION, DOMAIN, SERVICE_CLOSE_COVER, SERVICE_OPEN_COVER, SERVICE_SET_COVER_POSITION, intent as cover_intent, ) from homeassistant.const import STATE_CLOSED, STATE_OPEN from homeassistant.core import HomeAssistant from homeassistant.helpers import intent from homeassistant.setup import async_setup_component from tests.common import async_mock_service @pytest.mark.parametrize( ("slots"), [ ({"name": {"value": "garage door"}}), ({"device_class": {"value": "garage"}}), ], ) async def test_open_cover_intent(hass: HomeAssistant, slots: dict[str, Any]) -> None: """Test HassOpenCover intent.""" await cover_intent.async_setup_intents(hass) hass.states.async_set( f"{DOMAIN}.garage_door", STATE_CLOSED, attributes={"device_class": "garage"} ) calls = async_mock_service(hass, DOMAIN, SERVICE_OPEN_COVER) response = await intent.async_handle( hass, "test", cover_intent.INTENT_OPEN_COVER, slots ) await hass.async_block_till_done() assert response.speech["plain"]["speech"] == "Opening garage door" assert len(calls) == 1 call = calls[0] assert call.domain == DOMAIN assert call.service == SERVICE_OPEN_COVER assert call.data == {"entity_id": f"{DOMAIN}.garage_door"} @pytest.mark.parametrize( ("slots"), [ ({"name": {"value": "garage door"}}), ({"device_class": {"value": "garage"}}), ], ) async def test_close_cover_intent(hass: HomeAssistant, slots: dict[str, Any]) -> None: """Test HassCloseCover intent.""" await cover_intent.async_setup_intents(hass) hass.states.async_set( f"{DOMAIN}.garage_door", STATE_OPEN, attributes={"device_class": "garage"} ) calls = async_mock_service(hass, DOMAIN, SERVICE_CLOSE_COVER) response = await intent.async_handle( hass, "test", cover_intent.INTENT_CLOSE_COVER, slots, ) await hass.async_block_till_done() assert response.speech["plain"]["speech"] == "Closing garage door" assert len(calls) == 1 call = calls[0] assert call.domain == DOMAIN assert call.service == SERVICE_CLOSE_COVER assert call.data == {"entity_id": f"{DOMAIN}.garage_door"} @pytest.mark.parametrize( ("slots"), [ ({"name": {"value": "test cover"}, "position": {"value": 50}}), ({"device_class": {"value": "shade"}, "position": {"value": 50}}), ], ) async def test_set_cover_position(hass: HomeAssistant, slots: dict[str, Any]) -> None: """Test HassSetPosition intent for covers.""" assert await async_setup_component(hass, "intent", {}) entity_id = f"{DOMAIN}.test_cover" hass.states.async_set( entity_id, STATE_CLOSED, attributes={ATTR_CURRENT_POSITION: 0, "device_class": "shade"}, ) calls = async_mock_service(hass, DOMAIN, SERVICE_SET_COVER_POSITION) response = await intent.async_handle( hass, "test", intent.INTENT_SET_POSITION, slots, ) await hass.async_block_till_done() assert response.response_type == intent.IntentResponseType.ACTION_DONE assert len(calls) == 1 call = calls[0] assert call.domain == DOMAIN assert call.service == SERVICE_SET_COVER_POSITION assert call.data == {"entity_id": entity_id, "position": 50}