From d4cdb8e7e89cb4bfde6da9219b299f2c503f13c3 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Tue, 11 May 2021 00:06:29 -0500
Subject: [PATCH] Include mac address in roomba device info when available
 (#50437)

---
 homeassistant/components/roomba/irobot_base.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/roomba/irobot_base.py b/homeassistant/components/roomba/irobot_base.py
index 9d6a0f5cafc..45a69d38576 100644
--- a/homeassistant/components/roomba/irobot_base.py
+++ b/homeassistant/components/roomba/irobot_base.py
@@ -21,6 +21,7 @@ from homeassistant.components.vacuum import (
     SUPPORT_STOP,
     StateVacuumEntity,
 )
+import homeassistant.helpers.device_registry as dr
 from homeassistant.helpers.entity import Entity
 
 from . import roomba_reported_state
@@ -92,13 +93,18 @@ class IRobotEntity(Entity):
     @property
     def device_info(self):
         """Return the device info of the vacuum cleaner."""
-        return {
+        info = {
             "identifiers": {(DOMAIN, self.robot_unique_id)},
             "manufacturer": "iRobot",
             "name": str(self._name),
             "sw_version": self._version,
             "model": self._sku,
         }
+        if mac_address := self.vacuum_state.get("hwPartsRev", {}).get(
+            "wlan0HwAddr", self.vacuum_state.get("mac")
+        ):
+            info["connections"] = {(dr.CONNECTION_NETWORK_MAC, mac_address)}
+        return info
 
     @property
     def _battery_level(self):