609d7ebea5
* Add event loop to the core * Add block_till_done to HA core object * Fix some tests * Linting core * Fix statemachine tests * Core test fixes * fix block_till_done to wait for loop and queue to empty * fix test_core for passing, and correct start/stop/block_till_done * Fix remote tests * Fix tests: block_till_done * Fix linting * Fix more tests * Fix final linting * Fix remote test * remove unnecessary import * reduce sleep to avoid slowing down the tests excessively * fix remaining tests to wait for non-threadsafe operations * Add async_ doc strings for event loop / coroutine info * Fix command line test to block for the right timeout * Fix py3.4.2 loop var access * Fix SERVICE_CALL_LIMIT being in effect for other tests * Fix lint errors * Fix lint error with proper placement * Fix slave start to not start a timer * Add asyncio compatible listeners. * Increase min Python version to 3.4.2 * Move async backports to util * Add backported async tests * Fix linting * Simplify Python version check * Fix lint * Remove unneeded try/except and queue listener appproriately. * Fix tuple vs. list unorderable error on version compare. * Fix version tests
81 lines
2.8 KiB
Python
81 lines
2.8 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.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
|
|
}))
|