Move device registry into function declaration for tests in NUT (#143349)

This commit is contained in:
tdfountain 2025-04-22 16:11:14 -07:00 committed by GitHub
parent 0208188bb5
commit 0b2e5cd253
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -222,7 +222,10 @@ async def test_auth_fails(
assert flows[0]["context"]["source"] == "reauth"
async def test_serial_number(hass: HomeAssistant) -> None:
async def test_serial_number(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
) -> None:
"""Test for serial number set on device."""
mock_serial_number = "A00000000000"
await async_init_integration(
@ -234,9 +237,6 @@ async def test_serial_number(hass: HomeAssistant) -> None:
list_commands_return_value=[],
)
device_registry = dr.async_get(hass)
assert device_registry is not None
device_entry = device_registry.async_get_device(
identifiers={(DOMAIN, mock_serial_number)}
)
@ -245,7 +245,10 @@ async def test_serial_number(hass: HomeAssistant) -> None:
assert device_entry.serial_number == mock_serial_number
async def test_device_location(hass: HomeAssistant) -> None:
async def test_device_location(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
) -> None:
"""Test for suggested location on device."""
mock_serial_number = "A00000000000"
mock_device_location = "XYZ Location"
@ -261,9 +264,6 @@ async def test_device_location(hass: HomeAssistant) -> None:
list_commands_return_value=[],
)
device_registry = dr.async_get(hass)
assert device_registry is not None
device_entry = device_registry.async_get_device(
identifiers={(DOMAIN, mock_serial_number)}
)