"""Tests of the light entity of the balboa integration."""
from __future__ import annotations

from unittest.mock import MagicMock

from pybalboa import SpaControl
from pybalboa.enums import OffOnState, UnknownState
import pytest

from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNKNOWN
from homeassistant.core import HomeAssistant

from . import client_update, init_integration

from tests.components.light import common

ENTITY_LIGHT = "light.fakespa_light"


@pytest.fixture
def mock_light(client: MagicMock):
    """Return a mock light."""
    light = MagicMock(SpaControl)

    async def set_state(state: OffOnState):
        light.state = state

    light.client = client
    light.index = 0
    light.state = OffOnState.OFF
    light.set_state = set_state
    light.options = list(OffOnState)
    client.lights.append(light)

    return light


async def test_light(hass: HomeAssistant, client: MagicMock, mock_light) -> None:
    """Test spa light."""
    await init_integration(hass)

    # check if the initial state is off
    state = hass.states.get(ENTITY_LIGHT)
    assert state.state == STATE_OFF

    # test calling turn on
    await common.async_turn_on(hass, ENTITY_LIGHT)
    state = await client_update(hass, client, ENTITY_LIGHT)
    assert state.state == STATE_ON

    # test calling turn off
    await common.async_turn_off(hass, ENTITY_LIGHT)
    state = await client_update(hass, client, ENTITY_LIGHT)
    assert state.state == STATE_OFF


async def test_light_unknown_state(
    hass: HomeAssistant, client: MagicMock, mock_light
) -> None:
    """Tests spa light with unknown state."""
    await init_integration(hass)

    mock_light.state = UnknownState.UNKNOWN
    state = await client_update(hass, client, ENTITY_LIGHT)
    assert state.state == STATE_UNKNOWN