diff --git a/tests/components/conftest.py b/tests/components/conftest.py index 9e029e159a1..f153263cbc6 100644 --- a/tests/components/conftest.py +++ b/tests/components/conftest.py @@ -1,5 +1,6 @@ """Fixtures for component testing.""" -from unittest.mock import patch +from collections.abc import Generator +from unittest.mock import AsyncMock, patch import pytest @@ -22,3 +23,13 @@ def prevent_io(): return_value=[], ): yield + + +@pytest.fixture +def entity_registry_enabled_by_default() -> Generator[AsyncMock, None, None]: + """Test fixture that ensures all entities are enabled in the registry.""" + with patch( + "homeassistant.helpers.entity.Entity.entity_registry_enabled_default", + return_value=True, + ) as mock_entity_registry_enabled_by_default: + yield mock_entity_registry_enabled_by_default diff --git a/tests/components/wled/test_sensor.py b/tests/components/wled/test_sensor.py index bf3ef3b060b..8d16407b88d 100644 --- a/tests/components/wled/test_sensor.py +++ b/tests/components/wled/test_sensor.py @@ -1,6 +1,6 @@ """Tests for the WLED sensor platform.""" from datetime import datetime -from unittest.mock import MagicMock, patch +from unittest.mock import AsyncMock, MagicMock, patch import pytest @@ -28,61 +28,12 @@ async def test_sensors( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_wled: MagicMock, + entity_registry_enabled_by_default: AsyncMock, ) -> None: """Test the creation and values of the WLED sensors.""" registry = er.async_get(hass) - - # Pre-create registry entries for disabled by default sensors - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_uptime", - suggested_object_id="wled_rgb_light_uptime", - disabled_by=None, - ) - - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_free_heap", - suggested_object_id="wled_rgb_light_free_memory", - disabled_by=None, - ) - - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_wifi_signal", - suggested_object_id="wled_rgb_light_wifi_signal", - disabled_by=None, - ) - - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_wifi_rssi", - suggested_object_id="wled_rgb_light_wifi_rssi", - disabled_by=None, - ) - - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_wifi_channel", - suggested_object_id="wled_rgb_light_wifi_channel", - disabled_by=None, - ) - - registry.async_get_or_create( - SENSOR_DOMAIN, - DOMAIN, - "aabbccddeeff_wifi_bssid", - suggested_object_id="wled_rgb_light_wifi_bssid", - disabled_by=None, - ) - - # Setup mock_config_entry.add_to_hass(hass) + test_time = datetime(2019, 11, 11, 9, 10, 32, tzinfo=dt_util.UTC) with patch("homeassistant.components.wled.sensor.utcnow", return_value=test_time): await hass.config_entries.async_setup(mock_config_entry.entry_id) @@ -124,19 +75,19 @@ async def test_sensors( assert entry.unique_id == "aabbccddeeff_free_heap" assert entry.entity_category is EntityCategory.DIAGNOSTIC - state = hass.states.get("sensor.wled_rgb_light_wifi_signal") + state = hass.states.get("sensor.wled_rgb_light_wi_fi_signal") assert state assert state.attributes.get(ATTR_ICON) == "mdi:wifi" assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE assert state.state == "76" assert entry.entity_category is EntityCategory.DIAGNOSTIC - entry = registry.async_get("sensor.wled_rgb_light_wifi_signal") + entry = registry.async_get("sensor.wled_rgb_light_wi_fi_signal") assert entry assert entry.unique_id == "aabbccddeeff_wifi_signal" assert entry.entity_category is EntityCategory.DIAGNOSTIC - state = hass.states.get("sensor.wled_rgb_light_wifi_rssi") + state = hass.states.get("sensor.wled_rgb_light_wi_fi_rssi") assert state assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.SIGNAL_STRENGTH assert ( @@ -145,29 +96,29 @@ async def test_sensors( ) assert state.state == "-62" - entry = registry.async_get("sensor.wled_rgb_light_wifi_rssi") + entry = registry.async_get("sensor.wled_rgb_light_wi_fi_rssi") assert entry assert entry.unique_id == "aabbccddeeff_wifi_rssi" assert entry.entity_category is EntityCategory.DIAGNOSTIC - state = hass.states.get("sensor.wled_rgb_light_wifi_channel") + state = hass.states.get("sensor.wled_rgb_light_wi_fi_channel") assert state assert state.attributes.get(ATTR_ICON) == "mdi:wifi" assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None assert state.state == "11" - entry = registry.async_get("sensor.wled_rgb_light_wifi_channel") + entry = registry.async_get("sensor.wled_rgb_light_wi_fi_channel") assert entry assert entry.unique_id == "aabbccddeeff_wifi_channel" assert entry.entity_category is EntityCategory.DIAGNOSTIC - state = hass.states.get("sensor.wled_rgb_light_wifi_bssid") + state = hass.states.get("sensor.wled_rgb_light_wi_fi_bssid") assert state assert state.attributes.get(ATTR_ICON) == "mdi:wifi" assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None assert state.state == "AA:AA:AA:AA:AA:BB" - entry = registry.async_get("sensor.wled_rgb_light_wifi_bssid") + entry = registry.async_get("sensor.wled_rgb_light_wi_fi_bssid") assert entry assert entry.unique_id == "aabbccddeeff_wifi_bssid" assert entry.entity_category is EntityCategory.DIAGNOSTIC