From 414a6dbcc80f7ba085a4f755f8c9ba51ac8f3a3a Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Mon, 23 Jun 2025 13:56:20 +0200
Subject: [PATCH] fix mock

---
 tests/components/esphome/test_manager.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/components/esphome/test_manager.py b/tests/components/esphome/test_manager.py
index f3b96434a2a..b0bbea03fe6 100644
--- a/tests/components/esphome/test_manager.py
+++ b/tests/components/esphome/test_manager.py
@@ -1629,6 +1629,9 @@ async def test_sub_device_cleanup(
         devices=sub_devices_updated,
     )
 
+    # Update the mock client to return the new device info
+    mock_client.device_info = AsyncMock(return_value=device.device_info)
+
     # Simulate reconnection which triggers device registry update
     await device.mock_connect()
     await hass.async_block_till_done()