From 6c41540ad8e10391d6591f36c2bfd4b1d3e0b0b6 Mon Sep 17 00:00:00 2001
From: Josef Zweck <24647999+zweckj@users.noreply.github.com>
Date: Sun, 4 Feb 2024 11:29:32 +0100
Subject: [PATCH] Bugfix lamarzocco issue (#109596)

---
 homeassistant/components/lamarzocco/number.py              | 2 +-
 homeassistant/components/lamarzocco/sensor.py              | 2 ++
 tests/components/lamarzocco/fixtures/current_status.json   | 7 ++-----
 .../components/lamarzocco/snapshots/test_diagnostics.ambr  | 5 +----
 tests/components/lamarzocco/snapshots/test_sensor.ambr     | 6 ++++++
 5 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/homeassistant/components/lamarzocco/number.py b/homeassistant/components/lamarzocco/number.py
index 76632d4a5b8..bf866872f5b 100644
--- a/homeassistant/components/lamarzocco/number.py
+++ b/homeassistant/components/lamarzocco/number.py
@@ -79,7 +79,7 @@ ENTITIES: tuple[LaMarzoccoNumberEntityDescription, ...] = (
         set_value_fn=lambda coordinator, value: coordinator.lm.set_dose_hot_water(
             value=int(value)
         ),
-        native_value_fn=lambda lm: lm.current_status["dose_k5"],
+        native_value_fn=lambda lm: lm.current_status["dose_hot_water"],
         supported_fn=lambda coordinator: coordinator.lm.model_name
         in (
             LaMarzoccoModel.GS3_AV,
diff --git a/homeassistant/components/lamarzocco/sensor.py b/homeassistant/components/lamarzocco/sensor.py
index c46b965850c..ea5a5e184e1 100644
--- a/homeassistant/components/lamarzocco/sensor.py
+++ b/homeassistant/components/lamarzocco/sensor.py
@@ -62,6 +62,7 @@ ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
         key="current_temp_coffee",
         translation_key="current_temp_coffee",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+        suggested_display_precision=1,
         state_class=SensorStateClass.MEASUREMENT,
         device_class=SensorDeviceClass.TEMPERATURE,
         value_fn=lambda lm: lm.current_status.get("coffee_temp", 0),
@@ -70,6 +71,7 @@ ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
         key="current_temp_steam",
         translation_key="current_temp_steam",
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+        suggested_display_precision=1,
         state_class=SensorStateClass.MEASUREMENT,
         device_class=SensorDeviceClass.TEMPERATURE,
         value_fn=lambda lm: lm.current_status.get("steam_temp", 0),
diff --git a/tests/components/lamarzocco/fixtures/current_status.json b/tests/components/lamarzocco/fixtures/current_status.json
index 4f208607c17..f99c3d5c331 100644
--- a/tests/components/lamarzocco/fixtures/current_status.json
+++ b/tests/components/lamarzocco/fixtures/current_status.json
@@ -43,7 +43,7 @@
   "dose_k2": 1023,
   "dose_k3": 1023,
   "dose_k4": 1023,
-  "dose_k5": 1023,
+  "dose_hot_water": 1023,
   "prebrewing_ton_k1": 3,
   "prebrewing_toff_k1": 5,
   "prebrewing_ton_k2": 3,
@@ -52,11 +52,8 @@
   "prebrewing_toff_k3": 5,
   "prebrewing_ton_k4": 3,
   "prebrewing_toff_k4": 5,
-  "prebrewing_ton_k5": 3,
-  "prebrewing_toff_k5": 5,
   "preinfusion_k1": 4,
   "preinfusion_k2": 4,
   "preinfusion_k3": 4,
-  "preinfusion_k4": 4,
-  "preinfusion_k5": 4
+  "preinfusion_k4": 4
 }
diff --git a/tests/components/lamarzocco/snapshots/test_diagnostics.ambr b/tests/components/lamarzocco/snapshots/test_diagnostics.ambr
index 2462d4a125d..ec44100fe1e 100644
--- a/tests/components/lamarzocco/snapshots/test_diagnostics.ambr
+++ b/tests/components/lamarzocco/snapshots/test_diagnostics.ambr
@@ -198,11 +198,11 @@
       'coffee_boiler_on': True,
       'coffee_set_temp': 95,
       'coffee_temp': 93,
+      'dose_hot_water': 1023,
       'dose_k1': 1023,
       'dose_k2': 1023,
       'dose_k3': 1023,
       'dose_k4': 1023,
-      'dose_k5': 1023,
       'drinks_k1': 13,
       'drinks_k2': 2,
       'drinks_k3': 42,
@@ -221,17 +221,14 @@
       'prebrewing_toff_k2': 5,
       'prebrewing_toff_k3': 5,
       'prebrewing_toff_k4': 5,
-      'prebrewing_toff_k5': 5,
       'prebrewing_ton_k1': 3,
       'prebrewing_ton_k2': 3,
       'prebrewing_ton_k3': 3,
       'prebrewing_ton_k4': 3,
-      'prebrewing_ton_k5': 3,
       'preinfusion_k1': 4,
       'preinfusion_k2': 4,
       'preinfusion_k3': 4,
       'preinfusion_k4': 4,
-      'preinfusion_k5': 4,
       'sat_auto': 'Disabled',
       'sat_off_time': '00:00',
       'sat_on_time': '00:00',
diff --git a/tests/components/lamarzocco/snapshots/test_sensor.ambr b/tests/components/lamarzocco/snapshots/test_sensor.ambr
index 4228252f526..e0b04289f7c 100644
--- a/tests/components/lamarzocco/snapshots/test_sensor.ambr
+++ b/tests/components/lamarzocco/snapshots/test_sensor.ambr
@@ -20,6 +20,9 @@
     'id': <ANY>,
     'name': None,
     'options': dict({
+      'sensor': dict({
+        'suggested_display_precision': 1,
+      }),
     }),
     'original_device_class': <SensorDeviceClass.TEMPERATURE: 'temperature'>,
     'original_icon': None,
@@ -68,6 +71,9 @@
     'id': <ANY>,
     'name': None,
     'options': dict({
+      'sensor': dict({
+        'suggested_display_precision': 1,
+      }),
     }),
     'original_device_class': <SensorDeviceClass.TEMPERATURE: 'temperature'>,
     'original_icon': None,