From a665e152a95872f648e2ec96e6795811492312b8 Mon Sep 17 00:00:00 2001 From: Martin Hjelmare Date: Thu, 12 Nov 2020 12:11:05 +0100 Subject: [PATCH] Fix aurora config flow tests (#43128) --- tests/components/aurora/test_config_flow.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/components/aurora/test_config_flow.py b/tests/components/aurora/test_config_flow.py index 4d611bd3272..2f4b457a9dd 100644 --- a/tests/components/aurora/test_config_flow.py +++ b/tests/components/aurora/test_config_flow.py @@ -35,11 +35,11 @@ async def test_form(hass): result["flow_id"], DATA, ) + await hass.async_block_till_done() assert result2["type"] == "create_entry" assert result2["title"] == "Aurora - Home" assert result2["data"] == DATA - await hass.async_block_till_done() assert len(mock_setup.mock_calls) == 1 assert len(mock_setup_entry.mock_calls) == 1 @@ -95,6 +95,8 @@ async def test_option_flow(hass): assert not entry.options with patch("homeassistant.components.aurora.async_setup_entry", return_value=True): + await hass.config_entries.async_setup(entry.entry_id) + await hass.async_block_till_done() result = await hass.config_entries.options.async_init( entry.entry_id, data=None,