diff --git a/tests/components/mqtt/test_discovery.py b/tests/components/mqtt/test_discovery.py index 9acd15eea7c..3f8269c7281 100644 --- a/tests/components/mqtt/test_discovery.py +++ b/tests/components/mqtt/test_discovery.py @@ -1532,6 +1532,7 @@ async def test_mqtt_integration_discovery_subscribe_unsubscribe( await asyncio.sleep(0) await hass.async_block_till_done() await hass.async_block_till_done() + await hass.async_block_till_done() mqtt_client_mock.unsubscribe.assert_called_once_with(["comp/discovery/#"]) mqtt_client_mock.unsubscribe.reset_mock() @@ -1539,6 +1540,7 @@ async def test_mqtt_integration_discovery_subscribe_unsubscribe( await asyncio.sleep(0) await hass.async_block_till_done() await hass.async_block_till_done() + await hass.async_block_till_done() assert not mqtt_client_mock.unsubscribe.called