core/tests/components/mill/test_init.py

"""Tests for Mill init."""

import asyncio
from unittest.mock import patch

from homeassistant.components import mill
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component

from tests.common import MockConfigEntry


async def test_setup_with_cloud_config(hass: HomeAssistant) -> None:
    """Test setup of cloud config."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_USERNAME: "user",
            mill.CONF_PASSWORD: "pswd",
            mill.CONNECTION_TYPE: mill.CLOUD,
        },
    )
    entry.add_to_hass(hass)
    with (
        patch("mill.Mill.fetch_heater_and_sensor_data", return_value={}) as mock_fetch,
        patch("mill.Mill.connect", return_value=True) as mock_connect,
    ):
        assert await async_setup_component(hass, "mill", {})
    assert len(mock_fetch.mock_calls) == 1
    assert len(mock_connect.mock_calls) == 1


async def test_setup_with_cloud_config_fails(hass: HomeAssistant) -> None:
    """Test setup of cloud config."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_USERNAME: "user",
            mill.CONF_PASSWORD: "pswd",
            mill.CONNECTION_TYPE: mill.CLOUD,
        },
    )
    entry.add_to_hass(hass)
    with patch("mill.Mill.connect", return_value=False):
        assert await async_setup_component(hass, "mill", {})
    assert entry.state is ConfigEntryState.SETUP_RETRY


async def test_setup_with_cloud_config_times_out(hass: HomeAssistant) -> None:
    """Test setup of cloud config will retry if timed out."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_USERNAME: "user",
            mill.CONF_PASSWORD: "pswd",
            mill.CONNECTION_TYPE: mill.CLOUD,
        },
    )
    entry.add_to_hass(hass)
    with patch("mill.Mill.connect", side_effect=asyncio.TimeoutError):
        await hass.config_entries.async_setup(entry.entry_id)
    assert entry.state is ConfigEntryState.SETUP_RETRY


async def test_setup_with_old_cloud_config(hass: HomeAssistant) -> None:
    """Test setup of old cloud config."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_USERNAME: "user",
            mill.CONF_PASSWORD: "pswd",
        },
    )
    entry.add_to_hass(hass)
    with (
        patch("mill.Mill.fetch_heater_and_sensor_data", return_value={}),
        patch("mill.Mill.connect", return_value=True) as mock_connect,
    ):
        assert await async_setup_component(hass, "mill", {})

    assert len(mock_connect.mock_calls) == 1


async def test_setup_with_local_config(hass: HomeAssistant) -> None:
    """Test setup of local config."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_IP_ADDRESS: "192.168.1.59",
            mill.CONNECTION_TYPE: mill.LOCAL,
        },
    )
    entry.add_to_hass(hass)
    with (
        patch(
            "mill_local.Mill.fetch_heater_and_sensor_data",
            return_value={
                "ambient_temperature": 20,
                "set_temperature": 22,
                "current_power": 0,
                "control_signal": 0,
                "raw_ambient_temperature": 19,
            },
        ) as mock_fetch,
        patch(
            "mill_local.Mill.connect",
            return_value={
                "name": "panel heater gen. 3",
                "version": "0x210927",
                "operation_key": "",
                "status": "ok",
            },
        ) as mock_connect,
    ):
        assert await async_setup_component(hass, "mill", {})

    assert len(mock_fetch.mock_calls) == 1
    assert len(mock_connect.mock_calls) == 1


async def test_unload_entry(hass: HomeAssistant) -> None:
    """Test removing mill client."""
    entry = MockConfigEntry(
        domain=mill.DOMAIN,
        data={
            mill.CONF_USERNAME: "user",
            mill.CONF_PASSWORD: "pswd",
            mill.CONNECTION_TYPE: mill.CLOUD,
        },
    )
    entry.add_to_hass(hass)

    with (
        patch.object(
            hass.config_entries,
            "async_forward_entry_unload",
            return_value=True,
        ) as unload_entry,
        patch("mill.Mill.fetch_heater_and_sensor_data", return_value={}),
        patch(
            "mill.Mill.connect",
            return_value=True,
        ),
    ):
        assert await async_setup_component(hass, "mill", {})

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

        assert unload_entry.call_count == 2
        assert entry.entry_id not in hass.data[mill.DOMAIN]