108 lines
3.8 KiB
Python
108 lines
3.8 KiB
Python
|
"""
|
||
|
|
test.test_component_group
|
||
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
|
||
|
|
Tests the group compoments.
|
||
|
|
"""
|
||
|
|
# pylint: disable=protected-access,too-many-public-methods
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import homeassistant as ha
|
||
|
|
import homeassistant.components as comps
|
||
|
|
import homeassistant.components.group as group
|
||
|
|
|
||
|
|
|
||
|
|
class TestComponentsGroup(unittest.TestCase):
|
||
|
|
""" Tests homeassistant.components.group module. """
|
||
|
|
|
||
|
|
def setUp(self): # pylint: disable=invalid-name
|
||
|
|
""" Init needed objects. """
|
||
|
|
self.hass = ha.HomeAssistant()
|
||
|
|
|
||
|
|
self.hass.states.set('light.Bowl', comps.STATE_ON)
|
||
|
|
self.hass.states.set('light.Ceiling', comps.STATE_OFF)
|
||
|
|
group.setup_group(self.hass, 'init_group',
|
||
|
|
['light.Bowl', 'light.Ceiling'], False)
|
||
|
|
|
||
|
|
self.group_name = group.ENTITY_ID_FORMAT.format('init_group')
|
||
|
|
|
||
|
|
def tearDown(self): # pylint: disable=invalid-name
|
||
|
|
""" Stop down stuff we started. """
|
||
|
|
self.hass.stop()
|
||
|
|
|
||
|
|
def test_setup_and_monitor_group(self):
|
||
|
|
""" Test setup_group method. """
|
||
|
|
|
||
|
|
# Test if group setup in our init mode is ok
|
||
|
|
self.assertIn(self.group_name, self.hass.states.entity_ids)
|
||
|
|
|
||
|
|
group_state = self.hass.states.get(self.group_name)
|
||
|
|
self.assertEqual(comps.STATE_ON, group_state.state)
|
||
|
|
self.assertTrue(group_state.attributes[group.ATTR_AUTO])
|
||
|
|
|
||
|
|
# Turn the Bowl off and see if group turns off
|
||
|
|
self.hass.states.set('light.Bowl', comps.STATE_OFF)
|
||
|
|
|
||
|
|
self.hass._pool.block_till_done()
|
||
|
|
|
||
|
|
group_state = self.hass.states.get(self.group_name)
|
||
|
|
self.assertEqual(comps.STATE_OFF, group_state.state)
|
||
|
|
|
||
|
|
# Turn the Ceiling on and see if group turns on
|
||
|
|
self.hass.states.set('light.Ceiling', comps.STATE_ON)
|
||
|
|
|
||
|
|
self.hass._pool.block_till_done()
|
||
|
|
|
||
|
|
group_state = self.hass.states.get(self.group_name)
|
||
|
|
self.assertEqual(comps.STATE_ON, group_state.state)
|
||
|
|
|
||
|
|
def test__get_group_type(self):
|
||
|
|
""" Test _get_group_type method. """
|
||
|
|
self.assertEqual('on_off', group._get_group_type(comps.STATE_ON))
|
||
|
|
self.assertEqual('on_off', group._get_group_type(comps.STATE_OFF))
|
||
|
|
self.assertEqual('home_not_home',
|
||
|
|
group._get_group_type(comps.STATE_HOME))
|
||
|
|
self.assertEqual('home_not_home',
|
||
|
|
group._get_group_type(comps.STATE_NOT_HOME))
|
||
|
|
|
||
|
|
def test_is_on(self):
|
||
|
|
""" Test is_on method. """
|
||
|
|
self.assertTrue(group.is_on(self.hass, self.group_name))
|
||
|
|
self.hass.states.set('light.Bowl', comps.STATE_OFF)
|
||
|
|
self.hass._pool.block_till_done()
|
||
|
|
self.assertFalse(group.is_on(self.hass, self.group_name))
|
||
|
|
|
||
|
|
def test_expand_entity_ids(self):
|
||
|
|
""" Test expand_entity_ids method. """
|
||
|
|
self.assertEqual(sorted(['light.Ceiling', 'light.Bowl']),
|
||
|
|
sorted(group.expand_entity_ids(
|
||
|
|
self.hass, [self.group_name])))
|
||
|
|
|
||
|
|
self.assertEqual(
|
||
|
|
sorted(['light.Ceiling', 'light.Bowl']),
|
||
|
|
sorted(group.expand_entity_ids(
|
||
|
|
self.hass, [self.group_name, 'light.Ceiling'])))
|
||
|
|
|
||
|
|
def test_get_entity_ids(self):
|
||
|
|
""" Test get_entity_ids method. """
|
||
|
|
self.assertEqual(
|
||
|
|
sorted(['light.Ceiling', 'light.Bowl']),
|
||
|
|
sorted(group.get_entity_ids(self.hass, self.group_name)))
|
||
|
|
|
||
|
|
def test_setup(self):
|
||
|
|
""" Test setup method. """
|
||
|
|
self.assertTrue(
|
||
|
|
group.setup(
|
||
|
|
self.hass,
|
||
|
|
{
|
||
|
|
group.DOMAIN: {
|
||
|
|
'second_group': '{},light.Bowl'.format(self.group_name)
|
||
|
|
}
|
||
|
|
}))
|
||
|
|
|
||
|
|
group_state = self.hass.states.get(
|
||
|
|
group.ENTITY_ID_FORMAT.format('second_group'))
|
||
|
|
|
||
|
|
self.assertEqual(comps.STATE_ON, group_state.state)
|
||
|
|
self.assertFalse(group_state.attributes[group.ATTR_AUTO])
|