diff --git a/homeassistant/components/energyzero/coordinator.py b/homeassistant/components/energyzero/coordinator.py index 35054f7b3b7..122c2f76deb 100644 --- a/homeassistant/components/energyzero/coordinator.py +++ b/homeassistant/components/energyzero/coordinator.py @@ -56,11 +56,11 @@ class EnergyZeroDataUpdateCoordinator(DataUpdateCoordinator[EnergyZeroData]): energy_tomorrow = None try: - energy_today = await self.energyzero.energy_prices( + energy_today = await self.energyzero.get_electricity_prices_legacy( start_date=today, end_date=today ) try: - gas_today = await self.energyzero.gas_prices( + gas_today = await self.energyzero.get_gas_prices_legacy( start_date=today, end_date=today ) except EnergyZeroNoDataError: @@ -69,8 +69,10 @@ class EnergyZeroDataUpdateCoordinator(DataUpdateCoordinator[EnergyZeroData]): if dt_util.utcnow().hour >= THRESHOLD_HOUR: tomorrow = today + timedelta(days=1) try: - energy_tomorrow = await self.energyzero.energy_prices( - start_date=tomorrow, end_date=tomorrow + energy_tomorrow = ( + await self.energyzero.get_electricity_prices_legacy( + start_date=tomorrow, end_date=tomorrow + ) ) except EnergyZeroNoDataError: LOGGER.debug("No data for tomorrow for EnergyZero integration") diff --git a/homeassistant/components/energyzero/manifest.json b/homeassistant/components/energyzero/manifest.json index b647faebe1d..dff678c0fe3 100644 --- a/homeassistant/components/energyzero/manifest.json +++ b/homeassistant/components/energyzero/manifest.json @@ -6,6 +6,6 @@ "documentation": "https://www.home-assistant.io/integrations/energyzero", "integration_type": "service", "iot_class": "cloud_polling", - "requirements": ["energyzero==2.1.1"], + "requirements": ["energyzero==4.0.1"], "single_config_entry": true } diff --git a/homeassistant/components/energyzero/services.py b/homeassistant/components/energyzero/services.py index c47958b670f..8609cb745f3 100644 --- a/homeassistant/components/energyzero/services.py +++ b/homeassistant/components/energyzero/services.py @@ -128,13 +128,13 @@ async def __get_prices( data: Electricity | Gas if price_type == PriceType.GAS: - data = await coordinator.energyzero.gas_prices( + data = await coordinator.energyzero.get_gas_prices_legacy( start_date=start, end_date=end, vat=vat, ) else: - data = await coordinator.energyzero.energy_prices( + data = await coordinator.energyzero.get_electricity_prices_legacy( start_date=start, end_date=end, vat=vat, diff --git a/requirements_all.txt b/requirements_all.txt index 9c00d31b99b..a5599f22267 100644 --- a/requirements_all.txt +++ b/requirements_all.txt @@ -893,7 +893,7 @@ energyflip-client==0.2.2 energyid-webhooks==0.0.14 # homeassistant.components.energyzero -energyzero==2.1.1 +energyzero==4.0.1 # homeassistant.components.enocean enocean==0.50 diff --git a/requirements_test_all.txt b/requirements_test_all.txt index 1627aa69054..abc6292ea13 100644 --- a/requirements_test_all.txt +++ b/requirements_test_all.txt @@ -787,7 +787,7 @@ energyflip-client==0.2.2 energyid-webhooks==0.0.14 # homeassistant.components.energyzero -energyzero==2.1.1 +energyzero==4.0.1 # homeassistant.components.enocean enocean==0.50 diff --git a/tests/components/energyzero/conftest.py b/tests/components/energyzero/conftest.py index d861e1365f7..40fa50a9144 100644 --- a/tests/components/energyzero/conftest.py +++ b/tests/components/energyzero/conftest.py @@ -40,10 +40,10 @@ async def mock_energyzero(hass: HomeAssistant) -> AsyncGenerator[MagicMock]: "homeassistant.components.energyzero.coordinator.EnergyZero", autospec=True ) as energyzero_mock: client = energyzero_mock.return_value - client.energy_prices.return_value = Electricity.from_dict( + client.get_electricity_prices_legacy.return_value = Electricity.from_dict( await async_load_json_object_fixture(hass, "today_energy.json", DOMAIN) ) - client.gas_prices.return_value = Gas.from_dict( + client.get_gas_prices_legacy.return_value = Gas.from_dict( await async_load_json_object_fixture(hass, "today_gas.json", DOMAIN) ) yield client diff --git a/tests/components/energyzero/test_diagnostics.py b/tests/components/energyzero/test_diagnostics.py index 198f21822c7..53d58691bcc 100644 --- a/tests/components/energyzero/test_diagnostics.py +++ b/tests/components/energyzero/test_diagnostics.py @@ -45,7 +45,7 @@ async def test_diagnostics_no_gas_today( snapshot: SnapshotAssertion, ) -> None: """Test diagnostics, no gas sensors available.""" - mock_energyzero.gas_prices.side_effect = EnergyZeroNoDataError + mock_energyzero.get_gas_prices_legacy.side_effect = EnergyZeroNoDataError freezer.tick(SCAN_INTERVAL) async_fire_time_changed(hass) diff --git a/tests/components/energyzero/test_sensor.py b/tests/components/energyzero/test_sensor.py index d952ac77515..a5afb4b8d18 100644 --- a/tests/components/energyzero/test_sensor.py +++ b/tests/components/energyzero/test_sensor.py @@ -40,7 +40,7 @@ async def test_no_gas_today( freezer: FrozenDateTimeFactory, ) -> None: """Test the EnergyZero - No gas sensors available.""" - mock_energyzero.gas_prices.side_effect = EnergyZeroNoDataError + mock_energyzero.get_gas_prices_legacy.side_effect = EnergyZeroNoDataError freezer.tick(SCAN_INTERVAL) async_fire_time_changed(hass)