mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 10:59:40 +00:00
809aced9cc
* Add scene platform * Add cover platform * Refactor receiving state * Fix wrong auto-merged code
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
"""Test fixtures for qbus."""
|
|
|
|
from collections.abc import Generator
|
|
import json
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.qbus.const import CONF_SERIAL_NUMBER, DOMAIN
|
|
from homeassistant.components.qbus.entity import QbusEntity
|
|
from homeassistant.const import CONF_ID
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.util.json import JsonObjectType
|
|
|
|
from .const import FIXTURE_PAYLOAD_CONFIG, TOPIC_CONFIG
|
|
|
|
from tests.common import (
|
|
MockConfigEntry,
|
|
async_fire_mqtt_message,
|
|
load_json_object_fixture,
|
|
)
|
|
from tests.typing import MqttMockHAClient
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry(hass: HomeAssistant) -> MockConfigEntry:
|
|
"""Return the default mocked config entry."""
|
|
config_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="000001",
|
|
data={
|
|
CONF_ID: "UL1",
|
|
CONF_SERIAL_NUMBER: "000001",
|
|
},
|
|
)
|
|
config_entry.add_to_hass(hass)
|
|
return config_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def payload_config() -> JsonObjectType:
|
|
"""Return the config topic payload."""
|
|
return load_json_object_fixture(FIXTURE_PAYLOAD_CONFIG, DOMAIN)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_publish_state() -> Generator[AsyncMock]:
|
|
"""Return a mocked publish state call."""
|
|
with patch.object(QbusEntity, "_async_publish_output_state") as mock:
|
|
yield mock
|
|
|
|
|
|
@pytest.fixture
|
|
async def setup_integration(
|
|
hass: HomeAssistant,
|
|
mqtt_mock: MqttMockHAClient,
|
|
mock_config_entry: MockConfigEntry,
|
|
payload_config: JsonObjectType,
|
|
) -> None:
|
|
"""Set up the integration."""
|
|
|
|
assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
async_fire_mqtt_message(hass, TOPIC_CONFIG, json.dumps(payload_config))
|
|
await hass.async_block_till_done()
|