51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""Test component helpers."""
|
|
# pylint: disable=protected-access
|
|
from collections import OrderedDict
|
|
import unittest
|
|
|
|
from homeassistant import helpers
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
class TestHelpers(unittest.TestCase):
|
|
"""Tests homeassistant.helpers module."""
|
|
|
|
# pylint: disable=invalid-name
|
|
def setUp(self):
|
|
"""Init needed objects."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
# pylint: disable=invalid-name
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_extract_domain_configs(self):
|
|
"""Test the extraction of domain configuration."""
|
|
config = {
|
|
'zone': None,
|
|
'zoner': None,
|
|
'zone ': None,
|
|
'zone Hallo': None,
|
|
'zone 100': None,
|
|
}
|
|
|
|
assert set(['zone', 'zone Hallo', 'zone 100']) == \
|
|
set(helpers.extract_domain_configs(config, 'zone'))
|
|
|
|
def test_config_per_platform(self):
|
|
"""Test config per platform method."""
|
|
config = OrderedDict([
|
|
('zone', {'platform': 'hello'}),
|
|
('zoner', None),
|
|
('zone Hallo', [1, {'platform': 'hello 2'}]),
|
|
('zone 100', None),
|
|
])
|
|
|
|
assert [
|
|
('hello', config['zone']),
|
|
(None, 1),
|
|
('hello 2', config['zone Hallo'][1]),
|
|
] == list(helpers.config_per_platform(config, 'zone'))
|