From db0c82b88eb28bda03dfae9f56c1e707fced2fea Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Fri, 26 Nov 2021 00:00:37 +0100
Subject: [PATCH] Fix slow config_flow test in Twinkly (#60374)

---
 tests/components/twinkly/test_config_flow.py | 27 ++++++++++----------
 1 file changed, 14 insertions(+), 13 deletions(-)

diff --git a/tests/components/twinkly/test_config_flow.py b/tests/components/twinkly/test_config_flow.py
index afbe608219f..46566bdf54b 100644
--- a/tests/components/twinkly/test_config_flow.py
+++ b/tests/components/twinkly/test_config_flow.py
@@ -11,23 +11,24 @@ from homeassistant.components.twinkly.const import (
     DOMAIN as TWINKLY_DOMAIN,
 )
 
-from tests.components.twinkly import TEST_MODEL, ClientMock
+from . import TEST_MODEL, ClientMock
 
 
 async def test_invalid_host(hass):
     """Test the failure when invalid host provided."""
-    result = await hass.config_entries.flow.async_init(
-        TWINKLY_DOMAIN, context={"source": config_entries.SOURCE_USER}
-    )
-
-    assert result["type"] == "form"
-    assert result["step_id"] == "user"
-    assert result["errors"] == {}
-
-    result = await hass.config_entries.flow.async_configure(
-        result["flow_id"],
-        {CONF_ENTRY_HOST: "dummy"},
-    )
+    client = ClientMock()
+    client.is_offline = True
+    with patch("twinkly_client.TwinklyClient", return_value=client):
+        result = await hass.config_entries.flow.async_init(
+            TWINKLY_DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+        assert result["type"] == "form"
+        assert result["step_id"] == "user"
+        assert result["errors"] == {}
+        result = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            {CONF_ENTRY_HOST: "dummy"},
+        )
 
     assert result["type"] == "form"
     assert result["step_id"] == "user"