"""Tests for Renault setup process."""
from unittest.mock import AsyncMock, patch

import aiohttp
import pytest
from renault_api.gigya.exceptions import InvalidCredentialsException
from renault_api.kamereon import schemas

from homeassistant.components.renault import (
    RenaultHub,
    async_setup_entry,
    async_unload_entry,
)
from homeassistant.components.renault.const import DOMAIN
from homeassistant.components.renault.renault_vehicle import RenaultVehicleProxy
from homeassistant.exceptions import ConfigEntryNotReady

from .const import MOCK_CONFIG

from tests.common import MockConfigEntry, load_fixture


async def test_setup_unload_and_reload_entry(hass):
    """Test entry setup and unload."""
    # Create a mock entry so we don't have to go through config flow
    config_entry = MockConfigEntry(
        domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=123456
    )
    renault_account = AsyncMock()
    renault_account.get_vehicles.return_value = (
        schemas.KamereonVehiclesResponseSchema.loads(
            load_fixture("renault/vehicle_zoe_40.json")
        )
    )

    with patch("renault_api.renault_session.RenaultSession.login"), patch(
        "renault_api.renault_client.RenaultClient.get_api_account",
        return_value=renault_account,
    ):
        # Set up the entry and assert that the values set during setup are where we expect
        # them to be.
        assert await async_setup_entry(hass, config_entry)
        assert DOMAIN in hass.data and config_entry.unique_id in hass.data[DOMAIN]
        assert isinstance(hass.data[DOMAIN][config_entry.unique_id], RenaultHub)

        renault_hub: RenaultHub = hass.data[DOMAIN][config_entry.unique_id]
        assert len(renault_hub.vehicles) == 1
        assert isinstance(
            renault_hub.vehicles["VF1AAAAA555777999"], RenaultVehicleProxy
        )

        # Unload the entry and verify that the data has been removed
        assert await async_unload_entry(hass, config_entry)
        assert config_entry.unique_id not in hass.data[DOMAIN]


async def test_setup_entry_bad_password(hass):
    """Test entry setup and unload."""
    # Create a mock entry so we don't have to go through config flow
    config_entry = MockConfigEntry(
        domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=123456
    )

    with patch(
        "renault_api.renault_session.RenaultSession.login",
        side_effect=InvalidCredentialsException(403042, "invalid loginID or password"),
    ):
        # Set up the entry and assert that the values set during setup are where we expect
        # them to be.
        assert not await async_setup_entry(hass, config_entry)


async def test_setup_entry_exception(hass):
    """Test ConfigEntryNotReady when API raises an exception during entry setup."""
    config_entry = MockConfigEntry(
        domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=123456
    )

    # In this case we are testing the condition where async_setup_entry raises
    # ConfigEntryNotReady.
    with patch(
        "renault_api.renault_session.RenaultSession.login",
        side_effect=aiohttp.ClientConnectionError,
    ), pytest.raises(ConfigEntryNotReady):
        assert await async_setup_entry(hass, config_entry)