137 lines
4.1 KiB
Python
137 lines
4.1 KiB
Python
"""Tests for Shelly button platform."""
|
|
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
|
|
from homeassistant.components.button.const import SERVICE_PRESS
|
|
from homeassistant.components.shelly.const import DOMAIN
|
|
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_registry import async_get
|
|
|
|
|
|
async def test_block_button(hass: HomeAssistant, coap_wrapper):
|
|
"""Test block device OTA button."""
|
|
assert coap_wrapper
|
|
|
|
entity_registry = async_get(hass)
|
|
entity_registry.async_get_or_create(
|
|
BUTTON_DOMAIN,
|
|
DOMAIN,
|
|
"test_name_ota_update_beta",
|
|
suggested_object_id="test_name_ota_update_beta",
|
|
disabled_by=None,
|
|
)
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(coap_wrapper.entry, BUTTON_DOMAIN)
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
# stable channel button
|
|
state = hass.states.get("button.test_name_ota_update")
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_ota_update"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert coap_wrapper.device.trigger_ota_update.call_count == 1
|
|
coap_wrapper.device.trigger_ota_update.assert_called_with(beta=False)
|
|
|
|
# beta channel button
|
|
state = hass.states.get("button.test_name_ota_update_beta")
|
|
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_ota_update_beta"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert coap_wrapper.device.trigger_ota_update.call_count == 2
|
|
coap_wrapper.device.trigger_ota_update.assert_called_with(beta=True)
|
|
|
|
# reboot button
|
|
state = hass.states.get("button.test_name_reboot")
|
|
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_reboot"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert coap_wrapper.device.trigger_reboot.call_count == 1
|
|
|
|
|
|
async def test_rpc_button(hass: HomeAssistant, rpc_wrapper):
|
|
"""Test rpc device OTA button."""
|
|
assert rpc_wrapper
|
|
|
|
entity_registry = async_get(hass)
|
|
entity_registry.async_get_or_create(
|
|
BUTTON_DOMAIN,
|
|
DOMAIN,
|
|
"test_name_ota_update_beta",
|
|
suggested_object_id="test_name_ota_update_beta",
|
|
disabled_by=None,
|
|
)
|
|
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(rpc_wrapper.entry, BUTTON_DOMAIN)
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
# stable channel button
|
|
state = hass.states.get("button.test_name_ota_update")
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_ota_update"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert rpc_wrapper.device.trigger_ota_update.call_count == 1
|
|
rpc_wrapper.device.trigger_ota_update.assert_called_with(beta=False)
|
|
|
|
# beta channel button
|
|
state = hass.states.get("button.test_name_ota_update_beta")
|
|
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_ota_update_beta"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert rpc_wrapper.device.trigger_ota_update.call_count == 2
|
|
rpc_wrapper.device.trigger_ota_update.assert_called_with(beta=True)
|
|
|
|
# reboot button
|
|
state = hass.states.get("button.test_name_reboot")
|
|
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.test_name_reboot"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert rpc_wrapper.device.trigger_reboot.call_count == 1
|