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()