mirror of
https://github.com/home-assistant/core.git
synced 2025-11-14 05:20:17 +00:00
Add Shelly tests coverage (#82642)
* Add Shelly tests coverage * Review comments * Remove leftovers
This commit is contained in:
Shay Levy
225
tests/components/shelly/test_utils.py
Normal file
225
tests/components/shelly/test_utils.py
Normal file
@@ -0,0 +1,225 @@
|
||||
"""Tests for Shelly utils."""
|
||||
from freezegun import freeze_time
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.shelly.utils import (
|
||||
get_block_channel_name,
|
||||
get_block_device_sleep_period,
|
||||
get_block_input_triggers,
|
||||
get_device_uptime,
|
||||
get_number_of_channels,
|
||||
get_rpc_channel_name,
|
||||
get_rpc_input_triggers,
|
||||
is_block_momentary_input,
|
||||
)
|
||||
from homeassistant.util import dt
|
||||
|
||||
DEVICE_BLOCK_ID = 4
|
||||
|
||||
|
||||
async def test_block_get_number_of_channels(mock_block_device, monkeypatch):
|
||||
"""Test block get number of channels."""
|
||||
monkeypatch.setattr(mock_block_device.blocks[DEVICE_BLOCK_ID], "type", "emeter")
|
||||
monkeypatch.setitem(mock_block_device.shelly, "num_emeters", 3)
|
||||
|
||||
assert (
|
||||
get_number_of_channels(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== 3
|
||||
)
|
||||
|
||||
monkeypatch.setitem(mock_block_device.shelly, "num_inputs", 4)
|
||||
monkeypatch.setattr(mock_block_device.blocks[DEVICE_BLOCK_ID], "type", "input")
|
||||
assert (
|
||||
get_number_of_channels(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== 4
|
||||
)
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHDM-2")
|
||||
assert (
|
||||
get_number_of_channels(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== 2
|
||||
)
|
||||
|
||||
|
||||
async def test_block_get_block_channel_name(mock_block_device, monkeypatch):
|
||||
"""Test block get block channel name."""
|
||||
monkeypatch.setattr(mock_block_device.blocks[DEVICE_BLOCK_ID], "type", "relay")
|
||||
|
||||
assert (
|
||||
get_block_channel_name(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== "Test name channel 1"
|
||||
)
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHEM-3")
|
||||
|
||||
assert (
|
||||
get_block_channel_name(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== "Test name channel A"
|
||||
)
|
||||
|
||||
monkeypatch.setitem(
|
||||
mock_block_device.settings, "relays", [{"name": "test-channel"}]
|
||||
)
|
||||
|
||||
assert (
|
||||
get_block_channel_name(
|
||||
mock_block_device,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
== "test-channel"
|
||||
)
|
||||
|
||||
|
||||
async def test_is_block_momentary_input(mock_block_device, monkeypatch):
|
||||
"""Test is block momentary input."""
|
||||
monkeypatch.setattr(mock_block_device.blocks[DEVICE_BLOCK_ID], "type", "relay")
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings, "mode", "roller")
|
||||
monkeypatch.setitem(
|
||||
mock_block_device.settings, "rollers", [{"button_type": "detached"}]
|
||||
)
|
||||
assert (
|
||||
is_block_momentary_input(
|
||||
mock_block_device.settings,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
is False
|
||||
)
|
||||
assert (
|
||||
is_block_momentary_input(
|
||||
mock_block_device.settings, mock_block_device.blocks[DEVICE_BLOCK_ID], True
|
||||
)
|
||||
is True
|
||||
)
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings, "mode", "relay")
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHSW-L")
|
||||
assert (
|
||||
is_block_momentary_input(
|
||||
mock_block_device.settings, mock_block_device.blocks[DEVICE_BLOCK_ID], True
|
||||
)
|
||||
is False
|
||||
)
|
||||
|
||||
monkeypatch.delitem(mock_block_device.settings, "relays")
|
||||
monkeypatch.delitem(mock_block_device.settings, "rollers")
|
||||
assert (
|
||||
is_block_momentary_input(
|
||||
mock_block_device.settings,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
is False
|
||||
)
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHBTN-2")
|
||||
|
||||
assert (
|
||||
is_block_momentary_input(
|
||||
mock_block_device.settings,
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
)
|
||||
is True
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"settings, sleep_period",
|
||||
[
|
||||
({}, 0),
|
||||
({"sleep_mode": {"period": 1000, "unit": "m"}}, 1000 * 60),
|
||||
({"sleep_mode": {"period": 5, "unit": "h"}}, 5 * 3600),
|
||||
],
|
||||
)
|
||||
async def test_get_block_device_sleep_period(settings, sleep_period):
|
||||
"""Test get block device sleep period."""
|
||||
assert get_block_device_sleep_period(settings) == sleep_period
|
||||
|
||||
|
||||
@freeze_time("2019-01-10 18:43:00+00:00")
|
||||
async def test_get_device_uptime():
|
||||
"""Test block test get device uptime."""
|
||||
assert get_device_uptime(
|
||||
55, dt.as_utc(dt.parse_datetime("2019-01-10 18:42:00+00:00"))
|
||||
) == dt.as_utc(dt.parse_datetime("2019-01-10 18:42:00+00:00"))
|
||||
|
||||
assert get_device_uptime(
|
||||
50, dt.as_utc(dt.parse_datetime("2019-01-10 18:42:00+00:00"))
|
||||
) == dt.as_utc(dt.parse_datetime("2019-01-10 18:42:10+00:00"))
|
||||
|
||||
|
||||
async def test_get_block_input_triggers(mock_block_device, monkeypatch):
|
||||
"""Test get block input triggers."""
|
||||
monkeypatch.setattr(
|
||||
mock_block_device.blocks[DEVICE_BLOCK_ID],
|
||||
"sensor_ids",
|
||||
{"inputEvent": "S", "inputEventCnt": 0},
|
||||
)
|
||||
monkeypatch.setitem(
|
||||
mock_block_device.settings, "rollers", [{"button_type": "detached"}]
|
||||
)
|
||||
assert set(
|
||||
get_block_input_triggers(
|
||||
mock_block_device, mock_block_device.blocks[DEVICE_BLOCK_ID]
|
||||
)
|
||||
) == {("long", "button"), ("single", "button")}
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHBTN-1")
|
||||
assert set(
|
||||
get_block_input_triggers(
|
||||
mock_block_device, mock_block_device.blocks[DEVICE_BLOCK_ID]
|
||||
)
|
||||
) == {
|
||||
("long", "button"),
|
||||
("double", "button"),
|
||||
("single", "button"),
|
||||
("triple", "button"),
|
||||
}
|
||||
|
||||
monkeypatch.setitem(mock_block_device.settings["device"], "type", "SHIX3-1")
|
||||
assert set(
|
||||
get_block_input_triggers(
|
||||
mock_block_device, mock_block_device.blocks[DEVICE_BLOCK_ID]
|
||||
)
|
||||
) == {
|
||||
("long_single", "button"),
|
||||
("single_long", "button"),
|
||||
("triple", "button"),
|
||||
("long", "button"),
|
||||
("single", "button"),
|
||||
("double", "button"),
|
||||
}
|
||||
|
||||
|
||||
async def test_get_rpc_channel_name(mock_rpc_device):
|
||||
"""Test get RPC channel name."""
|
||||
assert get_rpc_channel_name(mock_rpc_device, "input:0") == "test switch_0"
|
||||
|
||||
|
||||
async def test_get_rpc_input_triggers(mock_rpc_device, monkeypatch):
|
||||
"""Test get RPC input triggers."""
|
||||
monkeypatch.setattr(mock_rpc_device, "config", {"input:0": {"type": "button"}})
|
||||
assert set(get_rpc_input_triggers(mock_rpc_device)) == {
|
||||
("long_push", "button1"),
|
||||
("single_push", "button1"),
|
||||
("btn_down", "button1"),
|
||||
("double_push", "button1"),
|
||||
("btn_up", "button1"),
|
||||
}
|
||||
|
||||
monkeypatch.setattr(mock_rpc_device, "config", {"input:0": {"type": "switch"}})
|
||||
assert not get_rpc_input_triggers(mock_rpc_device)
|
||||
Reference in New Issue
Block a user