Skip to content

Commit

Permalink
[powerwall] Skip backup reserve sensor if data is unavailable (#69637)
Browse files Browse the repository at this point in the history
  • Loading branch information
mdz authored and balloob committed Apr 8, 2022
1 parent 8017cb2 commit 7e317be
Show file tree
Hide file tree
Showing 4 changed files with 39 additions and 5 deletions.
7 changes: 6 additions & 1 deletion homeassistant/components/powerwall/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,13 +223,18 @@ def call_base_info(power_wall: Powerwall, host: str) -> PowerwallBaseInfo:

def _fetch_powerwall_data(power_wall: Powerwall) -> PowerwallData:
"""Process and update powerwall data."""
try:
backup_reserve = power_wall.get_backup_reserve_percentage()
except MissingAttributeError:
backup_reserve = None

return PowerwallData(
charge=power_wall.get_charge(),
site_master=power_wall.get_sitemaster(),
meters=power_wall.get_meters(),
grid_services_active=power_wall.is_grid_services_active(),
grid_status=power_wall.get_grid_status(),
backup_reserve=power_wall.get_backup_reserve_percentage(),
backup_reserve=backup_reserve,
)


Expand Down
2 changes: 1 addition & 1 deletion homeassistant/components/powerwall/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ class PowerwallData:
meters: MetersAggregates
grid_services_active: bool
grid_status: GridStatus
backup_reserve: float
backup_reserve: float | None


class PowerwallRuntimeData(TypedDict):
Expand Down
8 changes: 6 additions & 2 deletions homeassistant/components/powerwall/sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,11 @@ async def async_setup_entry(
data: PowerwallData = coordinator.data
entities: list[PowerWallEntity] = [
PowerWallChargeSensor(powerwall_data),
PowerWallBackupReserveSensor(powerwall_data),
]

if data.backup_reserve is not None:
entities.append(PowerWallBackupReserveSensor(powerwall_data))

for meter in data.meters.meters:
entities.append(PowerWallExportSensor(powerwall_data, meter))
entities.append(PowerWallImportSensor(powerwall_data, meter))
Expand Down Expand Up @@ -190,8 +192,10 @@ def unique_id(self) -> str:
return f"{self.base_unique_id}_backup_reserve"

@property
def native_value(self) -> int:
def native_value(self) -> int | None:
"""Get the current value in percentage."""
if self.data.backup_reserve is None:
return None
return round(self.data.backup_reserve)


Expand Down
27 changes: 26 additions & 1 deletion tests/components/powerwall/test_sensor.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
"""The sensor tests for the powerwall platform."""
from unittest.mock import patch
from unittest.mock import Mock, patch

from tesla_powerwall.error import MissingAttributeError

from homeassistant.components.powerwall.const import DOMAIN
from homeassistant.components.sensor import ATTR_STATE_CLASS
Expand Down Expand Up @@ -112,3 +114,26 @@ async def test_sensors(hass, entity_registry_enabled_by_default):
# HA changes the implementation and a new one appears
for key, value in expected_attributes.items():
assert state.attributes[key] == value


async def test_sensor_backup_reserve_unavailable(hass):
"""Confirm that backup reserve sensor is not added if data is unavailable from the device."""

mock_powerwall = await _mock_powerwall_with_fixtures(hass)
mock_powerwall.get_backup_reserve_percentage = Mock(
side_effect=MissingAttributeError(Mock(), "backup_reserve_percent", "operation")
)

config_entry = MockConfigEntry(domain=DOMAIN, data={CONF_IP_ADDRESS: "1.2.3.4"})
config_entry.add_to_hass(hass)
with patch(
"homeassistant.components.powerwall.config_flow.Powerwall",
return_value=mock_powerwall,
), patch(
"homeassistant.components.powerwall.Powerwall", return_value=mock_powerwall
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()

state = hass.states.get("sensor.powerwall_backup_reserve")
assert state is None

0 comments on commit 7e317be

Please sign in to comment.