From 7f49077ec67a87860733afff8f903f16dc5b96ab Mon Sep 17 00:00:00 2001
From: Robert Resch <robert@resch.dev>
Date: Thu, 30 May 2024 14:20:02 +0200
Subject: [PATCH] Set enity_category to config for airgradient select entities
 (#118477)

---
 homeassistant/components/airgradient/select.py          | 3 +++
 tests/components/airgradient/snapshots/test_select.ambr | 6 +++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/airgradient/select.py b/homeassistant/components/airgradient/select.py
index 8dc13fe0eba..41b5a48c686 100644
--- a/homeassistant/components/airgradient/select.py
+++ b/homeassistant/components/airgradient/select.py
@@ -8,6 +8,7 @@ from airgradient.models import ConfigurationControl, TemperatureUnit
 
 from homeassistant.components.select import SelectEntity, SelectEntityDescription
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ServiceValidationError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -30,6 +31,7 @@ CONFIG_CONTROL_ENTITY = AirGradientSelectEntityDescription(
     key="configuration_control",
     translation_key="configuration_control",
     options=[x.value for x in ConfigurationControl],
+    entity_category=EntityCategory.CONFIG,
     value_fn=lambda config: config.configuration_control,
     set_value_fn=lambda client, value: client.set_configuration_control(
         ConfigurationControl(value)
@@ -41,6 +43,7 @@ PROTECTED_SELECT_TYPES: tuple[AirGradientSelectEntityDescription, ...] = (
         key="display_temperature_unit",
         translation_key="display_temperature_unit",
         options=[x.value for x in TemperatureUnit],
+        entity_category=EntityCategory.CONFIG,
         value_fn=lambda config: config.temperature_unit,
         set_value_fn=lambda client, value: client.set_temperature_unit(
             TemperatureUnit(value)
diff --git a/tests/components/airgradient/snapshots/test_select.ambr b/tests/components/airgradient/snapshots/test_select.ambr
index e32b57758c1..986e3c6ebb8 100644
--- a/tests/components/airgradient/snapshots/test_select.ambr
+++ b/tests/components/airgradient/snapshots/test_select.ambr
@@ -16,7 +16,7 @@
     'device_id': <ANY>,
     'disabled_by': None,
     'domain': 'select',
-    'entity_category': None,
+    'entity_category': <EntityCategory.CONFIG: 'config'>,
     'entity_id': 'select.airgradient_configuration_source',
     'has_entity_name': True,
     'hidden_by': None,
@@ -72,7 +72,7 @@
     'device_id': <ANY>,
     'disabled_by': None,
     'domain': 'select',
-    'entity_category': None,
+    'entity_category': <EntityCategory.CONFIG: 'config'>,
     'entity_id': 'select.airgradient_display_temperature_unit',
     'has_entity_name': True,
     'hidden_by': None,
@@ -128,7 +128,7 @@
     'device_id': <ANY>,
     'disabled_by': None,
     'domain': 'select',
-    'entity_category': None,
+    'entity_category': <EntityCategory.CONFIG: 'config'>,
     'entity_id': 'select.airgradient_configuration_source',
     'has_entity_name': True,
     'hidden_by': None,