Update energyzero to v4.0.1 (home-assistant#158211)

Author: klaasnicolaas

homeassistant/components/energyzero/coordinator.py

@@ -56,11 +56,11 @@ async def _async_update_data(self) -> 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 @@ async def _async_update_data(self) -> 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")

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
 }

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,

requirements_all.txt

@@ -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

requirements_test_all.txt

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

tests/components/energyzero/conftest.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)