Update m* tests to use entity & device registry fixtures (#103980)

tests/components/nam/test_sensor.py

@@ -35,11 +35,9 @@ from . import INCOMPLETE_NAM_DATA, init_integration, nam_data
 from tests.common import async_fire_time_changed
 
 
-async def test_sensor(hass: HomeAssistant) -> None:
+async def test_sensor(hass: HomeAssistant, entity_registry: er.EntityRegistry) -> None:
     """Test states of the air_quality."""
-    registry = er.async_get(hass)
-
-    registry.async_get_or_create(
+    entity_registry.async_get_or_create(
         SENSOR_DOMAIN,
         DOMAIN,
         "aa:bb:cc:dd:ee:ff-signal",
@@ -47,7 +45,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         disabled_by=None,
     )
 
-    registry.async_get_or_create(
+    entity_registry.async_get_or_create(
         SENSOR_DOMAIN,
         DOMAIN,
         "aa:bb:cc:dd:ee:ff-uptime",
@@ -67,7 +65,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bme280_humidity")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bme280_humidity")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bme280_humidity"
 
@@ -78,7 +76,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bme280_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bme280_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bme280_temperature"
 
@@ -89,7 +87,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfPressure.HPA
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bme280_pressure")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bme280_pressure")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bme280_pressure"
 
@@ -100,7 +98,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bmp180_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bmp180_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bmp180_temperature"
 
@@ -111,7 +109,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfPressure.HPA
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bmp180_pressure")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bmp180_pressure")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bmp180_pressure"
 
@@ -122,7 +120,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bmp280_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bmp280_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bmp280_temperature"
 
@@ -133,7 +131,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfPressure.HPA
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_bmp280_pressure")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_bmp280_pressure")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-bmp280_pressure"
 
@@ -144,7 +142,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sht3x_humidity")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sht3x_humidity")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sht3x_humidity"
 
@@ -155,7 +153,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sht3x_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sht3x_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sht3x_temperature"
 
@@ -166,7 +164,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_dht22_humidity")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_dht22_humidity")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-dht22_humidity"
 
@@ -177,7 +175,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_dht22_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_dht22_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-dht22_temperature"
 
@@ -188,7 +186,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_heca_humidity")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_heca_humidity")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-heca_humidity"
 
@@ -199,7 +197,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UnitOfTemperature.CELSIUS
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_heca_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_heca_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-heca_temperature"
 
@@ -213,7 +211,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == SIGNAL_STRENGTH_DECIBELS_MILLIWATT
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_signal_strength")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_signal_strength")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-signal"
 
@@ -226,7 +224,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TIMESTAMP
     assert state.attributes.get(ATTR_STATE_CLASS) is None
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_uptime")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_uptime")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-uptime"
 
@@ -245,7 +243,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     ]
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_pmsx003_common_air_quality_index_level"
     )
     assert entry
@@ -259,7 +257,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.state == "19"
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_pmsx003_common_air_quality_index"
     )
     assert entry
@@ -275,7 +273,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm10")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm10")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-pms_p1"
 
@@ -289,7 +287,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm2_5")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm2_5")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-pms_p2"
 
@@ -303,7 +301,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm1")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_pmsx003_pm1")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-pms_p0"
 
@@ -317,7 +315,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sds011_pm10")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sds011_pm10")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sds011_p1"
 
@@ -328,7 +326,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.state == "19"
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_sds011_common_air_quality_index"
     )
     assert entry
@@ -349,7 +347,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     ]
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_sds011_common_air_quality_index_level"
     )
     assert entry
@@ -366,7 +364,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sds011_pm2_5")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sds011_pm2_5")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sds011_p2"
 
@@ -375,7 +373,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     assert state.state == "54"
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_sps30_common_air_quality_index"
     )
     assert entry
@@ -396,7 +394,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     ]
     assert state.attributes.get(ATTR_ICON) == "mdi:air-filter"
 
-    entry = registry.async_get(
+    entry = entity_registry.async_get(
         "sensor.nettigo_air_monitor_sps30_common_air_quality_index_level"
     )
     assert entry
@@ -413,7 +411,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sps30_pm1")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sps30_pm1")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sps30_p0"
 
@@ -427,7 +425,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sps30_pm10")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sps30_pm10")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sps30_p1"
 
@@ -441,7 +439,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
         == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
     )
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sps30_pm2_5")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sps30_pm2_5")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sps30_p2"
 
@@ -455,7 +453,7 @@ async def test_sensor(hass: HomeAssistant) -> None:
     )
     assert state.attributes.get(ATTR_ICON) == "mdi:molecule"
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_sps30_pm4")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_sps30_pm4")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-sps30_p4"
 
@@ -468,24 +466,27 @@ async def test_sensor(hass: HomeAssistant) -> None:
         state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
         == CONCENTRATION_PARTS_PER_MILLION
     )
-    entry = registry.async_get("sensor.nettigo_air_monitor_mh_z14a_carbon_dioxide")
+    entry = entity_registry.async_get(
+        "sensor.nettigo_air_monitor_mh_z14a_carbon_dioxide"
+    )
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-mhz14a_carbon_dioxide"
 
 
-async def test_sensor_disabled(hass: HomeAssistant) -> None:
+async def test_sensor_disabled(
+    hass: HomeAssistant, entity_registry: er.EntityRegistry
+) -> None:
     """Test sensor disabled by default."""
     await init_integration(hass)
-    registry = er.async_get(hass)
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_signal_strength")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_signal_strength")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-signal"
     assert entry.disabled
     assert entry.disabled_by is er.RegistryEntryDisabler.INTEGRATION
 
     # Test enabling entity
-    updated_entry = registry.async_update_entity(
+    updated_entry = entity_registry.async_update_entity(
         entry.entity_id, **{"disabled_by": None}
     )
 
@@ -574,11 +575,11 @@ async def test_manual_update_entity(hass: HomeAssistant) -> None:
     assert mock_get_data.call_count == 1
 
 
-async def test_unique_id_migration(hass: HomeAssistant) -> None:
+async def test_unique_id_migration(
+    hass: HomeAssistant, entity_registry: er.EntityRegistry
+) -> None:
     """Test states of the unique_id migration."""
-    registry = er.async_get(hass)
-
-    registry.async_get_or_create(
+    entity_registry.async_get_or_create(
         SENSOR_DOMAIN,
         DOMAIN,
         "aa:bb:cc:dd:ee:ff-temperature",
@@ -586,7 +587,7 @@ async def test_unique_id_migration(hass: HomeAssistant) -> None:
         disabled_by=None,
     )
 
-    registry.async_get_or_create(
+    entity_registry.async_get_or_create(
         SENSOR_DOMAIN,
         DOMAIN,
         "aa:bb:cc:dd:ee:ff-humidity",
@@ -596,10 +597,10 @@ async def test_unique_id_migration(hass: HomeAssistant) -> None:
 
     await init_integration(hass)
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_dht22_temperature")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_dht22_temperature")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-dht22_temperature"
 
-    entry = registry.async_get("sensor.nettigo_air_monitor_dht22_humidity")
+    entry = entity_registry.async_get("sensor.nettigo_air_monitor_dht22_humidity")
     assert entry
     assert entry.unique_id == "aa:bb:cc:dd:ee:ff-dht22_humidity"