From 4baa564f427981871f23f9c30971f63573ef42ee Mon Sep 17 00:00:00 2001
From: Bas Nijholt <bas@nijho.lt>
Date: Sun, 28 Aug 2022 14:39:53 -0700
Subject: [PATCH] Fix setting up lights

---
 tests/test_switch.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/test_switch.py b/tests/test_switch.py
index adfed407..33614e6a 100644
--- a/tests/test_switch.py
+++ b/tests/test_switch.py
@@ -140,11 +140,18 @@ async def setup_lights(hass):
             ct=240,
         ),
     ]
+    for light in lights:
+        light.hass = hass
+        slug = light.name.lower().replace(" ", "_")
+        light.entity_id = f"light.{slug}"
+        await light.async_update_ha_state()
+
     platform.ENTITIES.extend(lights)
     assert await async_setup_component(
         hass, LIGHT_DOMAIN, {LIGHT_DOMAIN: {CONF_PLATFORM: "test"}}
     )
     await hass.async_block_till_done()
+    assert all(hass.states.get(light.entity_id) is not None for light in lights)
     return lights