"""Test the Google Air Quality config flow."""

from unittest.mock import AsyncMock

from google_air_quality_api.exceptions import (
    GoogleAirQualityApiError,
    InvalidCustomLAQIConfigurationError,
)
import pytest

from homeassistant.components.google_air_quality.const import (
    CONF_ENABLE_CUSTOM_LAQI,
    CONF_REFERRER,
    CUSTOM_LAQI,
    CUSTOM_LOCAL_AQI_OPTIONS,
    DOMAIN,
    SECTION_API_KEY_OPTIONS,
)
from homeassistant.config_entries import SOURCE_USER
from homeassistant.const import (
    CONF_API_KEY,
    CONF_COUNTRY,
    CONF_LATITUDE,
    CONF_LOCATION,
    CONF_LONGITUDE,
    CONF_NAME,
)
from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType

from tests.common import MockConfigEntry, get_schema_suggested_value


def _assert_create_entry_result(
    result: dict, expected_referrer: str | None = None
) -> None:
    """Assert that the result is a create entry result."""
    assert result["type"] is FlowResultType.CREATE_ENTRY
    assert result["title"] == "Google Air Quality"
    assert result["data"] == {
        CONF_API_KEY: "test-api-key",
        CONF_REFERRER: expected_referrer,
    }
    assert len(result["subentries"]) == 1
    subentry = result["subentries"][0]
    assert subentry["subentry_type"] == "location"
    assert subentry["title"] == "test-name"
    assert subentry["data"] == {
        CONF_LATITUDE: 10.1,
        CONF_LONGITUDE: 20.1,
    }


async def test_create_entry(
    hass: HomeAssistant,
    mock_setup_entry: AsyncMock,
    mock_api: AsyncMock,
) -> None:
    """Test creating a config entry."""
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )
    assert result["type"] is FlowResultType.FORM
    assert result["errors"] == {}

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1,
                CONF_LONGITUDE: 20.1,
            },
        },
    )

    mock_api.async_get_current_conditions.assert_called_once_with(lat=10.1, lon=20.1)

    _assert_create_entry_result(result)
    assert len(mock_setup_entry.mock_calls) == 1


async def test_form_with_referrer(
    hass: HomeAssistant,
    mock_setup_entry: AsyncMock,
    mock_api: AsyncMock,
) -> None:
    """Test we get the form and optional referrer is specified."""
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )
    assert result["type"] is FlowResultType.FORM
    assert result["errors"] == {}

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            SECTION_API_KEY_OPTIONS: {
                CONF_REFERRER: "test-referrer",
            },
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1,
                CONF_LONGITUDE: 20.1,
            },
        },
    )

    mock_api.async_get_current_conditions.assert_called_once_with(lat=10.1, lon=20.1)

    _assert_create_entry_result(result, expected_referrer="test-referrer")
    assert len(mock_setup_entry.mock_calls) == 1


@pytest.mark.parametrize(
    ("api_exception", "expected_error"),
    [
        (InvalidCustomLAQIConfigurationError(), "mismatch_country_and_laqi"),
        (GoogleAirQualityApiError(), "cannot_connect"),
        (ValueError(), "unknown"),
    ],
)
async def test_form_exceptions(
    hass: HomeAssistant,
    mock_setup_entry: AsyncMock,
    mock_api: AsyncMock,
    api_exception,
    expected_error,
) -> None:
    """Test we handle exceptions."""
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )

    mock_api.async_get_current_conditions.side_effect = api_exception
    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1,
                CONF_LONGITUDE: 20.1,
            },
        },
    )

    assert result["type"] is FlowResultType.FORM
    assert result["errors"] == {"base": expected_error}
    # On error, the form should have the previous user input
    data_schema = result["data_schema"].schema
    assert get_schema_suggested_value(data_schema, CONF_NAME) == "test-name"
    assert get_schema_suggested_value(data_schema, CONF_API_KEY) == "test-api-key"
    assert get_schema_suggested_value(data_schema, CONF_LOCATION) == {
        CONF_LATITUDE: 10.1,
        CONF_LONGITUDE: 20.1,
    }

    # Make sure the config flow tests finish with either an
    # FlowResultType.CREATE_ENTRY or FlowResultType.ABORT so
    # we can show the config flow is able to recover from an error.

    mock_api.async_get_current_conditions.side_effect = None
    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1,
                CONF_LONGITUDE: 20.1,
            },
        },
    )

    _assert_create_entry_result(result)
    assert len(mock_setup_entry.mock_calls) == 1


async def test_form_api_key_already_configured(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test user input for config_entry with API key that already exists."""
    mock_config_entry.add_to_hass(hass)
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.2,
                CONF_LONGITUDE: 20.2,
            },
        },
    )

    assert result["type"] is FlowResultType.ABORT
    assert result["reason"] == "already_configured"
    assert mock_api.async_get_current_conditions.call_count == 0


async def test_form_location_already_configured(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test user input for a location that already exists."""
    mock_config_entry.add_to_hass(hass)
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "another-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1001,
                CONF_LONGITUDE: 20.0999,
            },
        },
    )

    assert result["type"] is FlowResultType.ABORT
    assert result["reason"] == "already_configured"
    assert mock_api.async_get_current_conditions.call_count == 0


async def test_form_not_already_configured(
    hass: HomeAssistant,
    mock_setup_entry: AsyncMock,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test user input for config_entry different than the existing one."""
    mock_config_entry.add_to_hass(hass)
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "new-test-name",
            CONF_API_KEY: "new-test-api-key",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1002,
                CONF_LONGITUDE: 20.0998,
            },
        },
    )

    mock_api.async_get_current_conditions.assert_called_once_with(
        lat=10.1002, lon=20.0998
    )

    assert result["type"] is FlowResultType.CREATE_ENTRY
    assert result["title"] == "Google Air Quality"
    assert result["data"] == {
        CONF_API_KEY: "new-test-api-key",
        CONF_REFERRER: None,
    }
    assert len(result["subentries"]) == 1
    subentry = result["subentries"][0]
    assert subentry["subentry_type"] == "location"
    assert subentry["title"] == "new-test-name"
    assert subentry["data"] == {
        CONF_LATITUDE: 10.1002,
        CONF_LONGITUDE: 20.0998,
    }
    assert len(mock_setup_entry.mock_calls) == 2


async def test_subentry_flow(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test creating a location subentry."""
    mock_config_entry.add_to_hass(hass)
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    await hass.async_block_till_done()

    # After initial setup for 1 subentry, each API is called once
    assert mock_api.async_get_current_conditions.call_count == 1

    result = await hass.config_entries.subentries.async_init(
        (mock_config_entry.entry_id, "location"),
        context={"source": "user"},
    )

    assert result["type"] is FlowResultType.FORM
    assert result["step_id"] == "location"

    result = await hass.config_entries.subentries.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "Work",
            CONF_LOCATION: {
                CONF_LATITUDE: 30.1,
                CONF_LONGITUDE: 40.1,
            },
            CUSTOM_LOCAL_AQI_OPTIONS: {
                CONF_COUNTRY: "DE",
                CUSTOM_LAQI: "deu_uba",
                CONF_ENABLE_CUSTOM_LAQI: False,
            },
        },
    )
    await hass.async_block_till_done()

    assert result["type"] is FlowResultType.CREATE_ENTRY
    assert result["title"] == "Work"
    assert result["data"] == {
        CONF_LATITUDE: 30.1,
        CONF_LONGITUDE: 40.1,
    }

    # Initial setup: 1 of each API call
    # Subentry flow validation: 1 current conditions call
    # Reload with 2 subentries: 2 of each API call
    assert mock_api.async_get_current_conditions.call_count == 1 + 1 + 2

    entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
    assert len(entry.subentries) == 2


async def test_subentry_flow_location_already_configured(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test user input for a location that already exists."""
    mock_config_entry.add_to_hass(hass)
    await hass.config_entries.async_setup(mock_config_entry.entry_id)

    result = await hass.config_entries.subentries.async_init(
        (mock_config_entry.entry_id, "location"),
        context={"source": "user"},
    )

    assert result["type"] is FlowResultType.FORM
    assert result["step_id"] == "location"

    result = await hass.config_entries.subentries.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "Work",
            CONF_LOCATION: {
                CONF_LATITUDE: 10.1,
                CONF_LONGITUDE: 20.1,
            },
        },
    )

    assert result["type"] is FlowResultType.FORM
    assert result["errors"]["base"] == "location_already_configured"

    entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
    assert len(entry.subentries) == 1


async def test_subentry_flow_location_name_already_configured(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test user input for a location name that already exists."""
    mock_config_entry.add_to_hass(hass)
    await hass.config_entries.async_setup(mock_config_entry.entry_id)

    result = await hass.config_entries.subentries.async_init(
        (mock_config_entry.entry_id, "location"),
        context={"source": "user"},
    )

    assert result["type"] is FlowResultType.FORM
    assert result["step_id"] == "location"

    result = await hass.config_entries.subentries.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "Home",
            CONF_LOCATION: {
                CONF_LATITUDE: 30.1,
                CONF_LONGITUDE: 40.1,
            },
        },
    )

    assert result["type"] is FlowResultType.FORM
    assert result["errors"]["base"] == "location_name_already_configured"

    entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
    assert len(entry.subentries) == 1


async def test_subentry_flow_entry_not_loaded(
    hass: HomeAssistant, mock_config_entry: MockConfigEntry
) -> None:
    """Test creating a location subentry when the parent entry is not loaded."""
    mock_config_entry.add_to_hass(hass)

    result = await hass.config_entries.subentries.async_init(
        (mock_config_entry.entry_id, "location"),
        context={"source": SOURCE_USER},
    )

    assert result["type"] is FlowResultType.ABORT
    assert result["reason"] == "entry_not_loaded"


async def test_create_entry_with_custom_laqi(
    hass: HomeAssistant,
    mock_setup_entry: AsyncMock,
    mock_api: AsyncMock,
) -> None:
    """Test creating a config entry with custom laqi."""
    result = await hass.config_entries.flow.async_init(
        DOMAIN, context={"source": SOURCE_USER}
    )
    assert result["type"] is FlowResultType.FORM
    assert result["errors"] == {}

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {CONF_LATITUDE: 10.1, CONF_LONGITUDE: 20.1},
            CUSTOM_LOCAL_AQI_OPTIONS: {
                CONF_COUNTRY: "DE",
                CONF_ENABLE_CUSTOM_LAQI: True,
            },
        },
    )
    assert result["type"] is FlowResultType.FORM
    assert result["errors"][CUSTOM_LOCAL_AQI_OPTIONS] == "missing_custom_laqi_options"

    result = await hass.config_entries.flow.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "test-name",
            CONF_API_KEY: "test-api-key",
            CONF_LOCATION: {CONF_LATITUDE: 10.1, CONF_LONGITUDE: 20.1},
            CUSTOM_LOCAL_AQI_OPTIONS: {
                CONF_COUNTRY: "DE",
                CUSTOM_LAQI: "deu_lubw",
                CONF_ENABLE_CUSTOM_LAQI: True,
            },
        },
    )
    mock_api.async_get_current_conditions.assert_called_once_with(
        lat=10.1, lon=20.1, region_code="DE", custom_local_aqi="deu_lubw"
    )
    assert result["type"] is FlowResultType.CREATE_ENTRY
    assert result["title"] == "Google Air Quality"
    assert result["data"] == {
        CONF_API_KEY: "test-api-key",
        CONF_REFERRER: None,
    }
    assert len(result["subentries"]) == 1
    subentry = result["subentries"][0]
    assert subentry["subentry_type"] == "location"
    assert subentry["title"] == "test-name"
    assert subentry["data"] == {
        CONF_LATITUDE: 10.1,
        CONF_LONGITUDE: 20.1,
        CUSTOM_LOCAL_AQI_OPTIONS: {
            CONF_COUNTRY: "DE",
            CUSTOM_LAQI: "deu_lubw",
            CONF_ENABLE_CUSTOM_LAQI: True,
        },
    }
    assert len(mock_setup_entry.mock_calls) == 1


async def test_subentry_flow_with_custom_laqi(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_api: AsyncMock,
) -> None:
    """Test creating a location subentry with a custom local AQI."""
    mock_config_entry.add_to_hass(hass)
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    await hass.async_block_till_done()

    assert mock_api.async_get_current_conditions.call_count == 1

    result = await hass.config_entries.subentries.async_init(
        (mock_config_entry.entry_id, "location"),
        context={"source": "user"},
    )

    assert result["type"] is FlowResultType.FORM
    assert result["step_id"] == "location"

    result = await hass.config_entries.subentries.async_configure(
        result["flow_id"],
        {
            CONF_NAME: "Work",
            CONF_LOCATION: {
                CONF_LATITUDE: 30.1,
                CONF_LONGITUDE: 40.1,
            },
            CUSTOM_LOCAL_AQI_OPTIONS: {
                CONF_COUNTRY: "DE",
                CUSTOM_LAQI: "deu_lubw",
                CONF_ENABLE_CUSTOM_LAQI: True,
            },
        },
    )
    await hass.async_block_till_done()

    assert result["type"] is FlowResultType.CREATE_ENTRY
    assert result["title"] == "Work"
    assert result["data"] == {
        CONF_LATITUDE: 30.1,
        CONF_LONGITUDE: 40.1,
        CUSTOM_LOCAL_AQI_OPTIONS: {
            CONF_COUNTRY: "DE",
            CUSTOM_LAQI: "deu_lubw",
            CONF_ENABLE_CUSTOM_LAQI: True,
        },
    }

    # Initial setup: 1 of each API call
    # Subentry flow validation: 1 current conditions call
    # Reload with 2 subentries: 2 of each API call
    assert mock_api.async_get_current_conditions.call_count == 1 + 1 + 2

    entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
    assert len(entry.subentries) == 2
