Ensure ReCollect Waste starts up even if no future pickup is found (h…

…ome-assistant#55349)
clarsen · Aug 27, 2021 · 1f37c21 · 1f37c21
1 parent 46d0523
commit 1f37c21
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/homeassistant/components/recollect_waste/config_flow.py b/homeassistant/components/recollect_waste/config_flow.py
@@ -59,7 +59,7 @@ async def async_step_user(
         )
 
         try:
-            await client.async_get_next_pickup_event()
+            await client.async_get_pickup_events()
         except RecollectError as err:
             LOGGER.error("Error during setup of integration: %s", err)
             return self.async_show_form(

diff --git a/tests/components/recollect_waste/test_config_flow.py b/tests/components/recollect_waste/test_config_flow.py
@@ -36,7 +36,7 @@ async def test_invalid_place_or_service_id(hass):
     conf = {CONF_PLACE_ID: "12345", CONF_SERVICE_ID: "12345"}
 
     with patch(
-        "aiorecollect.client.Client.async_get_next_pickup_event",
+        "aiorecollect.client.Client.async_get_pickup_events",
         side_effect=RecollectError,
     ):
         result = await hass.config_entries.flow.async_init(
@@ -87,9 +87,7 @@ async def test_step_import(hass):
 
     with patch(
         "homeassistant.components.recollect_waste.async_setup_entry", return_value=True
-    ), patch(
-        "aiorecollect.client.Client.async_get_next_pickup_event", return_value=True
-    ):
+    ), patch("aiorecollect.client.Client.async_get_pickup_events", return_value=True):
         result = await hass.config_entries.flow.async_init(
             DOMAIN, context={"source": SOURCE_IMPORT}, data=conf
         )
@@ -105,9 +103,7 @@ async def test_step_user(hass):
 
     with patch(
         "homeassistant.components.recollect_waste.async_setup_entry", return_value=True
-    ), patch(
-        "aiorecollect.client.Client.async_get_next_pickup_event", return_value=True
-    ):
+    ), patch("aiorecollect.client.Client.async_get_pickup_events", return_value=True):
         result = await hass.config_entries.flow.async_init(
             DOMAIN, context={"source": SOURCE_USER}, data=conf
         )