Use service_calls fixture in shelly tests (#120991)

This commit is contained in:
epenet 2024-07-02 10:36:50 +02:00 committed by GitHub
parent 07d80d5ad9
commit 3df3e6d081
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 20 additions and 26 deletions

View File

@ -11,11 +11,11 @@ from homeassistant.components.shelly.const import (
EVENT_SHELLY_CLICK, EVENT_SHELLY_CLICK,
REST_SENSORS_UPDATE_INTERVAL, REST_SENSORS_UPDATE_INTERVAL,
) )
from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.core import HomeAssistant
from . import MOCK_MAC from . import MOCK_MAC
from tests.common import async_capture_events, async_mock_service from tests.common import async_capture_events
MOCK_SETTINGS = { MOCK_SETTINGS = {
"name": "Test name", "name": "Test name",
@ -290,12 +290,6 @@ def mock_ws_server():
yield yield
@pytest.fixture
def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service."""
return async_mock_service(hass, "test", "automation")
@pytest.fixture @pytest.fixture
def events(hass: HomeAssistant): def events(hass: HomeAssistant):
"""Yield caught shelly_click events.""" """Yield caught shelly_click events."""

View File

@ -178,7 +178,7 @@ async def test_get_triggers_for_invalid_device_id(
async def test_if_fires_on_click_event_block_device( async def test_if_fires_on_click_event_block_device(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_block_device: Mock, mock_block_device: Mock,
) -> None: ) -> None:
"""Test for click_event trigger firing for block device.""" """Test for click_event trigger firing for block device."""
@ -215,14 +215,14 @@ async def test_if_fires_on_click_event_block_device(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single_click" assert service_calls[0].data["some"] == "test_trigger_single_click"
async def test_if_fires_on_click_event_rpc_device( async def test_if_fires_on_click_event_rpc_device(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_rpc_device: Mock, mock_rpc_device: Mock,
) -> None: ) -> None:
"""Test for click_event trigger firing for rpc device.""" """Test for click_event trigger firing for rpc device."""
@ -259,14 +259,14 @@ async def test_if_fires_on_click_event_rpc_device(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single_push" assert service_calls[0].data["some"] == "test_trigger_single_push"
async def test_validate_trigger_block_device_not_ready( async def test_validate_trigger_block_device_not_ready(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_block_device: Mock, mock_block_device: Mock,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
) -> None: ) -> None:
@ -304,14 +304,14 @@ async def test_validate_trigger_block_device_not_ready(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single_click" assert service_calls[0].data["some"] == "test_trigger_single_click"
async def test_validate_trigger_rpc_device_not_ready( async def test_validate_trigger_rpc_device_not_ready(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_rpc_device: Mock, mock_rpc_device: Mock,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
) -> None: ) -> None:
@ -349,8 +349,8 @@ async def test_validate_trigger_rpc_device_not_ready(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single_push" assert service_calls[0].data["some"] == "test_trigger_single_push"
async def test_validate_trigger_invalid_triggers( async def test_validate_trigger_invalid_triggers(
@ -391,7 +391,7 @@ async def test_validate_trigger_invalid_triggers(
async def test_rpc_no_runtime_data( async def test_rpc_no_runtime_data(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_rpc_device: Mock, mock_rpc_device: Mock,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
) -> None: ) -> None:
@ -429,14 +429,14 @@ async def test_rpc_no_runtime_data(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single_push" assert service_calls[0].data["some"] == "test_trigger_single_push"
async def test_block_no_runtime_data( async def test_block_no_runtime_data(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
calls: list[ServiceCall], service_calls: list[ServiceCall],
mock_block_device: Mock, mock_block_device: Mock,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
) -> None: ) -> None:
@ -474,5 +474,5 @@ async def test_block_no_runtime_data(
hass.bus.async_fire(EVENT_SHELLY_CLICK, message) hass.bus.async_fire(EVENT_SHELLY_CLICK, message)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(calls) == 1 assert len(service_calls) == 1
assert calls[0].data["some"] == "test_trigger_single" assert service_calls[0].data["some"] == "test_trigger_single"