"""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)
    device_registry = dr.async_get(hass)
    device_registry.async_get_or_create(
        config_entry_id=entry.entry_id,
        identifiers={(DOMAIN, "4886699_14F28854-E2")},
        manufacturer="SolarEdge",
        model="Power Optimizer",
        name="Optimizer 1.1.1",
        serial_number="14F28854-E2",
    )

    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")
    )
    energy_state = hass.states.get(energy_entity.entity_id)
    power_state = hass.states.get(power_entity.entity_id)
    assert energy_state.state == "1234.5"
    assert power_state.state == "160.7"
    assert "1.1.1" in energy_state.name
    assert "1.1.1" in power_state.name

    devices = dr.async_entries_for_config_entry(device_registry, entry.entry_id)
    assert len(devices) == 1
    assert devices[0].identifiers == {(DOMAIN, "4886699")}
    assert devices[0].name == "SolarEdge Site 4886699"
    assert all(entity.device_id == devices[0].id for entity in entities)
    assert entry.version == 2