Add account type to Forecast.Solar integration (home-assistant#55175)

jborup · Aug 25, 2021 · 49041b1 · 49041b1
1 parent db5e159
commit 49041b1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/homeassistant/components/forecast_solar/sensor.py b/homeassistant/components/forecast_solar/sensor.py
@@ -5,7 +5,12 @@
 
 from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN, SensorEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ATTR_IDENTIFIERS, ATTR_MANUFACTURER, ATTR_NAME
+from homeassistant.const import (
+    ATTR_IDENTIFIERS,
+    ATTR_MANUFACTURER,
+    ATTR_MODEL,
+    ATTR_NAME,
+)
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import StateType
@@ -56,6 +61,7 @@ def __init__(
             ATTR_IDENTIFIERS: {(DOMAIN, entry_id)},
             ATTR_NAME: "Solar Production Forecast",
             ATTR_MANUFACTURER: "Forecast.Solar",
+            ATTR_MODEL: coordinator.data.account_type.value,
             ATTR_ENTRY_TYPE: ENTRY_TYPE_SERVICE,
         }
 

diff --git a/tests/components/forecast_solar/conftest.py b/tests/components/forecast_solar/conftest.py
@@ -61,6 +61,7 @@ def mock_forecast_solar() -> Generator[None, MagicMock, None]:
         estimate = MagicMock(spec=models.Estimate)
         estimate.now.return_value = now
         estimate.timezone = "Europe/Amsterdam"
+        estimate.account_type.value = "public"
         estimate.energy_production_today = 100000
         estimate.energy_production_tomorrow = 200000
         estimate.power_production_now = 300000

diff --git a/tests/components/forecast_solar/test_sensor.py b/tests/components/forecast_solar/test_sensor.py
@@ -142,7 +142,7 @@ async def test_sensors(
     assert device_entry.manufacturer == "Forecast.Solar"
     assert device_entry.name == "Solar Production Forecast"
     assert device_entry.entry_type == ENTRY_TYPE_SERVICE
-    assert not device_entry.model
+    assert device_entry.model == "public"
     assert not device_entry.sw_version