Use entity_registry fixture

home-assistant · Nov 9, 2023 · 92fb190 · 92fb190
1 parent 37a1d13
commit 92fb190
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/components/discovergy/test_sensor.py b/tests/components/discovergy/test_sensor.py
@@ -27,11 +27,12 @@
 async def test_sensor(
     hass: HomeAssistant,
     setup_integration: ComponentSetup,
+    entity_registry: er.EntityRegistry,
 ) -> None:
     """Test sensor setup and update."""
     await setup_integration()
 
-    entry = er.async_get(hass).async_get(
+    entry = entity_registry.async_get(
         "sensor.electricity_teststrasse_1_total_consumption"
     )
     assert entry