diff --git a/test/config/custom_components/custom_one.py b/test/config/custom_components/custom_one.py new file mode 100644 index 00000000000..53371c59046 --- /dev/null +++ b/test/config/custom_components/custom_one.py @@ -0,0 +1,3 @@ +""" +Module to be loaded by the Loader test. +""" diff --git a/test/helper.py b/test/helper.py index 850510acdc1..3e5c9caa761 100644 --- a/test/helper.py +++ b/test/helper.py @@ -4,9 +4,24 @@ test.helper Helper method for writing tests. """ +import os + +import homeassistant as ha + + +def get_test_home_assistant(): + """ Returns a Home Assistant object pointing at test config dir. """ + hass = ha.HomeAssistant() + hass.config_dir = os.path.join(os.path.dirname(__file__), "config") + + return hass def mock_service(hass, domain, service): + """ + Sets up a fake service. + Returns a list that logs all calls to fake service. + """ calls = [] hass.services.register( diff --git a/test/test_loader.py b/test/test_loader.py index be0adbff621..8d9cad71303 100644 --- a/test/test_loader.py +++ b/test/test_loader.py @@ -12,12 +12,13 @@ import homeassistant.loader as loader import homeassistant.components.http as http import mock_toggledevice_platform +from helper import get_test_home_assistant class TestLoader(unittest.TestCase): """ Test the loader module. """ def setUp(self): # pylint: disable=invalid-name - self.hass = ha.HomeAssistant() + self.hass = get_test_home_assistant() loader.prepare(self.hass) def tearDown(self): # pylint: disable=invalid-name @@ -34,3 +35,5 @@ class TestLoader(unittest.TestCase): def test_get_component(self): """ Test if get_component works. """ self.assertEqual(http, loader.get_component('http')) + + self.assertIsNotNone(loader.get_component('custom_one'))