mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 10:59:40 +00:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""Test fixtures for air-Q."""
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from .common import TEST_BRIGHTNESS, TEST_DEVICE_DATA, TEST_DEVICE_INFO
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.airq.async_setup_entry", return_value=True
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_airq():
|
|
"""Mock the aioairq.AirQ object.
|
|
|
|
The integration imports it in two places: in coordinator and config_flow.
|
|
"""
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.airq.coordinator.AirQ",
|
|
autospec=True,
|
|
) as mock_airq_class,
|
|
patch(
|
|
"homeassistant.components.airq.config_flow.AirQ",
|
|
new=mock_airq_class,
|
|
),
|
|
):
|
|
airq = mock_airq_class.return_value
|
|
# Pre-configure default mock values for setup
|
|
airq.fetch_device_info = AsyncMock(return_value=TEST_DEVICE_INFO)
|
|
airq.get_latest_data = AsyncMock(return_value=TEST_DEVICE_DATA)
|
|
airq.get_current_brightness = AsyncMock(return_value=TEST_BRIGHTNESS)
|
|
yield airq
|