mock the get_zone_list

finity69x2 · Dec 13, 2021 · 4d58d08 · 4d58d08
1 parent 4b068f6
commit 4d58d08
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
@@ -39,12 +39,16 @@ async def test_form(
 ):
     """Test we get the form."""
     await setup.async_setup_component(hass, "persistent_notification", {})
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": config_entries.SOURCE_USER}
-    )
-    assert result["type"] == "form"
-    assert result["errors"] == {}
-    # assert result["title"] == title_1
+    with patch(
+        "custom_components.nws_alerts.config_flow._get_zone_list",
+        return_value=None):
+
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+        assert result["type"] == "form"
+        assert result["errors"] == {}
+        # assert result["title"] == title_1
 
     with patch(
         "custom_components.nws_alerts.async_setup_entry",