457 lines
16 KiB
Python
457 lines
16 KiB
Python
"""The tests for the group cover platform."""
|
|
from datetime import timedelta
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.cover import (
|
|
ATTR_CURRENT_POSITION,
|
|
ATTR_CURRENT_TILT_POSITION,
|
|
ATTR_POSITION,
|
|
ATTR_TILT_POSITION,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.components.group.cover import DEFAULT_NAME
|
|
from homeassistant.const import (
|
|
ATTR_ASSUMED_STATE,
|
|
ATTR_ENTITY_ID,
|
|
ATTR_FRIENDLY_NAME,
|
|
ATTR_SUPPORTED_FEATURES,
|
|
CONF_ENTITIES,
|
|
SERVICE_CLOSE_COVER,
|
|
SERVICE_CLOSE_COVER_TILT,
|
|
SERVICE_OPEN_COVER,
|
|
SERVICE_OPEN_COVER_TILT,
|
|
SERVICE_SET_COVER_POSITION,
|
|
SERVICE_SET_COVER_TILT_POSITION,
|
|
SERVICE_STOP_COVER,
|
|
SERVICE_STOP_COVER_TILT,
|
|
SERVICE_TOGGLE,
|
|
SERVICE_TOGGLE_COVER_TILT,
|
|
STATE_CLOSED,
|
|
STATE_OPEN,
|
|
)
|
|
from homeassistant.setup import async_setup_component
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from tests.common import assert_setup_component, async_fire_time_changed
|
|
|
|
COVER_GROUP = "cover.cover_group"
|
|
DEMO_COVER = "cover.kitchen_window"
|
|
DEMO_COVER_POS = "cover.hall_window"
|
|
DEMO_COVER_TILT = "cover.living_room_window"
|
|
DEMO_TILT = "cover.tilt_demo"
|
|
|
|
CONFIG = {
|
|
DOMAIN: [
|
|
{"platform": "demo"},
|
|
{
|
|
"platform": "group",
|
|
CONF_ENTITIES: [DEMO_COVER, DEMO_COVER_POS, DEMO_COVER_TILT, DEMO_TILT],
|
|
},
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
async def setup_comp(hass):
|
|
"""Set up group cover component."""
|
|
with assert_setup_component(2, DOMAIN):
|
|
await async_setup_component(hass, DOMAIN, CONFIG)
|
|
|
|
|
|
async def test_attributes(hass):
|
|
"""Test handling of state attributes."""
|
|
config = {
|
|
DOMAIN: {
|
|
"platform": "group",
|
|
CONF_ENTITIES: [DEMO_COVER, DEMO_COVER_POS, DEMO_COVER_TILT, DEMO_TILT],
|
|
}
|
|
}
|
|
|
|
with assert_setup_component(1, DOMAIN):
|
|
await async_setup_component(hass, DOMAIN, config)
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_CLOSED
|
|
assert state.attributes[ATTR_FRIENDLY_NAME] == DEFAULT_NAME
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 0
|
|
assert ATTR_CURRENT_POSITION not in state.attributes
|
|
assert ATTR_CURRENT_TILT_POSITION not in state.attributes
|
|
|
|
# Add Entity that supports open / close / stop
|
|
hass.states.async_set(DEMO_COVER, STATE_OPEN, {ATTR_SUPPORTED_FEATURES: 11})
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 11
|
|
assert ATTR_CURRENT_POSITION not in state.attributes
|
|
assert ATTR_CURRENT_TILT_POSITION not in state.attributes
|
|
|
|
# Add Entity that supports set_cover_position
|
|
hass.states.async_set(
|
|
DEMO_COVER_POS,
|
|
STATE_OPEN,
|
|
{ATTR_SUPPORTED_FEATURES: 4, ATTR_CURRENT_POSITION: 70},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 15
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 70
|
|
assert ATTR_CURRENT_TILT_POSITION not in state.attributes
|
|
|
|
# Add Entity that supports open tilt / close tilt / stop tilt
|
|
hass.states.async_set(DEMO_TILT, STATE_OPEN, {ATTR_SUPPORTED_FEATURES: 112})
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 127
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 70
|
|
assert ATTR_CURRENT_TILT_POSITION not in state.attributes
|
|
|
|
# Add Entity that supports set_tilt_position
|
|
hass.states.async_set(
|
|
DEMO_COVER_TILT,
|
|
STATE_OPEN,
|
|
{ATTR_SUPPORTED_FEATURES: 128, ATTR_CURRENT_TILT_POSITION: 60},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 255
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 70
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 60
|
|
|
|
# ### Test assumed state ###
|
|
# ##########################
|
|
|
|
# For covers
|
|
hass.states.async_set(
|
|
DEMO_COVER, STATE_OPEN, {ATTR_SUPPORTED_FEATURES: 4, ATTR_CURRENT_POSITION: 100}
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_ASSUMED_STATE] is True
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 244
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 100
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 60
|
|
|
|
hass.states.async_remove(DEMO_COVER)
|
|
hass.states.async_remove(DEMO_COVER_POS)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 240
|
|
assert ATTR_CURRENT_POSITION not in state.attributes
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 60
|
|
|
|
# For tilts
|
|
hass.states.async_set(
|
|
DEMO_TILT,
|
|
STATE_OPEN,
|
|
{ATTR_SUPPORTED_FEATURES: 128, ATTR_CURRENT_TILT_POSITION: 100},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_ASSUMED_STATE] is True
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 128
|
|
assert ATTR_CURRENT_POSITION not in state.attributes
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
|
|
hass.states.async_remove(DEMO_COVER_TILT)
|
|
hass.states.async_set(DEMO_TILT, STATE_CLOSED)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_CLOSED
|
|
assert ATTR_ASSUMED_STATE not in state.attributes
|
|
assert state.attributes[ATTR_SUPPORTED_FEATURES] == 0
|
|
assert ATTR_CURRENT_POSITION not in state.attributes
|
|
assert ATTR_CURRENT_TILT_POSITION not in state.attributes
|
|
|
|
hass.states.async_set(DEMO_TILT, STATE_CLOSED, {ATTR_ASSUMED_STATE: True})
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.attributes[ATTR_ASSUMED_STATE] is True
|
|
|
|
|
|
async def test_open_covers(hass, setup_comp):
|
|
"""Test open cover function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 100
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_OPEN
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 100
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 100
|
|
|
|
|
|
async def test_close_covers(hass, setup_comp):
|
|
"""Test close cover function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_CLOSE_COVER, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_CLOSED
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 0
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_CLOSED
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 0
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 0
|
|
|
|
|
|
async def test_toggle_covers(hass, setup_comp):
|
|
"""Test toggle cover function."""
|
|
# Start covers in open state
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
|
|
# Toggle will close covers
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TOGGLE, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_CLOSED
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 0
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_CLOSED
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 0
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 0
|
|
|
|
# Toggle again will open covers
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TOGGLE, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 100
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_OPEN
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 100
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 100
|
|
|
|
|
|
async def test_stop_covers(hass, setup_comp):
|
|
"""Test stop cover function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_STOP_COVER, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 100
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_OPEN
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 20
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 80
|
|
|
|
|
|
async def test_set_cover_position(hass, setup_comp):
|
|
"""Test set cover position function."""
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_COVER_POSITION,
|
|
{ATTR_ENTITY_ID: COVER_GROUP, ATTR_POSITION: 50},
|
|
blocking=True,
|
|
)
|
|
for _ in range(4):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_POSITION] == 50
|
|
|
|
assert hass.states.get(DEMO_COVER).state == STATE_CLOSED
|
|
assert hass.states.get(DEMO_COVER_POS).attributes[ATTR_CURRENT_POSITION] == 50
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_POSITION] == 50
|
|
|
|
|
|
async def test_open_tilts(hass, setup_comp):
|
|
"""Test open tilt function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(5):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
|
|
assert (
|
|
hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
)
|
|
|
|
|
|
async def test_close_tilts(hass, setup_comp):
|
|
"""Test close tilt function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_CLOSE_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(5):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 0
|
|
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 0
|
|
|
|
|
|
async def test_toggle_tilts(hass, setup_comp):
|
|
"""Test toggle tilt function."""
|
|
# Start tilted open
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
|
|
assert (
|
|
hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
)
|
|
|
|
# Toggle will tilt closed
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TOGGLE_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 0
|
|
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 0
|
|
|
|
# Toggle again will tilt open
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TOGGLE_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
for _ in range(10):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
|
|
assert (
|
|
hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 100
|
|
)
|
|
|
|
|
|
async def test_stop_tilts(hass, setup_comp):
|
|
"""Test stop tilts function."""
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_STOP_COVER_TILT, {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True
|
|
)
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 60
|
|
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 60
|
|
|
|
|
|
async def test_set_tilt_positions(hass, setup_comp):
|
|
"""Test set tilt position function."""
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_COVER_TILT_POSITION,
|
|
{ATTR_ENTITY_ID: COVER_GROUP, ATTR_TILT_POSITION: 80},
|
|
blocking=True,
|
|
)
|
|
for _ in range(3):
|
|
future = dt_util.utcnow() + timedelta(seconds=1)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(COVER_GROUP)
|
|
assert state.state == STATE_OPEN
|
|
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 80
|
|
|
|
assert hass.states.get(DEMO_COVER_TILT).attributes[ATTR_CURRENT_TILT_POSITION] == 80
|