mirror of
https://github.com/home-assistant/core.git
synced 2025-07-16 01:37:08 +00:00
Add tests for Shelly sensor platform (#82385)
This commit is contained in:
parent
6de887b9b9
commit
8bd7f59b93
@ -1113,7 +1113,6 @@ omit =
|
||||
homeassistant/components/shelly/coordinator.py
|
||||
homeassistant/components/shelly/entity.py
|
||||
homeassistant/components/shelly/number.py
|
||||
homeassistant/components/shelly/sensor.py
|
||||
homeassistant/components/shelly/utils.py
|
||||
homeassistant/components/shiftr/*
|
||||
homeassistant/components/shodan/sensor.py
|
||||
|
@ -63,10 +63,16 @@ def mock_light_set_state(
|
||||
|
||||
MOCK_BLOCKS = [
|
||||
Mock(
|
||||
sensor_ids={"inputEvent": "S", "inputEventCnt": 2, "overpower": 0},
|
||||
sensor_ids={
|
||||
"inputEvent": "S",
|
||||
"inputEventCnt": 2,
|
||||
"overpower": 0,
|
||||
"power": 53.4,
|
||||
},
|
||||
channel="0",
|
||||
type="relay",
|
||||
overpower=0,
|
||||
power=53.4,
|
||||
description="relay_0",
|
||||
set_state=AsyncMock(side_effect=lambda turn: {"ison": turn == "on"}),
|
||||
),
|
||||
@ -91,8 +97,9 @@ MOCK_BLOCKS = [
|
||||
set_state=AsyncMock(side_effect=mock_light_set_state),
|
||||
),
|
||||
Mock(
|
||||
sensor_ids={"motion": 0},
|
||||
sensor_ids={"motion": 0, "temp": 22.1},
|
||||
motion=0,
|
||||
temp=22.1,
|
||||
description="sensor_0",
|
||||
type="sensor",
|
||||
),
|
||||
@ -138,6 +145,7 @@ MOCK_STATUS_COAP = {
|
||||
"old_version": "some_old_version",
|
||||
},
|
||||
"uptime": 5 * REST_SENSORS_UPDATE_INTERVAL,
|
||||
"wifi_sta": {"rssi": -64},
|
||||
}
|
||||
|
||||
|
||||
@ -150,6 +158,7 @@ MOCK_STATUS_RPC = {
|
||||
"current_pos": 50,
|
||||
"apower": 85.3,
|
||||
},
|
||||
"temperature:0": {"tC": 22.9},
|
||||
"sys": {
|
||||
"available_updates": {
|
||||
"beta": {"version": "some_beta_version"},
|
||||
|
161
tests/components/shelly/test_sensor.py
Normal file
161
tests/components/shelly/test_sensor.py
Normal file
@ -0,0 +1,161 @@
|
||||
"""Tests for Shelly sensor platform."""
|
||||
|
||||
|
||||
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
||||
from homeassistant.core import State
|
||||
|
||||
from . import (
|
||||
init_integration,
|
||||
mock_rest_update,
|
||||
mutate_rpc_device_status,
|
||||
register_device,
|
||||
register_entity,
|
||||
)
|
||||
|
||||
from tests.common import mock_restore_cache
|
||||
|
||||
RELAY_BLOCK_ID = 0
|
||||
SENSOR_BLOCK_ID = 3
|
||||
|
||||
|
||||
async def test_block_sensor(hass, mock_block_device, monkeypatch):
|
||||
"""Test block sensor."""
|
||||
entity_id = f"{SENSOR_DOMAIN}.test_name_channel_1_power"
|
||||
await init_integration(hass, 1)
|
||||
|
||||
assert hass.states.get(entity_id).state == "53.4"
|
||||
|
||||
monkeypatch.setattr(mock_block_device.blocks[RELAY_BLOCK_ID], "power", 60.1)
|
||||
mock_block_device.mock_update()
|
||||
|
||||
assert hass.states.get(entity_id).state == "60.1"
|
||||
|
||||
|
||||
async def test_block_rest_sensor(hass, mock_block_device, monkeypatch):
|
||||
"""Test block REST sensor."""
|
||||
entity_id = register_entity(hass, SENSOR_DOMAIN, "test_name_rssi", "rssi")
|
||||
await init_integration(hass, 1)
|
||||
|
||||
assert hass.states.get(entity_id).state == "-64"
|
||||
|
||||
monkeypatch.setitem(mock_block_device.status["wifi_sta"], "rssi", -71)
|
||||
await mock_rest_update(hass)
|
||||
|
||||
assert hass.states.get(entity_id).state == "-71"
|
||||
|
||||
|
||||
async def test_block_sleeping_sensor(hass, mock_block_device, monkeypatch):
|
||||
"""Test block sleeping sensor."""
|
||||
entity_id = f"{SENSOR_DOMAIN}.test_name_temperature"
|
||||
await init_integration(hass, 1, sleep_period=1000)
|
||||
|
||||
# Sensor should be created when device is online
|
||||
assert hass.states.get(entity_id) is None
|
||||
|
||||
# Make device online
|
||||
mock_block_device.mock_update()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "22.1"
|
||||
|
||||
monkeypatch.setattr(mock_block_device.blocks[SENSOR_BLOCK_ID], "temp", 23.4)
|
||||
mock_block_device.mock_update()
|
||||
|
||||
assert hass.states.get(entity_id).state == "23.4"
|
||||
|
||||
|
||||
async def test_block_restored_sleeping_sensor(
|
||||
hass, mock_block_device, device_reg, monkeypatch
|
||||
):
|
||||
"""Test block restored sleeping sensor."""
|
||||
entry = await init_integration(hass, 1, sleep_period=1000, skip_setup=True)
|
||||
register_device(device_reg, entry)
|
||||
entity_id = register_entity(
|
||||
hass, SENSOR_DOMAIN, "test_name_temperature", "sensor_0-temp", entry
|
||||
)
|
||||
mock_restore_cache(hass, [State(entity_id, "20.4")])
|
||||
monkeypatch.setattr(mock_block_device, "initialized", False)
|
||||
await hass.config_entries.async_setup(entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "20.4"
|
||||
|
||||
# Make device online
|
||||
monkeypatch.setattr(mock_block_device, "initialized", True)
|
||||
mock_block_device.mock_update()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "22.1"
|
||||
|
||||
|
||||
async def test_rpc_sensor(hass, mock_rpc_device, monkeypatch) -> None:
|
||||
"""Test RPC sensor."""
|
||||
entity_id = f"{SENSOR_DOMAIN}.test_cover_0_power"
|
||||
await init_integration(hass, 2)
|
||||
|
||||
assert hass.states.get(entity_id).state == "85.3"
|
||||
|
||||
mutate_rpc_device_status(monkeypatch, mock_rpc_device, "cover:0", "apower", "88.2")
|
||||
mock_rpc_device.mock_update()
|
||||
|
||||
assert hass.states.get(entity_id).state == "88.2"
|
||||
|
||||
|
||||
async def test_rpc_sleeping_sensor(
|
||||
hass, mock_rpc_device, device_reg, monkeypatch
|
||||
) -> None:
|
||||
"""Test RPC online sleeping sensor."""
|
||||
entity_id = f"{SENSOR_DOMAIN}.test_name_temperature"
|
||||
entry = await init_integration(hass, 2, sleep_period=1000)
|
||||
|
||||
# Sensor should be created when device is online
|
||||
assert hass.states.get(entity_id) is None
|
||||
|
||||
register_entity(
|
||||
hass,
|
||||
SENSOR_DOMAIN,
|
||||
"test_name_temperature",
|
||||
"temperature:0-temperature_0",
|
||||
entry,
|
||||
)
|
||||
|
||||
# Make device online
|
||||
mock_rpc_device.mock_update()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "22.9"
|
||||
|
||||
mutate_rpc_device_status(monkeypatch, mock_rpc_device, "temperature:0", "tC", 23.4)
|
||||
mock_rpc_device.mock_update()
|
||||
|
||||
assert hass.states.get(entity_id).state == "23.4"
|
||||
|
||||
|
||||
async def test_rpc_restored_sleeping_sensor(
|
||||
hass, mock_rpc_device, device_reg, monkeypatch
|
||||
):
|
||||
"""Test RPC restored sensor."""
|
||||
entry = await init_integration(hass, 2, sleep_period=1000, skip_setup=True)
|
||||
register_device(device_reg, entry)
|
||||
entity_id = register_entity(
|
||||
hass,
|
||||
SENSOR_DOMAIN,
|
||||
"test_name_temperature",
|
||||
"temperature:0-temperature_0",
|
||||
entry,
|
||||
)
|
||||
|
||||
mock_restore_cache(hass, [State(entity_id, "21.0")])
|
||||
monkeypatch.setattr(mock_rpc_device, "initialized", False)
|
||||
|
||||
await hass.config_entries.async_setup(entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "21.0"
|
||||
|
||||
# Make device online
|
||||
monkeypatch.setattr(mock_rpc_device, "initialized", True)
|
||||
mock_rpc_device.mock_update()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get(entity_id).state == "22.9"
|
Loading…
x
Reference in New Issue
Block a user