core/tests/components/smartthings/test_fan.py

"""Test for the SmartThings fan platform."""

from unittest.mock import AsyncMock

from pysmartthings import Capability, Command
import pytest
from syrupy import SnapshotAssertion

from homeassistant.components.fan import (
    ATTR_PERCENTAGE,
    ATTR_PRESET_MODE,
    DOMAIN as FAN_DOMAIN,
    SERVICE_SET_PERCENTAGE,
    SERVICE_SET_PRESET_MODE,
)
from homeassistant.components.smartthings import MAIN
from homeassistant.const import (
    ATTR_ENTITY_ID,
    SERVICE_TURN_OFF,
    SERVICE_TURN_ON,
    Platform,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from . import setup_integration, snapshot_smartthings_entities

from tests.common import MockConfigEntry


async def test_all_entities(
    hass: HomeAssistant,
    snapshot: SnapshotAssertion,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
    entity_registry: er.EntityRegistry,
) -> None:
    """Test all entities."""
    await setup_integration(hass, mock_config_entry)

    snapshot_smartthings_entities(hass, entity_registry, snapshot, Platform.FAN)


@pytest.mark.parametrize("device_fixture", ["fake_fan"])
@pytest.mark.parametrize(
    ("action", "command"),
    [
        (SERVICE_TURN_ON, Command.ON),
        (SERVICE_TURN_OFF, Command.OFF),
    ],
)
async def test_turn_on_off(
    hass: HomeAssistant,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
    action: str,
    command: Command,
) -> None:
    """Test turning on and off."""
    await setup_integration(hass, mock_config_entry)

    await hass.services.async_call(
        FAN_DOMAIN,
        action,
        {ATTR_ENTITY_ID: "fan.fake_fan"},
        blocking=True,
    )
    devices.execute_device_command.assert_called_once_with(
        "f1af21a2-d5a1-437c-b10a-b34a87394b71",
        Capability.SWITCH,
        command,
        MAIN,
    )


@pytest.mark.parametrize("device_fixture", ["fake_fan"])
async def test_set_percentage(
    hass: HomeAssistant,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting the speed percentage of the fan."""
    await setup_integration(hass, mock_config_entry)

    await hass.services.async_call(
        FAN_DOMAIN,
        SERVICE_SET_PERCENTAGE,
        {ATTR_ENTITY_ID: "fan.fake_fan", ATTR_PERCENTAGE: 50},
        blocking=True,
    )
    devices.execute_device_command.assert_called_once_with(
        "f1af21a2-d5a1-437c-b10a-b34a87394b71",
        Capability.FAN_SPEED,
        Command.SET_FAN_SPEED,
        MAIN,
        argument=2,
    )


@pytest.mark.parametrize("device_fixture", ["fake_fan"])
async def test_set_percentage_off(
    hass: HomeAssistant,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting the speed percentage of the fan."""
    await setup_integration(hass, mock_config_entry)

    await hass.services.async_call(
        FAN_DOMAIN,
        SERVICE_SET_PERCENTAGE,
        {ATTR_ENTITY_ID: "fan.fake_fan", ATTR_PERCENTAGE: 0},
        blocking=True,
    )
    devices.execute_device_command.assert_called_once_with(
        "f1af21a2-d5a1-437c-b10a-b34a87394b71",
        Capability.SWITCH,
        Command.OFF,
        MAIN,
    )


@pytest.mark.parametrize("device_fixture", ["fake_fan"])
async def test_set_percentage_on(
    hass: HomeAssistant,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting the speed percentage of the fan."""
    await setup_integration(hass, mock_config_entry)

    await hass.services.async_call(
        FAN_DOMAIN,
        SERVICE_TURN_ON,
        {ATTR_ENTITY_ID: "fan.fake_fan", ATTR_PERCENTAGE: 50},
        blocking=True,
    )
    devices.execute_device_command.assert_called_once_with(
        "f1af21a2-d5a1-437c-b10a-b34a87394b71",
        Capability.FAN_SPEED,
        Command.SET_FAN_SPEED,
        MAIN,
        argument=2,
    )


@pytest.mark.parametrize("device_fixture", ["fake_fan"])
async def test_set_preset_mode(
    hass: HomeAssistant,
    devices: AsyncMock,
    mock_config_entry: MockConfigEntry,
) -> None:
    """Test setting the speed percentage of the fan."""
    await setup_integration(hass, mock_config_entry)

    await hass.services.async_call(
        FAN_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: "fan.fake_fan", ATTR_PRESET_MODE: "turbo"},
        blocking=True,
    )
    devices.execute_device_command.assert_called_once_with(
        "f1af21a2-d5a1-437c-b10a-b34a87394b71",
        Capability.AIR_CONDITIONER_FAN_MODE,
        Command.SET_FAN_MODE,
        MAIN,
        argument="turbo",
    )