hass-core/tests/components/light/test_mochad.py

"""The tests for the mochad light platform."""
import unittest
import unittest.mock as mock

import pytest

from homeassistant.components import light
from homeassistant.components.light import 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.light.mochad.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',
                    },
                ],
            }
        }
        self.assertTrue(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."""
        self.assertEqual('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')