609d7ebea5
* Add event loop to the core * Add block_till_done to HA core object * Fix some tests * Linting core * Fix statemachine tests * Core test fixes * fix block_till_done to wait for loop and queue to empty * fix test_core for passing, and correct start/stop/block_till_done * Fix remote tests * Fix tests: block_till_done * Fix linting * Fix more tests * Fix final linting * Fix remote test * remove unnecessary import * reduce sleep to avoid slowing down the tests excessively * fix remaining tests to wait for non-threadsafe operations * Add async_ doc strings for event loop / coroutine info * Fix command line test to block for the right timeout * Fix py3.4.2 loop var access * Fix SERVICE_CALL_LIMIT being in effect for other tests * Fix lint errors * Fix lint error with proper placement * Fix slave start to not start a timer * Add asyncio compatible listeners. * Increase min Python version to 3.4.2 * Move async backports to util * Add backported async tests * Fix linting * Simplify Python version check * Fix lint * Remove unneeded try/except and queue listener appproriately. * Fix tuple vs. list unorderable error on version compare. * Fix version tests
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
"""The tests for the Switch component."""
|
|
# pylint: disable=too-many-public-methods,protected-access
|
|
import unittest
|
|
|
|
from homeassistant import loader
|
|
from homeassistant.components import switch
|
|
from homeassistant.const import STATE_ON, STATE_OFF, CONF_PLATFORM
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
class TestSwitch(unittest.TestCase):
|
|
"""Test the switch module."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
platform = loader.get_component('switch.test')
|
|
platform.init()
|
|
self.assertTrue(switch.setup(
|
|
self.hass, {switch.DOMAIN: {CONF_PLATFORM: 'test'}}
|
|
))
|
|
|
|
# Switch 1 is ON, switch 2 is OFF
|
|
self.switch_1, self.switch_2, self.switch_3 = \
|
|
platform.DEVICES
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_methods(self):
|
|
"""Test is_on, turn_on, turn_off methods."""
|
|
self.assertTrue(switch.is_on(self.hass))
|
|
self.assertEqual(
|
|
STATE_ON,
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state)
|
|
self.assertTrue(switch.is_on(self.hass, self.switch_1.entity_id))
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_2.entity_id))
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_3.entity_id))
|
|
|
|
switch.turn_off(self.hass, self.switch_1.entity_id)
|
|
switch.turn_on(self.hass, self.switch_2.entity_id)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
self.assertTrue(switch.is_on(self.hass))
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_1.entity_id))
|
|
self.assertTrue(switch.is_on(self.hass, self.switch_2.entity_id))
|
|
|
|
# Turn all off
|
|
switch.turn_off(self.hass)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
self.assertFalse(switch.is_on(self.hass))
|
|
self.assertEqual(
|
|
STATE_OFF,
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state)
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_1.entity_id))
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_2.entity_id))
|
|
self.assertFalse(switch.is_on(self.hass, self.switch_3.entity_id))
|
|
|
|
# Turn all on
|
|
switch.turn_on(self.hass)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
self.assertTrue(switch.is_on(self.hass))
|
|
self.assertEqual(
|
|
STATE_ON,
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state)
|
|
self.assertTrue(switch.is_on(self.hass, self.switch_1.entity_id))
|
|
self.assertTrue(switch.is_on(self.hass, self.switch_2.entity_id))
|
|
self.assertTrue(switch.is_on(self.hass, self.switch_3.entity_id))
|
|
|
|
def test_setup_two_platforms(self):
|
|
"""Test with bad configuration."""
|
|
# Test if switch component returns 0 switches
|
|
test_platform = loader.get_component('switch.test')
|
|
test_platform.init(True)
|
|
|
|
loader.set_component('switch.test2', test_platform)
|
|
test_platform.init(False)
|
|
|
|
self.assertTrue(switch.setup(
|
|
self.hass, {
|
|
switch.DOMAIN: {CONF_PLATFORM: 'test'},
|
|
'{} 2'.format(switch.DOMAIN): {CONF_PLATFORM: 'test2'},
|
|
}
|
|
))
|