mirror of
https://github.com/home-assistant/core.git
synced 2026-07-04 13:15:29 +01:00
epenet
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
"""Tests for iNELS integration."""
|
|
|
|
from unittest.mock import ANY, AsyncMock, patch
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import HA_INELS_PATH
|
|
from .common import DOMAIN
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.typing import MqttMockHAClient
|
|
|
|
|
|
async def test_ha_mqtt_publish(
|
|
hass: HomeAssistant, mqtt_mock: MqttMockHAClient
|
|
) -> None:
|
|
"""Test that MQTT publish function works correctly."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data={})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with (
|
|
patch(f"{HA_INELS_PATH}.InelsDiscovery") as mock_discovery_class,
|
|
patch("homeassistant.components.inels.PLATFORMS", []),
|
|
):
|
|
mock_discovery = AsyncMock()
|
|
mock_discovery.start.return_value = []
|
|
mock_discovery_class.return_value = mock_discovery
|
|
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
topic, payload, qos, retain = "test/topic", "test_payload", 1, True
|
|
|
|
await config_entry.runtime_data.mqtt.publish(topic, payload, qos, retain)
|
|
mqtt_mock.async_publish.assert_called_once_with(
|
|
topic, payload, qos, retain, message_expiry_interval=None
|
|
)
|
|
|
|
|
|
async def test_ha_mqtt_subscribe(
|
|
hass: HomeAssistant, mqtt_mock: MqttMockHAClient
|
|
) -> None:
|
|
"""Test that MQTT subscribe function works correctly."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data={})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with (
|
|
patch(f"{HA_INELS_PATH}.InelsDiscovery") as mock_discovery_class,
|
|
patch("homeassistant.components.inels.PLATFORMS", []),
|
|
):
|
|
mock_discovery = AsyncMock()
|
|
mock_discovery.start.return_value = []
|
|
mock_discovery_class.return_value = mock_discovery
|
|
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
topic = "test/topic"
|
|
|
|
await config_entry.runtime_data.mqtt.subscribe(topic)
|
|
mqtt_mock.async_subscribe.assert_any_call(topic, ANY, 0, "utf-8", None)
|
|
|
|
|
|
async def test_ha_mqtt_not_available(
|
|
hass: HomeAssistant, mqtt_mock: MqttMockHAClient
|
|
) -> None:
|
|
"""Test that ConfigEntryNotReady is raised when MQTT is not available."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data={})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.mqtt.async_wait_for_mqtt_client",
|
|
return_value=False,
|
|
):
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
assert config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_unload_entry(
|
|
hass: HomeAssistant, mqtt_mock: MqttMockHAClient, mock_mqtt
|
|
) -> None:
|
|
"""Test unload entry."""
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data={})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
with (
|
|
patch(f"{HA_INELS_PATH}.InelsDiscovery") as mock_discovery_class,
|
|
patch("homeassistant.components.inels.PLATFORMS", []),
|
|
):
|
|
mock_discovery = AsyncMock()
|
|
mock_discovery.start.return_value = []
|
|
mock_discovery_class.return_value = mock_discovery
|
|
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
result = await hass.config_entries.async_unload(config_entry.entry_id)
|
|
|
|
assert result is True
|
|
mock_mqtt.unsubscribe_topics.assert_called_once()
|
|
mock_mqtt.unsubscribe_listeners.assert_called_once()
|