""" ha_tests.test_loader ~~~~~~~~~~~~~~~~~~~~~~ Provides tests to verify that we can load components. """ # pylint: disable=too-many-public-methods,protected-access import unittest import homeassistant.loader as loader import homeassistant.components.http as http from helpers import get_test_home_assistant, MockModule class TestLoader(unittest.TestCase): """ Test the loader module. """ def setUp(self): # pylint: disable=invalid-name self.hass = get_test_home_assistant() loader.prepare(self.hass) def tearDown(self): # pylint: disable=invalid-name """ Stop down stuff we started. """ self.hass.stop() def test_set_component(self): """ Test if set_component works. """ loader.set_component('switch.test', http) self.assertEqual(http, loader.get_component('switch.test')) def test_get_component(self): """ Test if get_component works. """ self.assertEqual(http, loader.get_component('http')) self.assertIsNotNone(loader.get_component('switch.test')) def test_load_order_component(self): """ Test if we can get the proper load order of components. """ loader.set_component('mod1', MockModule('mod1')) loader.set_component('mod2', MockModule('mod2', ['mod1'])) loader.set_component('mod3', MockModule('mod3', ['mod2'])) self.assertEqual( ['mod1', 'mod2', 'mod3'], loader.load_order_component('mod3')) # Create circular dependency loader.set_component('mod1', MockModule('mod1', ['mod3'])) self.assertEqual([], loader.load_order_component('mod3')) # Depend on non-existing component loader.set_component('mod1', MockModule('mod1', ['nonexisting'])) self.assertEqual([], loader.load_order_component('mod1')) # Try to get load order for non-existing component self.assertEqual([], loader.load_order_component('mod1')) def test_load_order_components(self): loader.set_component('mod1', MockModule('mod1', ['group'])) loader.set_component('mod2', MockModule('mod2', ['mod1', 'sun'])) loader.set_component('mod3', MockModule('mod3', ['mod2'])) loader.set_component('mod4', MockModule('mod4', ['group'])) self.assertEqual( ['group', 'mod4', 'mod1', 'sun', 'mod2', 'mod3'], loader.load_order_components(['mod4', 'mod3', 'mod2'])) loader.set_component('mod1', MockModule('mod1')) loader.set_component('mod2', MockModule('mod2', ['group'])) self.assertEqual( ['mod1', 'group', 'mod2'], loader.load_order_components(['mod2', 'mod1'])) # Add a non existing one self.assertEqual( ['mod1', 'group', 'mod2'], loader.load_order_components(['mod2', 'nonexisting', 'mod1'])) # Depend on a non existing one loader.set_component('mod1', MockModule('mod1', ['nonexisting'])) self.assertEqual( ['group', 'mod2'], loader.load_order_components(['mod2', 'mod1']))