attempt to fix tests

finity69x2 · Dec 13, 2021 · 4b068f6 · 4b068f6
1 parent 0bc5a9f
commit 4b068f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/custom_components/nws_alerts/config_flow.py b/custom_components/nws_alerts/config_flow.py
@@ -49,7 +49,7 @@ def _get_default(key):
     )
 
 
-async def _get_zone_list(self):
+async def _get_zone_list(self) -> list | None:
     """Return list of zone by lat/lon"""
 
     data = None

diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
@@ -49,7 +49,9 @@ async def test_form(
     with patch(
         "custom_components.nws_alerts.async_setup_entry",
         return_value=True,
-    ) as mock_setup_entry:
+    ) as mock_setup_entry, patch(
+        "custom_components.nws_alerts.config_flow._get_zone_list",
+        return_value=None):
 
         result2 = await hass.config_entries.flow.async_configure(
             result["flow_id"], input