mirror of
https://github.com/home-assistant/core.git
synced 2025-08-15 02:21:40 +02:00
fix mock
This commit is contained in:
@@ -1629,6 +1629,9 @@ async def test_sub_device_cleanup(
|
|||||||
devices=sub_devices_updated,
|
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
|
# Simulate reconnection which triggers device registry update
|
||||||
await device.mock_connect()
|
await device.mock_connect()
|
||||||
await hass.async_block_till_done()
|
await hass.async_block_till_done()
|
||||||
|
Reference in New Issue
Block a user