home-assistant
/
core
mirror of https://github.com/home-assistant/core
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
core/tests/components/esphome/test_fan.py

332 lines
9.6 KiB
Python
Raw Permalink Blame History

"""Test ESPHome fans."""
from unittest.mock import call
from aioesphomeapi import (
APIClient,
APIVersion,
FanDirection,
FanInfo,
FanSpeed,
FanState,
)
from homeassistant.components.fan import (
ATTR_DIRECTION,
ATTR_OSCILLATING,
ATTR_PERCENTAGE,
ATTR_PRESET_MODE,
DOMAIN as FAN_DOMAIN,
SERVICE_DECREASE_SPEED,
SERVICE_INCREASE_SPEED,
SERVICE_OSCILLATE,
SERVICE_SET_DIRECTION,
SERVICE_SET_PERCENTAGE,
SERVICE_SET_PRESET_MODE,
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
STATE_ON,
)
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant
async def test_fan_entity_with_all_features_old_api(
hass: HomeAssistant, mock_client: APIClient, mock_generic_device_entry
) -> None:
"""Test a generic fan entity that uses the old api and has all features."""
entity_info = [
FanInfo(
object_id="myfan",
key=1,
name="my fan",
unique_id="my_fan",
supports_direction=True,
supports_speed=True,
supports_oscillation=True,
)
]
states = [
FanState(
key=1,
state=True,
oscillating=True,
speed=FanSpeed.MEDIUM,
direction=FanDirection.REVERSE,
)
]
user_service = []
await mock_generic_device_entry(
mock_client=mock_client,
entity_info=entity_info,
user_service=user_service,
states=states,
)
state = hass.states.get("fan.test_myfan")
assert state is not None
assert state.state == STATE_ON
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 20},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, speed=FanSpeed.LOW, state=True)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 50},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, speed=FanSpeed.MEDIUM, state=True)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_DECREASE_SPEED,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, speed=FanSpeed.LOW, state=True)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_INCREASE_SPEED,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, speed=FanSpeed.HIGH, state=True)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_OFF,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, state=False)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 100},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, speed=FanSpeed.HIGH, state=True)]
)
mock_client.fan_command.reset_mock()
async def test_fan_entity_with_all_features_new_api(
hass: HomeAssistant, mock_client: APIClient, mock_generic_device_entry
) -> None:
"""Test a generic fan entity that uses the new api and has all features."""
mock_client.api_version = APIVersion(1, 4)
entity_info = [
FanInfo(
object_id="myfan",
key=1,
name="my fan",
unique_id="my_fan",
supported_speed_levels=4,
supports_direction=True,
supports_speed=True,
supports_oscillation=True,
supported_preset_modes=["Preset1", "Preset2"],
)
]
states = [
FanState(
key=1,
state=True,
oscillating=True,
speed_level=3,
direction=FanDirection.REVERSE,
preset_mode=None,
)
]
user_service = []
await mock_generic_device_entry(
mock_client=mock_client,
entity_info=entity_info,
user_service=user_service,
states=states,
)
state = hass.states.get("fan.test_myfan")
assert state is not None
assert state.state == STATE_ON
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 20},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, speed_level=1, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 50},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, speed_level=2, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_DECREASE_SPEED,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, speed_level=2, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_INCREASE_SPEED,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, speed_level=4, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_OFF,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, state=False)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 100},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, speed_level=4, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PERCENTAGE: 0},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, state=False)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_OSCILLATE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_OSCILLATING: True},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, oscillating=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_OSCILLATE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_OSCILLATING: False},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, oscillating=False)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_DIRECTION,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_DIRECTION: "forward"},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, direction=FanDirection.FORWARD)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_DIRECTION,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_DIRECTION: "reverse"},
blocking=True,
)
mock_client.fan_command.assert_has_calls(
[call(key=1, direction=FanDirection.REVERSE)]
)
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PRESET_MODE,
{ATTR_ENTITY_ID: "fan.test_myfan", ATTR_PRESET_MODE: "Preset1"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, preset_mode="Preset1")])
mock_client.fan_command.reset_mock()
async def test_fan_entity_with_no_features_new_api(
hass: HomeAssistant, mock_client: APIClient, mock_generic_device_entry
) -> None:
"""Test a generic fan entity that uses the new api and has no features."""
mock_client.api_version = APIVersion(1, 4)
entity_info = [
FanInfo(
object_id="myfan",
key=1,
name="my fan",
unique_id="my_fan",
supports_direction=False,
supports_speed=False,
supports_oscillation=False,
supported_preset_modes=[],
)
]
states = [FanState(key=1, state=True)]
user_service = []
await mock_generic_device_entry(
mock_client=mock_client,
entity_info=entity_info,
user_service=user_service,
states=states,
)
state = hass.states.get("fan.test_myfan")
assert state is not None
assert state.state == STATE_ON
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, state=True)])
mock_client.fan_command.reset_mock()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_OFF,
{ATTR_ENTITY_ID: "fan.test_myfan"},
blocking=True,
)
mock_client.fan_command.assert_has_calls([call(key=1, state=False)])
mock_client.fan_command.reset_mock()
Reference in New Issue View Git Blame Copy Permalink