245 lines
6.7 KiB
Python
245 lines
6.7 KiB
Python
"""Test for RFlink sensor components.
|
|
|
|
Test setup of rflink sensor component/platform. Verify manual and
|
|
automatic sensor creation.
|
|
|
|
"""
|
|
|
|
from homeassistant.components.rflink import (
|
|
CONF_RECONNECT_INTERVAL, TMP_ENTITY, DATA_ENTITY_LOOKUP,
|
|
EVENT_KEY_COMMAND, EVENT_KEY_SENSOR)
|
|
from homeassistant.const import STATE_UNKNOWN
|
|
from ..test_rflink import mock_rflink
|
|
|
|
DOMAIN = 'sensor'
|
|
|
|
CONFIG = {
|
|
'rflink': {
|
|
'port': '/dev/ttyABC0',
|
|
'ignore_devices': ['ignore_wildcard_*', 'ignore_sensor'],
|
|
},
|
|
DOMAIN: {
|
|
'platform': 'rflink',
|
|
'devices': {
|
|
'test': {
|
|
'name': 'test',
|
|
'sensor_type': 'temperature',
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
async def test_default_setup(hass, monkeypatch):
|
|
"""Test all basic functionality of the rflink sensor component."""
|
|
# setup mocking rflink module
|
|
event_callback, create, _, _ = await mock_rflink(
|
|
hass, CONFIG, DOMAIN, monkeypatch)
|
|
|
|
# make sure arguments are passed
|
|
assert create.call_args_list[0][1]['ignore']
|
|
|
|
# test default state of sensor loaded from config
|
|
config_sensor = hass.states.get('sensor.test')
|
|
assert config_sensor
|
|
assert config_sensor.state == 'unknown'
|
|
assert config_sensor.attributes['unit_of_measurement'] == '°C'
|
|
|
|
# test event for config sensor
|
|
event_callback({
|
|
'id': 'test',
|
|
'sensor': 'temperature',
|
|
'value': 1,
|
|
'unit': '°C',
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get('sensor.test').state == '1'
|
|
|
|
# test event for new unconfigured sensor
|
|
event_callback({
|
|
'id': 'test2',
|
|
'sensor': 'temperature',
|
|
'value': 0,
|
|
'unit': '°C',
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
# test state of new sensor
|
|
new_sensor = hass.states.get('sensor.test2')
|
|
assert new_sensor
|
|
assert new_sensor.state == '0'
|
|
assert new_sensor.attributes['unit_of_measurement'] == '°C'
|
|
assert new_sensor.attributes['icon'] == 'mdi:thermometer'
|
|
|
|
|
|
async def test_disable_automatic_add(hass, monkeypatch):
|
|
"""If disabled new devices should not be automatically added."""
|
|
config = {
|
|
'rflink': {
|
|
'port': '/dev/ttyABC0',
|
|
},
|
|
DOMAIN: {
|
|
'platform': 'rflink',
|
|
'automatic_add': False,
|
|
},
|
|
}
|
|
|
|
# setup mocking rflink module
|
|
event_callback, _, _, _ = await mock_rflink(
|
|
hass, config, DOMAIN, monkeypatch)
|
|
|
|
# test event for new unconfigured sensor
|
|
event_callback({
|
|
'id': 'test2',
|
|
'sensor': 'temperature',
|
|
'value': 0,
|
|
'unit': '°C',
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
# make sure new device is not added
|
|
assert not hass.states.get('sensor.test2')
|
|
|
|
|
|
async def test_entity_availability(hass, monkeypatch):
|
|
"""If Rflink device is disconnected, entities should become unavailable."""
|
|
# Make sure Rflink mock does not 'recover' to quickly from the
|
|
# disconnect or else the unavailability cannot be measured
|
|
config = CONFIG
|
|
failures = [True, True]
|
|
config[CONF_RECONNECT_INTERVAL] = 60
|
|
|
|
# Create platform and entities
|
|
_, _, _, disconnect_callback = await mock_rflink(
|
|
hass, config, DOMAIN, monkeypatch, failures=failures)
|
|
|
|
# Entities are available by default
|
|
assert hass.states.get('sensor.test').state == STATE_UNKNOWN
|
|
|
|
# Mock a disconnect of the Rflink device
|
|
disconnect_callback()
|
|
|
|
# Wait for dispatch events to propagate
|
|
await hass.async_block_till_done()
|
|
|
|
# Entity should be unavailable
|
|
assert hass.states.get('sensor.test').state == 'unavailable'
|
|
|
|
# Reconnect the Rflink device
|
|
disconnect_callback()
|
|
|
|
# Wait for dispatch events to propagate
|
|
await hass.async_block_till_done()
|
|
|
|
# Entities should be available again
|
|
assert hass.states.get('sensor.test').state == STATE_UNKNOWN
|
|
|
|
|
|
async def test_aliasses(hass, monkeypatch):
|
|
"""Validate the response to sensor's alias (with aliasses)."""
|
|
config = {
|
|
'rflink': {
|
|
'port': '/dev/ttyABC0',
|
|
},
|
|
DOMAIN: {
|
|
'platform': 'rflink',
|
|
'devices': {
|
|
'test_02': {
|
|
'name': 'test_02',
|
|
'sensor_type': 'humidity',
|
|
'aliasses': ['test_alias_02_0'],
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
# setup mocking rflink module
|
|
event_callback, _, _, _ = await mock_rflink(
|
|
hass, config, DOMAIN, monkeypatch)
|
|
|
|
# test default state of sensor loaded from config
|
|
config_sensor = hass.states.get('sensor.test_02')
|
|
assert config_sensor
|
|
assert config_sensor.state == 'unknown'
|
|
|
|
# test event for config sensor
|
|
event_callback({
|
|
'id': 'test_alias_02_0',
|
|
'sensor': 'humidity',
|
|
'value': 65,
|
|
'unit': '%',
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
# test state of new sensor
|
|
updated_sensor = hass.states.get('sensor.test_02')
|
|
assert updated_sensor
|
|
assert updated_sensor.state == '65'
|
|
assert updated_sensor.attributes['unit_of_measurement'] == '%'
|
|
|
|
|
|
async def test_race_condition(hass, monkeypatch):
|
|
"""Test race condition for unknown components."""
|
|
config = {
|
|
'rflink': {
|
|
'port': '/dev/ttyABC0',
|
|
},
|
|
DOMAIN: {
|
|
'platform': 'rflink',
|
|
},
|
|
}
|
|
tmp_entity = TMP_ENTITY.format('test3')
|
|
|
|
# setup mocking rflink module
|
|
event_callback, _, _, _ = await mock_rflink(
|
|
hass, config, DOMAIN, monkeypatch)
|
|
|
|
# test event for new unconfigured sensor
|
|
event_callback({
|
|
'id': 'test3',
|
|
'sensor': 'battery',
|
|
'value': 'ok',
|
|
'unit': '',
|
|
})
|
|
event_callback({
|
|
'id': 'test3',
|
|
'sensor': 'battery',
|
|
'value': 'ko',
|
|
'unit': '',
|
|
})
|
|
|
|
# tmp_entity added to EVENT_KEY_SENSOR
|
|
assert tmp_entity in hass.data[DATA_ENTITY_LOOKUP][
|
|
EVENT_KEY_SENSOR]['test3']
|
|
# tmp_entity must no be added to EVENT_KEY_COMMAND
|
|
assert tmp_entity not in hass.data[DATA_ENTITY_LOOKUP][
|
|
EVENT_KEY_COMMAND]['test3']
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
# test state of new sensor
|
|
updated_sensor = hass.states.get('sensor.test3')
|
|
assert updated_sensor
|
|
|
|
# test state of new sensor
|
|
new_sensor = hass.states.get(DOMAIN+'.test3')
|
|
assert new_sensor
|
|
assert new_sensor.state == 'ok'
|
|
|
|
event_callback({
|
|
'id': 'test3',
|
|
'sensor': 'battery',
|
|
'value': 'ko',
|
|
'unit': '',
|
|
})
|
|
await hass.async_block_till_done()
|
|
# tmp_entity must be deleted from EVENT_KEY_COMMAND
|
|
assert tmp_entity not in hass.data[DATA_ENTITY_LOOKUP][
|
|
EVENT_KEY_SENSOR]['test3']
|
|
|
|
# test state of new sensor
|
|
new_sensor = hass.states.get(DOMAIN+'.test3')
|
|
assert new_sensor
|
|
assert new_sensor.state == 'ko'
|