Battery status for Roborock Q7 (#165886)

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

homeassistant/components/roborock/coordinator.py

@@ -550,6 +550,7 @@ class RoborockB01Q7UpdateCoordinator(RoborockDataUpdateCoordinatorB01):
             RoborockB01Props.WIND,
             RoborockB01Props.WATER,
             RoborockB01Props.MODE,
+            RoborockB01Props.QUANTITY,
         ]
 
     async def _async_update_data(

homeassistant/components/roborock/sensor.py

@@ -366,6 +366,13 @@ Q7_B01_SENSOR_DESCRIPTIONS = [
         translation_key="main_brush_time_left",
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
+    RoborockSensorDescriptionB01(
+        key="battery",
+        value_fn=lambda data: data.battery,
+        device_class=SensorDeviceClass.BATTERY,
+        native_unit_of_measurement=PERCENTAGE,
+        entity_category=EntityCategory.DIAGNOSTIC,
+    ),
     RoborockSensorDescriptionB01(
         key="side_brush_time_left",
         value_fn=lambda data: data.side_brush_time_left,

tests/components/roborock/mock_data.py

@@ -1564,6 +1564,7 @@ Q7_B01_PROPS = B01Props(
     main_sensor=500,
     mop_life=1200,
     real_clean_time=3000,
+    quantity=100,
 )
 
 Q10_STATUS = Q10Status(

tests/components/roborock/snapshots/test_sensor.ambr

@@ -407,6 +407,58 @@
     'state': '3.55',
   })
 # ---
+# name: test_sensors[sensor.roborock_q7_battery-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': list([
+      None,
+    ]),
+    'area_id': None,
+    'capabilities': None,
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'sensor',
+    'entity_category': <EntityCategory.DIAGNOSTIC: 'diagnostic'>,
+    'entity_id': 'sensor.roborock_q7_battery',
+    'has_entity_name': True,
+    'hidden_by': None,
+    'icon': None,
+    'id': <ANY>,
+    'labels': set({
+    }),
+    'name': None,
+    'object_id_base': 'Battery',
+    'options': dict({
+    }),
+    'original_device_class': <SensorDeviceClass.BATTERY: 'battery'>,
+    'original_icon': None,
+    'original_name': 'Battery',
+    'platform': 'roborock',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': None,
+    'unique_id': 'battery_q7_duid',
+    'unit_of_measurement': '%',
+  })
+# ---
+# name: test_sensors[sensor.roborock_q7_battery-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'device_class': 'battery',
+      'friendly_name': 'Roborock Q7 Battery',
+      'unit_of_measurement': '%',
+    }),
+    'context': <ANY>,
+    'entity_id': 'sensor.roborock_q7_battery',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': '100',
+  })
+# ---
 # name: test_sensors[sensor.roborock_q7_filter_time_left-entry]
   EntityRegistryEntrySnapshot({
     'aliases': list([