hass-core/tests/components/test_updater.py
Josh Wright 2e9bf42688 Disable updater.updater on dev versions
The Updater component doesn't make much sense on dev versions. If you

want to run a production config with the updater enabled, you end up

with an 'Update Available' badge pointing you to the last release

version. This change disables the Updater component on dev and updates

the tests to use a faked version number.



A warning is emitted if the Updater component is disabled to ensure

there is no confusion.
2016-04-06 18:46:48 -07:00

81 lines
2.9 KiB
Python

"""The tests for the Updater component."""
import unittest
from unittest.mock import patch
import requests
from homeassistant.components import updater
import homeassistant.util.dt as dt_util
from tests.common import fire_time_changed, get_test_home_assistant
NEW_VERSION = '10000.0'
# We need to use a 'real' looking version number to load the updater component
MOCK_CURRENT_VERSION = '10.0'
class TestUpdater(unittest.TestCase):
"""Test the Updater component."""
def setUp(self): # pylint: disable=invalid-name
"""Setup things to be run when tests are started."""
self.hass = get_test_home_assistant()
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
@patch('homeassistant.components.updater.get_newest_version')
def test_new_version_shows_entity_on_start(self, mock_get_newest_version):
"""Test if new entity is created if new version is available."""
mock_get_newest_version.return_value = NEW_VERSION
updater.CURRENT_VERSION = MOCK_CURRENT_VERSION
self.assertTrue(updater.setup(self.hass, {
'updater': None
}))
self.assertTrue(self.hass.states.is_state(updater.ENTITY_ID,
NEW_VERSION))
@patch('homeassistant.components.updater.get_newest_version')
def test_no_entity_on_same_version(self, mock_get_newest_version):
"""Test if no entity is created if same version."""
mock_get_newest_version.return_value = MOCK_CURRENT_VERSION
updater.CURRENT_VERSION = MOCK_CURRENT_VERSION
self.assertTrue(updater.setup(self.hass, {
'updater': None
}))
self.assertIsNone(self.hass.states.get(updater.ENTITY_ID))
mock_get_newest_version.return_value = NEW_VERSION
fire_time_changed(self.hass,
dt_util.utcnow().replace(hour=0, minute=0, second=0))
self.hass.pool.block_till_done()
self.assertTrue(self.hass.states.is_state(updater.ENTITY_ID,
NEW_VERSION))
@patch('homeassistant.components.updater.requests.get')
def test_errors_while_fetching_new_version(self, mock_get):
"""Test for errors while fetching the new version."""
mock_get.side_effect = requests.RequestException
self.assertIsNone(updater.get_newest_version())
mock_get.side_effect = ValueError
self.assertIsNone(updater.get_newest_version())
mock_get.side_effect = KeyError
self.assertIsNone(updater.get_newest_version())
def test_updater_disabled_on_dev(self):
"""Test if the updater component is disabled on dev."""
updater.CURRENT_VERSION = MOCK_CURRENT_VERSION + 'dev'
self.assertFalse(updater.setup(self.hass, {
'updater': None
}))