"""Configuration for Saunum Leil integration tests."""

from collections.abc import Generator
from datetime import timedelta
from unittest.mock import AsyncMock, MagicMock, patch

from pysaunum import SaunumData
import pytest

from homeassistant.components.saunum.const import DOMAIN
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry


@pytest.fixture(autouse=True)
def patch_delayed_refresh_seconds() -> Generator[None]:
    """Patch DELAYED_REFRESH_SECONDS to 0 to avoid delays in tests."""
    with patch(
        "homeassistant.components.saunum.climate.DELAYED_REFRESH_SECONDS",
        timedelta(seconds=0),
    ):
        yield


@pytest.fixture
def platforms() -> list[Platform]:
    """Fixture to specify platforms to test."""
    return [Platform.CLIMATE, Platform.LIGHT]


@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
    """Return the default mocked config entry."""
    return MockConfigEntry(
        entry_id="01K98T2T85R5GN0ZHYV25VFMMA",
        title="Saunum Leil Sauna",
        domain=DOMAIN,
        data={CONF_HOST: "192.168.1.100"},
    )


@pytest.fixture
def mock_saunum_data() -> SaunumData:
    """Return default mock Saunum data."""
    return SaunumData(
        session_active=False,
        sauna_type=0,
        sauna_duration=120,
        fan_duration=10,
        target_temperature=80,
        fan_speed=2,
        light_on=False,
        current_temperature=75.0,
        on_time=3600,
        heater_elements_active=0,
        door_open=False,
        alarm_door_open=False,
        alarm_door_sensor=False,
        alarm_thermal_cutoff=False,
        alarm_internal_temp=False,
        alarm_temp_sensor_short=False,
        alarm_temp_sensor_open=False,
    )


@pytest.fixture
def mock_saunum_client_class(mock_saunum_data: SaunumData) -> Generator[MagicMock]:
    """Return a mocked Saunum client class for config flow and integration tests."""
    with (
        patch(
            "homeassistant.components.saunum.config_flow.SaunumClient", autospec=True
        ) as mock_client_class,
        patch("homeassistant.components.saunum.SaunumClient", new=mock_client_class),
    ):
        mock_client = mock_client_class.return_value
        mock_client.is_connected = True

        mock_client_class.create = AsyncMock(return_value=mock_client)
        mock_client.async_get_data.return_value = mock_saunum_data

        yield mock_client_class


@pytest.fixture
def mock_saunum_client(mock_saunum_client_class: MagicMock) -> MagicMock:
    """Return a mocked Saunum client instance."""
    return mock_saunum_client_class.return_value


@pytest.fixture
async def init_integration(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    mock_saunum_client: MagicMock,
    platforms: list[Platform],
) -> MockConfigEntry:
    """Set up the integration for testing."""
    mock_config_entry.add_to_hass(hass)

    with patch("homeassistant.components.saunum.PLATFORMS", platforms):
        await hass.config_entries.async_setup(mock_config_entry.entry_id)
        await hass.async_block_till_done()

    return mock_config_entry


@pytest.fixture
def mock_setup_entry() -> Generator[MagicMock]:
    """Mock Saunum setup entry."""
    with patch(
        "homeassistant.components.saunum.async_setup_entry", autospec=True
    ) as mock_setup_entry:
        mock_setup_entry.return_value = True
        yield mock_setup_entry
