Set correct entity categories

homeassistant/components/devolo_home_network/button.py

@@ -15,6 +15,7 @@
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from .const import DOMAIN, IDENTIFY, PAIRING, RESTART, START_WPS
@@ -40,6 +41,7 @@ class DevoloButtonEntityDescription(
 BUTTON_TYPES: dict[str, DevoloButtonEntityDescription] = {
     IDENTIFY: DevoloButtonEntityDescription(
         key=IDENTIFY,
+        entity_category=EntityCategory.DIAGNOSTIC,
         icon="mdi:led-on",
         name="Identify device with a blinking LED",
         press_func=lambda device: device.plcnet.async_identify_device_start(),  # type: ignore[union-attr]
@@ -53,6 +55,7 @@ class DevoloButtonEntityDescription(
     RESTART: DevoloButtonEntityDescription(
         key=RESTART,
         device_class=ButtonDeviceClass.RESTART,
+        entity_category=EntityCategory.CONFIG,
         name="Restart device",
         press_func=lambda device: device.device.async_restart(),  # type: ignore[union-attr]
     ),

tests/components/devolo_home_network/test_button.py

@@ -14,6 +14,8 @@
 from homeassistant.config_entries import SOURCE_REAUTH
 from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers import entity_registry
+from homeassistant.helpers.entity import EntityCategory
 
 from . import configure_integration
 from .mock import MockDevice
@@ -44,13 +46,14 @@ async def test_identify_device(hass: HomeAssistant, mock_device: MockDevice):
     entry = configure_integration(hass)
     device_name = entry.title.replace(" ", "_").lower()
     state_key = f"{PLATFORM}.{device_name}_identify_device_with_a_blinking_led"
-
+    er = entity_registry.async_get(hass)
     await hass.config_entries.async_setup(entry.entry_id)
     await hass.async_block_till_done()
 
     state = hass.states.get(state_key)
     assert state is not None
     assert state.state == STATE_UNKNOWN
+    assert er.async_get(state_key).entity_category is EntityCategory.DIAGNOSTIC
 
     # Emulate button press
     await hass.services.async_call(
@@ -74,7 +77,6 @@ async def test_start_plc_pairing(hass: HomeAssistant, mock_device: MockDevice):
     entry = configure_integration(hass)
     device_name = entry.title.replace(" ", "_").lower()
     state_key = f"{PLATFORM}.{device_name}_start_plc_pairing"
-
     await hass.config_entries.async_setup(entry.entry_id)
     await hass.async_block_till_done()
 
@@ -104,14 +106,15 @@ async def test_restart(hass: HomeAssistant, mock_device: MockDevice):
     entry = configure_integration(hass)
     device_name = entry.title.replace(" ", "_").lower()
     state_key = f"{PLATFORM}.{device_name}_restart_device"
-
+    er = entity_registry.async_get(hass)
     await hass.config_entries.async_setup(entry.entry_id)
     await hass.async_block_till_done()
 
     state = hass.states.get(state_key)
     assert state is not None
     assert state.state == STATE_UNKNOWN
     assert state.attributes["device_class"] == ButtonDeviceClass.RESTART
+    assert er.async_get(state_key).entity_category is EntityCategory.CONFIG
 
     # Emulate button press
     await hass.services.async_call(