From b35d7a348bbcb01d1f202e2840558098fc44797c Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Mon, 19 Feb 2024 00:36:32 -0600 Subject: [PATCH] Fix mqtt discovery subscribe unsubscribe test with one less task (#110914) This test currently relies on event loop overhead and since #110899 will remove another task in the config entry init flow, it needs another async_block_till_done. In the future it should be refactored to have better synchronization without having to rely on iterations of the event loop. --- tests/components/mqtt/test_discovery.py | 2 ++ 1 file changed, 2 insertions(+) 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