76537305e2
* Add logbook and device trigger platforms to Shelly Add `logbook` platform for describing “shelly.click” event Add `device_trigger` platform for adding automation based on click events: Example of logbook event: Shelly 'single' click event for Test I3 channel 3 was fired. (Test I3 is the name of the device) Example of automation triggers: First button triple clicked First button long clicked and then single clicked First button double clicked First button long clicked First button single clicked and then long clicked First button single clicked Second button triple clicked .. Second button single clicked * Fix codespell * Remove pylint added for debug * Add tests * Rebase * Fix Rebase & Apply PR review suggestions Fix tests after rebasing Use `INPUTS_EVENTS_DICT` for input triggers Apply PR suggestions
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
"""Test configuration for Shelly."""
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.shelly import ShellyDeviceWrapper
|
|
from homeassistant.components.shelly.const import (
|
|
COAP,
|
|
DATA_CONFIG_ENTRY,
|
|
DOMAIN,
|
|
EVENT_SHELLY_CLICK,
|
|
)
|
|
from homeassistant.core import callback as ha_callback
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry, async_mock_service, mock_device_registry
|
|
|
|
MOCK_SETTINGS = {
|
|
"name": "Test name",
|
|
"device": {
|
|
"mac": "test-mac",
|
|
"hostname": "test-host",
|
|
"type": "SHSW-25",
|
|
"num_outputs": 2,
|
|
},
|
|
"coiot": {"update_period": 15},
|
|
"fw": "20201124-092159/v1.9.0@57ac4ad8",
|
|
"relays": [{"btn_type": "momentary"}, {"btn_type": "toggle"}],
|
|
}
|
|
|
|
MOCK_BLOCKS = [
|
|
Mock(sensor_ids={"inputEvent": "S", "inputEventCnt": 2}, channel="0", type="relay")
|
|
]
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_coap():
|
|
"""Mock out coap."""
|
|
with patch("homeassistant.components.shelly.get_coap_context"):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def device_reg(hass):
|
|
"""Return an empty, loaded, registry."""
|
|
return mock_device_registry(hass)
|
|
|
|
|
|
@pytest.fixture
|
|
def calls(hass):
|
|
"""Track calls to a mock service."""
|
|
return async_mock_service(hass, "test", "automation")
|
|
|
|
|
|
@pytest.fixture
|
|
def events(hass):
|
|
"""Yield caught shelly_click events."""
|
|
ha_events = []
|
|
hass.bus.async_listen(EVENT_SHELLY_CLICK, ha_callback(ha_events.append))
|
|
yield ha_events
|
|
|
|
|
|
@pytest.fixture
|
|
async def coap_wrapper(hass):
|
|
"""Setups a coap wrapper with mocked device."""
|
|
await async_setup_component(hass, "shelly", {})
|
|
|
|
config_entry = MockConfigEntry(domain=DOMAIN, data={})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
device = Mock(blocks=MOCK_BLOCKS, settings=MOCK_SETTINGS)
|
|
|
|
hass.data[DOMAIN] = {DATA_CONFIG_ENTRY: {}}
|
|
hass.data[DOMAIN][DATA_CONFIG_ENTRY][config_entry.entry_id] = {}
|
|
wrapper = hass.data[DOMAIN][DATA_CONFIG_ENTRY][config_entry.entry_id][
|
|
COAP
|
|
] = ShellyDeviceWrapper(hass, config_entry, device)
|
|
|
|
await wrapper.async_setup()
|
|
|
|
return wrapper
|