"""Test Agent DVR integration."""
from unittest.mock import AsyncMock, patch

from agent import AgentError

from homeassistant.components.agent_dvr.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant

from . import CONF_DATA, create_entry

from tests.components.agent_dvr import init_integration
from tests.test_util.aiohttp import AiohttpClientMocker


async def _create_mocked_agent(available: bool = True):
    mocked_agent = AsyncMock()
    mocked_agent.is_available = available
    return mocked_agent


def _patch_init_agent(mocked_agent):
    return patch(
        "homeassistant.components.agent_dvr.Agent",
        return_value=mocked_agent,
    )


async def test_setup_config_and_unload(
    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
):
    """Test setup and unload."""
    entry = await init_integration(hass, aioclient_mock)
    assert entry.state == ConfigEntryState.LOADED
    assert len(hass.config_entries.async_entries(DOMAIN)) == 1
    assert entry.data == CONF_DATA

    assert await hass.config_entries.async_unload(entry.entry_id)
    await hass.async_block_till_done()

    assert entry.state is ConfigEntryState.NOT_LOADED
    assert not hass.data.get(DOMAIN)


async def test_async_setup_entry_not_ready(hass: HomeAssistant):
    """Test that it throws ConfigEntryNotReady when exception occurs during setup."""
    entry = create_entry(hass)
    with patch(
        "homeassistant.components.agent_dvr.Agent.update",
        side_effect=AgentError,
    ):
        await hass.config_entries.async_setup(entry.entry_id)
    assert entry.state == ConfigEntryState.SETUP_RETRY
    with _patch_init_agent(await _create_mocked_agent(available=False)):
        await hass.config_entries.async_reload(entry.entry_id)
    assert entry.state == ConfigEntryState.SETUP_RETRY