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.
This commit is contained in:
J. Nick Koston 2024-02-19 00:36:32 -06:00 committed by GitHub
parent 795d90eb1b
commit b35d7a348b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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