mirror of
https://github.com/home-assistant/core.git
synced 2026-09-17 14:08:42 +01:00
305 lines
9.0 KiB
Python
305 lines
9.0 KiB
Python
"""Test Lutron light platform."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.light import (
|
|
ATTR_BRIGHTNESS,
|
|
ATTR_FLASH,
|
|
ATTR_TRANSITION,
|
|
DOMAIN as LIGHT_DOMAIN,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_OFF,
|
|
STATE_ON,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from tests.common import MockConfigEntry, snapshot_platform
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_platforms():
|
|
"""Patch PLATFORMS for all tests in this file."""
|
|
with patch("homeassistant.components.lutron.PLATFORMS", [Platform.LIGHT]):
|
|
yield
|
|
|
|
|
|
async def test_light_setup(
|
|
hass: HomeAssistant,
|
|
mock_lutron: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
entity_registry: er.EntityRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test light setup."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.level = 0
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
|
|
|
|
|
|
async def test_light_turn_on_off(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test light turn on and off."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.level = 0
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
|
|
# Turn on
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_BRIGHTNESS: 128},
|
|
blocking=True,
|
|
)
|
|
light.set_level.assert_called_with(new_level=pytest.approx(50.196, rel=1e-3))
|
|
|
|
# Turn off
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
light.set_level.assert_called_with(new_level=0)
|
|
|
|
|
|
async def test_light_update(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test light state update from library."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.level = 0
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
assert hass.states.get(entity_id).state == STATE_OFF
|
|
|
|
# Simulate update from library
|
|
light.last_level.return_value = 100
|
|
# The library calls the callback registered with subscribe
|
|
callback = light.subscribe.call_args[0][0]
|
|
callback(light, None, None, None)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(entity_id).state == STATE_ON
|
|
assert hass.states.get(entity_id).attributes[ATTR_BRIGHTNESS] == 255
|
|
|
|
|
|
async def test_light_transition(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test light turn on/off with transition."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.level = 0
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
|
|
# Turn on with transition
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_TRANSITION: 2.5},
|
|
blocking=True,
|
|
)
|
|
# Default brightness is used if not specified (DEFAULT_DIMMER_LEVEL is 50%)
|
|
light.set_level.assert_called_with(
|
|
new_level=pytest.approx(50.0, abs=0.5), fade_time_seconds=2.5
|
|
)
|
|
|
|
# Turn off with transition
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_TRANSITION: 3.0},
|
|
blocking=True,
|
|
)
|
|
light.set_level.assert_called_with(new_level=0, fade_time_seconds=3.0)
|
|
|
|
|
|
async def test_light_flash(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test light flash."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
|
|
# Short flash
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_FLASH: "short"},
|
|
blocking=True,
|
|
)
|
|
light.flash.assert_called_with(0.5)
|
|
|
|
# Long flash
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_FLASH: "long"},
|
|
blocking=True,
|
|
)
|
|
light.flash.assert_called_with(1.5)
|
|
|
|
|
|
async def test_light_brightness_restore(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test light brightness restore logic."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.level = 0
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
|
|
# Turn on first time - uses default (50%)
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
light.set_level.assert_called_with(new_level=pytest.approx(50.0, abs=0.5))
|
|
|
|
# Simulate update to 50% (Lutron level 50 -> HA level 127)
|
|
light.last_level.return_value = 50
|
|
callback = light.subscribe.call_args[0][0]
|
|
callback(light, None, None, None)
|
|
await hass.async_block_till_done()
|
|
|
|
# Turn off
|
|
light.last_level.return_value = 0
|
|
callback(light, None, None, None)
|
|
await hass.async_block_till_done()
|
|
|
|
# Turn on again - should restore ~50%
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
# HA level 127 -> Lutron level ~49.8
|
|
light.set_level.assert_called_with(new_level=pytest.approx(50.0, abs=0.5))
|
|
|
|
|
|
async def test_light_state_published_without_waiting_for_a_report(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""State follows the command without waiting for the controller to report.
|
|
|
|
The subscription still delivers the authoritative value later, but it can
|
|
be slow, and on a HomeWorks QS system driven by keypad scenes it never
|
|
arrives for zone levels at all.
|
|
"""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
assert hass.states.get(entity_id).state == STATE_OFF
|
|
|
|
# No callback is fired here: nothing reports the change back.
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_BRIGHTNESS: 128},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state.state == STATE_ON
|
|
assert state.attributes[ATTR_BRIGHTNESS] == 128
|
|
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(entity_id).state == STATE_OFF
|
|
|
|
|
|
async def test_a_later_report_wins_over_the_commanded_value(
|
|
hass: HomeAssistant, mock_lutron: MagicMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""The controller remains authoritative when it does report."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
light = mock_lutron.areas[0].outputs[0]
|
|
light.last_level.return_value = 0
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "light.test_area_test_light"
|
|
|
|
await hass.services.async_call(
|
|
LIGHT_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id, ATTR_BRIGHTNESS: 128},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get(entity_id).attributes[ATTR_BRIGHTNESS] == 128
|
|
|
|
# The controller says the zone actually landed on 100%.
|
|
light.last_level.return_value = 100
|
|
callback = light.subscribe.call_args[0][0]
|
|
callback(light, None, None, None)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(entity_id).attributes[ATTR_BRIGHTNESS] == 255
|