67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
"""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
|