"""Tests for SolarEdge optimizer sensors."""

from __future__ import annotations

from copy import deepcopy
from unittest.mock import AsyncMock, patch

from homeassistant.const import CONF_URL, STATE_UNAVAILABLE
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_AREA_MAPPING,
    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},
        options={CONF_AREA_MAPPING: {"west": ["1.1.1", "1.1.2"]}},
    )
    entry.add_to_hass(hass)
    device_registry = dr.async_get(hass)
    entity_registry = er.async_get(hass)
    stale_area_entity = entity_registry.async_get_or_create(
        "sensor",
        DOMAIN,
        "4886699_area_old_daily_energy",
        config_entry=entry,
        original_name="Old area energy",
    )
    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()

    entities = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
    assert len(entities) == 6
    assert entity_registry.async_get(stale_area_entity.entity_id) is None

    energy_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_14F28854-E2_daily_energy"
    )
    power_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_14F28854-E2_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

    area_energy_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_area_west_daily_energy"
    )
    area_power_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_area_west_current_power"
    )
    area_energy_state = hass.states.get(area_energy_entity.entity_id)
    area_power_state = hass.states.get(area_power_entity.entity_id)
    assert area_energy_state.state == "2468.5"
    assert area_power_state.state == "321.4"
    assert area_energy_state.attributes["optimizer_count"] == 2
    assert area_energy_state.attributes["optimizer_ids"] == ["1.1.1", "1.1.2"]

    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


async def test_area_energy_unavailable_when_one_optimizer_failed(
    hass: HomeAssistant,
) -> None:
    """Do not publish a misleading partial area sum."""
    payload = deepcopy(SAMPLE_PAYLOAD)
    payload["optimizers"][1]["dailyEnergyWh"] = None
    payload["optimizers"][1]["error"] = "Request failed"
    payload["successfulOptimizerCount"] = 1
    payload["failedOptimizerCount"] = 1

    entry = MockConfigEntry(
        domain=DOMAIN,
        title="SolarEdge Site 4886699",
        unique_id="4886699",
        data={CONF_URL: "http://app:8099", CONF_SCAN_INTERVAL: 600},
        options={CONF_AREA_MAPPING: {"west": ["1.1.1", "1.1.2"]}},
        version=2,
    )
    entry.add_to_hass(hass)

    with patch(
        "custom_components.solaredge_optimizers.api."
        "SolarEdgeOptimizerApiClient.async_get_optimizers",
        AsyncMock(return_value=parse_snapshot(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)
    area_energy_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_area_west_daily_energy"
    )
    area_power_entity = next(
        entity
        for entity in entities
        if entity.unique_id == "4886699_area_west_current_power"
    )

    assert hass.states.get(area_energy_entity.entity_id).state == STATE_UNAVAILABLE
    assert hass.states.get(area_power_entity.entity_id).state == "321.4"