core/tests/components/incomfort/test_init.py

"""Tests for Intergas InComfort integration."""

from datetime import timedelta
from unittest.mock import AsyncMock, MagicMock, patch

from aiohttp import ClientResponseError, RequestInfo
from freezegun.api import FrozenDateTimeFactory
from incomfortclient import InvalidGateway, InvalidHeaterList
import pytest

from homeassistant.components.incomfort.coordinator import UPDATE_INTERVAL
from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from tests.common import async_fire_time_changed


@pytest.mark.usefixtures("entity_registry_enabled_by_default")
async def test_setup_platforms(
    hass: HomeAssistant,
    mock_incomfort: MagicMock,
    entity_registry: er.EntityRegistry,
    mock_config_entry: ConfigEntry,
) -> None:
    """Test the incomfort integration is set up correctly."""
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    assert mock_config_entry.state is ConfigEntryState.LOADED


@pytest.mark.usefixtures("entity_registry_enabled_by_default")
async def test_coordinator_updates(
    hass: HomeAssistant,
    mock_incomfort: MagicMock,
    freezer: FrozenDateTimeFactory,
    mock_config_entry: ConfigEntry,
) -> None:
    """Test the incomfort coordinator is updating."""
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    state = hass.states.get("climate.thermostat_1")
    assert state is not None
    assert state.attributes["current_temperature"] == 21.4
    mock_incomfort().mock_room_status["room_temp"] = 20.91

    state = hass.states.get("sensor.boiler_pressure")
    assert state is not None
    assert state.state == "1.86"
    mock_incomfort().mock_heater_status["pressure"] = 1.84

    freezer.tick(timedelta(seconds=UPDATE_INTERVAL + 5))
    async_fire_time_changed(hass)
    await hass.async_block_till_done(wait_background_tasks=True)

    state = hass.states.get("climate.thermostat_1")
    assert state is not None
    assert state.attributes["current_temperature"] == 20.9

    state = hass.states.get("sensor.boiler_pressure")
    assert state is not None
    assert state.state == "1.84"


@pytest.mark.usefixtures("entity_registry_enabled_by_default")
@pytest.mark.parametrize(
    "exc",
    [
        ClientResponseError(
            RequestInfo(
                url="http://example.com",
                method="GET",
                headers=[],
                real_url="http://example.com",
            ),
            None,
            status=401,
        ),
        InvalidHeaterList,
        ClientResponseError(
            RequestInfo(
                url="http://example.com",
                method="GET",
                headers=[],
                real_url="http://example.com",
            ),
            None,
            status=500,
        ),
        TimeoutError,
    ],
)
async def test_coordinator_update_fails(
    hass: HomeAssistant,
    mock_incomfort: MagicMock,
    freezer: FrozenDateTimeFactory,
    exc: Exception,
    mock_config_entry: ConfigEntry,
) -> None:
    """Test the incomfort coordinator update fails."""
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    state = hass.states.get("sensor.boiler_pressure")
    assert state is not None
    assert state.state == "1.86"

    with patch.object(
        mock_incomfort().heaters.return_value[0], "update", side_effect=exc
    ):
        freezer.tick(timedelta(seconds=UPDATE_INTERVAL + 5))
        async_fire_time_changed(hass)
        await hass.async_block_till_done(wait_background_tasks=True)

    state = hass.states.get("sensor.boiler_pressure")
    assert state is not None
    assert state.state == STATE_UNAVAILABLE


@pytest.mark.usefixtures("entity_registry_enabled_by_default")
@pytest.mark.parametrize(
    ("exc", "config_entry_state"),
    [
        (
            InvalidGateway,
            ConfigEntryState.SETUP_ERROR,
        ),
        (InvalidHeaterList, ConfigEntryState.SETUP_RETRY),
        (
            ClientResponseError(
                RequestInfo(
                    url="http://example.com",
                    method="GET",
                    headers=[],
                    real_url="http://example.com",
                ),
                None,
                status=404,
            ),
            ConfigEntryState.SETUP_ERROR,
        ),
        (
            ClientResponseError(
                RequestInfo(
                    url="http://example.com",
                    method="GET",
                    headers=[],
                    real_url="http://example.com",
                ),
                None,
                status=500,
            ),
            ConfigEntryState.SETUP_RETRY,
        ),
        (TimeoutError, ConfigEntryState.SETUP_RETRY),
    ],
)
async def test_entry_setup_fails(
    hass: HomeAssistant,
    mock_incomfort: MagicMock,
    freezer: FrozenDateTimeFactory,
    mock_config_entry: ConfigEntry,
    exc: Exception,
    config_entry_state: ConfigEntryState,
) -> None:
    """Test the incomfort coordinator entry setup fails."""
    with patch(
        "homeassistant.components.incomfort.async_connect_gateway",
        AsyncMock(side_effect=exc),
    ):
        await hass.config_entries.async_setup(mock_config_entry.entry_id)
    state = hass.states.get("sensor.boiler_pressure")
    assert state is None
    assert mock_config_entry.state is config_entry_state