fix test commented in #20678 (#20680)

This commit is contained in:
Diogo Gomes 2019-02-02 06:11:50 +00:00 committed by Paulus Schoutsen
parent fcccf133ba
commit 384a9625c9

View File

@ -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