Use registry fixtures in tests (tailscale) (#118301)

This commit is contained in:
epenet 2024-05-28 13:12:51 +02:00 committed by GitHub
parent 21f5ac7715
commit 01be006d40
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 6 deletions

View file

@ -15,12 +15,11 @@ from tests.common import MockConfigEntry
async def test_tailscale_binary_sensors(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
init_integration: MockConfigEntry,
) -> None:
"""Test the Tailscale binary sensors."""
entity_registry = er.async_get(hass)
device_registry = dr.async_get(hass)
state = hass.states.get("binary_sensor.frencks_iphone_client")
entry = entity_registry.async_get("binary_sensor.frencks_iphone_client")
assert entry

View file

@ -11,12 +11,11 @@ from tests.common import MockConfigEntry
async def test_tailscale_sensors(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
init_integration: MockConfigEntry,
) -> None:
"""Test the Tailscale sensors."""
entity_registry = er.async_get(hass)
device_registry = dr.async_get(hass)
state = hass.states.get("sensor.router_expires")
entry = entity_registry.async_get("sensor.router_expires")
assert entry