mirror of
https://github.com/home-assistant/core.git
synced 2025-08-02 18:18:21 +00:00
Add tests
This commit is contained in:
parent
286467c7f8
commit
5da2f5e96d
@ -8,6 +8,12 @@ import pytest
|
|||||||
from syrupy.assertion import SnapshotAssertion
|
from syrupy.assertion import SnapshotAssertion
|
||||||
from tuya_sharing import CustomerDevice
|
from tuya_sharing import CustomerDevice
|
||||||
|
|
||||||
|
from homeassistant.components.humidifier import (
|
||||||
|
DOMAIN as HUMIDIFIER_DOMAIN,
|
||||||
|
SERVICE_SET_HUMIDITY,
|
||||||
|
SERVICE_TURN_OFF,
|
||||||
|
SERVICE_TURN_ON,
|
||||||
|
)
|
||||||
from homeassistant.components.tuya import ManagerCompat
|
from homeassistant.components.tuya import ManagerCompat
|
||||||
from homeassistant.const import Platform
|
from homeassistant.const import Platform
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
@ -54,3 +60,87 @@ async def test_platform_setup_no_discovery(
|
|||||||
assert not er.async_entries_for_config_entry(
|
assert not er.async_entries_for_config_entry(
|
||||||
entity_registry, mock_config_entry.entry_id
|
entity_registry, mock_config_entry.entry_id
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mock_device_code",
|
||||||
|
["cs_arete_two_12l_dehumidifier_air_purifier"],
|
||||||
|
)
|
||||||
|
async def test_turn_on(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mock_manager: ManagerCompat,
|
||||||
|
mock_config_entry: MockConfigEntry,
|
||||||
|
mock_device: CustomerDevice,
|
||||||
|
) -> None:
|
||||||
|
"""Test turn on service."""
|
||||||
|
entity_id = "humidifier.dehumidifier"
|
||||||
|
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(
|
||||||
|
HUMIDIFIER_DOMAIN,
|
||||||
|
SERVICE_TURN_ON,
|
||||||
|
{"entity_id": entity_id},
|
||||||
|
)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
mock_manager.send_commands.assert_called_once_with(
|
||||||
|
mock_device.id, [{"code": "switch", "value": True}]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mock_device_code",
|
||||||
|
["cs_arete_two_12l_dehumidifier_air_purifier"],
|
||||||
|
)
|
||||||
|
async def test_turn_off(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mock_manager: ManagerCompat,
|
||||||
|
mock_config_entry: MockConfigEntry,
|
||||||
|
mock_device: CustomerDevice,
|
||||||
|
) -> None:
|
||||||
|
"""Test turn off service."""
|
||||||
|
entity_id = "humidifier.dehumidifier"
|
||||||
|
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(
|
||||||
|
HUMIDIFIER_DOMAIN,
|
||||||
|
SERVICE_TURN_OFF,
|
||||||
|
{"entity_id": entity_id},
|
||||||
|
)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
mock_manager.send_commands.assert_called_once_with(
|
||||||
|
mock_device.id, [{"code": "switch", "value": False}]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mock_device_code",
|
||||||
|
["cs_arete_two_12l_dehumidifier_air_purifier"],
|
||||||
|
)
|
||||||
|
async def test_set_humidity(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mock_manager: ManagerCompat,
|
||||||
|
mock_config_entry: MockConfigEntry,
|
||||||
|
mock_device: CustomerDevice,
|
||||||
|
) -> None:
|
||||||
|
"""Test set humidity service."""
|
||||||
|
entity_id = "humidifier.dehumidifier"
|
||||||
|
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(
|
||||||
|
HUMIDIFIER_DOMAIN,
|
||||||
|
SERVICE_SET_HUMIDITY,
|
||||||
|
{
|
||||||
|
"entity_id": entity_id,
|
||||||
|
"humidity": 50,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
mock_manager.send_commands.assert_called_once_with(
|
||||||
|
mock_device.id, [{"code": "dehumidify_set_value", "value": 50}]
|
||||||
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user