diff --git a/tests/components/numato/test_switch.py b/tests/components/numato/test_switch.py index 42102ea4869..a5967efc9b1 100644 --- a/tests/components/numato/test_switch.py +++ b/tests/components/numato/test_switch.py @@ -42,6 +42,7 @@ async def test_regular_hass_operations(hass: HomeAssistant, numato_fixture) -> N {ATTR_ENTITY_ID: "switch.numato_switch_mock_port5"}, blocking=True, ) + await hass.async_block_till_done() assert hass.states.get("switch.numato_switch_mock_port5").state == "on" assert numato_fixture.devices[0].values[5] == 1 await hass.services.async_call( @@ -50,6 +51,7 @@ async def test_regular_hass_operations(hass: HomeAssistant, numato_fixture) -> N {ATTR_ENTITY_ID: "switch.numato_switch_mock_port6"}, blocking=True, ) + await hass.async_block_till_done() assert hass.states.get("switch.numato_switch_mock_port6").state == "on" assert numato_fixture.devices[0].values[6] == 1 await hass.services.async_call( @@ -58,6 +60,7 @@ async def test_regular_hass_operations(hass: HomeAssistant, numato_fixture) -> N {ATTR_ENTITY_ID: "switch.numato_switch_mock_port5"}, blocking=True, ) + await hass.async_block_till_done() assert hass.states.get("switch.numato_switch_mock_port5").state == "off" assert numato_fixture.devices[0].values[5] == 0 await hass.services.async_call( @@ -66,6 +69,7 @@ async def test_regular_hass_operations(hass: HomeAssistant, numato_fixture) -> N {ATTR_ENTITY_ID: "switch.numato_switch_mock_port6"}, blocking=True, ) + await hass.async_block_till_done() assert hass.states.get("switch.numato_switch_mock_port6").state == "off" assert numato_fixture.devices[0].values[6] == 0