"""Tests for the AdGuard Home config flow."""
from unittest.mock import patch

import aiohttp

from homeassistant import data_entry_flow, config_entries
from homeassistant.components.adguard import config_flow
from homeassistant.components.adguard.const import DOMAIN
from homeassistant.const import (
    CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_SSL, CONF_USERNAME,
    CONF_VERIFY_SSL)

from tests.common import MockConfigEntry, mock_coro

FIXTURE_USER_INPUT = {
    CONF_HOST: '127.0.0.1',
    CONF_PORT: 3000,
    CONF_USERNAME: 'user',
    CONF_PASSWORD: 'pass',
    CONF_SSL: True,
    CONF_VERIFY_SSL: True,
}


async def test_show_authenticate_form(hass):
    """Test that the setup form is served."""
    flow = config_flow.AdGuardHomeFlowHandler()
    flow.hass = hass
    result = await flow.async_step_user(user_input=None)

    assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
    assert result['step_id'] == 'user'


async def test_connection_error(hass, aioclient_mock):
    """Test we show user form on AdGuard Home connection error."""
    aioclient_mock.get(
        "{}://{}:{}/control/status".format(
            'https' if FIXTURE_USER_INPUT[CONF_SSL] else 'http',
            FIXTURE_USER_INPUT[CONF_HOST],
            FIXTURE_USER_INPUT[CONF_PORT],
        ),
        exc=aiohttp.ClientError,
    )

    flow = config_flow.AdGuardHomeFlowHandler()
    flow.hass = hass
    result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT)

    assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
    assert result['step_id'] == 'user'
    assert result['errors'] == {'base': 'connection_error'}


async def test_full_flow_implementation(hass, aioclient_mock):
    """Test registering an integration and finishing flow works."""
    aioclient_mock.get(
        "{}://{}:{}/control/status".format(
            'https' if FIXTURE_USER_INPUT[CONF_SSL] else 'http',
            FIXTURE_USER_INPUT[CONF_HOST],
            FIXTURE_USER_INPUT[CONF_PORT],
        ),
        json={'version': '1.0'},
        headers={'Content-Type': 'application/json'},
    )

    flow = config_flow.AdGuardHomeFlowHandler()
    flow.hass = hass
    result = await flow.async_step_user(user_input=None)
    assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
    assert result['step_id'] == 'user'

    result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT)
    assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
    assert result['title'] == FIXTURE_USER_INPUT[CONF_HOST]
    assert result['data'][CONF_HOST] == FIXTURE_USER_INPUT[CONF_HOST]
    assert result['data'][CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD]
    assert result['data'][CONF_PORT] == FIXTURE_USER_INPUT[CONF_PORT]
    assert result['data'][CONF_SSL] == FIXTURE_USER_INPUT[CONF_SSL]
    assert result['data'][CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME]
    assert (
        result['data'][CONF_VERIFY_SSL] == FIXTURE_USER_INPUT[CONF_VERIFY_SSL]
    )


async def test_integration_already_exists(hass):
    """Test we only allow a single config flow."""
    MockConfigEntry(domain=DOMAIN).add_to_hass(hass)

    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={'source': 'user'}
    )
    assert result['type'] == 'abort'
    assert result['reason'] == 'single_instance_allowed'


async def test_hassio_single_instance(hass):
    """Test we only allow a single config flow."""
    MockConfigEntry(domain='adguard', data={
        'host': 'mock-adguard',
        'port': '3000'
    }).add_to_hass(hass)

    result = await hass.config_entries.flow.async_init(
        'adguard',
        data={
            'addon': 'AdGuard Home Addon',
            'host': 'mock-adguard',
            'port': '3000',
        },
        context={'source': 'hassio'}
    )
    assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
    assert result['reason'] == 'single_instance_allowed'


async def test_hassio_update_instance_not_running(hass):
    """Test we only allow a single config flow."""
    entry = MockConfigEntry(domain='adguard', data={
        'host': 'mock-adguard',
        'port': '3000'
    })
    entry.add_to_hass(hass)
    assert entry.state == config_entries.ENTRY_STATE_NOT_LOADED

    result = await hass.config_entries.flow.async_init(
        'adguard',
        data={
            'addon': 'AdGuard Home Addon',
            'host': 'mock-adguard-updated',
            'port': '3000',
        },
        context={'source': 'hassio'}
    )
    assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
    assert result['reason'] == 'existing_instance_updated'


async def test_hassio_update_instance_running(hass):
    """Test we only allow a single config flow."""
    entry = MockConfigEntry(domain='adguard', data={
        'host': 'mock-adguard',
        'port': '3000',
        'verify_ssl': False,
        'username': None,
        'password': None,
        'ssl': False,
    })
    entry.add_to_hass(hass)

    with patch.object(
        hass.config_entries, 'async_forward_entry_setup',
        side_effect=lambda *_: mock_coro(True)
    ) as mock_load:
        assert await hass.config_entries.async_setup(entry.entry_id)
        assert entry.state == config_entries.ENTRY_STATE_LOADED
        assert len(mock_load.mock_calls) == 2

    with patch.object(
        hass.config_entries, 'async_forward_entry_unload',
        side_effect=lambda *_: mock_coro(True)
    ) as mock_unload, patch.object(
        hass.config_entries, 'async_forward_entry_setup',
        side_effect=lambda *_: mock_coro(True)
    ) as mock_load:
        result = await hass.config_entries.flow.async_init(
            'adguard',
            data={
                'addon': 'AdGuard Home Addon',
                'host': 'mock-adguard-updated',
                'port': '3000',
            },
            context={'source': 'hassio'}
        )
        assert len(mock_unload.mock_calls) == 2
        assert len(mock_load.mock_calls) == 2

    assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
    assert result['reason'] == 'existing_instance_updated'
    assert entry.data['host'] == 'mock-adguard-updated'


async def test_hassio_confirm(hass, aioclient_mock):
    """Test we can finish a config flow."""
    aioclient_mock.get(
        "http://mock-adguard:3000/control/status",
        json={'version': '1.0'},
        headers={'Content-Type': 'application/json'},
    )

    result = await hass.config_entries.flow.async_init(
        'adguard',
        data={
            'addon': 'AdGuard Home Addon',
            'host': 'mock-adguard',
            'port': 3000,
        },
        context={'source': 'hassio'},
    )
    assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
    assert result['step_id'] == 'hassio_confirm'
    assert result['description_placeholders'] == {
        'addon': 'AdGuard Home Addon'
    }

    result = await hass.config_entries.flow.async_configure(
        result['flow_id'], {}
    )
    assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
    assert result['title'] == 'AdGuard Home Addon'
    assert result['data'][CONF_HOST] == 'mock-adguard'
    assert result['data'][CONF_PASSWORD] is None
    assert result['data'][CONF_PORT] == 3000
    assert result['data'][CONF_SSL] is False
    assert result['data'][CONF_USERNAME] is None
    assert result['data'][CONF_VERIFY_SSL]


async def test_hassio_connection_error(hass, aioclient_mock):
    """Test we show hassio confirm form on AdGuard Home connection error."""
    aioclient_mock.get(
        "http://mock-adguard:3000/control/status",
        exc=aiohttp.ClientError,
    )

    result = await hass.config_entries.flow.async_init(
        'adguard',
        data={
            'addon': 'AdGuard Home Addon',
            'host': 'mock-adguard',
            'port': 3000,
        },
        context={'source': 'hassio'},
    )

    result = await hass.config_entries.flow.async_configure(
        result['flow_id'], {}
    )

    assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
    assert result['step_id'] == 'hassio_confirm'
    assert result['errors'] == {'base': 'connection_error'}