add None usecase

marcolivierarsenault · Aug 30, 2024 · 0f3f26d · 0f3f26d
1 parent 0b0b936
commit 0f3f26d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/custom_components/moonraker/sensor.py b/custom_components/moonraker/sensor.py
@@ -215,6 +215,8 @@ class MoonrakerSensorDescription(SensorEntityDescription):
             sensor.coordinator.data["status"]["print_stats"]["info"]["total_layer"]
             if "total_layer" in sensor.coordinator.data["status"]["print_stats"]["info"]
             and sensor.coordinator.data["status"]["print_stats"]["info"]["total_layer"]
+            is not None
+            and sensor.coordinator.data["status"]["print_stats"]["info"]["total_layer"]
             > 0
             else sensor.coordinator.data["layer_count"]
         ),

diff --git a/tests/test_sensor.py b/tests/test_sensor.py
@@ -467,6 +467,19 @@ async def test_total_layer_in_info_0(hass, get_data):
     assert state.state == "313"
 
 
+async def test_total_layer_in_info_is_none(hass, get_data):
+    """Test."""
+    get_data["status"]["print_stats"]["info"]["total_layer"] = None
+
+    config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_CONFIG, entry_id="test")
+    config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    state = hass.states.get("sensor.mainsail_total_layer")
+    assert state.state == "313"
+
+
 async def test_current_layer_calculated():
     """Test."""
     data = {