mirror of
https://github.com/home-assistant/core.git
synced 2025-11-08 18:39:30 +00:00
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
"""Satel Integra tests configuration."""
|
|
|
|
from collections.abc import Generator
|
|
from copy import deepcopy
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.satel_integra.const import DOMAIN
|
|
|
|
from . import (
|
|
MOCK_CONFIG_DATA,
|
|
MOCK_CONFIG_OPTIONS,
|
|
MOCK_ENTRY_ID,
|
|
MOCK_OUTPUT_SUBENTRY,
|
|
MOCK_PARTITION_SUBENTRY,
|
|
MOCK_SWITCHABLE_OUTPUT_SUBENTRY,
|
|
MOCK_ZONE_SUBENTRY,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Override integration setup."""
|
|
with patch(
|
|
"homeassistant.components.satel_integra.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_satel() -> Generator[AsyncMock]:
|
|
"""Override the satel test."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.satel_integra.AsyncSatel",
|
|
autospec=True,
|
|
) as client,
|
|
patch(
|
|
"homeassistant.components.satel_integra.config_flow.AsyncSatel", new=client
|
|
),
|
|
):
|
|
client.return_value.partition_states = {}
|
|
client.return_value.violated_outputs = []
|
|
client.return_value.violated_zones = []
|
|
client.return_value.connect.return_value = True
|
|
|
|
yield client
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Mock satel configuration entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="192.168.0.2",
|
|
data=MOCK_CONFIG_DATA,
|
|
options=MOCK_CONFIG_OPTIONS,
|
|
entry_id=MOCK_ENTRY_ID,
|
|
version=2,
|
|
minor_version=1,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry_with_subentries(
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> MockConfigEntry:
|
|
"""Mock satel configuration entry."""
|
|
mock_config_entry.subentries = deepcopy(
|
|
{
|
|
MOCK_PARTITION_SUBENTRY.subentry_id: MOCK_PARTITION_SUBENTRY,
|
|
MOCK_ZONE_SUBENTRY.subentry_id: MOCK_ZONE_SUBENTRY,
|
|
MOCK_OUTPUT_SUBENTRY.subentry_id: MOCK_OUTPUT_SUBENTRY,
|
|
MOCK_SWITCHABLE_OUTPUT_SUBENTRY.subentry_id: MOCK_SWITCHABLE_OUTPUT_SUBENTRY,
|
|
}
|
|
)
|
|
return mock_config_entry
|