e13d5bdc10
* refactoring for multi platform * adopted test_bridge to refactoring * refactoring tests for multi-platform additional coverage in config and init * comment for clarity * more specific imports from lib * library version bump * removed async_update * changed parameter order to start with hass * removed pylint disable * unsubscribe from signal dispatcher inherit from Entity * use device.unique_id * changed hass_obj to hass * added test for remove entity bug fix from the test * removed the polling try_connect. hate polling... it is now part of the async_setup() significantly makes the code clearer and simplifies the tests * removed leftover debug logs in the library * changed tests to get the entry_id from hass * changed place to assign hass.data only after success * fixes for test_init * removed assert * removed device_info * removed bridge internal from common * modified test_bridge to work without the bridge directly * removed bridge from test_existing_update * changed update to not use bridge internals * dyn_bridge fixture no longer used - removed
85 lines
3.1 KiB
Python
Executable file
85 lines
3.1 KiB
Python
Executable file
"""Test Dynalite bridge."""
|
|
|
|
from asynctest import CoroutineMock, Mock, patch
|
|
from dynalite_devices_lib.const import CONF_ALL
|
|
|
|
from homeassistant.components import dynalite
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_update_device(hass):
|
|
"""Test that update works."""
|
|
host = "1.2.3.4"
|
|
entry = MockConfigEntry(domain=dynalite.DOMAIN, data={dynalite.CONF_HOST: host})
|
|
entry.add_to_hass(hass)
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices"
|
|
) as mock_dyn_dev:
|
|
mock_dyn_dev().async_setup = CoroutineMock(return_value=True)
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
# Not waiting so it add the devices before registration
|
|
update_device_func = mock_dyn_dev.mock_calls[1][2]["updateDeviceFunc"]
|
|
device = Mock()
|
|
device.unique_id = "abcdef"
|
|
wide_func = Mock()
|
|
async_dispatcher_connect(hass, f"dynalite-update-{host}", wide_func)
|
|
specific_func = Mock()
|
|
async_dispatcher_connect(
|
|
hass, f"dynalite-update-{host}-{device.unique_id}", specific_func
|
|
)
|
|
update_device_func(CONF_ALL)
|
|
await hass.async_block_till_done()
|
|
wide_func.assert_called_once()
|
|
specific_func.assert_not_called()
|
|
update_device_func(device)
|
|
await hass.async_block_till_done()
|
|
wide_func.assert_called_once()
|
|
specific_func.assert_called_once()
|
|
|
|
|
|
async def test_add_devices_then_register(hass):
|
|
"""Test that add_devices work."""
|
|
host = "1.2.3.4"
|
|
entry = MockConfigEntry(domain=dynalite.DOMAIN, data={dynalite.CONF_HOST: host})
|
|
entry.add_to_hass(hass)
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices"
|
|
) as mock_dyn_dev:
|
|
mock_dyn_dev().async_setup = CoroutineMock(return_value=True)
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
# Not waiting so it add the devices before registration
|
|
new_device_func = mock_dyn_dev.mock_calls[1][2]["newDeviceFunc"]
|
|
# Now with devices
|
|
device1 = Mock()
|
|
device1.category = "light"
|
|
device1.name = "NAME"
|
|
device2 = Mock()
|
|
device2.category = "switch"
|
|
new_device_func([device1, device2])
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get("light.name")
|
|
|
|
|
|
async def test_register_then_add_devices(hass):
|
|
"""Test that add_devices work after register_add_entities."""
|
|
host = "1.2.3.4"
|
|
entry = MockConfigEntry(domain=dynalite.DOMAIN, data={dynalite.CONF_HOST: host})
|
|
entry.add_to_hass(hass)
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices"
|
|
) as mock_dyn_dev:
|
|
mock_dyn_dev().async_setup = CoroutineMock(return_value=True)
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
new_device_func = mock_dyn_dev.mock_calls[1][2]["newDeviceFunc"]
|
|
# Now with devices
|
|
device1 = Mock()
|
|
device1.category = "light"
|
|
device1.name = "NAME"
|
|
device2 = Mock()
|
|
device2.category = "switch"
|
|
new_device_func([device1, device2])
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get("light.name")
|