45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
|
"""Test Dynalite light."""
|
||
|
|
from unittest.mock import Mock, call, patch
|
||
|
|
|
||
|
|
from homeassistant.components.dynalite import DOMAIN
|
||
|
|
from homeassistant.components.dynalite.light import DynaliteLight, async_setup_entry
|
||
|
|
|
||
|
|
from tests.common import mock_coro
|
||
|
|
|
||
|
|
|
||
|
|
async def test_light_setup():
|
||
|
|
"""Test a successful setup."""
|
||
|
|
hass = Mock()
|
||
|
|
entry = Mock()
|
||
|
|
async_add = Mock()
|
||
|
|
bridge = Mock()
|
||
|
|
hass.data = {DOMAIN: {entry.entry_id: bridge}}
|
||
|
|
await async_setup_entry(hass, entry, async_add)
|
||
|
|
bridge.register_add_entities.assert_called_once()
|
||
|
|
assert bridge.register_add_entities.mock_calls[0] == call(async_add)
|
||
|
|
|
||
|
|
|
||
|
|
async def test_light():
|
||
|
|
"""Test the light entity."""
|
||
|
|
device = Mock()
|
||
|
|
device.async_turn_on = Mock(return_value=mock_coro(Mock()))
|
||
|
|
device.async_turn_off = Mock(return_value=mock_coro(Mock()))
|
||
|
|
bridge = Mock()
|
||
|
|
dyn_light = DynaliteLight(device, bridge)
|
||
|
|
assert dyn_light.name is device.name
|
||
|
|
assert dyn_light.unique_id is device.unique_id
|
||
|
|
assert dyn_light.available is device.available
|
||
|
|
assert dyn_light.hidden is device.hidden
|
||
|
|
await dyn_light.async_update() # does nothing
|
||
|
|
assert dyn_light.device_info is device.device_info
|
||
|
|
assert dyn_light.brightness is device.brightness
|
||
|
|
assert dyn_light.is_on is device.is_on
|
||
|
|
await dyn_light.async_turn_on(aaa="bbb")
|
||
|
|
assert device.async_turn_on.mock_calls[0] == call(aaa="bbb")
|
||
|
|
await dyn_light.async_turn_off(ccc="ddd")
|
||
|
|
assert device.async_turn_off.mock_calls[0] == call(ccc="ddd")
|
||
|
|
with patch.object(dyn_light, "hass"):
|
||
|
|
with patch.object(dyn_light, "schedule_update_ha_state") as update_ha:
|
||
|
|
dyn_light.try_schedule_ha()
|
||
|
|
update_ha.assert_called_once()
|