Fix yr test

This commit is contained in:
Philip Lundrigan 2016-01-08 13:30:16 -07:00
parent d69c1b848a
commit 3db6faab4d

View File

@ -4,12 +4,14 @@ tests.components.sensor.test_yr
Tests Yr sensor. Tests Yr sensor.
""" """
from datetime import datetime
from unittest.mock import patch from unittest.mock import patch
import pytest import pytest
import homeassistant.core as ha import homeassistant.core as ha
import homeassistant.components.sensor as sensor import homeassistant.components.sensor as sensor
import homeassistant.util.dt as dt_util
@pytest.mark.usefixtures('betamax_session') @pytest.mark.usefixtures('betamax_session')
@ -26,14 +28,18 @@ class TestSensorYr:
self.hass.stop() self.hass.stop()
def test_default_setup(self, betamax_session): def test_default_setup(self, betamax_session):
now = datetime(2016, 1, 5, 1, tzinfo=dt_util.UTC)
with patch('homeassistant.components.sensor.yr.requests.Session', with patch('homeassistant.components.sensor.yr.requests.Session',
return_value=betamax_session): return_value=betamax_session):
assert sensor.setup(self.hass, { with patch('homeassistant.components.sensor.yr.dt_util.utcnow',
'sensor': { return_value=now):
'platform': 'yr', assert sensor.setup(self.hass, {
'elevation': 0, 'sensor': {
} 'platform': 'yr',
}) 'elevation': 0,
}
})
state = self.hass.states.get('sensor.yr_symbol') state = self.hass.states.get('sensor.yr_symbol')