Improve timing mqtt integration discovery test (#110960)

* Improve timing mqtt integration discovery test

* Remove condition
This commit is contained in:
Jan Bouwhuis 2024-02-19 20:16:06 +01:00 committed by GitHub
parent 1bc0263ea4
commit f4fd3633b4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1523,25 +1523,20 @@ async def test_mqtt_integration_discovery_subscribe_unsubscribe(
"""Test mqtt step."""
return self.async_abort(reason="already_configured")
with mock_config_flow("comp", TestFlow):
await asyncio.sleep(0)
assert ("comp/discovery/#", 0) in help_all_subscribe_calls(mqtt_client_mock)
assert not mqtt_client_mock.unsubscribe.called
assert not mqtt_client_mock.unsubscribe.called
wait_unsub = asyncio.Event()
def _mock_unsubscribe(topics: list[str]) -> tuple[int, int]:
wait_unsub.set()
return (0, 0)
with mock_config_flow("comp", TestFlow), patch.object(
mqtt_client_mock, "unsubscribe", side_effect=_mock_unsubscribe
):
async_fire_mqtt_message(hass, "comp/discovery/bla/config", "")
await asyncio.sleep(0)
await hass.async_block_till_done()
await hass.async_block_till_done()
await hass.async_block_till_done()
await wait_unsub.wait()
mqtt_client_mock.unsubscribe.assert_called_once_with(["comp/discovery/#"])
mqtt_client_mock.unsubscribe.reset_mock()
async_fire_mqtt_message(hass, "comp/discovery/bla/config", "")
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
@patch("homeassistant.components.mqtt.PLATFORMS", [])