Template sensor change flow / add restore (#6336)

tests/components/sensor/test_template.py

@@ -1,7 +1,12 @@
 """The test for the Template sensor platform."""
-from homeassistant.bootstrap import setup_component
+import asyncio
 
-from tests.common import get_test_home_assistant, assert_setup_component
+from homeassistant.core import CoreState, State
+from homeassistant.bootstrap import setup_component, async_setup_component
+from homeassistant.helpers.restore_state import DATA_RESTORE_CACHE
+
+from tests.common import (
+    get_test_home_assistant, assert_setup_component, mock_component)
 
 
 class TestTemplateSensor:
@@ -33,6 +38,8 @@ class TestTemplateSensor:
                 }
             })
 
+        self.hass.start()
+
         state = self.hass.states.get('sensor.test_template_sensor')
         assert state.state == 'It .'
 
@@ -60,6 +67,8 @@ class TestTemplateSensor:
                 }
             })
 
+        self.hass.start()
+
         state = self.hass.states.get('sensor.test_template_sensor')
         assert 'icon' not in state.attributes
 
@@ -82,6 +91,8 @@ class TestTemplateSensor:
                     }
                 }
             })
+
+        self.hass.start()
         assert self.hass.states.all() == []
 
     def test_template_attribute_missing(self):
@@ -99,6 +110,8 @@ class TestTemplateSensor:
                 }
             })
 
+        self.hass.start()
+
         state = self.hass.states.get('sensor.test_template_sensor')
         assert state.state == 'unknown'
 
@@ -116,6 +129,8 @@ class TestTemplateSensor:
                     }
                 }
             })
+
+        self.hass.start()
         assert self.hass.states.all() == []
 
     def test_invalid_sensor_does_not_create(self):
@@ -129,6 +144,8 @@ class TestTemplateSensor:
                     }
                 }
             })
+
+        self.hass.start()
         assert self.hass.states.all() == []
 
     def test_no_sensors_does_not_create(self):
@@ -139,6 +156,8 @@ class TestTemplateSensor:
                     'platform': 'template'
                 }
             })
+
+        self.hass.start()
         assert self.hass.states.all() == []
 
     def test_missing_template_does_not_create(self):
@@ -155,4 +174,39 @@ class TestTemplateSensor:
                     }
                 }
             })
+
+        self.hass.start()
         assert self.hass.states.all() == []
+
+
+@asyncio.coroutine
+def test_restore_state(hass):
+    """Ensure states are restored on startup."""
+    hass.data[DATA_RESTORE_CACHE] = {
+        'sensor.test_template_sensor':
+            State('sensor.test_template_sensor', 'It Test.'),
+    }
+
+    hass.state = CoreState.starting
+    mock_component(hass, 'recorder')
+
+    yield from async_setup_component(hass, 'sensor', {
+        'sensor': {
+            'platform': 'template',
+            'sensors': {
+                'test_template_sensor': {
+                    'value_template':
+                        "It {{ states.sensor.test_state.state }}."
+                }
+            }
+        }
+    })
+
+    state = hass.states.get('sensor.test_template_sensor')
+    assert state.state == 'It Test.'
+
+    yield from hass.async_start()
+    yield from hass.async_block_till_done()
+
+    state = hass.states.get('sensor.test_template_sensor')
+    assert state.state == 'It .'