From 384a9625c9d30c1b3be811a25e044267a679720a Mon Sep 17 00:00:00 2001
From: Diogo Gomes <diogogomes@gmail.com>
Date: Sat, 2 Feb 2019 06:11:50 +0000
Subject: [PATCH] fix test commented in #20678 (#20680)

---
 tests/components/sensor/test_history_stats.py | 37 +++++++++++--------
 1 file changed, 21 insertions(+), 16 deletions(-)

diff --git a/tests/components/sensor/test_history_stats.py b/tests/components/sensor/test_history_stats.py
index a739325847f..28d01de4b34 100644
--- a/tests/components/sensor/test_history_stats.py
+++ b/tests/components/sensor/test_history_stats.py
@@ -1,8 +1,11 @@
 """The test for the History Statistics sensor platform."""
 # pylint: disable=protected-access
-from datetime import timedelta
+from datetime import datetime, timedelta
 import unittest
 from unittest.mock import patch
+import pytest
+import pytz
+from homeassistant.helpers import template
 
 from homeassistant.const import STATE_UNKNOWN
 from homeassistant.setup import setup_component
@@ -12,7 +15,6 @@ from homeassistant.helpers.template import Template
 import homeassistant.util.dt as dt_util
 
 from tests.common import init_recorder_component, get_test_home_assistant
-import pytest
 
 
 class TestHistoryStatsSensor(unittest.TestCase):
@@ -50,19 +52,22 @@ class TestHistoryStatsSensor(unittest.TestCase):
 
     def test_period_parsing(self):
         """Test the conversion from templates to period."""
-        today = Template('{{ now().replace(hour=0).replace(minute=0)'
-                         '.replace(second=0) }}', self.hass)
-        duration = timedelta(hours=2, minutes=1)
+        now = datetime(2019, 1, 1, 23, 30, 0, tzinfo=pytz.utc)
+        with patch.dict(template.ENV.globals, {'now': lambda: now}):
+            print(dt_util.now())
+            today = Template('{{ now().replace(hour=0).replace(minute=0)'
+                             '.replace(second=0) }}', self.hass)
+            duration = timedelta(hours=2, minutes=1)
 
-        sensor1 = HistoryStatsSensor(
-            self.hass, 'test', 'on', today, None, duration, 'time', 'test')
-        sensor2 = HistoryStatsSensor(
-            self.hass, 'test', 'on', None, today, duration, 'time', 'test')
+            sensor1 = HistoryStatsSensor(
+                self.hass, 'test', 'on', today, None, duration, 'time', 'test')
+            sensor2 = HistoryStatsSensor(
+                self.hass, 'test', 'on', None, today, duration, 'time', 'test')
 
-        sensor1.update_period()
-        sensor1_start, sensor1_end = sensor1._period
-        sensor2.update_period()
-        sensor2_start, sensor2_end = sensor2._period
+            sensor1.update_period()
+            sensor1_start, sensor1_end = sensor1._period
+            sensor2.update_period()
+            sensor2_start, sensor2_end = sensor2._period
 
         # Start = 00:00:00
         assert sensor1_start.hour == 0
@@ -70,9 +75,9 @@ class TestHistoryStatsSensor(unittest.TestCase):
         assert sensor1_start.second == 0
 
         # End = 02:01:00
-        # assert sensor1_end.hour == 2
-        # assert sensor1_end.minute == 1
-        # assert sensor1_end.second == 0
+        assert sensor1_end.hour == 2
+        assert sensor1_end.minute == 1
+        assert sensor1_end.second == 0
 
         # Start = 21:59:00
         assert sensor2_start.hour == 21