"""The tests for the Rfxtrx light platform."""
import unittest

import pytest

from homeassistant.bootstrap import setup_component
from homeassistant.components import rfxtrx as rfxtrx_core

from tests.common import get_test_home_assistant


@pytest.mark.skipif("os.environ.get('RFXTRX') != 'RUN'")
class TestLightRfxtrx(unittest.TestCase):
    """Test the Rfxtrx light platform."""

    def setUp(self):
        """Setup things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        self.hass.config.components = set(['rfxtrx'])

    def tearDown(self):
        """Stop everything that was started."""
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS = []
        rfxtrx_core.RFX_DEVICES = {}
        if rfxtrx_core.RFXOBJECT:
            rfxtrx_core.RFXOBJECT.close_connection()
        self.hass.stop()

    def test_valid_config(self):
        """Test configuration."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'automatic_add': True,
                      'devices':
                      {'0b1100cd0213c7f210010f51': {
                               'name': 'Test',
                               rfxtrx_core.ATTR_FIREEVENT: True}}}}))

        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'automatic_add': True,
                      'devices':
                      {'213c7f216': {
                               'name': 'Test',
                               'packetid': '0b1100cd0213c7f210010f51',
                               'signal_repetitions': 3}}}}))

    def test_invalid_config(self):
        """Test configuration."""
        self.assertFalse(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'automatic_add': True,
                      'invalid_key': 'afda',
                      'devices':
                      {'213c7f216': {
                               'name': 'Test',
                               'packetid': '0b1100cd0213c7f210010f51',
                               rfxtrx_core.ATTR_FIREEVENT: True}}}}))

    def test_default_config(self):
        """Test with 0 switches."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'devices': {}}}))
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

    def test_old_config(self):
        """Test with 1 light."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'devices':
                      {'123efab1': {
                               'name': 'Test',
                               'packetid': '0b1100cd0213c7f210010f51'}}}}))

        import RFXtrx as rfxtrxmod
        rfxtrx_core.RFXOBJECT =\
            rfxtrxmod.Core("", transport_protocol=rfxtrxmod.DummyTransport)

        self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
        entity = rfxtrx_core.RFX_DEVICES['213c7f216']
        self.assertEqual('Test', entity.name)
        self.assertEqual('off', entity.state)
        self.assertTrue(entity.assumed_state)
        self.assertEqual(entity.signal_repetitions, 1)
        self.assertFalse(entity.should_fire_event)
        self.assertFalse(entity.should_poll)

        self.assertFalse(entity.is_on)

        entity.turn_on()
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 255)

        entity.turn_off()
        self.assertFalse(entity.is_on)
        self.assertEqual(entity.brightness, 0)

        entity.turn_on(brightness=100)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 100)

        entity.turn_on(brightness=10)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 10)

        entity.turn_on(brightness=255)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 255)

    def test_one_light(self):
        """Test with 1 light."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'devices':
                      {'0b1100cd0213c7f210010f51': {
                               'name': 'Test'}}}}))

        import RFXtrx as rfxtrxmod
        rfxtrx_core.RFXOBJECT =\
            rfxtrxmod.Core("", transport_protocol=rfxtrxmod.DummyTransport)

        self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
        entity = rfxtrx_core.RFX_DEVICES['213c7f216']
        self.assertEqual('Test', entity.name)
        self.assertEqual('off', entity.state)
        self.assertTrue(entity.assumed_state)
        self.assertEqual(entity.signal_repetitions, 1)
        self.assertFalse(entity.should_fire_event)
        self.assertFalse(entity.should_poll)

        self.assertFalse(entity.is_on)

        entity.turn_on()
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 255)

        entity.turn_off()
        self.assertFalse(entity.is_on)
        self.assertEqual(entity.brightness, 0)

        entity.turn_on(brightness=100)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 100)

        entity.turn_on(brightness=10)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 10)

        entity.turn_on(brightness=255)
        self.assertTrue(entity.is_on)
        self.assertEqual(entity.brightness, 255)

        entity.turn_off()
        entity_id = rfxtrx_core.RFX_DEVICES['213c7f216'].entity_id
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('Test', entity_hass.name)
        self.assertEqual('off', entity_hass.state)

        entity.turn_on()
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('on', entity_hass.state)

        entity.turn_off()
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('off', entity_hass.state)

        entity.turn_on(brightness=100)
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('on', entity_hass.state)

        entity.turn_on(brightness=10)
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('on', entity_hass.state)

        entity.turn_on(brightness=255)
        entity_hass = self.hass.states.get(entity_id)
        self.assertEqual('on', entity_hass.state)

    def test_several_lights(self):
        """Test with 3 lights."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'signal_repetitions': 3,
                      'devices':
                      {'0b1100cd0213c7f230010f71': {
                               'name': 'Test'},
                        '0b1100100118cdea02010f70': {
                            'name': 'Bath'},
                        '0b1100101118cdea02010f70': {
                            'name': 'Living'}}}}))

        self.assertEqual(3, len(rfxtrx_core.RFX_DEVICES))
        device_num = 0
        for id in rfxtrx_core.RFX_DEVICES:
            entity = rfxtrx_core.RFX_DEVICES[id]
            self.assertEqual(entity.signal_repetitions, 3)
            if entity.name == 'Living':
                device_num = device_num + 1
                self.assertEqual('off', entity.state)
                self.assertEqual('<Entity Living: off>', entity.__str__())
            elif entity.name == 'Bath':
                device_num = device_num + 1
                self.assertEqual('off', entity.state)
                self.assertEqual('<Entity Bath: off>', entity.__str__())
            elif entity.name == 'Test':
                device_num = device_num + 1
                self.assertEqual('off', entity.state)
                self.assertEqual('<Entity Test: off>', entity.__str__())

        self.assertEqual(3, device_num)

    def test_discover_light(self):
        """Test with discovery of lights."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'automatic_add': True,
                      'devices': {}}}))

        event = rfxtrx_core.get_rfx_object('0b11009e00e6116202020070')
        event.data = bytearray(b'\x0b\x11\x00\x9e\x00\xe6\x11b\x02\x02\x00p')

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        entity = rfxtrx_core.RFX_DEVICES['0e611622']
        self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
        self.assertEqual('<Entity 0b11009e00e6116202020070: on>',
                         entity.__str__())

        event = rfxtrx_core.get_rfx_object('0b11009e00e6116201010070')
        event.data = bytearray(b'\x0b\x11\x00\x9e\x00\xe6\x11b\x01\x01\x00p')

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))

        event = rfxtrx_core.get_rfx_object('0b1100120118cdea02020070')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x12, 0x01, 0x18,
                                0xcd, 0xea, 0x02, 0x02, 0x00, 0x70])

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        entity = rfxtrx_core.RFX_DEVICES['118cdea2']
        self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
        self.assertEqual('<Entity 0b1100120118cdea02020070: on>',
                         entity.__str__())

        # trying to add a sensor
        event = rfxtrx_core.get_rfx_object('0a52085e070100b31b0279')
        event.data = bytearray(b'\nR\x08^\x07\x01\x00\xb3\x1b\x02y')
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))

        # trying to add a swicth
        event = rfxtrx_core.get_rfx_object('0b1100100118cdea02010f70')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x10, 0x01, 0x18,
                                0xcd, 0xea, 0x01, 0x01, 0x0f, 0x70])

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))

        # Trying to add a rollershutter
        event = rfxtrx_core.get_rfx_object('0a1400adf394ab020e0060')
        event.data = bytearray([0x0A, 0x14, 0x00, 0xAD, 0xF3, 0x94,
                                0xAB, 0x02, 0x0E, 0x00, 0x60])
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))

    def test_discover_light_noautoadd(self):
        """Test with discover of light when auto add is False."""
        self.assertTrue(setup_component(self.hass, 'light', {
            'light': {'platform': 'rfxtrx',
                      'automatic_add': False,
                      'devices': {}}}))

        event = rfxtrx_core.get_rfx_object('0b1100120118cdea02020070')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x12, 0x01, 0x18,
                                0xcd, 0xea, 0x02, 0x02, 0x00, 0x70])

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

        event = rfxtrx_core.get_rfx_object('0b1100120118cdea02010070')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x12, 0x01, 0x18,
                                0xcd, 0xea, 0x02, 0x01, 0x00, 0x70])

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

        event = rfxtrx_core.get_rfx_object('0b1100120118cdea02020070')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x12, 0x01, 0x18,
                                0xcd, 0xea, 0x02, 0x02, 0x00, 0x70])

        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

        # Trying to add a sensor
        event = rfxtrx_core.get_rfx_object('0a52085e070100b31b0279')
        event.data = bytearray(b'\nR\x08^\x07\x01\x00\xb3\x1b\x02y')
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

        # Trying to add a switch
        event = rfxtrx_core.get_rfx_object('0b1100100118cdea02010f70')
        event.data = bytearray([0x0b, 0x11, 0x00, 0x10, 0x01, 0x18,
                                0xcd, 0xea, 0x01, 0x01, 0x0f, 0x70])
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))

        # Trying to add a rollershutter
        event = rfxtrx_core.get_rfx_object('0a1400adf394ab020e0060')
        event.data = bytearray([0x0A, 0x14, 0x00, 0xAD, 0xF3, 0x94,
                                0xAB, 0x02, 0x0E, 0x00, 0x60])
        rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
        self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))