jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 15:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve timing mqtt integration discovery test (#110960)

Browse Source 
* 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
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/components/mqtt/test_discovery.py
View File

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