Use setup_test_component_platform helper for alarm_control_panel entity component tests instead of hass.components (#114301)

tests/components/alarm_control_panel/common.py

@@ -4,7 +4,11 @@ All containing methods are legacy helpers that should not be used by new
 components. Instead call the service directly.
 """
 
-from homeassistant.components.alarm_control_panel import DOMAIN
+from homeassistant.components.alarm_control_panel import (
+    DOMAIN,
+    AlarmControlPanelEntity,
+    AlarmControlPanelEntityFeature,
+)
 from homeassistant.const import (
     ATTR_CODE,
     ATTR_ENTITY_ID,
@@ -16,8 +20,16 @@ from homeassistant.const import (
     SERVICE_ALARM_ARM_VACATION,
     SERVICE_ALARM_DISARM,
     SERVICE_ALARM_TRIGGER,
+    STATE_ALARM_ARMED_AWAY,
+    STATE_ALARM_ARMED_HOME,
+    STATE_ALARM_ARMED_NIGHT,
+    STATE_ALARM_ARMED_VACATION,
+    STATE_ALARM_DISARMED,
+    STATE_ALARM_TRIGGERED,
 )
 
+from tests.common import MockEntity
+
 
 async def async_alarm_disarm(hass, code=None, entity_id=ENTITY_MATCH_ALL):
     """Send the alarm the command for disarm."""
@@ -98,3 +110,51 @@ async def async_alarm_arm_custom_bypass(hass, code=None, entity_id=ENTITY_MATCH_
     await hass.services.async_call(
         DOMAIN, SERVICE_ALARM_ARM_CUSTOM_BYPASS, data, blocking=True
     )
+
+
+class MockAlarm(MockEntity, AlarmControlPanelEntity):
+    """Mock Alarm control panel class."""
+
+    _attr_supported_features = (
+        AlarmControlPanelEntityFeature.ARM_HOME
+        | AlarmControlPanelEntityFeature.ARM_AWAY
+        | AlarmControlPanelEntityFeature.ARM_NIGHT
+        | AlarmControlPanelEntityFeature.TRIGGER
+        | AlarmControlPanelEntityFeature.ARM_VACATION
+    )
+
+    @property
+    def code_arm_required(self):
+        """Whether the code is required for arm actions."""
+        return self._handle("code_arm_required")
+
+    def alarm_arm_away(self, code=None):
+        """Send arm away command."""
+        self._attr_state = STATE_ALARM_ARMED_AWAY
+        self.schedule_update_ha_state()
+
+    def alarm_arm_home(self, code=None):
+        """Send arm home command."""
+        self._attr_state = STATE_ALARM_ARMED_HOME
+        self.schedule_update_ha_state()
+
+    def alarm_arm_night(self, code=None):
+        """Send arm night command."""
+        self._attr_state = STATE_ALARM_ARMED_NIGHT
+        self.schedule_update_ha_state()
+
+    def alarm_arm_vacation(self, code=None):
+        """Send arm night command."""
+        self._attr_state = STATE_ALARM_ARMED_VACATION
+        self.schedule_update_ha_state()
+
+    def alarm_disarm(self, code=None):
+        """Send disarm command."""
+        if code == "1234":
+            self._attr_state = STATE_ALARM_DISARMED
+            self.schedule_update_ha_state()
+
+    def alarm_trigger(self, code=None):
+        """Send alarm trigger command."""
+        self._attr_state = STATE_ALARM_TRIGGERED
+        self.schedule_update_ha_state()