From 49a4101e925e7607a0a2051bbcd36743e85f3a6b Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Fri, 7 Aug 2026 13:42:58 +0200 Subject: [PATCH] Migrate calls to async_get_device in ruckus_unleashed tests (#178417) --- tests/components/ruckus_unleashed/test_init.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/components/ruckus_unleashed/test_init.py b/tests/components/ruckus_unleashed/test_init.py index c77dff769255..4f5d7eaa5e44 100644 --- a/tests/components/ruckus_unleashed/test_init.py +++ b/tests/components/ruckus_unleashed/test_init.py @@ -60,13 +60,12 @@ async def test_router_device_setup( hass: HomeAssistant, device_registry: dr.DeviceRegistry ) -> None: """Test a router device is created.""" - await init_integration(hass) + entry = await init_integration(hass) device_info = DEFAULT_AP_INFO[0] - device = device_registry.async_get_device( - identifiers={(CONNECTION_NETWORK_MAC, device_info[API_AP_MAC])}, - connections={(CONNECTION_NETWORK_MAC, device_info[API_AP_MAC])}, + device = device_registry.async_get_device_by_connection( + (CONNECTION_NETWORK_MAC, device_info[API_AP_MAC]), entry.entry_id ) assert device