diff --git a/homeassistant/components/dlink/entity.py b/homeassistant/components/dlink/entity.py index 33302f7fffa..bfe16abd780 100644 --- a/homeassistant/components/dlink/entity.py +++ b/homeassistant/components/dlink/entity.py @@ -1,7 +1,7 @@ """Entity representing a D-Link Power Plug device.""" from __future__ import annotations -from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry +from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_CONNECTIONS from homeassistant.helpers import device_registry as dr from homeassistant.helpers.entity import DeviceInfo, Entity, EntityDescription @@ -14,6 +14,7 @@ class DLinkEntity(Entity): """Representation of a D-Link Power Plug entity.""" _attr_attribution = ATTRIBUTION + _attr_has_entity_name = True def __init__( self, @@ -24,10 +25,6 @@ class DLinkEntity(Entity): """Initialize a D-Link Power Plug entity.""" self.data = data self.entity_description = description - if config_entry.source == SOURCE_IMPORT: - self._attr_name = config_entry.title - else: - self._attr_has_entity_name = True self._attr_unique_id = f"{config_entry.entry_id}_{description.key}" self._attr_device_info = DeviceInfo( identifiers={(DOMAIN, config_entry.entry_id)}, diff --git a/homeassistant/components/dlink/switch.py b/homeassistant/components/dlink/switch.py index e6b9a4c7883..f568800645d 100644 --- a/homeassistant/components/dlink/switch.py +++ b/homeassistant/components/dlink/switch.py @@ -47,10 +47,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( } ) -SWITCH_TYPE = SwitchEntityDescription( - key="switch", - name="Switch", -) +SWITCH_TYPE = SwitchEntityDescription(key="") def setup_platform( diff --git a/tests/components/dlink/test_switch.py b/tests/components/dlink/test_switch.py index 66c40892a66..7db2a22d406 100644 --- a/tests/components/dlink/test_switch.py +++ b/tests/components/dlink/test_switch.py @@ -44,7 +44,7 @@ async def test_switch_state( await setup_integration() - entity_id = "switch.mock_title_switch" + entity_id = "switch.mock_title" state = hass.states.get(entity_id) assert state.state == STATE_OFF assert state.attributes["total_consumption"] == 1040.0 @@ -71,7 +71,7 @@ async def test_switch_no_value( """Test we handle 'N/A' being passed by the pypi package.""" await setup_integration_legacy() - state = hass.states.get("switch.mock_title_switch") + state = hass.states.get("switch.mock_title") assert state.state == STATE_OFF assert state.attributes["total_consumption"] is None assert state.attributes["temperature"] is None