Add battery_count parameter in api/batteries endpoint

homewizard_energy/models.py

@@ -631,6 +631,7 @@ class Mode(StrEnum):
     target_power_w: float = field()
     max_consumption_w: float = field()
     max_production_w: float = field()
+    battery_count: int | None = field(default=None)
 
 
 @dataclass(kw_only=True)

tests/v2/__snapshots__/test_v2_batteries.ambr

@@ -1,12 +1,12 @@
 # serializer version: 1
 # name: test_batteries[HWE-KWH1-fixtures1]
-  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0)
+  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0, battery_count=2)
 # ---
 # name: test_batteries[HWE-KWH3-fixtures2]
-  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0)
+  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0, battery_count=2)
 # ---
 # name: test_batteries[HWE-P1-fixtures0]
-  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0)
+  Batteries(mode=<Mode.ZERO: 'zero'>, power_w=-404.0, target_power_w=-400.0, max_consumption_w=1600.0, max_production_w=800.0, battery_count=2)
 # ---
 # name: test_batteries_update[standby]
   dict({

tests/v2/fixtures/HWE-KWH1/batteries.json

@@ -1,5 +1,6 @@
 {
     "mode": "zero",
+    "battery_count": 2,
     "power_w": -404,
     "target_power_w": -400,
     "max_consumption_w": 1600,

tests/v2/fixtures/HWE-KWH3/batteries.json

@@ -1,5 +1,6 @@
 {
     "mode": "zero",
+    "battery_count": 2,
     "power_w": -404,
     "target_power_w": -400,
     "max_consumption_w": 1600,

tests/v2/fixtures/HWE-P1/batteries.json

@@ -1,5 +1,6 @@
 {
     "mode": "zero",
+    "battery_count": 2,
     "power_w": -404,
     "target_power_w": -400,
     "max_consumption_w": 1600,

tests/v2/test_v2_batteries.py

@@ -28,6 +28,28 @@ async def test_batteries(model: str, fixtures: str, snapshot: SnapshotAssertion)
         assert snapshot == data
 
 
+def test_batteries_battery_count_optional():
+    """Test Batteries model with and without battery_count."""
+    b1 = Batteries(
+        mode=Batteries.Mode.ZERO,
+        power_w=100.0,
+        target_power_w=200.0,
+        max_consumption_w=300.0,
+        max_production_w=400.0,
+        battery_count=None,
+    )
+    b2 = Batteries(
+        mode=Batteries.Mode.TO_FULL,
+        power_w=110.0,
+        target_power_w=210.0,
+        max_consumption_w=310.0,
+        max_production_w=410.0,
+        battery_count=2,
+    )
+    assert b1.battery_count is None
+    assert b2.battery_count == 2
+
+
 @pytest.mark.parametrize(
     ("mode"),
     [