From bcfedeb79717980eea6198d2ae19ca3ebae7aaca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= <mail@dahoiv.net>
Date: Mon, 6 Sep 2021 22:36:45 +0200
Subject: [PATCH] Surepetcare, bug fix (#55842)

---
 homeassistant/components/surepetcare/__init__.py |  2 +-
 tests/components/surepetcare/conftest.py         | 12 ++++++++----
 tests/components/surepetcare/test_sensor.py      |  2 +-
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/surepetcare/__init__.py b/homeassistant/components/surepetcare/__init__.py
index 58890090d57..00c45701423 100644
--- a/homeassistant/components/surepetcare/__init__.py
+++ b/homeassistant/components/surepetcare/__init__.py
@@ -144,7 +144,7 @@ class SurePetcareAPI:
         """Get the latest data from Sure Petcare."""
 
         try:
-            self.states = await self.surepy.get_entities()
+            self.states = await self.surepy.get_entities(refresh=True)
         except SurePetcareError as error:
             _LOGGER.error("Unable to fetch data: %s", error)
             return
diff --git a/tests/components/surepetcare/conftest.py b/tests/components/surepetcare/conftest.py
index 43738f22587..cecdaababa9 100644
--- a/tests/components/surepetcare/conftest.py
+++ b/tests/components/surepetcare/conftest.py
@@ -7,12 +7,16 @@ from surepy import MESTART_RESOURCE
 from . import MOCK_API_DATA
 
 
+async def _mock_call(method, resource):
+    if method == "GET" and resource == MESTART_RESOURCE:
+        return {"data": MOCK_API_DATA}
+
+
 @pytest.fixture
 async def surepetcare():
     """Mock the SurePetcare for easier testing."""
-    with patch("surepy.SureAPIClient", autospec=True) as mock_client_class, patch(
-        "surepy.find_token"
-    ):
+    with patch("surepy.SureAPIClient", autospec=True) as mock_client_class:
         client = mock_client_class.return_value
-        client.resources = {MESTART_RESOURCE: {"data": MOCK_API_DATA}}
+        client.resources = {}
+        client.call = _mock_call
         yield client
diff --git a/tests/components/surepetcare/test_sensor.py b/tests/components/surepetcare/test_sensor.py
index 8e7160364ea..cbf69bb97dc 100644
--- a/tests/components/surepetcare/test_sensor.py
+++ b/tests/components/surepetcare/test_sensor.py
@@ -12,7 +12,7 @@ EXPECTED_ENTITY_IDS = {
 }
 
 
-async def test_binary_sensors(hass, surepetcare) -> None:
+async def test_sensors(hass, surepetcare) -> None:
     """Test the generation of unique ids."""
     assert await async_setup_component(hass, DOMAIN, MOCK_CONFIG)
     await hass.async_block_till_done()