mirror of
				https://github.com/home-assistant/core.git
				synced 2025-11-03 16:09:36 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			1024 lines
		
	
	
		
			32 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			1024 lines
		
	
	
		
			32 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""Test the Z-Wave JS fan platform."""
 | 
						|
 | 
						|
import copy
 | 
						|
 | 
						|
import pytest
 | 
						|
from voluptuous.error import MultipleInvalid
 | 
						|
from zwave_js_server.const import CommandClass
 | 
						|
from zwave_js_server.event import Event
 | 
						|
from zwave_js_server.model.node import Node
 | 
						|
 | 
						|
from homeassistant.components.fan import (
 | 
						|
    ATTR_PERCENTAGE,
 | 
						|
    ATTR_PERCENTAGE_STEP,
 | 
						|
    ATTR_PRESET_MODE,
 | 
						|
    ATTR_PRESET_MODES,
 | 
						|
    DOMAIN as FAN_DOMAIN,
 | 
						|
    SERVICE_SET_PERCENTAGE,
 | 
						|
    SERVICE_SET_PRESET_MODE,
 | 
						|
    FanEntityFeature,
 | 
						|
    NotValidPresetModeError,
 | 
						|
)
 | 
						|
from homeassistant.components.zwave_js.fan import ATTR_FAN_STATE
 | 
						|
from homeassistant.const import (
 | 
						|
    ATTR_ENTITY_ID,
 | 
						|
    ATTR_SUPPORTED_FEATURES,
 | 
						|
    SERVICE_TURN_OFF,
 | 
						|
    SERVICE_TURN_ON,
 | 
						|
    STATE_OFF,
 | 
						|
    STATE_ON,
 | 
						|
    STATE_UNAVAILABLE,
 | 
						|
    STATE_UNKNOWN,
 | 
						|
)
 | 
						|
from homeassistant.core import HomeAssistant
 | 
						|
from homeassistant.exceptions import HomeAssistantError
 | 
						|
from homeassistant.helpers import entity_registry as er
 | 
						|
 | 
						|
 | 
						|
async def test_generic_fan(
 | 
						|
    hass: HomeAssistant, client, fan_generic, integration
 | 
						|
) -> None:
 | 
						|
    """Test the fan entity for a generic fan that lacks specific speed configuration."""
 | 
						|
    node = fan_generic
 | 
						|
    entity_id = "fan.generic_fan_controller"
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
 | 
						|
    assert state
 | 
						|
    assert state.state == STATE_OFF
 | 
						|
 | 
						|
    # Test turn on no speed
 | 
						|
    await hass.services.async_call(
 | 
						|
        "fan",
 | 
						|
        "turn_on",
 | 
						|
        {"entity_id": entity_id},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 17
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": 38,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "targetValue",
 | 
						|
    }
 | 
						|
    assert args["value"] == 255
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Due to optimistic updates, the state should be on even though the Z-Wave state
 | 
						|
    # hasn't been updated yet
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
 | 
						|
    assert state
 | 
						|
    assert state.state == STATE_ON
 | 
						|
 | 
						|
    # Test turn on setting speed
 | 
						|
    await hass.services.async_call(
 | 
						|
        "fan",
 | 
						|
        "turn_on",
 | 
						|
        {"entity_id": entity_id, "percentage": 66},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 17
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": 38,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "targetValue",
 | 
						|
    }
 | 
						|
    assert args["value"] == 66
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test setting unknown speed
 | 
						|
    with pytest.raises(MultipleInvalid):
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "set_percentage",
 | 
						|
            {"entity_id": entity_id, "percentage": "bad"},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test turning off
 | 
						|
    await hass.services.async_call(
 | 
						|
        "fan",
 | 
						|
        "turn_off",
 | 
						|
        {"entity_id": entity_id},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 17
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": 38,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "targetValue",
 | 
						|
    }
 | 
						|
    assert args["value"] == 0
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test speed update from value updated event
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 17,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Multilevel Switch",
 | 
						|
                "commandClass": 38,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "currentValue",
 | 
						|
                "newValue": 99,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "currentValue",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_ON
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE] == 100
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 17,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Multilevel Switch",
 | 
						|
                "commandClass": 38,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "currentValue",
 | 
						|
                "newValue": 0,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "currentValue",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_OFF
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE] == 0
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test setting percentage to 0
 | 
						|
    await hass.services.async_call(
 | 
						|
        "fan",
 | 
						|
        SERVICE_SET_PERCENTAGE,
 | 
						|
        {"entity_id": entity_id, "percentage": 0},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 17
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": 38,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "targetValue",
 | 
						|
    }
 | 
						|
    assert args["value"] == 0
 | 
						|
 | 
						|
    # Test value is None
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 17,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Multilevel Switch",
 | 
						|
                "commandClass": 38,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "currentValue",
 | 
						|
                "newValue": None,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "currentValue",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_UNKNOWN
 | 
						|
 | 
						|
 | 
						|
async def test_configurable_speeds_fan(
 | 
						|
    hass: HomeAssistant, client, hs_fc200, integration
 | 
						|
) -> None:
 | 
						|
    """Test a fan entity with configurable speeds."""
 | 
						|
    node = hs_fc200
 | 
						|
    node_id = 39
 | 
						|
    entity_id = "fan.scene_capable_fan_control_switch"
 | 
						|
 | 
						|
    async def get_zwave_speed_from_percentage(percentage):
 | 
						|
        """Set the fan to a particular percentage and get the resulting Zwave speed."""
 | 
						|
        client.async_send_command.reset_mock()
 | 
						|
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "percentage": percentage},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
        assert len(client.async_send_command.call_args_list) == 1
 | 
						|
        args = client.async_send_command.call_args[0][0]
 | 
						|
        assert args["command"] == "node.set_value"
 | 
						|
        assert args["nodeId"] == node_id
 | 
						|
        return args["value"]
 | 
						|
 | 
						|
    async def get_percentage_from_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed and get the resulting percentage."""
 | 
						|
        event = Event(
 | 
						|
            type="value updated",
 | 
						|
            data={
 | 
						|
                "source": "node",
 | 
						|
                "event": "value updated",
 | 
						|
                "nodeId": node_id,
 | 
						|
                "args": {
 | 
						|
                    "commandClassName": "Multilevel Switch",
 | 
						|
                    "commandClass": 38,
 | 
						|
                    "endpoint": 0,
 | 
						|
                    "property": "currentValue",
 | 
						|
                    "newValue": zwave_speed,
 | 
						|
                    "prevValue": 0,
 | 
						|
                    "propertyName": "currentValue",
 | 
						|
                },
 | 
						|
            },
 | 
						|
        )
 | 
						|
        node.receive_event(event)
 | 
						|
        state = hass.states.get(entity_id)
 | 
						|
        return state.attributes[ATTR_PERCENTAGE]
 | 
						|
 | 
						|
    # In 3-speed mode, the speeds are:
 | 
						|
    # low = 1-33, med=34-66, high=67-99
 | 
						|
    percentages_to_zwave_speeds = [
 | 
						|
        [[0], [0]],
 | 
						|
        [range(1, 34), range(1, 34)],
 | 
						|
        [range(34, 68), range(34, 67)],
 | 
						|
        [range(68, 101), range(67, 100)],
 | 
						|
    ]
 | 
						|
 | 
						|
    for percentages, zwave_speeds in percentages_to_zwave_speeds:
 | 
						|
        for percentage in percentages:
 | 
						|
            actual_zwave_speed = await get_zwave_speed_from_percentage(percentage)
 | 
						|
            assert actual_zwave_speed in zwave_speeds
 | 
						|
        for zwave_speed in zwave_speeds:
 | 
						|
            actual_percentage = await get_percentage_from_zwave_speed(zwave_speed)
 | 
						|
            assert actual_percentage in percentages
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE_STEP] == pytest.approx(33.3333, rel=1e-3)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODES] == []
 | 
						|
 | 
						|
 | 
						|
async def test_configurable_speeds_fan_with_missing_config_value(
 | 
						|
    hass: HomeAssistant, client, hs_fc200_state, integration
 | 
						|
) -> None:
 | 
						|
    """Test a fan entity with configurable speeds."""
 | 
						|
    entity_id = "fan.scene_capable_fan_control_switch"
 | 
						|
 | 
						|
    # Attach a modified version of the node with a bad config
 | 
						|
    bad_node_data = copy.deepcopy(hs_fc200_state)
 | 
						|
    fan_type_value = next(
 | 
						|
        (
 | 
						|
            v
 | 
						|
            for v in bad_node_data["values"]
 | 
						|
            if v["endpoint"] == 0 and v["commandClass"] == 112 and v["property"] == 5
 | 
						|
        ),
 | 
						|
        None,
 | 
						|
    )
 | 
						|
    assert fan_type_value is not None
 | 
						|
    bad_node_data["values"].remove(fan_type_value)
 | 
						|
 | 
						|
    node = Node(client, bad_node_data)
 | 
						|
    event = {"node": node}
 | 
						|
    client.driver.controller.emit("node added", event)
 | 
						|
    await hass.async_block_till_done()
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_UNAVAILABLE
 | 
						|
 | 
						|
 | 
						|
async def test_configurable_speeds_fan_with_bad_config_value(
 | 
						|
    hass: HomeAssistant, client, hs_fc200_state, integration
 | 
						|
) -> None:
 | 
						|
    """Test a fan entity with configurable speeds."""
 | 
						|
    entity_id = "fan.scene_capable_fan_control_switch"
 | 
						|
 | 
						|
    # Attach a modified version of the node with a bad config
 | 
						|
    bad_node_data = copy.deepcopy(hs_fc200_state)
 | 
						|
    fan_type_value = next(
 | 
						|
        (
 | 
						|
            v
 | 
						|
            for v in bad_node_data["values"]
 | 
						|
            if v["endpoint"] == 0 and v["commandClass"] == 112 and v["property"] == 5
 | 
						|
        ),
 | 
						|
        None,
 | 
						|
    )
 | 
						|
    assert fan_type_value is not None
 | 
						|
 | 
						|
    # 42 is not a valid configuration option with this device
 | 
						|
    fan_type_value["value"] = 42
 | 
						|
 | 
						|
    node = Node(client, bad_node_data)
 | 
						|
    event = {"node": node}
 | 
						|
    client.driver.controller.emit("node added", event)
 | 
						|
    await hass.async_block_till_done()
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_UNAVAILABLE
 | 
						|
 | 
						|
 | 
						|
async def test_ge_12730_fan(hass: HomeAssistant, client, ge_12730, integration) -> None:
 | 
						|
    """Test a GE 12730 fan with 3 fixed speeds."""
 | 
						|
    node = ge_12730
 | 
						|
    node_id = 24
 | 
						|
    entity_id = "fan.in_wall_smart_fan_control"
 | 
						|
 | 
						|
    async def get_zwave_speed_from_percentage(percentage):
 | 
						|
        """Set the fan to a particular percentage and get the resulting Zwave speed."""
 | 
						|
        client.async_send_command.reset_mock()
 | 
						|
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "percentage": percentage},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
        assert len(client.async_send_command.call_args_list) == 1
 | 
						|
        args = client.async_send_command.call_args[0][0]
 | 
						|
        assert args["command"] == "node.set_value"
 | 
						|
        assert args["nodeId"] == node_id
 | 
						|
        return args["value"]
 | 
						|
 | 
						|
    async def get_percentage_from_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed and get the resulting percentage."""
 | 
						|
        event = Event(
 | 
						|
            type="value updated",
 | 
						|
            data={
 | 
						|
                "source": "node",
 | 
						|
                "event": "value updated",
 | 
						|
                "nodeId": node_id,
 | 
						|
                "args": {
 | 
						|
                    "commandClassName": "Multilevel Switch",
 | 
						|
                    "commandClass": 38,
 | 
						|
                    "endpoint": 0,
 | 
						|
                    "property": "currentValue",
 | 
						|
                    "newValue": zwave_speed,
 | 
						|
                    "prevValue": 0,
 | 
						|
                    "propertyName": "currentValue",
 | 
						|
                },
 | 
						|
            },
 | 
						|
        )
 | 
						|
        node.receive_event(event)
 | 
						|
        state = hass.states.get(entity_id)
 | 
						|
        return state.attributes[ATTR_PERCENTAGE]
 | 
						|
 | 
						|
    # This device has the speeds:
 | 
						|
    # low = 1-33, med = 34-67, high = 68-99
 | 
						|
    percentages_to_zwave_speeds = [
 | 
						|
        [[0], [0]],
 | 
						|
        [range(1, 34), range(1, 34)],
 | 
						|
        [range(34, 68), range(34, 68)],
 | 
						|
        [range(68, 101), range(68, 100)],
 | 
						|
    ]
 | 
						|
 | 
						|
    for percentages, zwave_speeds in percentages_to_zwave_speeds:
 | 
						|
        for percentage in percentages:
 | 
						|
            actual_zwave_speed = await get_zwave_speed_from_percentage(percentage)
 | 
						|
            assert actual_zwave_speed in zwave_speeds
 | 
						|
        for zwave_speed in zwave_speeds:
 | 
						|
            actual_percentage = await get_percentage_from_zwave_speed(zwave_speed)
 | 
						|
            assert actual_percentage in percentages
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE_STEP] == pytest.approx(33.3333, rel=1e-3)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODES] == []
 | 
						|
 | 
						|
    # Test value is None
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": node_id,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Multilevel Switch",
 | 
						|
                "commandClass": 38,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "currentValue",
 | 
						|
                "newValue": None,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "currentValue",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_UNKNOWN
 | 
						|
 | 
						|
 | 
						|
async def test_inovelli_lzw36(
 | 
						|
    hass: HomeAssistant, client, inovelli_lzw36, integration
 | 
						|
) -> None:
 | 
						|
    """Test an LZW36."""
 | 
						|
    node = inovelli_lzw36
 | 
						|
    node_id = 19
 | 
						|
    entity_id = "fan.family_room_combo_2"
 | 
						|
 | 
						|
    async def get_zwave_speed_from_percentage(percentage):
 | 
						|
        """Set the fan to a particular percentage and get the resulting Zwave speed."""
 | 
						|
        client.async_send_command.reset_mock()
 | 
						|
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "percentage": percentage},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
        assert len(client.async_send_command.call_args_list) == 1
 | 
						|
        args = client.async_send_command.call_args[0][0]
 | 
						|
        assert args["command"] == "node.set_value"
 | 
						|
        assert args["nodeId"] == node_id
 | 
						|
        return args["value"]
 | 
						|
 | 
						|
    async def set_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed."""
 | 
						|
        event = Event(
 | 
						|
            type="value updated",
 | 
						|
            data={
 | 
						|
                "source": "node",
 | 
						|
                "event": "value updated",
 | 
						|
                "nodeId": node_id,
 | 
						|
                "args": {
 | 
						|
                    "commandClassName": "Multilevel Switch",
 | 
						|
                    "commandClass": 38,
 | 
						|
                    "endpoint": 2,
 | 
						|
                    "property": "currentValue",
 | 
						|
                    "newValue": zwave_speed,
 | 
						|
                    "prevValue": 0,
 | 
						|
                    "propertyName": "currentValue",
 | 
						|
                },
 | 
						|
            },
 | 
						|
        )
 | 
						|
        node.receive_event(event)
 | 
						|
 | 
						|
    async def get_percentage_from_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed and get the resulting percentage."""
 | 
						|
        await set_zwave_speed(zwave_speed)
 | 
						|
        state = hass.states.get(entity_id)
 | 
						|
        return state.attributes[ATTR_PERCENTAGE]
 | 
						|
 | 
						|
    # This device has the speeds:
 | 
						|
    # low = 2-33, med = 34-66, high = 67-99
 | 
						|
    percentages_to_zwave_speeds = [
 | 
						|
        [[0], [0]],
 | 
						|
        [range(1, 34), range(2, 34)],
 | 
						|
        [range(34, 68), range(34, 67)],
 | 
						|
        [range(68, 101), range(67, 100)],
 | 
						|
    ]
 | 
						|
 | 
						|
    for percentages, zwave_speeds in percentages_to_zwave_speeds:
 | 
						|
        for percentage in percentages:
 | 
						|
            actual_zwave_speed = await get_zwave_speed_from_percentage(percentage)
 | 
						|
            assert actual_zwave_speed in zwave_speeds
 | 
						|
        for zwave_speed in zwave_speeds:
 | 
						|
            actual_percentage = await get_percentage_from_zwave_speed(zwave_speed)
 | 
						|
            assert actual_percentage in percentages
 | 
						|
 | 
						|
    # Check static entity properties
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE_STEP] == pytest.approx(33.3333, rel=1e-3)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODES] == ["breeze"]
 | 
						|
 | 
						|
    # This device has one preset, where a device level of "1" is the
 | 
						|
    # "breeze" mode
 | 
						|
    await set_zwave_speed(1)
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODE] == "breeze"
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE] is None
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    await hass.services.async_call(
 | 
						|
        "fan",
 | 
						|
        "turn_on",
 | 
						|
        {"entity_id": entity_id, "preset_mode": "breeze"},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == node_id
 | 
						|
    assert args["value"] == 1
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
    with pytest.raises(NotValidPresetModeError) as exc:
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "preset_mode": "wheeze"},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
    assert exc.value.translation_key == "not_valid_preset_mode"
 | 
						|
    assert len(client.async_send_command.call_args_list) == 0
 | 
						|
 | 
						|
 | 
						|
async def test_leviton_zw4sf_fan(
 | 
						|
    hass: HomeAssistant, client, leviton_zw4sf, integration
 | 
						|
) -> None:
 | 
						|
    """Test a Leviton ZW4SF fan with 4 fixed speeds."""
 | 
						|
    node = leviton_zw4sf
 | 
						|
    node_id = 88
 | 
						|
    entity_id = "fan.4_speed_fan_controller"
 | 
						|
 | 
						|
    async def get_zwave_speed_from_percentage(percentage):
 | 
						|
        """Set the fan to a particular percentage and get the resulting Zwave speed."""
 | 
						|
        client.async_send_command.reset_mock()
 | 
						|
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "percentage": percentage},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
        assert len(client.async_send_command.call_args_list) == 1
 | 
						|
        args = client.async_send_command.call_args[0][0]
 | 
						|
        assert args["command"] == "node.set_value"
 | 
						|
        assert args["nodeId"] == node_id
 | 
						|
        return args["value"]
 | 
						|
 | 
						|
    async def get_percentage_from_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed and get the resulting percentage."""
 | 
						|
        event = Event(
 | 
						|
            type="value updated",
 | 
						|
            data={
 | 
						|
                "source": "node",
 | 
						|
                "event": "value updated",
 | 
						|
                "nodeId": node_id,
 | 
						|
                "args": {
 | 
						|
                    "commandClassName": "Multilevel Switch",
 | 
						|
                    "commandClass": 38,
 | 
						|
                    "endpoint": 0,
 | 
						|
                    "property": "currentValue",
 | 
						|
                    "newValue": zwave_speed,
 | 
						|
                    "prevValue": 0,
 | 
						|
                    "propertyName": "currentValue",
 | 
						|
                },
 | 
						|
            },
 | 
						|
        )
 | 
						|
        node.receive_event(event)
 | 
						|
        state = hass.states.get(entity_id)
 | 
						|
        return state.attributes[ATTR_PERCENTAGE]
 | 
						|
 | 
						|
    # This device has the speeds:
 | 
						|
    # 1 = 1-25, 2 = 26-49, 3 = 50-74, 4 = 75-99
 | 
						|
    percentages_to_zwave_speeds = [
 | 
						|
        [[0], [0]],
 | 
						|
        [range(1, 26), range(1, 26)],
 | 
						|
        [range(26, 51), range(26, 51)],
 | 
						|
        [range(51, 76), range(51, 76)],
 | 
						|
        [range(76, 101), range(76, 100)],
 | 
						|
    ]
 | 
						|
 | 
						|
    for percentages, zwave_speeds in percentages_to_zwave_speeds:
 | 
						|
        for percentage in percentages:
 | 
						|
            actual_zwave_speed = await get_zwave_speed_from_percentage(percentage)
 | 
						|
            assert actual_zwave_speed in zwave_speeds
 | 
						|
        for zwave_speed in zwave_speeds:
 | 
						|
            actual_percentage = await get_percentage_from_zwave_speed(zwave_speed)
 | 
						|
            assert actual_percentage in percentages
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE_STEP] == pytest.approx(25, rel=1e-3)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODES] == []
 | 
						|
 | 
						|
 | 
						|
async def test_thermostat_fan(
 | 
						|
    hass: HomeAssistant,
 | 
						|
    client,
 | 
						|
    climate_adc_t3000,
 | 
						|
    integration,
 | 
						|
    entity_registry: er.EntityRegistry,
 | 
						|
) -> None:
 | 
						|
    """Test the fan entity for a z-wave fan."""
 | 
						|
    node = climate_adc_t3000
 | 
						|
    entity_id = "fan.adc_t3000"
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state is None
 | 
						|
 | 
						|
    entry = entity_registry.async_get(entity_id)
 | 
						|
    assert entry
 | 
						|
    assert entry.disabled
 | 
						|
    assert entry.disabled_by is er.RegistryEntryDisabler.INTEGRATION
 | 
						|
 | 
						|
    # Test enabling entity
 | 
						|
    updated_entry = entity_registry.async_update_entity(entity_id, disabled_by=None)
 | 
						|
    assert updated_entry != entry
 | 
						|
    assert updated_entry.disabled is False
 | 
						|
 | 
						|
    await hass.config_entries.async_reload(integration.entry_id)
 | 
						|
    await hass.async_block_till_done()
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state
 | 
						|
    assert state.state == STATE_ON
 | 
						|
    assert state.attributes.get(ATTR_FAN_STATE) == "Idle / off"
 | 
						|
    assert state.attributes.get(ATTR_PRESET_MODE) == "Auto low"
 | 
						|
    assert (
 | 
						|
        state.attributes.get(ATTR_SUPPORTED_FEATURES)
 | 
						|
        == FanEntityFeature.PRESET_MODE
 | 
						|
        | FanEntityFeature.TURN_OFF
 | 
						|
        | FanEntityFeature.TURN_ON
 | 
						|
    )
 | 
						|
 | 
						|
    # Test setting preset mode
 | 
						|
    await hass.services.async_call(
 | 
						|
        FAN_DOMAIN,
 | 
						|
        SERVICE_SET_PRESET_MODE,
 | 
						|
        {ATTR_ENTITY_ID: entity_id, ATTR_PRESET_MODE: "Low"},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 68
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "mode",
 | 
						|
    }
 | 
						|
    assert args["value"] == 1
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test setting unknown preset mode
 | 
						|
    with pytest.raises(NotValidPresetModeError) as exc:
 | 
						|
        await hass.services.async_call(
 | 
						|
            FAN_DOMAIN,
 | 
						|
            SERVICE_SET_PRESET_MODE,
 | 
						|
            {ATTR_ENTITY_ID: entity_id, ATTR_PRESET_MODE: "Turbo"},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
    assert exc.value.translation_key == "not_valid_preset_mode"
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test turning off
 | 
						|
    await hass.services.async_call(
 | 
						|
        FAN_DOMAIN,
 | 
						|
        SERVICE_TURN_OFF,
 | 
						|
        {ATTR_ENTITY_ID: entity_id},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 68
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "off",
 | 
						|
    }
 | 
						|
    assert args["value"]
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test turning on
 | 
						|
    await hass.services.async_call(
 | 
						|
        FAN_DOMAIN,
 | 
						|
        SERVICE_TURN_ON,
 | 
						|
        {ATTR_ENTITY_ID: entity_id},
 | 
						|
        blocking=True,
 | 
						|
    )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 1
 | 
						|
    args = client.async_send_command.call_args[0][0]
 | 
						|
    assert args["command"] == "node.set_value"
 | 
						|
    assert args["nodeId"] == 68
 | 
						|
    assert args["valueId"] == {
 | 
						|
        "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
        "endpoint": 0,
 | 
						|
        "property": "off",
 | 
						|
    }
 | 
						|
    assert not args["value"]
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test fan state update from value updated event
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 68,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Thermostat Fan State",
 | 
						|
                "commandClass": CommandClass.THERMOSTAT_FAN_STATE.value,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "state",
 | 
						|
                "newValue": 4,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "state",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes.get(ATTR_FAN_STATE) == "Circulation mode"
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test unknown fan state update from value updated event
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 68,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Thermostat Fan State",
 | 
						|
                "commandClass": CommandClass.THERMOSTAT_FAN_STATE.value,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "state",
 | 
						|
                "newValue": 99,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "state",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert not state.attributes.get(ATTR_FAN_STATE)
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test fan mode update from value updated event
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 68,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Thermostat Fan Mode",
 | 
						|
                "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "mode",
 | 
						|
                "newValue": 1,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "mode",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes.get(ATTR_PRESET_MODE) == "Low"
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test fan mode update from value updated event for an unknown mode
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 68,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Thermostat Fan Mode",
 | 
						|
                "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "mode",
 | 
						|
                "newValue": 79,
 | 
						|
                "prevValue": 0,
 | 
						|
                "propertyName": "mode",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert not state.attributes.get(ATTR_PRESET_MODE)
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test fan mode turned off update from value updated event
 | 
						|
    event = Event(
 | 
						|
        type="value updated",
 | 
						|
        data={
 | 
						|
            "source": "node",
 | 
						|
            "event": "value updated",
 | 
						|
            "nodeId": 68,
 | 
						|
            "args": {
 | 
						|
                "commandClassName": "Thermostat Fan Mode",
 | 
						|
                "commandClass": CommandClass.THERMOSTAT_FAN_MODE.value,
 | 
						|
                "endpoint": 0,
 | 
						|
                "property": "off",
 | 
						|
                "newValue": True,
 | 
						|
                "prevValue": False,
 | 
						|
                "propertyName": "off",
 | 
						|
            },
 | 
						|
        },
 | 
						|
    )
 | 
						|
    node.receive_event(event)
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.state == STATE_OFF
 | 
						|
 | 
						|
 | 
						|
async def test_thermostat_fan_without_off(
 | 
						|
    hass: HomeAssistant,
 | 
						|
    client,
 | 
						|
    climate_radio_thermostat_ct100_plus,
 | 
						|
    integration,
 | 
						|
    entity_registry: er.EntityRegistry,
 | 
						|
) -> None:
 | 
						|
    """Test the fan entity for a z-wave fan without "off" property."""
 | 
						|
    entity_id = "fan.z_wave_thermostat"
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state is None
 | 
						|
 | 
						|
    entry = entity_registry.async_get(entity_id)
 | 
						|
    assert entry
 | 
						|
    assert entry.disabled
 | 
						|
    assert entry.disabled_by is er.RegistryEntryDisabler.INTEGRATION
 | 
						|
 | 
						|
    # Test enabling entity
 | 
						|
    updated_entry = entity_registry.async_update_entity(entity_id, disabled_by=None)
 | 
						|
    assert updated_entry != entry
 | 
						|
    assert updated_entry.disabled is False
 | 
						|
 | 
						|
    await hass.config_entries.async_reload(integration.entry_id)
 | 
						|
    await hass.async_block_till_done()
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state
 | 
						|
    assert state.state == STATE_UNKNOWN
 | 
						|
 | 
						|
    # Test turning off
 | 
						|
    with pytest.raises(HomeAssistantError):
 | 
						|
        await hass.services.async_call(
 | 
						|
            FAN_DOMAIN,
 | 
						|
            SERVICE_TURN_OFF,
 | 
						|
            {ATTR_ENTITY_ID: entity_id},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 0
 | 
						|
    assert state.state == STATE_UNKNOWN
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
    # Test turning on
 | 
						|
    with pytest.raises(HomeAssistantError):
 | 
						|
        await hass.services.async_call(
 | 
						|
            FAN_DOMAIN,
 | 
						|
            SERVICE_TURN_ON,
 | 
						|
            {ATTR_ENTITY_ID: entity_id},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
    assert len(client.async_send_command.call_args_list) == 0
 | 
						|
    assert state.state == STATE_UNKNOWN
 | 
						|
 | 
						|
    client.async_send_command.reset_mock()
 | 
						|
 | 
						|
 | 
						|
async def test_thermostat_fan_without_preset_modes(
 | 
						|
    hass: HomeAssistant,
 | 
						|
    client,
 | 
						|
    climate_adc_t3000_missing_fan_mode_states,
 | 
						|
    integration,
 | 
						|
    entity_registry: er.EntityRegistry,
 | 
						|
) -> None:
 | 
						|
    """Test the fan entity for a z-wave fan without "states" metadata."""
 | 
						|
    entity_id = "fan.adc_t3000_missing_fan_mode_states"
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state is None
 | 
						|
 | 
						|
    entry = entity_registry.async_get(entity_id)
 | 
						|
    assert entry
 | 
						|
    assert entry.disabled
 | 
						|
    assert entry.disabled_by is er.RegistryEntryDisabler.INTEGRATION
 | 
						|
 | 
						|
    # Test enabling entity
 | 
						|
    updated_entry = entity_registry.async_update_entity(entity_id, disabled_by=None)
 | 
						|
    assert updated_entry != entry
 | 
						|
    assert updated_entry.disabled is False
 | 
						|
 | 
						|
    await hass.config_entries.async_reload(integration.entry_id)
 | 
						|
    await hass.async_block_till_done()
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state
 | 
						|
 | 
						|
    assert not state.attributes.get(ATTR_PRESET_MODE)
 | 
						|
    assert not state.attributes.get(ATTR_PRESET_MODES)
 | 
						|
 | 
						|
 | 
						|
async def test_honeywell_39358_fan(
 | 
						|
    hass: HomeAssistant, client, fan_honeywell_39358, integration
 | 
						|
) -> None:
 | 
						|
    """Test a Honeywell 39358 fan with 3 fixed speeds."""
 | 
						|
    node = fan_honeywell_39358
 | 
						|
    node_id = 61
 | 
						|
    entity_id = "fan.honeywell_in_wall_smart_fan_control"
 | 
						|
 | 
						|
    async def get_zwave_speed_from_percentage(percentage):
 | 
						|
        """Set the fan to a particular percentage and get the resulting Zwave speed."""
 | 
						|
        client.async_send_command.reset_mock()
 | 
						|
 | 
						|
        await hass.services.async_call(
 | 
						|
            "fan",
 | 
						|
            "turn_on",
 | 
						|
            {"entity_id": entity_id, "percentage": percentage},
 | 
						|
            blocking=True,
 | 
						|
        )
 | 
						|
 | 
						|
        assert len(client.async_send_command.call_args_list) == 1
 | 
						|
        args = client.async_send_command.call_args[0][0]
 | 
						|
        assert args["command"] == "node.set_value"
 | 
						|
        assert args["nodeId"] == node_id
 | 
						|
        return args["value"]
 | 
						|
 | 
						|
    async def get_percentage_from_zwave_speed(zwave_speed):
 | 
						|
        """Set the underlying device speed and get the resulting percentage."""
 | 
						|
        event = Event(
 | 
						|
            type="value updated",
 | 
						|
            data={
 | 
						|
                "source": "node",
 | 
						|
                "event": "value updated",
 | 
						|
                "nodeId": node_id,
 | 
						|
                "args": {
 | 
						|
                    "commandClassName": "Multilevel Switch",
 | 
						|
                    "commandClass": 38,
 | 
						|
                    "endpoint": 0,
 | 
						|
                    "property": "currentValue",
 | 
						|
                    "newValue": zwave_speed,
 | 
						|
                    "prevValue": 0,
 | 
						|
                    "propertyName": "currentValue",
 | 
						|
                },
 | 
						|
            },
 | 
						|
        )
 | 
						|
        node.receive_event(event)
 | 
						|
        state = hass.states.get(entity_id)
 | 
						|
        return state.attributes[ATTR_PERCENTAGE]
 | 
						|
 | 
						|
    # This device has the speeds:
 | 
						|
    # low = 1-32, med = 33-66, high = 67-99
 | 
						|
    percentages_to_zwave_speeds = [
 | 
						|
        [[0], [0]],
 | 
						|
        [range(1, 34), range(1, 33)],
 | 
						|
        [range(34, 68), range(33, 67)],
 | 
						|
        [range(68, 101), range(67, 100)],
 | 
						|
    ]
 | 
						|
 | 
						|
    for percentages, zwave_speeds in percentages_to_zwave_speeds:
 | 
						|
        for percentage in percentages:
 | 
						|
            actual_zwave_speed = await get_zwave_speed_from_percentage(percentage)
 | 
						|
            assert actual_zwave_speed in zwave_speeds
 | 
						|
        for zwave_speed in zwave_speeds:
 | 
						|
            actual_percentage = await get_percentage_from_zwave_speed(zwave_speed)
 | 
						|
            assert actual_percentage in percentages
 | 
						|
 | 
						|
    state = hass.states.get(entity_id)
 | 
						|
    assert state.attributes[ATTR_PERCENTAGE_STEP] == pytest.approx(33.3333, rel=1e-3)
 | 
						|
    assert state.attributes[ATTR_PRESET_MODES] == []
 |