Add SolarEdge Optimizers Home Assistant integration

tests/test_sensor.py

@@ -0,0 +1,56 @@
+"""Tests for SolarEdge optimizer sensors."""
+
+from __future__ import annotations
+
+from unittest.mock import AsyncMock, patch
+
+from homeassistant.const import CONF_URL
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers import entity_registry as er
+from pytest_homeassistant_custom_component.common import MockConfigEntry
+
+from custom_components.solaredge_optimizers.api import parse_snapshot
+from custom_components.solaredge_optimizers.const import CONF_SCAN_INTERVAL, DOMAIN
+
+from .sample_data import SAMPLE_PAYLOAD
+
+
+async def test_optimizer_sensors(hass: HomeAssistant) -> None:
+    """Create energy and power sensors for every optimizer."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        title="SolarEdge Site 4886699",
+        unique_id="4886699",
+        data={CONF_URL: "http://app:8099", CONF_SCAN_INTERVAL: 600},
+    )
+    entry.add_to_hass(hass)
+
+    with patch(
+        "custom_components.solaredge_optimizers.api."
+        "SolarEdgeOptimizerApiClient.async_get_optimizers",
+        AsyncMock(return_value=parse_snapshot(SAMPLE_PAYLOAD)),
+    ):
+        assert await hass.config_entries.async_setup(entry.entry_id)
+        await hass.async_block_till_done()
+
+    entity_registry = er.async_get(hass)
+    entities = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
+    assert len(entities) == 4
+
+    energy_entity = next(
+        entity for entity in entities if entity.unique_id.endswith("_daily_energy")
+    )
+    power_entity = next(
+        entity for entity in entities if entity.unique_id.endswith("_current_power")
+    )
+    assert hass.states.get(energy_entity.entity_id).state == "1234.5"
+    assert hass.states.get(power_entity.entity_id).state == "160.7"
+
+    device_registry = dr.async_get(hass)
+    devices = dr.async_entries_for_config_entry(device_registry, entry.entry_id)
+    assert len(devices) == 2
+    assert {device.serial_number for device in devices} == {
+        "14F28854-E2",
+        "14F28855-E2",
+    }