From f82ec4d233e6ba34acda3708f8fc6921ba725342 Mon Sep 17 00:00:00 2001 From: rappenze Date: Thu, 26 May 2022 20:52:30 +0200 Subject: [PATCH] Address issues from late review in fibaro config flow tests (#72553) --- tests/components/fibaro/test_config_flow.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/tests/components/fibaro/test_config_flow.py b/tests/components/fibaro/test_config_flow.py index 6f3e035a2f7..f056f484a58 100644 --- a/tests/components/fibaro/test_config_flow.py +++ b/tests/components/fibaro/test_config_flow.py @@ -53,7 +53,12 @@ async def test_config_flow_user_initiated_success(hass): login_mock = Mock() login_mock.get.return_value = Mock(status=True) - with patch("fiblary3.client.v4.client.Client.login", login_mock, create=True): + with patch( + "fiblary3.client.v4.client.Client.login", login_mock, create=True + ), patch( + "homeassistant.components.fibaro.async_setup_entry", + return_value=True, + ): result = await hass.config_entries.flow.async_configure( result["flow_id"], { @@ -182,7 +187,12 @@ async def test_config_flow_import(hass): """Test for importing config from configuration.yaml.""" login_mock = Mock() login_mock.get.return_value = Mock(status=True) - with patch("fiblary3.client.v4.client.Client.login", login_mock, create=True): + with patch( + "fiblary3.client.v4.client.Client.login", login_mock, create=True + ), patch( + "homeassistant.components.fibaro.async_setup_entry", + return_value=True, + ): result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_IMPORT},