mirror of https://github.com/home-assistant/core
135 lines
3.3 KiB
Python
135 lines
3.3 KiB
Python
"""Tests for the WiLight integration."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import pywilight
|
|
|
|
from homeassistant.components.cover import (
|
|
ATTR_CURRENT_POSITION,
|
|
ATTR_POSITION,
|
|
DOMAIN as COVER_DOMAIN,
|
|
CoverState,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
SERVICE_CLOSE_COVER,
|
|
SERVICE_OPEN_COVER,
|
|
SERVICE_SET_COVER_POSITION,
|
|
SERVICE_STOP_COVER,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import (
|
|
HOST,
|
|
UPNP_MAC_ADDRESS,
|
|
UPNP_MODEL_NAME_COVER,
|
|
UPNP_MODEL_NUMBER,
|
|
UPNP_SERIAL,
|
|
WILIGHT_ID,
|
|
setup_integration,
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="dummy_device_from_host_cover")
|
|
def mock_dummy_device_from_host_light_fan():
|
|
"""Mock a valid api_devce."""
|
|
|
|
device = pywilight.wilight_from_discovery(
|
|
f"http://{HOST}:45995/wilight.xml",
|
|
UPNP_MAC_ADDRESS,
|
|
UPNP_MODEL_NAME_COVER,
|
|
UPNP_SERIAL,
|
|
UPNP_MODEL_NUMBER,
|
|
)
|
|
|
|
device.set_dummy(True)
|
|
|
|
with patch(
|
|
"pywilight.device_from_host",
|
|
return_value=device,
|
|
):
|
|
yield device
|
|
|
|
|
|
async def test_loading_cover(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
dummy_device_from_host_cover,
|
|
) -> None:
|
|
"""Test the WiLight configuration entry loading."""
|
|
|
|
entry = await setup_integration(hass)
|
|
assert entry
|
|
assert entry.unique_id == WILIGHT_ID
|
|
|
|
# First segment of the strip
|
|
state = hass.states.get("cover.wl000000000099_1")
|
|
assert state
|
|
assert state.state == CoverState.CLOSED
|
|
|
|
entry = entity_registry.async_get("cover.wl000000000099_1")
|
|
assert entry
|
|
assert entry.unique_id == "WL000000000099_0"
|
|
|
|
|
|
async def test_open_close_cover_state(
|
|
hass: HomeAssistant, dummy_device_from_host_cover
|
|
) -> None:
|
|
"""Test the change of state of the cover."""
|
|
await setup_integration(hass)
|
|
|
|
# Open
|
|
await hass.services.async_call(
|
|
COVER_DOMAIN,
|
|
SERVICE_OPEN_COVER,
|
|
{ATTR_ENTITY_ID: "cover.wl000000000099_1"},
|
|
blocking=True,
|
|
)
|
|
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("cover.wl000000000099_1")
|
|
assert state
|
|
assert state.state == CoverState.OPENING
|
|
|
|
# Close
|
|
await hass.services.async_call(
|
|
COVER_DOMAIN,
|
|
SERVICE_CLOSE_COVER,
|
|
{ATTR_ENTITY_ID: "cover.wl000000000099_1"},
|
|
blocking=True,
|
|
)
|
|
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("cover.wl000000000099_1")
|
|
assert state
|
|
assert state.state == CoverState.CLOSING
|
|
|
|
# Set position
|
|
await hass.services.async_call(
|
|
COVER_DOMAIN,
|
|
SERVICE_SET_COVER_POSITION,
|
|
{ATTR_POSITION: 50, ATTR_ENTITY_ID: "cover.wl000000000099_1"},
|
|
blocking=True,
|
|
)
|
|
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("cover.wl000000000099_1")
|
|
assert state
|
|
assert state.state == CoverState.OPEN
|
|
assert state.attributes.get(ATTR_CURRENT_POSITION) == 50
|
|
|
|
# Stop
|
|
await hass.services.async_call(
|
|
COVER_DOMAIN,
|
|
SERVICE_STOP_COVER,
|
|
{ATTR_ENTITY_ID: "cover.wl000000000099_1"},
|
|
blocking=True,
|
|
)
|
|
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("cover.wl000000000099_1")
|
|
assert state
|
|
assert state.state == CoverState.OPEN
|