155 lines
4.8 KiB
Python
155 lines
4.8 KiB
Python
"""The tests for the mochad light platform."""
|
|
import unittest
|
|
import unittest.mock as mock
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import light
|
|
from homeassistant.components.mochad import light as mochad
|
|
from homeassistant.setup import setup_component
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def pymochad_mock():
|
|
"""Mock pymochad."""
|
|
with mock.patch.dict('sys.modules', {
|
|
'pymochad': mock.MagicMock(),
|
|
}):
|
|
yield
|
|
|
|
|
|
class TestMochadSwitchSetup(unittest.TestCase):
|
|
"""Test the mochad light."""
|
|
|
|
PLATFORM = mochad
|
|
COMPONENT = light
|
|
THING = 'light'
|
|
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
@mock.patch('homeassistant.components.mochad.light.MochadLight')
|
|
def test_setup_adds_proper_devices(self, mock_light):
|
|
"""Test if setup adds devices."""
|
|
good_config = {
|
|
'mochad': {},
|
|
'light': {
|
|
'platform': 'mochad',
|
|
'devices': [
|
|
{
|
|
'name': 'Light1',
|
|
'address': 'a1',
|
|
},
|
|
],
|
|
}
|
|
}
|
|
assert setup_component(self.hass, light.DOMAIN, good_config)
|
|
|
|
|
|
class TestMochadLight(unittest.TestCase):
|
|
"""Test for mochad light platform."""
|
|
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
controller_mock = mock.MagicMock()
|
|
dev_dict = {'address': 'a1', 'name': 'fake_light',
|
|
'brightness_levels': 32}
|
|
self.light = mochad.MochadLight(self.hass, controller_mock,
|
|
dev_dict)
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_name(self):
|
|
"""Test the name."""
|
|
assert 'fake_light' == self.light.name
|
|
|
|
def test_turn_on_with_no_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on()
|
|
self.light.light.send_cmd.assert_called_once_with('on')
|
|
|
|
def test_turn_on_with_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on(brightness=45)
|
|
self.light.light.send_cmd.assert_has_calls(
|
|
[mock.call('on'), mock.call('dim 25')])
|
|
|
|
def test_turn_off(self):
|
|
"""Test turn_off."""
|
|
self.light.turn_off()
|
|
self.light.light.send_cmd.assert_called_once_with('off')
|
|
|
|
|
|
class TestMochadLight256Levels(unittest.TestCase):
|
|
"""Test for mochad light platform."""
|
|
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
controller_mock = mock.MagicMock()
|
|
dev_dict = {'address': 'a1', 'name': 'fake_light',
|
|
'brightness_levels': 256}
|
|
self.light = mochad.MochadLight(self.hass, controller_mock,
|
|
dev_dict)
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_turn_on_with_no_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on()
|
|
self.light.light.send_cmd.assert_called_once_with('xdim 255')
|
|
|
|
def test_turn_on_with_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on(brightness=45)
|
|
self.light.light.send_cmd.assert_called_once_with('xdim 45')
|
|
|
|
def test_turn_off(self):
|
|
"""Test turn_off."""
|
|
self.light.turn_off()
|
|
self.light.light.send_cmd.assert_called_once_with('off')
|
|
|
|
|
|
class TestMochadLight64Levels(unittest.TestCase):
|
|
"""Test for mochad light platform."""
|
|
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
controller_mock = mock.MagicMock()
|
|
dev_dict = {'address': 'a1', 'name': 'fake_light',
|
|
'brightness_levels': 64}
|
|
self.light = mochad.MochadLight(self.hass, controller_mock,
|
|
dev_dict)
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_turn_on_with_no_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on()
|
|
self.light.light.send_cmd.assert_called_once_with('xdim 63')
|
|
|
|
def test_turn_on_with_brightness(self):
|
|
"""Test turn_on."""
|
|
self.light.turn_on(brightness=45)
|
|
self.light.light.send_cmd.assert_called_once_with('xdim 11')
|
|
|
|
def test_turn_off(self):
|
|
"""Test turn_off."""
|
|
self.light.turn_off()
|
|
self.light.light.send_cmd.assert_called_once_with('off')
|