home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2026-05-05 04:14:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use pytest.parametrize in Tuya cover tests (#156921)

Browse Source
This commit is contained in:
epenet
2025-11-20 10:38:14 +01:00
committed by GitHub
parent e7e13ecc74
commit 3ad1c6a47a
1 changed files with 37 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/components/tuya/test_cover.py
+37 -73
View File
@@ -2,6 +2,7 @@
from __future__ import annotations
from typing import Any
from unittest.mock import patch
import pytest
@@ -53,48 +54,47 @@ async def test_platform_setup_and_discovery(
"mock_device_code",
["cl_zah67ekd"],
)
@patch("homeassistant.components.tuya.PLATFORMS", [Platform.COVER])
async def test_open_service(
hass: HomeAssistant,
mock_manager: Manager,
mock_config_entry: MockConfigEntry,
mock_device: CustomerDevice,
) -> None:
"""Test open service."""
entity_id = "cover.kitchen_blinds_curtain"
await initialize_entry(hass, mock_manager, mock_config_entry, mock_device)
state = hass.states.get(entity_id)
assert state is not None, f"{entity_id} does not exist"
await hass.services.async_call(
COVER_DOMAIN,
SERVICE_OPEN_COVER,
{
ATTR_ENTITY_ID: entity_id,
},
blocking=True,
)
mock_manager.send_commands.assert_called_once_with(
mock_device.id,
[
{"code": "control", "value": "open"},
{"code": "percent_control", "value": 0},
],
)
@pytest.mark.parametrize(
"mock_device_code",
["cl_zah67ekd"],
("service", "service_data", "expected_commands"),
[
(
SERVICE_OPEN_COVER,
{},
[
{"code": "control", "value": "open"},
{"code": "percent_control", "value": 0},
],
),
(
SERVICE_CLOSE_COVER,
{},
[
{"code": "control", "value": "close"},
{"code": "percent_control", "value": 100},
],
),
(
SERVICE_SET_COVER_POSITION,
{
ATTR_POSITION: 25,
},
[
{"code": "percent_control", "value": 75},
],
),
],
)
@patch("homeassistant.components.tuya.PLATFORMS", [Platform.COVER])
async def test_close_service(
async def test_action(
hass: HomeAssistant,
mock_manager: Manager,
mock_config_entry: MockConfigEntry,
mock_device: CustomerDevice,
service: str,
service_data: dict[str, Any],
expected_commands: list[dict[str, Any]],
) -> None:
"""Test close service."""
"""Test cover action."""
entity_id = "cover.kitchen_blinds_curtain"
await initialize_entry(hass, mock_manager, mock_config_entry, mock_device)
@@ -102,51 +102,15 @@ async def test_close_service(
assert state is not None, f"{entity_id} does not exist"
await hass.services.async_call(
COVER_DOMAIN,
SERVICE_CLOSE_COVER,
service,
{
ATTR_ENTITY_ID: entity_id,
**service_data,
},
blocking=True,
)
mock_manager.send_commands.assert_called_once_with(
mock_device.id,
[
{"code": "control", "value": "close"},
{"code": "percent_control", "value": 100},
],
)
@pytest.mark.parametrize(
"mock_device_code",
["cl_zah67ekd"],
)
async def test_set_position(
hass: HomeAssistant,
mock_manager: Manager,
mock_config_entry: MockConfigEntry,
mock_device: CustomerDevice,
) -> None:
"""Test set position service (not available on this device)."""
entity_id = "cover.kitchen_blinds_curtain"
await initialize_entry(hass, mock_manager, mock_config_entry, mock_device)
state = hass.states.get(entity_id)
assert state is not None, f"{entity_id} does not exist"
await hass.services.async_call(
COVER_DOMAIN,
SERVICE_SET_COVER_POSITION,
{
ATTR_ENTITY_ID: entity_id,
ATTR_POSITION: 25,
},
blocking=True,
)
mock_manager.send_commands.assert_called_once_with(
mock_device.id,
[
{"code": "percent_control", "value": 75},
],
mock_device.id, expected_commands
)