Reorganize trigger code (#38655)

tests/components/homeassistant/triggers/test_homeassistant.py

@@ -0,0 +1,68 @@
+"""The tests for the Event automation."""
+import homeassistant.components.automation as automation
+from homeassistant.core import CoreState
+from homeassistant.setup import async_setup_component
+
+from tests.async_mock import AsyncMock, patch
+from tests.common import async_mock_service
+
+
+async def test_if_fires_on_hass_start(hass):
+    """Test the firing when Home Assistant starts."""
+    calls = async_mock_service(hass, "test", "automation")
+    hass.state = CoreState.not_running
+    config = {
+        automation.DOMAIN: {
+            "alias": "hello",
+            "trigger": {"platform": "homeassistant", "event": "start"},
+            "action": {"service": "test.automation"},
+        }
+    }
+
+    assert await async_setup_component(hass, automation.DOMAIN, config)
+    assert automation.is_on(hass, "automation.hello")
+    assert len(calls) == 0
+
+    await hass.async_start()
+    await hass.async_block_till_done()
+    assert automation.is_on(hass, "automation.hello")
+    assert len(calls) == 1
+
+    with patch(
+        "homeassistant.config.async_hass_config_yaml", AsyncMock(return_value=config),
+    ):
+        await hass.services.async_call(
+            automation.DOMAIN, automation.SERVICE_RELOAD, blocking=True
+        )
+
+    assert automation.is_on(hass, "automation.hello")
+    assert len(calls) == 1
+
+
+async def test_if_fires_on_hass_shutdown(hass):
+    """Test the firing when Home Assistant shuts down."""
+    calls = async_mock_service(hass, "test", "automation")
+    hass.state = CoreState.not_running
+
+    assert await async_setup_component(
+        hass,
+        automation.DOMAIN,
+        {
+            automation.DOMAIN: {
+                "alias": "hello",
+                "trigger": {"platform": "homeassistant", "event": "shutdown"},
+                "action": {"service": "test.automation"},
+            }
+        },
+    )
+    assert automation.is_on(hass, "automation.hello")
+    assert len(calls) == 0
+
+    await hass.async_start()
+    assert automation.is_on(hass, "automation.hello")
+    await hass.async_block_till_done()
+    assert len(calls) == 0
+
+    with patch.object(hass.loop, "stop"):
+        await hass.async_stop()
+    assert len(calls) == 1