Change default charge.energy_cost from None to 0

mattrayner · Nov 12, 2022 · 13fcee9 · 13fcee9
1 parent 3d24b80
commit 13fcee9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/podpointclient/charge.py b/podpointclient/charge.py
@@ -23,7 +23,7 @@ def __init__(self, data: Dict[str, Any]):
         self.duration: int       = data.get('duration', 0)
         self.starts_at: datetime = lazy_convert_to_datetime(data.get('starts_at', None))
         self.ends_at: datetime   = lazy_convert_to_datetime(data.get('ends_at', None))
-        self.energy_cost: int    = data.get('energy_cost', None)
+        self.energy_cost: int    = data.get('energy_cost', 0)
 
         charging_duration_data = data.get('charging_duration', {})
         self.charging_duration = self.ChargingDuration(

diff --git a/podpointclient/version.py b/podpointclient/version.py
@@ -1,3 +1,3 @@
 """Version for the podpointclient library"""
 
-__version__ = "0.3.0-beta6"
+__version__ = "0.3.0-beta7"
diff --git a/tests/test_charge.py b/tests/test_charge.py
@@ -7,6 +7,9 @@ def test_charge():
     ongoing_charge_data = charges_data[0]
     completed_charge_data = charges_data[1]
 
+    empty_charge = Charge(data={})
+    assert empty_charge.energy_cost == 0
+
     ongoing_charge = Charge(data=ongoing_charge_data)
     assert 1 == ongoing_charge.id
     assert 12.2 == ongoing_charge.kwh_used