297 lines
13 KiB
Python
297 lines
13 KiB
Python
"""The tests for the Rfxtrx sensor platform."""
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
from homeassistant.setup import setup_component
|
|
from homeassistant.components import rfxtrx as rfxtrx_core
|
|
from homeassistant.const import TEMP_CELSIUS
|
|
|
|
from tests.common import get_test_home_assistant, mock_component
|
|
|
|
|
|
@pytest.mark.skipif("os.environ.get('RFXTRX') != 'RUN'")
|
|
class TestSensorRfxtrx(unittest.TestCase):
|
|
"""Test the Rfxtrx sensor platform."""
|
|
|
|
def setUp(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
mock_component(self.hass, 'rfxtrx')
|
|
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS = []
|
|
rfxtrx_core.RFX_DEVICES = {}
|
|
if rfxtrx_core.RFXOBJECT:
|
|
rfxtrx_core.RFXOBJECT.close_connection()
|
|
self.hass.stop()
|
|
|
|
def test_default_config(self):
|
|
"""Test with 0 sensor."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{}}})
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
def test_old_config_sensor(self):
|
|
"""Test with 1 sensor."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'sensor_0502': {
|
|
'name': 'Test',
|
|
'packetid': '0a52080705020095220269',
|
|
'data_type': 'Temperature'}}}})
|
|
|
|
assert 1 == len(rfxtrx_core.RFX_DEVICES)
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
assert 'Test' == entity.name
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert entity.state is None
|
|
|
|
def test_one_sensor(self):
|
|
"""Test with 1 sensor."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'}}}})
|
|
|
|
assert 1 == len(rfxtrx_core.RFX_DEVICES)
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
assert 'Test' == entity.name
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert entity.state is None
|
|
|
|
def test_one_sensor_no_datatype(self):
|
|
"""Test with 1 sensor."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test'}}}})
|
|
|
|
assert 1 == len(rfxtrx_core.RFX_DEVICES)
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
assert 'Test' == entity.name
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert entity.state is None
|
|
|
|
entity_id = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']\
|
|
.entity_id
|
|
entity = self.hass.states.get(entity_id)
|
|
assert 'Test' == entity.name
|
|
assert 'unknown' == entity.state
|
|
|
|
def test_several_sensors(self):
|
|
"""Test with 3 sensors."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'},
|
|
'0a520802060100ff0e0269': {
|
|
'name': 'Bath',
|
|
'data_type': ['Temperature', 'Humidity']
|
|
}}}})
|
|
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
device_num = 0
|
|
for id in rfxtrx_core.RFX_DEVICES:
|
|
if id == 'sensor_0601':
|
|
device_num = device_num + 1
|
|
assert len(rfxtrx_core.RFX_DEVICES[id]) == 2
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
assert '%' == _entity_hum.unit_of_measurement
|
|
assert 'Bath' == _entity_hum.__str__()
|
|
assert _entity_hum.state is None
|
|
assert TEMP_CELSIUS == \
|
|
_entity_temp.unit_of_measurement
|
|
assert 'Bath' == _entity_temp.__str__()
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
assert entity.state is None
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert 'Test' == entity.__str__()
|
|
|
|
assert 2 == device_num
|
|
|
|
def test_discover_sensor(self):
|
|
"""Test with discovery of sensor."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'automatic_add': True,
|
|
'devices': {}}})
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a520801070100b81b0279')
|
|
event.data = bytearray(b'\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0701']['Temperature']
|
|
assert 1 == len(rfxtrx_core.RFX_DEVICES)
|
|
assert {'Humidity status': 'normal',
|
|
'Temperature': 18.4,
|
|
'Rssi numeric': 7, 'Humidity': 27,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2} == \
|
|
entity.device_state_attributes
|
|
assert '0a520801070100b81b0279' == \
|
|
entity.__str__()
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 1 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a52080405020095240279')
|
|
event.data = bytearray(b'\nR\x08\x04\x05\x02\x00\x95$\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
assert {'Humidity status': 'normal',
|
|
'Temperature': 14.9,
|
|
'Rssi numeric': 7, 'Humidity': 36,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2} == \
|
|
entity.device_state_attributes
|
|
assert '0a52080405020095240279' == \
|
|
entity.__str__()
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a52085e070100b31b0279')
|
|
event.data = bytearray(b'\nR\x08^\x07\x01\x00\xb3\x1b\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0701']['Temperature']
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
assert {'Humidity status': 'normal',
|
|
'Temperature': 17.9,
|
|
'Rssi numeric': 7, 'Humidity': 27,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2} == \
|
|
entity.device_state_attributes
|
|
assert '0a520801070100b81b0279' == \
|
|
entity.__str__()
|
|
|
|
# trying to add a switch
|
|
event = rfxtrx_core.get_rfx_object('0b1100cd0213c7f210010f70')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
def test_discover_sensor_noautoadd(self):
|
|
"""Test with discover of sensor when auto add is False."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'automatic_add': False,
|
|
'devices': {}}})
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a520801070100b81b0279')
|
|
event.data = bytearray(b'\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y')
|
|
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a52080405020095240279')
|
|
event.data = bytearray(b'\nR\x08\x04\x05\x02\x00\x95$\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a52085e070100b31b0279')
|
|
event.data = bytearray(b'\nR\x08^\x07\x01\x00\xb3\x1b\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
assert 0 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
def test_update_of_sensors(self):
|
|
"""Test with 3 sensors."""
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'},
|
|
'0a520802060100ff0e0269': {
|
|
'name': 'Bath',
|
|
'data_type': ['Temperature', 'Humidity']
|
|
}}}})
|
|
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
device_num = 0
|
|
for id in rfxtrx_core.RFX_DEVICES:
|
|
if id == 'sensor_0601':
|
|
device_num = device_num + 1
|
|
assert len(rfxtrx_core.RFX_DEVICES[id]) == 2
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
assert '%' == _entity_hum.unit_of_measurement
|
|
assert 'Bath' == _entity_hum.__str__()
|
|
assert _entity_temp.state is None
|
|
assert TEMP_CELSIUS == \
|
|
_entity_temp.unit_of_measurement
|
|
assert 'Bath' == _entity_temp.__str__()
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
assert entity.state is None
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert 'Test' == entity.__str__()
|
|
|
|
assert 2 == device_num
|
|
|
|
event = rfxtrx_core.get_rfx_object('0a520802060101ff0f0269')
|
|
event.data = bytearray(b'\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
event = rfxtrx_core.get_rfx_object('0a52080705020085220269')
|
|
event.data = bytearray(b'\nR\x08\x04\x05\x02\x00\x95$\x02y')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|
|
|
|
device_num = 0
|
|
for id in rfxtrx_core.RFX_DEVICES:
|
|
if id == 'sensor_0601':
|
|
device_num = device_num + 1
|
|
assert len(rfxtrx_core.RFX_DEVICES[id]) == 2
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
assert '%' == _entity_hum.unit_of_measurement
|
|
assert 15 == _entity_hum.state
|
|
assert {'Battery numeric': 9, 'Temperature': 51.1,
|
|
'Humidity': 15, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6} == \
|
|
_entity_hum.device_state_attributes
|
|
assert 'Bath' == _entity_hum.__str__()
|
|
|
|
assert TEMP_CELSIUS == \
|
|
_entity_temp.unit_of_measurement
|
|
assert 51.1 == _entity_temp.state
|
|
assert {'Battery numeric': 9, 'Temperature': 51.1,
|
|
'Humidity': 15, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6} == \
|
|
_entity_temp.device_state_attributes
|
|
assert 'Bath' == _entity_temp.__str__()
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
assert TEMP_CELSIUS == entity.unit_of_measurement
|
|
assert 13.3 == entity.state
|
|
assert {'Humidity status': 'normal',
|
|
'Temperature': 13.3,
|
|
'Rssi numeric': 6, 'Humidity': 34,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2} == \
|
|
entity.device_state_attributes
|
|
assert 'Test' == entity.__str__()
|
|
|
|
assert 2 == device_num
|
|
assert 2 == len(rfxtrx_core.RFX_DEVICES)
|