core/tests/components/hue_ble/test_init.py

"""Test hue_ble setup process."""

from unittest.mock import patch

from bleak.backends.device import BLEDevice
from HueBLE import ConnectionError, HueBleError
import pytest

from homeassistant.components.hue_ble.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component

from . import TEST_DEVICE_MAC, TEST_DEVICE_NAME

from tests.common import MockConfigEntry
from tests.components.bluetooth import generate_ble_device


@pytest.mark.parametrize(
    (
        "ble_device",
        "reachability_reason",
        "connect_result",
        "poll_state_result",
        "message",
    ),
    [
        (
            None,
            "Bad vibes",
            True,
            None,
            f"The light {TEST_DEVICE_NAME} ({TEST_DEVICE_MAC}) was not found: Bad vibes",
        ),
        (
            generate_ble_device(TEST_DEVICE_MAC, TEST_DEVICE_NAME),
            None,
            False,
            ConnectionError,
            "Device found but unable to connect.",
        ),
        (
            generate_ble_device(TEST_DEVICE_MAC, TEST_DEVICE_NAME),
            None,
            True,
            HueBleError,
            "Device found and connected but unable to poll values from it.",
        ),
    ],
    ids=["no_device", "error_connect", "error_poll"],
)
async def test_setup_error(
    hass: HomeAssistant,
    caplog: pytest.LogCaptureFixture,
    reachability_reason: str,
    ble_device: BLEDevice | None,
    connect_result: Exception | None,
    poll_state_result: Exception | None,
    message: str,
) -> None:
    """Test that ConfigEntryNotReady is raised if there is an error condition."""

    entry = MockConfigEntry(
        domain=DOMAIN, title=TEST_DEVICE_NAME, unique_id=TEST_DEVICE_MAC, data={}
    )
    entry.add_to_hass(hass)
    with (
        patch(
            "homeassistant.components.hue_ble.async_ble_device_from_address",
            return_value=ble_device,
        ),
        patch(
            "homeassistant.components.hue_ble.bluetooth.async_address_reachability_diagnostics",
            return_value=reachability_reason,
        ),
        patch(
            "homeassistant.components.hue_ble.HueBleLight.connect",
            side_effect=[connect_result],
        ),
        patch(
            "homeassistant.components.hue_ble.HueBleLight.poll_state",
            side_effect=[poll_state_result],
        ),
    ):
        assert await async_setup_component(hass, DOMAIN, {}) is True
        await hass.async_block_till_done()
        assert entry.state is ConfigEntryState.SETUP_RETRY
        assert message in caplog.text


async def test_setup(
    hass: HomeAssistant,
) -> None:
    """Test that the config is loaded if there are no errors."""

    entry = MockConfigEntry(domain=DOMAIN, unique_id="abcd", data={})
    entry.add_to_hass(hass)
    with (
        patch(
            "homeassistant.components.hue_ble.async_ble_device_from_address",
            return_value=generate_ble_device(TEST_DEVICE_MAC, TEST_DEVICE_NAME),
        ),
        patch(
            "homeassistant.components.hue_ble.HueBleLight.connect",
            return_value=None,
        ),
        patch(
            "homeassistant.components.hue_ble.HueBleLight.poll_state",
            return_value=None,
        ),
    ):
        assert await async_setup_component(hass, DOMAIN, {}) is True
        await hass.async_block_till_done()
        assert entry.state is ConfigEntryState.LOADED