Initial commit: Main Gate Controller HACS integration

tests/test_sensor.py

@@ -0,0 +1,76 @@
+"""Tests for the countdown phase of the cycle."""
+
+from __future__ import annotations
+
+import asyncio
+
+from homeassistant.core import HomeAssistant
+from pytest_homeassistant_custom_component.common import MockConfigEntry
+
+from custom_components.main_gate_controller.const import (
+    ATTR_DURATION,
+    ATTR_REMAINING_SECONDS,
+    ATTR_RUNNING,
+    DOMAIN,
+    GATE_STATE_OPEN,
+)
+from tests.conftest import make_entry_data
+
+
+async def test_countdown_attributes_decrease(hass: HomeAssistant, service_calls):
+    """remaining_seconds decreases over the open phase and finishes at zero."""
+    switch = "switch.fast_gate_relay"
+    hass.states.async_set(switch, "off")
+
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        title="Main Gate",
+        data=make_entry_data(
+            switch_entity_id=switch,
+            name="Main Gate",
+            open_duration=2,
+            pulse_duration=0.1,
+            notify_enabled=False,
+            notify_entities=[],
+        ),
+        unique_id=switch,
+    )
+    entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(entry.entry_id)
+    await hass.async_block_till_done()
+
+    controller = entry.runtime_data
+
+    # Manually step into the open phase and capture attribute changes.
+    assert controller.async_trigger() is True
+
+    # Wait until the cycle is in the ``open`` state.
+    while controller.status != GATE_STATE_OPEN:
+        await asyncio.sleep(0.05)
+    snapshot1 = controller.extra_state_attributes.copy()
+    assert snapshot1[ATTR_DURATION] == 2
+    assert snapshot1[ATTR_RUNNING] is True
+    first_remaining = snapshot1[ATTR_REMAINING_SECONDS]
+    assert first_remaining is not None and first_remaining >= 1
+
+    # Let the countdown progress for ~1 second before re-checking.
+    await asyncio.sleep(1.0)
+    snapshot2 = controller.extra_state_attributes.copy()
+    assert snapshot2[ATTR_REMAINING_SECONDS] < first_remaining
+    assert snapshot2[ATTR_RUNNING] is True
+
+    # Wait for completion.
+    await controller.async_wait_until_idle()
+
+    after = controller.extra_state_attributes
+    assert after[ATTR_RUNNING] is False
+    assert after[ATTR_REMAINING_SECONDS] is None
+
+
+async def test_duration_reflects_configured_value(
+    hass: HomeAssistant, loaded_entry
+):
+    """The ``duration`` attribute follows the configuration value."""
+    controller = loaded_entry.runtime_data
+    attrs = controller.extra_state_attributes
+    assert attrs[ATTR_DURATION] == 20