"""The tests for the group cover platform."""
from datetime import timedelta

import pytest

from homeassistant.components.cover import (
    ATTR_CURRENT_POSITION, ATTR_CURRENT_TILT_POSITION, ATTR_POSITION,
    ATTR_TILT_POSITION, DOMAIN)
from homeassistant.components.cover.group import DEFAULT_NAME
from homeassistant.const import (
    ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME,
    ATTR_SUPPORTED_FEATURES, CONF_ENTITIES,
    SERVICE_CLOSE_COVER, SERVICE_CLOSE_COVER_TILT,
    SERVICE_OPEN_COVER, SERVICE_OPEN_COVER_TILT, SERVICE_SET_COVER_POSITION,
    SERVICE_SET_COVER_TILT_POSITION, SERVICE_STOP_COVER,
    SERVICE_STOP_COVER_TILT, STATE_OPEN, STATE_CLOSED)
from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util

from tests.common import assert_setup_component, async_fire_time_changed

COVER_GROUP = 'cover.cover_group'
DEMO_COVER = 'cover.kitchen_window'
DEMO_COVER_POS = 'cover.hall_window'
DEMO_COVER_TILT = 'cover.living_room_window'
DEMO_TILT = 'cover.tilt_demo'

CONFIG = {
    DOMAIN: [
        {'platform': 'demo'},
        {'platform': 'group',
         CONF_ENTITIES: [
             DEMO_COVER, DEMO_COVER_POS, DEMO_COVER_TILT, DEMO_TILT]}
    ]
}


@pytest.fixture
async def setup_comp(hass):
    """Set up group cover component."""
    with assert_setup_component(2, DOMAIN):
        await async_setup_component(hass, DOMAIN, CONFIG)


async def test_attributes(hass):
    """Test handling of state attributes."""
    config = {DOMAIN: {'platform': 'group', CONF_ENTITIES: [
        DEMO_COVER, DEMO_COVER_POS, DEMO_COVER_TILT, DEMO_TILT]}}

    with assert_setup_component(1, DOMAIN):
        await async_setup_component(hass, DOMAIN, config)

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_CLOSED
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == DEFAULT_NAME
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 0
    assert state.attributes.get(ATTR_CURRENT_POSITION) is None
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) is None

    # Add Entity that supports open / close / stop
    hass.states.async_set(
        DEMO_COVER, STATE_OPEN, {ATTR_SUPPORTED_FEATURES: 11})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 11
    assert state.attributes.get(ATTR_CURRENT_POSITION) is None
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) is None

    # Add Entity that supports set_cover_position
    hass.states.async_set(
        DEMO_COVER_POS, STATE_OPEN,
        {ATTR_SUPPORTED_FEATURES: 4, ATTR_CURRENT_POSITION: 70})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 15
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 70
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) is None

    # Add Entity that supports open tilt / close tilt / stop tilt
    hass.states.async_set(
        DEMO_TILT, STATE_OPEN, {ATTR_SUPPORTED_FEATURES: 112})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 127
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 70
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) is None

    # Add Entity that supports set_tilt_position
    hass.states.async_set(
        DEMO_COVER_TILT, STATE_OPEN,
        {ATTR_SUPPORTED_FEATURES: 128, ATTR_CURRENT_TILT_POSITION: 60})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 255
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 70
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 60

    # ### Test assumed state ###
    # ##########################

    # For covers
    hass.states.async_set(
        DEMO_COVER, STATE_OPEN,
        {ATTR_SUPPORTED_FEATURES: 4, ATTR_CURRENT_POSITION: 100})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is True
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 244
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 100
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 60

    hass.states.async_remove(DEMO_COVER)
    hass.states.async_remove(DEMO_COVER_POS)
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 240
    assert state.attributes.get(ATTR_CURRENT_POSITION) is None
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 60

    # For tilts
    hass.states.async_set(
        DEMO_TILT, STATE_OPEN,
        {ATTR_SUPPORTED_FEATURES: 128, ATTR_CURRENT_TILT_POSITION: 100})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_ASSUMED_STATE) is True
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 128
    assert state.attributes.get(ATTR_CURRENT_POSITION) is None
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 100

    hass.states.async_remove(DEMO_COVER_TILT)
    hass.states.async_set(DEMO_TILT, STATE_CLOSED)
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_CLOSED
    assert state.attributes.get(ATTR_ASSUMED_STATE) is None
    assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 0
    assert state.attributes.get(ATTR_CURRENT_POSITION) is None
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) is None

    hass.states.async_set(
        DEMO_TILT, STATE_CLOSED, {ATTR_ASSUMED_STATE: True})
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.attributes.get(ATTR_ASSUMED_STATE) is True


async def test_open_covers(hass, setup_comp):
    """Test open cover function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_OPEN_COVER,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    for _ in range(10):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 100

    assert hass.states.get(DEMO_COVER).state == STATE_OPEN
    assert hass.states.get(DEMO_COVER_POS) \
               .attributes.get(ATTR_CURRENT_POSITION) == 100
    assert hass.states.get(DEMO_COVER_TILT) \
               .attributes.get(ATTR_CURRENT_POSITION) == 100


async def test_close_covers(hass, setup_comp):
    """Test close cover function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_CLOSE_COVER,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    for _ in range(10):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_CLOSED
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 0

    assert hass.states.get(DEMO_COVER).state == STATE_CLOSED
    assert hass.states.get(DEMO_COVER_POS) \
               .attributes.get(ATTR_CURRENT_POSITION) == 0
    assert hass.states.get(DEMO_COVER_TILT) \
               .attributes.get(ATTR_CURRENT_POSITION) == 0


async def test_stop_covers(hass, setup_comp):
    """Test stop cover function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_OPEN_COVER,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    future = dt_util.utcnow() + timedelta(seconds=1)
    async_fire_time_changed(hass, future)
    await hass.async_block_till_done()

    await hass.services.async_call(
        DOMAIN, SERVICE_STOP_COVER,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    future = dt_util.utcnow() + timedelta(seconds=1)
    async_fire_time_changed(hass, future)
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 100

    assert hass.states.get(DEMO_COVER).state == STATE_OPEN
    assert hass.states.get(DEMO_COVER_POS) \
        .attributes.get(ATTR_CURRENT_POSITION) == 20
    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_POSITION) == 80


async def test_set_cover_position(hass, setup_comp):
    """Test set cover position function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_SET_COVER_POSITION,
        {ATTR_ENTITY_ID: COVER_GROUP, ATTR_POSITION: 50}, blocking=True)
    for _ in range(4):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_POSITION) == 50

    assert hass.states.get(DEMO_COVER).state == STATE_CLOSED
    assert hass.states.get(DEMO_COVER_POS) \
        .attributes.get(ATTR_CURRENT_POSITION) == 50
    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_POSITION) == 50


async def test_open_tilts(hass, setup_comp):
    """Test open tilt function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_OPEN_COVER_TILT,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    for _ in range(5):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 100

    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_TILT_POSITION) == 100


async def test_close_tilts(hass, setup_comp):
    """Test close tilt function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_CLOSE_COVER_TILT,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    for _ in range(5):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 0

    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_TILT_POSITION) == 0


async def test_stop_tilts(hass, setup_comp):
    """Test stop tilts function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_OPEN_COVER_TILT,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    future = dt_util.utcnow() + timedelta(seconds=1)
    async_fire_time_changed(hass, future)
    await hass.async_block_till_done()

    await hass.services.async_call(
        DOMAIN, SERVICE_STOP_COVER_TILT,
        {ATTR_ENTITY_ID: COVER_GROUP}, blocking=True)
    future = dt_util.utcnow() + timedelta(seconds=1)
    async_fire_time_changed(hass, future)
    await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 60

    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_TILT_POSITION) == 60


async def test_set_tilt_positions(hass, setup_comp):
    """Test set tilt position function."""
    await hass.services.async_call(
        DOMAIN, SERVICE_SET_COVER_TILT_POSITION,
        {ATTR_ENTITY_ID: COVER_GROUP, ATTR_TILT_POSITION: 80}, blocking=True)
    for _ in range(3):
        future = dt_util.utcnow() + timedelta(seconds=1)
        async_fire_time_changed(hass, future)
        await hass.async_block_till_done()

    state = hass.states.get(COVER_GROUP)
    assert state.state == STATE_OPEN
    assert state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 80

    assert hass.states.get(DEMO_COVER_TILT) \
        .attributes.get(ATTR_CURRENT_TILT_POSITION) == 80