mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 12:17:07 +00:00
Move device registry into function declaration for tests in NUT (#143349)
This commit is contained in:
parent
0208188bb5
commit
0b2e5cd253
@ -222,7 +222,10 @@ async def test_auth_fails(
|
|||||||
assert flows[0]["context"]["source"] == "reauth"
|
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."""
|
"""Test for serial number set on device."""
|
||||||
mock_serial_number = "A00000000000"
|
mock_serial_number = "A00000000000"
|
||||||
await async_init_integration(
|
await async_init_integration(
|
||||||
@ -234,9 +237,6 @@ async def test_serial_number(hass: HomeAssistant) -> None:
|
|||||||
list_commands_return_value=[],
|
list_commands_return_value=[],
|
||||||
)
|
)
|
||||||
|
|
||||||
device_registry = dr.async_get(hass)
|
|
||||||
assert device_registry is not None
|
|
||||||
|
|
||||||
device_entry = device_registry.async_get_device(
|
device_entry = device_registry.async_get_device(
|
||||||
identifiers={(DOMAIN, mock_serial_number)}
|
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
|
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."""
|
"""Test for suggested location on device."""
|
||||||
mock_serial_number = "A00000000000"
|
mock_serial_number = "A00000000000"
|
||||||
mock_device_location = "XYZ Location"
|
mock_device_location = "XYZ Location"
|
||||||
@ -261,9 +264,6 @@ async def test_device_location(hass: HomeAssistant) -> None:
|
|||||||
list_commands_return_value=[],
|
list_commands_return_value=[],
|
||||||
)
|
)
|
||||||
|
|
||||||
device_registry = dr.async_get(hass)
|
|
||||||
assert device_registry is not None
|
|
||||||
|
|
||||||
device_entry = device_registry.async_get_device(
|
device_entry = device_registry.async_get_device(
|
||||||
identifiers={(DOMAIN, mock_serial_number)}
|
identifiers={(DOMAIN, mock_serial_number)}
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user