jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 05:07:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve zwave.cover tests (#93458)

Browse Source 
* Improve zwave.cover tests

* Update conftest.py
This commit is contained in:
Raman Gupta 2023-05-24 12:51:15 -04:00 committed by GitHub
parent 08c8c22fac
commit 3e84c374f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 76 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/components/zwave_js/conftest.py
View File

@ -946,7 +946,7 @@ def motorized_barrier_cover_fixture(client, gdc_zw062_state):
@pytest.fixture(name="iblinds_v2") @pytest.fixture(name="iblinds_v2")
def iblinds_cover_fixture(client, iblinds_v2_state): def iblinds_v2_cover_fixture(client, iblinds_v2_state):
"""Mock an iBlinds v2.0 window cover node.""" """Mock an iBlinds v2.0 window cover node."""
node = Node(client, copy.deepcopy(iblinds_v2_state)) node = Node(client, copy.deepcopy(iblinds_v2_state))
client.driver.controller.nodes[node.node_id] = node client.driver.controller.nodes[node.node_id] = node

121
tests/components/zwave_js/test_cover.py
View File

@ -10,14 +10,22 @@ from zwave_js_server.model.node import Node
from homeassistant.components.cover import ( from homeassistant.components.cover import (
ATTR_CURRENT_POSITION, ATTR_CURRENT_POSITION,
ATTR_CURRENT_TILT_POSITION, ATTR_CURRENT_TILT_POSITION,
ATTR_POSITION,
ATTR_TILT_POSITION,
DOMAIN, DOMAIN,
SERVICE_CLOSE_COVER, SERVICE_CLOSE_COVER,
SERVICE_CLOSE_COVER_TILT,
SERVICE_OPEN_COVER, SERVICE_OPEN_COVER,
SERVICE_OPEN_COVER_TILT,
SERVICE_SET_COVER_POSITION,
SERVICE_SET_COVER_TILT_POSITION,
SERVICE_STOP_COVER,
CoverDeviceClass, CoverDeviceClass,
) )
from homeassistant.components.zwave_js.helpers import ZwaveValueMatcher from homeassistant.components.zwave_js.helpers import ZwaveValueMatcher
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ATTR_ENTITY_ID,
STATE_CLOSED, STATE_CLOSED,
STATE_CLOSING, STATE_CLOSING,
STATE_OPEN, STATE_OPEN,
@ -46,14 +54,14 @@ async def test_window_cover(
assert state assert state
assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.WINDOW assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.WINDOW
assert state.state == "closed" assert state.state == STATE_CLOSED
assert state.attributes[ATTR_CURRENT_POSITION] == 0 assert state.attributes[ATTR_CURRENT_POSITION] == 0
# Test setting position # Test setting position
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"set_cover_position", SERVICE_SET_COVER_POSITION,
{"entity_id": WINDOW_COVER_ENTITY, "position": 50}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY, ATTR_POSITION: 50},
blocking=True, blocking=True,
) )
@ -72,9 +80,9 @@ async def test_window_cover(
# Test setting position # Test setting position
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"set_cover_position", SERVICE_SET_COVER_POSITION,
{"entity_id": WINDOW_COVER_ENTITY, "position": 0}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY, ATTR_POSITION: 0},
blocking=True, blocking=True,
) )
@ -93,9 +101,9 @@ async def test_window_cover(
# Test opening # Test opening
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"open_cover", SERVICE_OPEN_COVER,
{"entity_id": WINDOW_COVER_ENTITY}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -113,9 +121,9 @@ async def test_window_cover(
client.async_send_command.reset_mock() client.async_send_command.reset_mock()
# Test stop after opening # Test stop after opening
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"stop_cover", SERVICE_STOP_COVER,
{"entity_id": WINDOW_COVER_ENTITY}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -152,13 +160,13 @@ async def test_window_cover(
client.async_send_command.reset_mock() client.async_send_command.reset_mock()
state = hass.states.get(WINDOW_COVER_ENTITY) state = hass.states.get(WINDOW_COVER_ENTITY)
assert state.state == "open" assert state.state == STATE_OPEN
# Test closing # Test closing
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"close_cover", SERVICE_CLOSE_COVER,
{"entity_id": WINDOW_COVER_ENTITY}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY},
blocking=True, blocking=True,
) )
assert len(client.async_send_command.call_args_list) == 1 assert len(client.async_send_command.call_args_list) == 1
@ -176,9 +184,9 @@ async def test_window_cover(
# Test stop after closing # Test stop after closing
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"stop_cover", SERVICE_STOP_COVER,
{"entity_id": WINDOW_COVER_ENTITY}, {ATTR_ENTITY_ID: WINDOW_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -215,10 +223,10 @@ async def test_window_cover(
node.receive_event(event) node.receive_event(event)
state = hass.states.get(WINDOW_COVER_ENTITY) state = hass.states.get(WINDOW_COVER_ENTITY)
assert state.state == "closed" assert state.state == STATE_CLOSED
async def test_fibaro_FGR222_shutter_cover( async def test_fibaro_fgr222_shutter_cover(
hass: HomeAssistant, client, fibaro_fgr222_shutter, integration hass: HomeAssistant, client, fibaro_fgr222_shutter, integration
) -> None: ) -> None:
"""Test tilt function of the Fibaro Shutter devices.""" """Test tilt function of the Fibaro Shutter devices."""
@ -226,14 +234,14 @@ async def test_fibaro_FGR222_shutter_cover(
assert state assert state
assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.SHUTTER assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.SHUTTER
assert state.state == "open" assert state.state == STATE_OPEN
assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 0 assert state.attributes[ATTR_CURRENT_TILT_POSITION] == 0
# Test opening tilts # Test opening tilts
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"open_cover_tilt", SERVICE_OPEN_COVER_TILT,
{"entity_id": FIBARO_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: FIBARO_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -252,9 +260,9 @@ async def test_fibaro_FGR222_shutter_cover(
client.async_send_command.reset_mock() client.async_send_command.reset_mock()
# Test closing tilts # Test closing tilts
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"close_cover_tilt", SERVICE_CLOSE_COVER_TILT,
{"entity_id": FIBARO_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: FIBARO_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -270,6 +278,27 @@ async def test_fibaro_FGR222_shutter_cover(
} }
assert args["value"] == 0 assert args["value"] == 0
client.async_send_command.reset_mock()
# Test setting tilt position
await hass.services.async_call(
DOMAIN,
SERVICE_SET_COVER_TILT_POSITION,
{ATTR_ENTITY_ID: FIBARO_SHUTTER_COVER_ENTITY, ATTR_TILT_POSITION: 12},
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"] == 42
assert args["valueId"] == {
"endpoint": 0,
"commandClass": 145,
"property": "fibaro",
"propertyKey": "venetianBlindsTilt",
}
assert args["value"] == 12
# Test some tilt # Test some tilt
event = Event( event = Event(
type="value updated", type="value updated",
@ -306,14 +335,14 @@ async def test_aeotec_nano_shutter_cover(
assert state assert state
assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.WINDOW assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.WINDOW
assert state.state == "closed" assert state.state == STATE_CLOSED
assert state.attributes[ATTR_CURRENT_POSITION] == 0 assert state.attributes[ATTR_CURRENT_POSITION] == 0
# Test opening # Test opening
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"open_cover", SERVICE_OPEN_COVER,
{"entity_id": AEOTEC_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: AEOTEC_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -331,9 +360,9 @@ async def test_aeotec_nano_shutter_cover(
client.async_send_command.reset_mock() client.async_send_command.reset_mock()
# Test stop after opening # Test stop after opening
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"stop_cover", SERVICE_STOP_COVER,
{"entity_id": AEOTEC_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: AEOTEC_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -371,13 +400,13 @@ async def test_aeotec_nano_shutter_cover(
client.async_send_command.reset_mock() client.async_send_command.reset_mock()
state = hass.states.get(AEOTEC_SHUTTER_COVER_ENTITY) state = hass.states.get(AEOTEC_SHUTTER_COVER_ENTITY)
assert state.state == "open" assert state.state == STATE_OPEN
# Test closing # Test closing
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"close_cover", SERVICE_CLOSE_COVER,
{"entity_id": AEOTEC_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: AEOTEC_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
assert len(client.async_send_command.call_args_list) == 1 assert len(client.async_send_command.call_args_list) == 1
@ -395,9 +424,9 @@ async def test_aeotec_nano_shutter_cover(
# Test stop after closing # Test stop after closing
await hass.services.async_call( await hass.services.async_call(
"cover", DOMAIN,
"stop_cover", SERVICE_STOP_COVER,
{"entity_id": AEOTEC_SHUTTER_COVER_ENTITY}, {ATTR_ENTITY_ID: AEOTEC_SHUTTER_COVER_ENTITY},
blocking=True, blocking=True,
) )
@ -447,7 +476,7 @@ async def test_motor_barrier_cover(
# Test open # Test open
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_OPEN_COVER, {"entity_id": GDC_COVER_ENTITY}, blocking=True DOMAIN, SERVICE_OPEN_COVER, {ATTR_ENTITY_ID: GDC_COVER_ENTITY}, blocking=True
) )
assert len(client.async_send_command.call_args_list) == 1 assert len(client.async_send_command.call_args_list) == 1
@ -469,7 +498,7 @@ async def test_motor_barrier_cover(
# Test close # Test close
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_CLOSE_COVER, {"entity_id": GDC_COVER_ENTITY}, blocking=True DOMAIN, SERVICE_CLOSE_COVER, {ATTR_ENTITY_ID: GDC_COVER_ENTITY}, blocking=True
) )
assert len(client.async_send_command.call_args_list) == 1 assert len(client.async_send_command.call_args_list) == 1
@ -631,7 +660,7 @@ async def test_motor_barrier_cover_no_primary_value(
assert ATTR_CURRENT_POSITION not in state.attributes assert ATTR_CURRENT_POSITION not in state.attributes
async def test_fibaro_FGR222_shutter_cover_no_tilt( async def test_fibaro_fgr222_shutter_cover_no_tilt(
hass: HomeAssistant, client, fibaro_fgr222_shutter_state, integration hass: HomeAssistant, client, fibaro_fgr222_shutter_state, integration
) -> None: ) -> None:
"""Test tilt function of the Fibaro Shutter devices with tilt value is None.""" """Test tilt function of the Fibaro Shutter devices with tilt value is None."""