Add for option for template triggers (#24330)

tests/components/automation/test_template.py

@@ -1,11 +1,15 @@
 """The tests for the Template automation."""
+from datetime import timedelta
+
 import pytest
 
 from homeassistant.core import Context
 from homeassistant.setup import async_setup_component
+import homeassistant.util.dt as dt_util
 import homeassistant.components.automation as automation
 
-from tests.common import (assert_setup_component, mock_component)
+from tests.common import (
+    async_fire_time_changed, assert_setup_component, mock_component)
 from tests.components.automation import common
 from tests.common import async_mock_service
 
@@ -434,3 +438,90 @@ async def test_wait_template_with_trigger(hass, calls):
     assert 1 == len(calls)
     assert 'template - test.entity - hello - world' == \
         calls[0].data['some']
+
+
+async def test_if_fires_on_change_with_for(hass, calls):
+    """Test for firing on change with for."""
+    assert await async_setup_component(hass, automation.DOMAIN, {
+        automation.DOMAIN: {
+            'trigger': {
+                'platform': 'template',
+                'value_template': "{{ is_state('test.entity', 'world') }}",
+                'for': {
+                    'seconds': 5
+                },
+            },
+            'action': {
+                'service': 'test.automation'
+            }
+        }
+    })
+
+    hass.states.async_set('test.entity', 'world')
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=10))
+    await hass.async_block_till_done()
+    assert 1 == len(calls)
+
+
+async def test_if_not_fires_on_change_with_for(hass, calls):
+    """Test for firing on change with for."""
+    assert await async_setup_component(hass, automation.DOMAIN, {
+        automation.DOMAIN: {
+            'trigger': {
+                'platform': 'template',
+                'value_template': "{{ is_state('test.entity', 'world') }}",
+                'for': {
+                    'seconds': 5
+                },
+            },
+            'action': {
+                'service': 'test.automation'
+            }
+        }
+    })
+
+    hass.states.async_set('test.entity', 'world')
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=4))
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    hass.states.async_set('test.entity', 'hello')
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=6))
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+
+
+async def test_if_not_fires_when_turned_off_with_for(hass, calls):
+    """Test for firing on change with for."""
+    assert await async_setup_component(hass, automation.DOMAIN, {
+        automation.DOMAIN: {
+            'trigger': {
+                'platform': 'template',
+                'value_template': "{{ is_state('test.entity', 'world') }}",
+                'for': {
+                    'seconds': 5
+                },
+            },
+            'action': {
+                'service': 'test.automation'
+            }
+        }
+    })
+
+    hass.states.async_set('test.entity', 'world')
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=4))
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    await common.async_turn_off(hass)
+    await hass.async_block_till_done()
+    assert 0 == len(calls)
+    async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=6))
+    await hass.async_block_till_done()
+    assert 0 == len(calls)