diff --git a/tests/components/tailscale/test_binary_sensor.py b/tests/components/tailscale/test_binary_sensor.py index b59d3872655..b2b593101d7 100644 --- a/tests/components/tailscale/test_binary_sensor.py +++ b/tests/components/tailscale/test_binary_sensor.py @@ -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 diff --git a/tests/components/tailscale/test_sensor.py b/tests/components/tailscale/test_sensor.py index aa2bc6c472a..776b707202b 100644 --- a/tests/components/tailscale/test_sensor.py +++ b/tests/components/tailscale/test_sensor.py @@ -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