home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-05 21:55:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
tdfountain
2025-04-22 16:11:14 -07:00
committed by GitHub
parent 0208188bb5
commit 0b2e5cd253
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/components/nut/test_init.py
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)}
)