6ba9ccf052
* Load dependencies from manifests. Fallback to current DEPENDENCIES * Fix typing * Ignore typing correctly * Split out dependency processing to a new method * Fix tests * Only pull from manifest if dependencies is non empty * Inline temporary function * Fix light tests [skip ci] * Fix tests/common * Fix some mqtt tests [skip ci] * Fix tests and component manifests which have only one platform * Fix rflink tests * Fix more tests and manifests * Readability over shorthand format * Fix demo/notify tests * Load dependencies from manifests. Fallback to current DEPENDENCIES * Load requirements from manifests. Fallback to current REQUIREMENTS * Fix typing * Ignore typing correctly * Split out dependency processing to a new method * Only pull from manifest if dependencies is non empty * Inline temporary function * Fix tests and component manifests which have only one platform * Fix rflink tests * Readability over shorthand format * Clean up requirements * Use integration to resolve deps/reqs * Lint * Lint * revert a change * Revert a test change * Fix types * Fix types * Add back cache for load component * Fix test_component_not_found * Move light.test and device_tracker.test into test package instead with manifest to fix tests * Fix broken device_tracker tests * Add docstrings to __init__ * Fix all of the light tests that I broke earlier * Embed the test.switch platform to fix other tests * Embed and fix the test.imagimage_processing platform * Fix tests for nx584 * Add dependencies from platform file's DEPENDENCIES * Try to setup component when entity_platform is setting up Fix tests in helpers folder * Rewrite test_setup * Simplify * Lint * Disable demo component if running in test Temp workaround to unblock CI tests * Skip demo tests * Fix config entry test * Fix repeat test * Clarify doc * One extra guard * Fix import * Lint * Workaround google tts
114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
"""The tests for the Switch component."""
|
|
# pylint: disable=protected-access
|
|
import unittest
|
|
|
|
from homeassistant.setup import setup_component, async_setup_component
|
|
from homeassistant import core, loader
|
|
from homeassistant.components import switch
|
|
from homeassistant.const import STATE_ON, STATE_OFF, CONF_PLATFORM
|
|
|
|
from tests.common import get_test_home_assistant
|
|
from tests.components.switch import common
|
|
|
|
|
|
class TestSwitch(unittest.TestCase):
|
|
"""Test the switch module."""
|
|
|
|
# pylint: disable=invalid-name
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
platform = loader.get_component(self.hass, 'test.switch')
|
|
platform.init()
|
|
# Switch 1 is ON, switch 2 is OFF
|
|
self.switch_1, self.switch_2, self.switch_3 = \
|
|
platform.DEVICES
|
|
|
|
# pylint: disable=invalid-name
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_methods(self):
|
|
"""Test is_on, turn_on, turn_off methods."""
|
|
assert setup_component(
|
|
self.hass, switch.DOMAIN, {switch.DOMAIN: {CONF_PLATFORM: 'test'}}
|
|
)
|
|
assert switch.is_on(self.hass)
|
|
assert STATE_ON == \
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
|
|
assert switch.is_on(self.hass, self.switch_1.entity_id)
|
|
assert not switch.is_on(self.hass, self.switch_2.entity_id)
|
|
assert not switch.is_on(self.hass, self.switch_3.entity_id)
|
|
|
|
common.turn_off(self.hass, self.switch_1.entity_id)
|
|
common.turn_on(self.hass, self.switch_2.entity_id)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
assert switch.is_on(self.hass)
|
|
assert not switch.is_on(self.hass, self.switch_1.entity_id)
|
|
assert switch.is_on(self.hass, self.switch_2.entity_id)
|
|
|
|
# Turn all off
|
|
common.turn_off(self.hass)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
assert not switch.is_on(self.hass)
|
|
assert STATE_OFF == \
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
|
|
assert not switch.is_on(self.hass, self.switch_1.entity_id)
|
|
assert not switch.is_on(self.hass, self.switch_2.entity_id)
|
|
assert not switch.is_on(self.hass, self.switch_3.entity_id)
|
|
|
|
# Turn all on
|
|
common.turn_on(self.hass)
|
|
|
|
self.hass.block_till_done()
|
|
|
|
assert switch.is_on(self.hass)
|
|
assert STATE_ON == \
|
|
self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
|
|
assert switch.is_on(self.hass, self.switch_1.entity_id)
|
|
assert switch.is_on(self.hass, self.switch_2.entity_id)
|
|
assert 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(self.hass, 'test.switch')
|
|
test_platform.init(True)
|
|
|
|
loader.set_component(self.hass, 'switch.test2', test_platform)
|
|
test_platform.init(False)
|
|
|
|
assert setup_component(
|
|
self.hass, switch.DOMAIN, {
|
|
switch.DOMAIN: {CONF_PLATFORM: 'test'},
|
|
'{} 2'.format(switch.DOMAIN): {CONF_PLATFORM: 'test2'},
|
|
}
|
|
)
|
|
|
|
|
|
async def test_switch_context(hass, hass_admin_user):
|
|
"""Test that switch context works."""
|
|
assert await async_setup_component(hass, 'switch', {
|
|
'switch': {
|
|
'platform': 'test'
|
|
}
|
|
})
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get('switch.ac')
|
|
assert state is not None
|
|
|
|
await hass.services.async_call('switch', 'toggle', {
|
|
'entity_id': state.entity_id,
|
|
}, True, core.Context(user_id=hass_admin_user.id))
|
|
|
|
state2 = hass.states.get('switch.ac')
|
|
assert state2 is not None
|
|
assert state.state != state2.state
|
|
assert state2.context.user_id == hass_admin_user.id
|