Add automatic optimizer string sensors

tests/test_sensor.py

@@ -23,12 +23,15 @@ from .sample_data import SAMPLE_PAYLOAD
 
 async def test_optimizer_sensors(hass: HomeAssistant) -> None:
     """Create energy and power sensors for every optimizer."""
+    payload = deepcopy(SAMPLE_PAYLOAD)
+    payload["optimizers"][1]["stringId"] = "1.2"
+    payload["optimizers"][1]["optimizerId"] = "1.2.1"
     entry = MockConfigEntry(
         domain=DOMAIN,
         title="SolarEdge Site 4886699",
         unique_id="4886699",
         data={CONF_URL: "http://app:8099", CONF_SCAN_INTERVAL: 600},
-        options={CONF_AREA_MAPPING: {"west": ["1.1.1", "1.1.2"]}},
+        options={CONF_AREA_MAPPING: {"west": ["1.1.1", "1.2.1"]}},
     )
     entry.add_to_hass(hass)
     device_registry = dr.async_get(hass)
@@ -40,6 +43,13 @@ async def test_optimizer_sensors(hass: HomeAssistant) -> None:
         config_entry=entry,
         original_name="Old area energy",
     )
+    stale_string_entity = entity_registry.async_get_or_create(
+        "sensor",
+        DOMAIN,
+        "4886699_string_old_daily_energy",
+        config_entry=entry,
+        original_name="Old string energy",
+    )
     device_registry.async_get_or_create(
         config_entry_id=entry.entry_id,
         identifiers={(DOMAIN, "4886699_14F28854-E2")},
@@ -52,14 +62,15 @@ async def test_optimizer_sensors(hass: HomeAssistant) -> None:
     with patch(
         "custom_components.solaredge_optimizers.api."
         "SolarEdgeOptimizerApiClient.async_get_optimizers",
-        AsyncMock(return_value=parse_snapshot(SAMPLE_PAYLOAD)),
+        AsyncMock(return_value=parse_snapshot(payload)),
     ):
         assert await hass.config_entries.async_setup(entry.entry_id)
         await hass.async_block_till_done()
 
     entities = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
-    assert len(entities) == 6
+    assert len(entities) == 10
     assert entity_registry.async_get(stale_area_entity.entity_id) is None
+    assert entity_registry.async_get(stale_string_entity.entity_id) is None
 
     energy_entity = next(
         entity
@@ -93,7 +104,26 @@ async def test_optimizer_sensors(hass: HomeAssistant) -> None:
     assert area_energy_state.state == "2468.5"
     assert area_power_state.state == "321.4"
     assert area_energy_state.attributes["optimizer_count"] == 2
-    assert area_energy_state.attributes["optimizer_ids"] == ["1.1.1", "1.1.2"]
+    assert area_energy_state.attributes["optimizer_ids"] == ["1.1.1", "1.2.1"]
+
+    string_1_energy_entity = next(
+        entity
+        for entity in entities
+        if entity.unique_id == "4886699_string_1_1_daily_energy"
+    )
+    string_2_power_entity = next(
+        entity
+        for entity in entities
+        if entity.unique_id == "4886699_string_1_2_current_power"
+    )
+    string_1_energy_state = hass.states.get(string_1_energy_entity.entity_id)
+    string_2_power_state = hass.states.get(string_2_power_entity.entity_id)
+    assert string_1_energy_state.state == "1234.5"
+    assert string_1_energy_state.attributes["string_id"] == "1.1"
+    assert string_1_energy_state.attributes["optimizer_ids"] == ["1.1.1"]
+    assert string_2_power_state.state == "160.7"
+    assert string_2_power_state.attributes["string_id"] == "1.2"
+    assert string_2_power_state.attributes["optimizer_ids"] == ["1.2.1"]
 
     devices = dr.async_entries_for_config_entry(device_registry, entry.entry_id)
     assert len(devices) == 1
@@ -143,6 +173,18 @@ async def test_area_energy_unavailable_when_one_optimizer_failed(
         for entity in entities
         if entity.unique_id == "4886699_area_west_current_power"
     )
+    string_energy_entity = next(
+        entity
+        for entity in entities
+        if entity.unique_id == "4886699_string_1_1_daily_energy"
+    )
+    string_power_entity = next(
+        entity
+        for entity in entities
+        if entity.unique_id == "4886699_string_1_1_current_power"
+    )
 
     assert hass.states.get(area_energy_entity.entity_id).state == STATE_UNAVAILABLE
     assert hass.states.get(area_power_entity.entity_id).state == "321.4"
+    assert hass.states.get(string_energy_entity.entity_id).state == STATE_UNAVAILABLE
+    assert hass.states.get(string_power_entity.entity_id).state == "321.4"