mirror of
https://github.com/home-assistant/core.git
synced 2026-06-03 06:04:06 +01:00
Åke Strandberg
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
"""Tests the services provided by the miele integration."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from aiohttp import ClientResponseError
|
|
import pytest
|
|
from voluptuous import MultipleInvalid
|
|
|
|
from homeassistant.components.miele.const import DOMAIN
|
|
from homeassistant.components.miele.services import ATTR_PROGRAM_ID, SERVICE_SET_PROGRAM
|
|
from homeassistant.const import ATTR_DEVICE_ID
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
|
|
from homeassistant.helpers.device_registry import DeviceRegistry
|
|
|
|
from . import setup_integration
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
TEST_APPLIANCE = "Dummy_Appliance_1"
|
|
|
|
|
|
async def test_services(
|
|
hass: HomeAssistant,
|
|
device_registry: DeviceRegistry,
|
|
mock_miele_client: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Tests that the custom services are correct."""
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
device = device_registry.async_get_device(identifiers={(DOMAIN, TEST_APPLIANCE)})
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_PROGRAM,
|
|
{
|
|
ATTR_DEVICE_ID: device.id,
|
|
ATTR_PROGRAM_ID: 24,
|
|
},
|
|
blocking=True,
|
|
)
|
|
mock_miele_client.set_program.assert_called_once_with(
|
|
TEST_APPLIANCE, {"programId": 24}
|
|
)
|
|
|
|
|
|
async def test_service_api_errors(
|
|
hass: HomeAssistant,
|
|
device_registry: DeviceRegistry,
|
|
mock_miele_client: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test service api errors."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
device = device_registry.async_get_device(identifiers={(DOMAIN, TEST_APPLIANCE)})
|
|
|
|
# Test http error
|
|
mock_miele_client.set_program.side_effect = ClientResponseError("TestInfo", "test")
|
|
with pytest.raises(HomeAssistantError, match="'Set program' action failed"):
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_PROGRAM,
|
|
{"device_id": device.id, ATTR_PROGRAM_ID: 1},
|
|
blocking=True,
|
|
)
|
|
mock_miele_client.set_program.assert_called_once_with(
|
|
TEST_APPLIANCE, {"programId": 1}
|
|
)
|
|
|
|
|
|
async def test_service_validation_errors(
|
|
hass: HomeAssistant,
|
|
device_registry: DeviceRegistry,
|
|
mock_miele_client: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Tests that the custom services handle bad data."""
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
device = device_registry.async_get_device(identifiers={(DOMAIN, TEST_APPLIANCE)})
|
|
|
|
# Test missing program_id
|
|
with pytest.raises(MultipleInvalid, match="required key not provided"):
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_PROGRAM,
|
|
{"device_id": device.id},
|
|
blocking=True,
|
|
)
|
|
mock_miele_client.set_program.assert_not_called()
|
|
|
|
# Test invalid program_id
|
|
with pytest.raises(MultipleInvalid, match="expected int for dictionary value"):
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_PROGRAM,
|
|
{"device_id": device.id, ATTR_PROGRAM_ID: "invalid"},
|
|
blocking=True,
|
|
)
|
|
mock_miele_client.set_program.assert_not_called()
|
|
|
|
# Test invalid device
|
|
with pytest.raises(ServiceValidationError, match="Invalid device targeted"):
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_SET_PROGRAM,
|
|
{"device_id": "invalid_device", ATTR_PROGRAM_ID: 1},
|
|
blocking=True,
|
|
)
|
|
mock_miele_client.set_program.assert_not_called()
|