2a6c0cfc17
* LiteJet: Unit tests and new trigger options held_more_than and held_less_than. * Unit tests for the LiteJet component and associated platforms. Coverage is almost 100% -- just misses one line. * The automation LiteJet trigger returns an empty "removal" function to ensure the automation base is happy with it. The pylitejet library doesn't actually support a real removal. * The automation LiteJet trigger can detect hold time and act appropriately to support things like short tap or long hold. * LiteJet: Fix indent in unit test source code. * LiteJet: Fix test_include_switches_* unit tests on Python 3.5 * LiteJet: Remove wait for state existence from unit tests. Recent fixes to discovery make this no longer necessary.
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
"""The tests for the litejet component."""
|
|
import logging
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from homeassistant import bootstrap
|
|
from homeassistant.components import litejet
|
|
from tests.common import get_test_home_assistant
|
|
import homeassistant.components.switch as switch
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ENTITY_SWITCH = 'switch.mock_switch_1'
|
|
ENTITY_SWITCH_NUMBER = 1
|
|
ENTITY_OTHER_SWITCH = 'switch.mock_switch_2'
|
|
ENTITY_OTHER_SWITCH_NUMBER = 2
|
|
|
|
|
|
class TestLiteJetSwitch(unittest.TestCase):
|
|
"""Test the litejet component."""
|
|
|
|
@mock.patch('pylitejet.LiteJet')
|
|
def setup_method(self, method, mock_pylitejet):
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
self.hass.start()
|
|
|
|
self.switch_pressed_callbacks = {}
|
|
self.switch_released_callbacks = {}
|
|
|
|
def get_switch_name(number):
|
|
return "Mock Switch #"+str(number)
|
|
|
|
def on_switch_pressed(number, callback):
|
|
self.switch_pressed_callbacks[number] = callback
|
|
|
|
def on_switch_released(number, callback):
|
|
self.switch_released_callbacks[number] = callback
|
|
|
|
self.mock_lj = mock_pylitejet.return_value
|
|
self.mock_lj.loads.return_value = range(0)
|
|
self.mock_lj.button_switches.return_value = range(1, 3)
|
|
self.mock_lj.all_switches.return_value = range(1, 6)
|
|
self.mock_lj.scenes.return_value = range(0)
|
|
self.mock_lj.get_switch_name.side_effect = get_switch_name
|
|
self.mock_lj.on_switch_pressed.side_effect = on_switch_pressed
|
|
self.mock_lj.on_switch_released.side_effect = on_switch_released
|
|
|
|
config = {
|
|
'litejet': {
|
|
'port': '/tmp/this_will_be_mocked',
|
|
}
|
|
}
|
|
if method == self.test_include_switches_False:
|
|
config['litejet']['include_switches'] = False
|
|
elif method != self.test_include_switches_unspecified:
|
|
config['litejet']['include_switches'] = True
|
|
|
|
assert bootstrap.setup_component(self.hass, litejet.DOMAIN, config)
|
|
self.hass.block_till_done()
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def switch(self):
|
|
return self.hass.states.get(ENTITY_SWITCH)
|
|
|
|
def other_switch(self):
|
|
return self.hass.states.get(ENTITY_OTHER_SWITCH)
|
|
|
|
def test_include_switches_unspecified(self):
|
|
"""Test that switches are ignored by default."""
|
|
|
|
self.mock_lj.button_switches.assert_not_called()
|
|
self.mock_lj.all_switches.assert_not_called()
|
|
|
|
def test_include_switches_False(self):
|
|
"""Test that switches can be explicitly ignored."""
|
|
|
|
self.mock_lj.button_switches.assert_not_called()
|
|
self.mock_lj.all_switches.assert_not_called()
|
|
|
|
def test_on_off(self):
|
|
"""Test turning the switch on and off."""
|
|
|
|
assert self.switch().state == 'off'
|
|
assert self.other_switch().state == 'off'
|
|
|
|
assert not switch.is_on(self.hass, ENTITY_SWITCH)
|
|
|
|
switch.turn_on(self.hass, ENTITY_SWITCH)
|
|
self.hass.block_till_done()
|
|
self.mock_lj.press_switch.assert_called_with(ENTITY_SWITCH_NUMBER)
|
|
|
|
switch.turn_off(self.hass, ENTITY_SWITCH)
|
|
self.hass.block_till_done()
|
|
self.mock_lj.release_switch.assert_called_with(ENTITY_SWITCH_NUMBER)
|
|
|
|
def test_pressed_event(self):
|
|
"""Test handling an event from LiteJet."""
|
|
|
|
# Switch 1
|
|
|
|
_LOGGER.info(self.switch_pressed_callbacks[ENTITY_SWITCH_NUMBER])
|
|
self.switch_pressed_callbacks[ENTITY_SWITCH_NUMBER]()
|
|
self.hass.block_till_done()
|
|
|
|
assert switch.is_on(self.hass, ENTITY_SWITCH)
|
|
assert not switch.is_on(self.hass, ENTITY_OTHER_SWITCH)
|
|
assert self.switch().state == 'on'
|
|
assert self.other_switch().state == 'off'
|
|
|
|
# Switch 2
|
|
|
|
self.switch_pressed_callbacks[ENTITY_OTHER_SWITCH_NUMBER]()
|
|
self.hass.block_till_done()
|
|
|
|
assert switch.is_on(self.hass, ENTITY_OTHER_SWITCH)
|
|
assert switch.is_on(self.hass, ENTITY_SWITCH)
|
|
assert self.other_switch().state == 'on'
|
|
assert self.switch().state == 'on'
|
|
|
|
def test_released_event(self):
|
|
"""Test handling an event from LiteJet."""
|
|
|
|
# Initial state is on.
|
|
|
|
self.switch_pressed_callbacks[ENTITY_OTHER_SWITCH_NUMBER]()
|
|
self.hass.block_till_done()
|
|
|
|
assert switch.is_on(self.hass, ENTITY_OTHER_SWITCH)
|
|
|
|
# Event indicates it is off now.
|
|
|
|
self.switch_released_callbacks[ENTITY_OTHER_SWITCH_NUMBER]()
|
|
self.hass.block_till_done()
|
|
|
|
assert not switch.is_on(self.hass, ENTITY_OTHER_SWITCH)
|
|
assert not switch.is_on(self.hass, ENTITY_SWITCH)
|
|
assert self.other_switch().state == 'off'
|
|
assert self.switch().state == 'off'
|