core/tests/components/test_updater.py

"""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
        }))