core/tests/components/motionblinds_ble/conftest.py

"""Setup the Motionblinds Bluetooth tests."""

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

from motionblindsble.const import MotionBlindType
import pytest

from homeassistant.components.bluetooth.models import BluetoothServiceInfoBleak
from homeassistant.components.motionblinds_ble.const import (
    CONF_BLIND_TYPE,
    CONF_LOCAL_NAME,
    CONF_MAC_CODE,
    DOMAIN,
)
from homeassistant.const import CONF_ADDRESS

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


@pytest.fixture
def address() -> str:
    """Address fixture."""
    return "cc:cc:cc:cc:cc:cc"


@pytest.fixture
def mac_code(address: str) -> str:
    """MAC code fixture."""
    return "".join(address.split(":")[-3:-1]).upper()


@pytest.fixture
def display_name(mac_code: str) -> str:
    """Display name fixture."""
    return f"Motionblind {mac_code.upper()}"


@pytest.fixture
def name(display_name: str) -> str:
    """Name fixture."""
    return display_name.lower().replace(" ", "_")


@pytest.fixture
def local_name(mac_code: str) -> str:
    """Local name fixture."""
    return f"MOTION_{mac_code.upper()}"


@pytest.fixture
def blind_type() -> MotionBlindType:
    """Blind type fixture."""
    return MotionBlindType.ROLLER


@pytest.fixture
def service_info(local_name: str, address: str) -> BluetoothServiceInfoBleak:
    """Service info fixture."""
    return BluetoothServiceInfoBleak(
        name=local_name,
        address=address,
        device=generate_ble_device(
            address=address,
            name=local_name,
        ),
        rssi=-61,
        manufacturer_data={000: b"test"},
        service_data={
            "test": bytearray(b"0000"),
        },
        service_uuids=[
            "test",
        ],
        source="local",
        advertisement=generate_advertisement_data(
            manufacturer_data={000: b"test"},
            service_uuids=["test"],
        ),
        connectable=True,
        time=0,
        tx_power=-127,
    )


@pytest.fixture
def mock_motion_device(
    blind_type: MotionBlindType, display_name: str
) -> Generator[AsyncMock]:
    """Mock a MotionDevice."""

    with patch(
        "homeassistant.components.motionblinds_ble.MotionDevice",
        autospec=True,
    ) as mock_device:
        device = mock_device.return_value
        device.ble_device = Mock()
        device.display_name = display_name
        device.blind_type = blind_type
        yield device


@pytest.fixture
def mock_config_entry(
    blind_type: MotionBlindType, address: str, display_name: str, mac_code: str
) -> MockConfigEntry:
    """Config entry fixture."""
    return MockConfigEntry(
        title="mock_title",
        domain=DOMAIN,
        unique_id=address,
        data={
            CONF_ADDRESS: address,
            CONF_LOCAL_NAME: display_name,
            CONF_MAC_CODE: mac_code,
            CONF_BLIND_TYPE: blind_type.name.lower(),
        },
    )


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


@pytest.fixture
def motionblinds_ble_connect(
    enable_bluetooth: None, local_name: str, address: str
) -> Generator[tuple[AsyncMock, Mock]]:
    """Mock motion blinds ble connection and entry setup."""
    device = Mock()
    device.name = local_name
    device.address = address

    bleak_scanner = AsyncMock()
    bleak_scanner.discover.return_value = [device]

    with (
        patch(
            "homeassistant.components.motionblinds_ble.config_flow.bluetooth.async_scanner_count",
            return_value=1,
        ),
        patch(
            "homeassistant.components.motionblinds_ble.config_flow.bluetooth.async_get_scanner",
            return_value=bleak_scanner,
        ),
    ):
        yield bleak_scanner, device