Make sure entity inherit disabled device (#60469)

tests/helpers/test_entity_platform.py

@@ -7,14 +7,14 @@ from unittest.mock import ANY, Mock, patch
 import pytest
 
 from homeassistant.const import EVENT_HOMEASSISTANT_STARTED, PERCENTAGE
-from homeassistant.core import CoreState, callback
+from homeassistant.core import CoreState, HomeAssistant, callback
 from homeassistant.exceptions import HomeAssistantError, PlatformNotReady
 from homeassistant.helpers import (
     device_registry as dr,
     entity_platform,
     entity_registry as er,
 )
-from homeassistant.helpers.entity import async_generate_entity_id
+from homeassistant.helpers.entity import DeviceInfo, async_generate_entity_id
 from homeassistant.helpers.entity_component import (
     DEFAULT_SCAN_INTERVAL,
     EntityComponent,
@@ -1080,6 +1080,44 @@ async def test_entity_disabled_by_integration(hass):
     assert entry_disabled.disabled_by == er.DISABLED_INTEGRATION
 
 
+async def test_entity_disabled_by_device(hass: HomeAssistant):
+    """Test entity disabled by device."""
+
+    connections = {(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}
+    entity_disabled = MockEntity(
+        unique_id="disabled", device_info=DeviceInfo(connections=connections)
+    )
+
+    async def async_setup_entry(hass, config_entry, async_add_entities):
+        """Mock setup entry method."""
+        async_add_entities([entity_disabled])
+        return True
+
+    platform = MockPlatform(async_setup_entry=async_setup_entry)
+    config_entry = MockConfigEntry(entry_id="super-mock-id", domain=DOMAIN)
+    entity_platform = MockEntityPlatform(
+        hass, platform_name=config_entry.domain, platform=platform
+    )
+
+    device_registry = dr.async_get(hass)
+    device_registry.async_get_or_create(
+        config_entry_id=config_entry.entry_id,
+        connections=connections,
+        disabled_by=dr.DeviceEntryDisabler.USER,
+    )
+
+    assert await entity_platform.async_setup_entry(config_entry)
+    await hass.async_block_till_done()
+
+    assert entity_disabled.hass is None
+    assert entity_disabled.platform is None
+
+    registry = er.async_get(hass)
+
+    entry_disabled = registry.async_get_or_create(DOMAIN, DOMAIN, "disabled")
+    assert entry_disabled.disabled_by == er.DISABLED_DEVICE
+
+
 async def test_entity_info_added_to_entity_registry(hass):
     """Test entity info is written to entity registry."""
     component = EntityComponent(_LOGGER, DOMAIN, hass, timedelta(seconds=20))