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

Add humidifier unit test for switchbot (#143207)

Browse Source
This commit is contained in:
Retha Runolfsson 2025-04-19 15:48:01 +08:00 committed by GitHub
parent 302dbc424b
commit 27b7fb6f91
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 150 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/components/switchbot/__init__.py
View File

@ -436,3 +436,30 @@ ROLLER_SHADE_SERVICE_INFO = BluetoothServiceInfoBleak(
connectable=True,
tx_power=-127,
)
HUMIDIFIER_SERVICE_INFO = BluetoothServiceInfoBleak(
name="Humidifier",
manufacturer_data={
741: b"\xacg\xb2\xcd\xfa\xbe",
},
service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"e\x80\x00\xf9\x80Bc\x00"},
service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
address="AA:BB:CC:DD:EE:FF",
rssi=-60,
source="local",
advertisement=generate_advertisement_data(
local_name="Humidifier",
manufacturer_data={
741: b"\xacg\xb2\xcd\xfa\xbe",
},
service_data={
"0000fd3d-0000-1000-8000-00805f9b34fb": b"e\x80\x00\xf9\x80Bc\x00"
},
service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
),
device=generate_ble_device("AA:BB:CC:DD:EE:FF", "Humidifier"),
time=0,
connectable=True,
tx_power=-127,
)

123
tests/components/switchbot/test_humidifier.py Normal file
View File

@ -0,0 +1,123 @@
"""Test the switchbot humidifiers."""
from collections.abc import Callable
from unittest.mock import AsyncMock, patch
import pytest
from homeassistant.components.humidifier import (
ATTR_HUMIDITY,
ATTR_MODE,
DOMAIN as HUMIDIFIER_DOMAIN,
MODE_AUTO,
MODE_NORMAL,
SERVICE_SET_HUMIDITY,
SERVICE_SET_MODE,
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
)
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant
from . import HUMIDIFIER_SERVICE_INFO
from tests.common import MockConfigEntry
from tests.components.bluetooth import inject_bluetooth_service_info
@pytest.mark.parametrize(
(
"service",
"service_data",
"mock_method",
"expected_args",
),
[
(
SERVICE_TURN_OFF,
{},
"turn_off",
(),
),
(
SERVICE_TURN_ON,
{},
"turn_on",
(),
),
(
SERVICE_SET_HUMIDITY,
{ATTR_HUMIDITY: 50},
"set_humidity_level",
(50,),
),
(
SERVICE_SET_MODE,
{ATTR_MODE: MODE_AUTO},
"set_auto_mode",
(),
),
(
SERVICE_SET_MODE,
{ATTR_MODE: MODE_NORMAL},
"set_manual_mode",
(),
),
],
)
async def test_humidifier_services(
hass: HomeAssistant,
mock_entry_factory: Callable[[str], MockConfigEntry],
service: str,
service_data: dict,
mock_method: str,
expected_args: tuple,
) -> None:
"""Test all humidifier services with proper parameters."""
inject_bluetooth_service_info(hass, HUMIDIFIER_SERVICE_INFO)
entry = mock_entry_factory(sensor_type="humidifier")
entry.add_to_hass(hass)
entity_id = "humidifier.test_name"
with (
patch(
"homeassistant.components.switchbot.humidifier.switchbot.SwitchbotHumidifier.set_level",
new=AsyncMock(return_value=True),
) as mock_set_humidity_level,
patch(
"homeassistant.components.switchbot.humidifier.switchbot.SwitchbotHumidifier.async_set_auto",
new=AsyncMock(return_value=True),
) as mock_set_auto_mode,
patch(
"homeassistant.components.switchbot.humidifier.switchbot.SwitchbotHumidifier.async_set_manual",
new=AsyncMock(return_value=True),
) as mock_set_manual_mode,
patch(
"homeassistant.components.switchbot.humidifier.switchbot.SwitchbotHumidifier.turn_off",
new=AsyncMock(return_value=True),
) as mock_turn_off,
patch(
"homeassistant.components.switchbot.humidifier.switchbot.SwitchbotHumidifier.turn_on",
new=AsyncMock(return_value=True),
) as mock_turn_on,
):
assert await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
await hass.services.async_call(
HUMIDIFIER_DOMAIN,
service,
{**service_data, ATTR_ENTITY_ID: entity_id},
blocking=True,
)
mock_map = {
"turn_off": mock_turn_off,
"turn_on": mock_turn_on,
"set_humidity_level": mock_set_humidity_level,
"set_auto_mode": mock_set_auto_mode,
"set_manual_mode": mock_set_manual_mode,
}
mock_instance = mock_map[mock_method]
mock_instance.assert_awaited_once_with(*expected_args)