From 85587f995b0fec052e882e79cd4e4f512eb58793 Mon Sep 17 00:00:00 2001
From: Jan-Philipp Benecke <jan-philipp@bnck.me>
Date: Sun, 18 Feb 2024 20:55:12 +0100
Subject: [PATCH] Use entity registry fixture in octoprint tests (#110887)

---
 tests/components/octoprint/test_sensor.py | 26 ++++++++++++-----------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/tests/components/octoprint/test_sensor.py b/tests/components/octoprint/test_sensor.py
index 3d3efd04da0..2a0e14e5ce0 100644
--- a/tests/components/octoprint/test_sensor.py
+++ b/tests/components/octoprint/test_sensor.py
@@ -9,7 +9,11 @@ from homeassistant.helpers import entity_registry as er
 from . import init_integration
 
 
-async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> None:
+async def test_sensors(
+    hass: HomeAssistant,
+    freezer: FrozenDateTimeFactory,
+    entity_registry: er.EntityRegistry,
+) -> None:
     """Test the underlying sensors."""
     printer = {
         "state": {
@@ -26,8 +30,6 @@ async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> N
     freezer.move_to(datetime(2020, 2, 20, 9, 10, 13, 543, tzinfo=UTC))
     await init_integration(hass, "sensor", printer=printer, job=job)
 
-    entity_registry = er.async_get(hass)
-
     state = hass.states.get("sensor.octoprint_job_percentage")
     assert state is not None
     assert state.state == "50"
@@ -79,7 +81,9 @@ async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> N
 
 
 async def test_sensors_no_target_temp(
-    hass: HomeAssistant, freezer: FrozenDateTimeFactory
+    hass: HomeAssistant,
+    freezer: FrozenDateTimeFactory,
+    entity_registry: er.EntityRegistry,
 ) -> None:
     """Test the underlying sensors."""
     printer = {
@@ -92,8 +96,6 @@ async def test_sensors_no_target_temp(
     freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
     await init_integration(hass, "sensor", printer=printer)
 
-    entity_registry = er.async_get(hass)
-
     state = hass.states.get("sensor.octoprint_actual_tool1_temp")
     assert state is not None
     assert state.state == "18.83"
@@ -110,7 +112,9 @@ async def test_sensors_no_target_temp(
 
 
 async def test_sensors_paused(
-    hass: HomeAssistant, freezer: FrozenDateTimeFactory
+    hass: HomeAssistant,
+    freezer: FrozenDateTimeFactory,
+    entity_registry: er.EntityRegistry,
 ) -> None:
     """Test the underlying sensors."""
     printer = {
@@ -128,8 +132,6 @@ async def test_sensors_paused(
     freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
     await init_integration(hass, "sensor", printer=printer, job=job)
 
-    entity_registry = er.async_get(hass)
-
     state = hass.states.get("sensor.octoprint_start_time")
     assert state is not None
     assert state.state == "unknown"
@@ -146,7 +148,9 @@ async def test_sensors_paused(
 
 
 async def test_sensors_printer_disconnected(
-    hass: HomeAssistant, freezer: FrozenDateTimeFactory
+    hass: HomeAssistant,
+    freezer: FrozenDateTimeFactory,
+    entity_registry: er.EntityRegistry,
 ) -> None:
     """Test the underlying sensors."""
     job = {
@@ -157,8 +161,6 @@ async def test_sensors_printer_disconnected(
     freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
     await init_integration(hass, "sensor", printer=None, job=job)
 
-    entity_registry = er.async_get(hass)
-
     state = hass.states.get("sensor.octoprint_job_percentage")
     assert state is not None
     assert state.state == "50"