From 4f59317ab5d8f846167ac0cea98449df99463834 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Mon, 23 Jun 2025 13:26:40 +0200
Subject: [PATCH] cover

---
 tests/components/esphome/test_entity.py  |  2 +-
 tests/components/esphome/test_manager.py | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/tests/components/esphome/test_entity.py b/tests/components/esphome/test_entity.py
index 3aafa7ded2e..44d2aeddc2f 100644
--- a/tests/components/esphome/test_entity.py
+++ b/tests/components/esphome/test_entity.py
@@ -764,7 +764,7 @@ async def test_entity_assignment_to_sub_device(
 
     # Check main device
     main_device = device_registry.async_get_device(
-        connections={(dr.CONNECTION_NETWORK_MAC, device.entry.unique_id)}
+        connections={(dr.CONNECTION_NETWORK_MAC, device.device_info.mac_address)}
     )
     assert main_device is not None
 
diff --git a/tests/components/esphome/test_manager.py b/tests/components/esphome/test_manager.py
index d3491a42bfe..f3b96434a2a 100644
--- a/tests/components/esphome/test_manager.py
+++ b/tests/components/esphome/test_manager.py
@@ -1597,19 +1597,19 @@ async def test_sub_device_cleanup(
     # Verify all sub devices exist
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_11111111")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_11111111")}
         )
         is not None
     )
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_22222222")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_22222222")}
         )
         is not None
     )
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_33333333")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_33333333")}
         )
         is not None
     )
@@ -1636,19 +1636,19 @@ async def test_sub_device_cleanup(
     # Verify device 2 was removed
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_11111111")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_11111111")}
         )
         is not None
     )
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_22222222")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_22222222")}
         )
         is None
     )  # Should be removed
     assert (
         device_registry.async_get_device(
-            identifiers={(DOMAIN, f"{device.entry.unique_id}_33333333")}
+            identifiers={(DOMAIN, f"{device.device_info.mac_address}_33333333")}
         )
         is not None
     )