71346760d0
* Upgrade pytest to 4.6.3 https://docs.pytest.org/en/latest/changelog.html#pytest-4-6-2-2019-06-03 https://docs.pytest.org/en/latest/changelog.html#pytest-4-6-3-2019-06-11 * Make litejet switch test work with pytest 4.6.3 Essentially reverts the corresponing change that was made for pytest 4.2 compatibility.
138 lines
4.9 KiB
Python
138 lines
4.9 KiB
Python
"""The tests for the litejet component."""
|
|
import logging
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from homeassistant import setup
|
|
from homeassistant.components import litejet
|
|
import homeassistant.components.switch as switch
|
|
|
|
from tests.common import get_test_home_assistant
|
|
from tests.components.switch import common
|
|
|
|
_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):
|
|
"""Set up 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 setup.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 the switch state."""
|
|
return self.hass.states.get(ENTITY_SWITCH)
|
|
|
|
def other_switch(self):
|
|
"""Return the other switch state."""
|
|
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)
|
|
|
|
common.turn_on(self.hass, ENTITY_SWITCH)
|
|
self.hass.block_till_done()
|
|
self.mock_lj.press_switch.assert_called_with(ENTITY_SWITCH_NUMBER)
|
|
|
|
common.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'
|