core/tests/components/led_infrared/conftest.py

"""Common fixtures for the LED Infrared tests."""

from collections.abc import Generator
from unittest.mock import AsyncMock, patch

import pytest

from homeassistant.components.led_infrared.const import (
    CONF_DEVICE_TYPE,
    CONF_INFRARED_ENTITY_ID,
    CONF_INFRARED_RECEIVER_ENTITY_ID,
    DOMAIN,
    LEDIrDeviceType,
)

from tests.common import MockConfigEntry
from tests.components.infrared import EMITTER_ENTITY_ID, RECEIVER_ENTITY_ID


@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
    """Override async_setup_entry."""
    with patch(
        "homeassistant.components.led_infrared.async_setup_entry", return_value=True
    ) as mock_setup_entry:
        yield mock_setup_entry


@pytest.fixture(name="config_entry")
def mock_config_entry(request: pytest.FixtureRequest) -> MockConfigEntry:
    """Return a mock config entry."""
    return MockConfigEntry(
        domain=DOMAIN,
        title="LED Infrared via Test IR emitter",
        entry_id="1234567890",
        data={
            CONF_DEVICE_TYPE: (
                request.param
                if hasattr(request, "param")
                else LEDIrDeviceType.GENERIC_24_KEY
            ),
            CONF_INFRARED_ENTITY_ID: EMITTER_ENTITY_ID,
            CONF_INFRARED_RECEIVER_ENTITY_ID: RECEIVER_ENTITY_ID,
        },
    )


@pytest.fixture(name="infrared_codes")
def mock_infrared_code_to_command() -> Generator[None]:
    """Patch to_command to return the code directly.

    This allows tests to assert on the high-level code enum value
    rather than the raw NEC timings.
    """
    with (
        patch(
            "infrared_protocols.codes.generic.led.Generic24KeyCode.to_command",
            autospec=True,
            side_effect=lambda self, **kwargs: self,
        ) as mock_to_command,
        patch(
            "infrared_protocols.codes.generic.led.Generic13KeyCode.to_command",
            new=mock_to_command,
        ),
        patch(
            "infrared_protocols.codes.generic.led.Generic40KeyCode.to_command",
            new=mock_to_command,
        ),
        patch(
            "infrared_protocols.codes.generic.led.Generic44KeyCode.to_command",
            new=mock_to_command,
        ),
    ):
        yield