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.
64 lines
1.9 KiB
Python
64 lines
1.9 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.scene as scene
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ENTITY_SCENE = 'scene.mock_scene_1'
|
|
ENTITY_SCENE_NUMBER = 1
|
|
ENTITY_OTHER_SCENE = 'scene.mock_scene_2'
|
|
ENTITY_OTHER_SCENE_NUMBER = 2
|
|
|
|
|
|
class TestLiteJetScene(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()
|
|
|
|
def get_scene_name(number):
|
|
return "Mock Scene #"+str(number)
|
|
|
|
self.mock_lj = mock_pylitejet.return_value
|
|
self.mock_lj.loads.return_value = range(0)
|
|
self.mock_lj.button_switches.return_value = range(0)
|
|
self.mock_lj.all_switches.return_value = range(0)
|
|
self.mock_lj.scenes.return_value = range(1, 3)
|
|
self.mock_lj.get_scene_name.side_effect = get_scene_name
|
|
|
|
assert bootstrap.setup_component(
|
|
self.hass,
|
|
litejet.DOMAIN,
|
|
{
|
|
'litejet': {
|
|
'port': '/tmp/this_will_be_mocked'
|
|
}
|
|
})
|
|
self.hass.block_till_done()
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def scene(self):
|
|
return self.hass.states.get(ENTITY_SCENE)
|
|
|
|
def other_scene(self):
|
|
return self.hass.states.get(ENTITY_OTHER_SCENE)
|
|
|
|
def test_activate(self):
|
|
"""Test activating the scene."""
|
|
|
|
scene.activate(self.hass, ENTITY_SCENE)
|
|
self.hass.block_till_done()
|
|
self.mock_lj.activate_scene.assert_called_once_with(
|
|
ENTITY_SCENE_NUMBER)
|