57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
"""Tests for SolarEdge optimizer sensors."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from homeassistant.const import CONF_URL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.helpers import entity_registry as er
|
|
from pytest_homeassistant_custom_component.common import MockConfigEntry
|
|
|
|
from custom_components.solaredge_optimizers.api import parse_snapshot
|
|
from custom_components.solaredge_optimizers.const import CONF_SCAN_INTERVAL, DOMAIN
|
|
|
|
from .sample_data import SAMPLE_PAYLOAD
|
|
|
|
|
|
async def test_optimizer_sensors(hass: HomeAssistant) -> None:
|
|
"""Create energy and power sensors for every optimizer."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="SolarEdge Site 4886699",
|
|
unique_id="4886699",
|
|
data={CONF_URL: "http://app:8099", CONF_SCAN_INTERVAL: 600},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"custom_components.solaredge_optimizers.api."
|
|
"SolarEdgeOptimizerApiClient.async_get_optimizers",
|
|
AsyncMock(return_value=parse_snapshot(SAMPLE_PAYLOAD)),
|
|
):
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_registry = er.async_get(hass)
|
|
entities = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
|
|
assert len(entities) == 4
|
|
|
|
energy_entity = next(
|
|
entity for entity in entities if entity.unique_id.endswith("_daily_energy")
|
|
)
|
|
power_entity = next(
|
|
entity for entity in entities if entity.unique_id.endswith("_current_power")
|
|
)
|
|
assert hass.states.get(energy_entity.entity_id).state == "1234.5"
|
|
assert hass.states.get(power_entity.entity_id).state == "160.7"
|
|
|
|
device_registry = dr.async_get(hass)
|
|
devices = dr.async_entries_for_config_entry(device_registry, entry.entry_id)
|
|
assert len(devices) == 2
|
|
assert {device.serial_number for device in devices} == {
|
|
"14F28854-E2",
|
|
"14F28855-E2",
|
|
}
|