From b6309e89d367e5f6909e7e0d8e46ba6f7f97f10b Mon Sep 17 00:00:00 2001 From: Bas Nijholt Date: Sun, 28 Aug 2022 14:44:24 -0700 Subject: [PATCH] Fix assert and block till done --- tests/test_switch.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_switch.py b/tests/test_switch.py index 33614e6a..77d96341 100644 --- a/tests/test_switch.py +++ b/tests/test_switch.py @@ -49,6 +49,7 @@ from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN from homeassistant.components.light import SERVICE_TURN_OFF from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN import homeassistant.config as config_util +from homeassistant.config_entries import ConfigEntryState from homeassistant.const import ( ATTR_ENTITY_ID, CONF_LIGHTS, @@ -109,6 +110,7 @@ async def setup_switch(hass, extra_data): entry.add_to_hass(hass) await hass.config_entries.async_setup(entry.entry_id) await hass.async_block_till_done() + assert entry.state is ConfigEntryState.LOADED switch = hass.data[DOMAIN][entry.entry_id][SWITCH_DOMAIN] return entry, switch @@ -587,6 +589,7 @@ async def test_apply_service(hass): }, blocking=True, ) + await hass.async_block_till_done() # Test turn on with defaults assert hass.states.get(entity_id).state == STATE_OFF