Guard for bad device info (#17238)

2018-10-08 12:53:51 +02:00 · 2018-10-08 12:53:51 +02:00 · d5f5273c31
commit d5f5273c31
parent af2402ea59
2 changed files with 5 additions and 3 deletions
--- a/tests/helpers/test_entity_platform.py
+++ b/tests/helpers/test_entity_platform.py
@ -719,6 +719,8 @@ async def test_device_info_called(hass):
    assert await entity_platform.async_setup_entry(config_entry)
    await hass.async_block_till_done()

+    assert len(hass.states.async_entity_ids()) == 2
+
    device = registry.async_get_device({('hue', '1234')}, set())
    assert device is not None
    assert device.identifiers == {('hue', '1234')}