Surepetcare, bug fix

homeassistant/components/surepetcare/__init__.py

@@ -144,7 +144,7 @@ async def async_update(self, _: Any = None) -> None:
         """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

tests/components/surepetcare/conftest.py

@@ -7,12 +7,16 @@
 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

tests/components/surepetcare/test_sensor.py

@@ -12,7 +12,7 @@
 }
 
 
-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()