mirror of
https://github.com/home-assistant/core.git
synced 2026-05-17 14:01:34 +01:00
DeerMaximum
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
"""Tests for the Nina integration."""
|
|
|
|
from copy import deepcopy
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from pynina import Warning
|
|
|
|
from homeassistant.components.nina.const import CONF_REGIONS
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def setup_platform(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
mock_nina_class: AsyncMock,
|
|
nina_warnings: list[Warning],
|
|
) -> None:
|
|
"""Set up the NINA platforms."""
|
|
mock_nina_class.warnings = {
|
|
region: deepcopy(nina_warnings)
|
|
for region in config_entry.data.get(CONF_REGIONS, {})
|
|
}
|
|
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.LOADED
|
|
|
|
|
|
async def setup_single_platform(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
platform: Platform | None,
|
|
mock_nina_class: AsyncMock,
|
|
nina_warnings: list[Warning],
|
|
) -> None:
|
|
"""Set up a single NINA platform."""
|
|
mock_nina_class.warnings = {
|
|
region: deepcopy(nina_warnings)
|
|
for region in config_entry.data.get(CONF_REGIONS, {})
|
|
}
|
|
|
|
platforms = [platform] if platform else []
|
|
|
|
with patch("homeassistant.components.nina.PLATFORMS", platforms):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert config_entry.state is ConfigEntryState.LOADED
|