"""The tests for the Scene component."""
import io
import unittest

from homeassistant.setup import setup_component
from homeassistant import loader
from homeassistant.components import light, scene
from homeassistant.util import yaml

from tests.common import get_test_home_assistant


class TestScene(unittest.TestCase):
    """Test the scene component."""

    def setUp(self):  # pylint: disable=invalid-name
        """Setup things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        test_light = loader.get_component('light.test')
        test_light.init()

        self.assertTrue(setup_component(self.hass, light.DOMAIN, {
            light.DOMAIN: {'platform': 'test'}
        }))

        self.light_1, self.light_2 = test_light.DEVICES[0:2]

        light.turn_off(
            self.hass, [self.light_1.entity_id, self.light_2.entity_id])

        self.hass.block_till_done()

        self.assertFalse(self.light_1.is_on)
        self.assertFalse(self.light_2.is_on)

    def tearDown(self):  # pylint: disable=invalid-name
        """Stop everything that was started."""
        self.hass.stop()

    def test_config_yaml_alias_anchor(self):
        """Test the usage of YAML aliases and anchors.

        The following test scene configuration is equivalent to:

        scene:
          - name: test
            entities:
              light_1: &light_1_state
                state: 'on'
                brightness: 100
              light_2: *light_1_state

        When encountering a YAML alias/anchor, the PyYAML parser will use a
        reference to the original dictionary, instead of creating a copy, so
        care needs to be taken to not modify the original.
        """
        entity_state = {
            'state': 'on',
            'brightness': 100,
        }
        self.assertTrue(setup_component(self.hass, scene.DOMAIN, {
            'scene': [{
                'name': 'test',
                'entities': {
                    self.light_1.entity_id: entity_state,
                    self.light_2.entity_id: entity_state,
                }
            }]
        }))

        scene.activate(self.hass, 'scene.test')
        self.hass.block_till_done()

        self.assertTrue(self.light_1.is_on)
        self.assertTrue(self.light_2.is_on)
        self.assertEqual(
            100, self.light_1.last_call('turn_on')[1].get('brightness'))
        self.assertEqual(
            100, self.light_2.last_call('turn_on')[1].get('brightness'))

    def test_config_yaml_bool(self):
        """Test parsing of booleans in yaml config."""
        config = (
            'scene:\n'
            '  - name: test\n'
            '    entities:\n'
            '      {0}: on\n'
            '      {1}:\n'
            '        state: on\n'
            '        brightness: 100\n').format(
                self.light_1.entity_id, self.light_2.entity_id)

        with io.StringIO(config) as file:
            doc = yaml.yaml.safe_load(file)

        self.assertTrue(setup_component(self.hass, scene.DOMAIN, doc))
        scene.activate(self.hass, 'scene.test')
        self.hass.block_till_done()

        self.assertTrue(self.light_1.is_on)
        self.assertTrue(self.light_2.is_on)
        self.assertEqual(
            100, self.light_2.last_call('turn_on')[1].get('brightness'))

    def test_activate_scene(self):
        """Test active scene."""
        self.assertTrue(setup_component(self.hass, scene.DOMAIN, {
            'scene': [{
                'name': 'test',
                'entities': {
                    self.light_1.entity_id: 'on',
                    self.light_2.entity_id: {
                        'state': 'on',
                        'brightness': 100,
                    }
                }
            }]
        }))

        scene.activate(self.hass, 'scene.test')
        self.hass.block_till_done()

        self.assertTrue(self.light_1.is_on)
        self.assertTrue(self.light_2.is_on)
        self.assertEqual(
            100, self.light_2.last_call('turn_on')[1].get('brightness'))