"""Tests for the Palazzetti sensor platform."""

from unittest.mock import AsyncMock, patch

from pypalazzetti.exceptions import CommunicationError, ValidationError
from pypalazzetti.fan import FanType
import pytest
from syrupy.assertion import SnapshotAssertion

from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN, SERVICE_SET_VALUE
from homeassistant.const import ATTR_ENTITY_ID, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import entity_registry as er

from . import setup_integration

from tests.common import MockConfigEntry, snapshot_platform

POWER_ENTITY_ID = "number.stove_combustion_power"
FAN_ENTITY_ID = "number.stove_left_fan_speed"


async def test_all_entities(
    hass: HomeAssistant,
    snapshot: SnapshotAssertion,
    mock_palazzetti_client: AsyncMock,
    mock_config_entry: MockConfigEntry,
    entity_registry: er.EntityRegistry,
) -> None:
    """Test all entities."""
    with patch("homeassistant.components.palazzetti.PLATFORMS", [Platform.NUMBER]):
        await setup_integration(hass, mock_config_entry)

    await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)


async def test_async_set_data_power(
    hass: HomeAssistant,
    mock_palazzetti_client: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting number data via service call."""
    await setup_integration(hass, mock_config_entry)

    # Set value: Success
    await hass.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: POWER_ENTITY_ID, "value": 1},
        blocking=True,
    )
    mock_palazzetti_client.set_power_mode.assert_called_once_with(1)
    mock_palazzetti_client.set_power_mode.reset_mock()

    # Set value: Error
    mock_palazzetti_client.set_power_mode.side_effect = CommunicationError()
    message = "Could not connect to the device"
    with pytest.raises(HomeAssistantError, match=message):
        await hass.services.async_call(
            NUMBER_DOMAIN,
            SERVICE_SET_VALUE,
            {ATTR_ENTITY_ID: POWER_ENTITY_ID, "value": 1},
            blocking=True,
        )
    mock_palazzetti_client.set_power_mode.reset_mock()

    mock_palazzetti_client.set_power_mode.side_effect = ValidationError()
    message = "Combustion power 1.0 is invalid"
    with pytest.raises(ServiceValidationError, match=message):
        await hass.services.async_call(
            NUMBER_DOMAIN,
            SERVICE_SET_VALUE,
            {ATTR_ENTITY_ID: POWER_ENTITY_ID, "value": 1},
            blocking=True,
        )


async def test_async_set_data_fan(
    hass: HomeAssistant,
    mock_palazzetti_client: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting number data via service call."""
    await setup_integration(hass, mock_config_entry)

    # Set value: Success
    await hass.services.async_call(
        NUMBER_DOMAIN,
        SERVICE_SET_VALUE,
        {ATTR_ENTITY_ID: FAN_ENTITY_ID, "value": 1},
        blocking=True,
    )
    mock_palazzetti_client.set_fan_speed.assert_called_once_with(1, FanType.LEFT)
    mock_palazzetti_client.set_on.reset_mock()

    # Set value: Error
    mock_palazzetti_client.set_fan_speed.side_effect = CommunicationError()
    message = "Could not connect to the device"
    with pytest.raises(HomeAssistantError, match=message):
        await hass.services.async_call(
            NUMBER_DOMAIN,
            SERVICE_SET_VALUE,
            {ATTR_ENTITY_ID: FAN_ENTITY_ID, "value": 1},
            blocking=True,
        )
    mock_palazzetti_client.set_on.reset_mock()

    mock_palazzetti_client.set_fan_speed.side_effect = ValidationError()
    message = "Fan left speed 1.0 is invalid"
    with pytest.raises(ServiceValidationError, match=message):
        await hass.services.async_call(
            NUMBER_DOMAIN,
            SERVICE_SET_VALUE,
            {ATTR_ENTITY_ID: FAN_ENTITY_ID, "value": 1},
            blocking=True,
        )
