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

Improve deCONZ fan tests (#122493)

Browse Source 
* Improve fan tests

* Use snapshots
This commit is contained in:
Robert Svensson 2024-07-24 09:01:09 +02:00 committed by GitHub
parent 99aa68c93f
commit 4c7828fd50
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 89 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
tests/components/deconz/snapshots/test_fan.ambr Normal file
View File

@ -0,0 +1,54 @@
# serializer version: 1
# name: test_fans[light_payload0][fan.ceiling_fan-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': dict({
'preset_modes': None,
}),
'config_entry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'fan',
'entity_category': None,
'entity_id': 'fan.ceiling_fan',
'has_entity_name': False,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': None,
'original_icon': None,
'original_name': 'Ceiling fan',
'platform': 'deconz',
'previous_unique_id': None,
'supported_features': <FanEntityFeature: 49>,
'translation_key': None,
'unique_id': '00:22:a3:00:00:27:8b:81-01',
'unit_of_measurement': None,
})
# ---
# name: test_fans[light_payload0][fan.ceiling_fan-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'friendly_name': 'Ceiling fan',
'percentage': 100,
'percentage_step': 1.0,
'preset_mode': None,
'preset_modes': None,
'supported_features': <FanEntityFeature: 49>,
}),
'context': <ANY>,
'entity_id': 'fan.ceiling_fan',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'on',
})
# ---

106
tests/components/deconz/test_fan.py
View File

@ -1,8 +1,10 @@
"""deCONZ fan platform tests."""
from collections.abc import Callable
from unittest.mock import patch
import pytest
from syrupy import SnapshotAssertion
from homeassistant.components.fan import (
ATTR_PERCENTAGE,
@ -12,11 +14,19 @@ from homeassistant.components.fan import (
SERVICE_TURN_ON,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, STATE_UNAVAILABLE
from homeassistant.const import (
ATTR_ENTITY_ID,
STATE_OFF,
STATE_ON,
STATE_UNAVAILABLE,
Platform,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .conftest import WebsocketDataType
from tests.common import snapshot_platform
from tests.test_util.aiohttp import AiohttpClientMocker
@ -44,33 +54,25 @@ from tests.test_util.aiohttp import AiohttpClientMocker
)
async def test_fans(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
snapshot: SnapshotAssertion,
aioclient_mock: AiohttpClientMocker,
config_entry_setup: ConfigEntry,
config_entry_factory: ConfigEntry,
mock_put_request: Callable[[str, str], AiohttpClientMocker],
light_ws_data: WebsocketDataType,
) -> None:
"""Test that all supported fan entities are created."""
assert len(hass.states.async_all()) == 2 # Light and fan
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == 100
with patch("homeassistant.components.deconz.PLATFORMS", [Platform.FAN]):
config_entry = await config_entry_factory()
await snapshot_platform(hass, entity_registry, snapshot, config_entry.entry_id)
# Test states
await light_ws_data({"state": {"speed": 1}})
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == 25
await light_ws_data({"state": {"speed": 2}})
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == 50
await light_ws_data({"state": {"speed": 3}})
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == 75
await light_ws_data({"state": {"speed": 4}})
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == 100
for speed, percent in (1, 25), (2, 50), (3, 75), (4, 100):
await light_ws_data({"state": {"speed": speed}})
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE] == percent
await light_ws_data({"state": {"speed": 0}})
assert hass.states.get("fan.ceiling_fan").state == STATE_OFF
@ -110,55 +112,17 @@ async def test_fans(
)
assert aioclient_mock.mock_calls[3][2] == {"speed": 1}
# Service set fan percentage to 20%
# Service set fan percentage
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: 20},
blocking=True,
)
assert aioclient_mock.mock_calls[4][2] == {"speed": 1}
# Service set fan percentage to 40%
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: 40},
blocking=True,
)
assert aioclient_mock.mock_calls[5][2] == {"speed": 2}
# Service set fan percentage to 60%
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: 60},
blocking=True,
)
assert aioclient_mock.mock_calls[6][2] == {"speed": 3}
# Service set fan percentage to 80%
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: 80},
blocking=True,
)
assert aioclient_mock.mock_calls[7][2] == {"speed": 4}
# Service set fan percentage to 0% does not equal off
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: 0},
blocking=True,
)
assert aioclient_mock.mock_calls[8][2] == {"speed": 0}
for percent, speed in (20, 1), (40, 2), (60, 3), (80, 4), (0, 0):
aioclient_mock.mock_calls.clear()
await hass.services.async_call(
FAN_DOMAIN,
SERVICE_SET_PERCENTAGE,
{ATTR_ENTITY_ID: "fan.ceiling_fan", ATTR_PERCENTAGE: percent},
blocking=True,
)
assert aioclient_mock.mock_calls[0][2] == {"speed": speed}
# Events with an unsupported speed does not get converted
@ -166,13 +130,13 @@ async def test_fans(
assert hass.states.get("fan.ceiling_fan").state == STATE_ON
assert not hass.states.get("fan.ceiling_fan").attributes[ATTR_PERCENTAGE]
await hass.config_entries.async_unload(config_entry_setup.entry_id)
await hass.config_entries.async_unload(config_entry.entry_id)
states = hass.states.async_all()
assert len(states) == 2
assert len(states) == 1
for state in states:
assert state.state == STATE_UNAVAILABLE
await hass.config_entries.async_remove(config_entry_setup.entry_id)
await hass.config_entries.async_remove(config_entry.entry_id)
await hass.async_block_till_done()
assert len(hass.states.async_all()) == 0