mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Improve timing mqtt integration discovery test (#110960)
* Improve timing mqtt integration discovery test * Remove condition
This commit is contained in:
parent
1bc0263ea4
commit
f4fd3633b4
@ -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", [])
|
||||||
|
Loading…
x
Reference in New Issue
Block a user