297 lines
14 KiB
Python
297 lines
14 KiB
Python
"""The tests for the Rfxtrx sensor platform."""
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
from homeassistant.bootstrap 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
|
|
|
|
|
|
@pytest.mark.skipif("os.environ.get('RFXTRX') != 'RUN'")
|
|
class TestSensorRfxtrx(unittest.TestCase):
|
|
"""Test the Rfxtrx sensor platform."""
|
|
|
|
def setUp(self):
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
self.hass.config.components = ['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."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{}}}))
|
|
self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))
|
|
|
|
def test_old_config_sensor(self):
|
|
"""Test with 1 sensor."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'sensor_0502': {
|
|
'name': 'Test',
|
|
'packetid': '0a52080705020095220269',
|
|
'data_type': 'Temperature'}}}}))
|
|
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
self.assertEqual('Test', entity.name)
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual(None, entity.state)
|
|
|
|
def test_one_sensor(self):
|
|
"""Test with 1 sensor."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'}}}}))
|
|
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
self.assertEqual('Test', entity.name)
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual(None, entity.state)
|
|
|
|
def test_one_sensor_no_datatype(self):
|
|
"""Test with 1 sensor."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test'}}}}))
|
|
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
entity = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']
|
|
self.assertEqual('Test', entity.name)
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual(None, entity.state)
|
|
|
|
entity_id = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']\
|
|
.entity_id
|
|
entity = self.hass.states.get(entity_id)
|
|
self.assertEqual('Test', entity.name)
|
|
self.assertEqual('unknown', entity.state)
|
|
|
|
def test_several_sensors(self):
|
|
"""Test with 3 sensors."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'},
|
|
'0a520802060100ff0e0269': {
|
|
'name': 'Bath',
|
|
'data_type': ['Temperature', 'Humidity']
|
|
}}}}))
|
|
|
|
self.assertEqual(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
|
|
self.assertEqual(len(rfxtrx_core.RFX_DEVICES[id]), 2)
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
self.assertEqual('%', _entity_hum.unit_of_measurement)
|
|
self.assertEqual('Bath', _entity_hum.__str__())
|
|
self.assertEqual(None, _entity_hum.state)
|
|
self.assertEqual(TEMP_CELSIUS,
|
|
_entity_temp.unit_of_measurement)
|
|
self.assertEqual('Bath', _entity_temp.__str__())
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
self.assertEqual(None, entity.state)
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual('Test', entity.__str__())
|
|
|
|
self.assertEqual(2, device_num)
|
|
|
|
def test_discover_sensor(self):
|
|
"""Test with discovery of sensor."""
|
|
self.assertTrue(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']
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual({'Humidity status': 'normal',
|
|
'Temperature': 18.4,
|
|
'Rssi numeric': 7, 'Humidity': 27,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('0a520801070100b81b0279',
|
|
entity.__str__())
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(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']
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual({'Humidity status': 'normal',
|
|
'Temperature': 14.9,
|
|
'Rssi numeric': 7, 'Humidity': 36,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('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']
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual({'Humidity status': 'normal',
|
|
'Temperature': 17.9,
|
|
'Rssi numeric': 7, 'Humidity': 27,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('0a520801070100b81b0279',
|
|
entity.__str__())
|
|
|
|
# trying to add a switch
|
|
event = rfxtrx_core.get_rfx_object('0b1100cd0213c7f210010f70')
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|
|
|
|
def test_discover_sensor_noautoadd(self):
|
|
"""Test with discover of sensor when auto add is False."""
|
|
self.assertTrue(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')
|
|
|
|
self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(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)
|
|
self.assertEqual(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)
|
|
self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))
|
|
|
|
def test_update_of_sensors(self):
|
|
"""Test with 3 sensors."""
|
|
self.assertTrue(setup_component(self.hass, 'sensor', {
|
|
'sensor': {'platform': 'rfxtrx',
|
|
'devices':
|
|
{'0a52080705020095220269': {
|
|
'name': 'Test',
|
|
'data_type': 'Temperature'},
|
|
'0a520802060100ff0e0269': {
|
|
'name': 'Bath',
|
|
'data_type': ['Temperature', 'Humidity']
|
|
}}}}))
|
|
|
|
self.assertEqual(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
|
|
self.assertEqual(len(rfxtrx_core.RFX_DEVICES[id]), 2)
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
self.assertEqual('%', _entity_hum.unit_of_measurement)
|
|
self.assertEqual('Bath', _entity_hum.__str__())
|
|
self.assertEqual(None, _entity_temp.state)
|
|
self.assertEqual(TEMP_CELSIUS,
|
|
_entity_temp.unit_of_measurement)
|
|
self.assertEqual('Bath', _entity_temp.__str__())
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
self.assertEqual(None, entity.state)
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual('Test', entity.__str__())
|
|
|
|
self.assertEqual(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)
|
|
|
|
self.assertEqual(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
|
|
self.assertEqual(len(rfxtrx_core.RFX_DEVICES[id]), 2)
|
|
_entity_temp = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
_entity_hum = rfxtrx_core.RFX_DEVICES[id]['Humidity']
|
|
self.assertEqual('%', _entity_hum.unit_of_measurement)
|
|
self.assertEqual(15, _entity_hum.state)
|
|
self.assertEqual({'Battery numeric': 9, 'Temperature': 51.1,
|
|
'Humidity': 15, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6},
|
|
_entity_hum.device_state_attributes)
|
|
self.assertEqual('Bath', _entity_hum.__str__())
|
|
|
|
self.assertEqual(TEMP_CELSIUS,
|
|
_entity_temp.unit_of_measurement)
|
|
self.assertEqual(51.1, _entity_temp.state)
|
|
self.assertEqual({'Battery numeric': 9, 'Temperature': 51.1,
|
|
'Humidity': 15, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6},
|
|
_entity_temp.device_state_attributes)
|
|
self.assertEqual('Bath', _entity_temp.__str__())
|
|
elif id == 'sensor_0502':
|
|
device_num = device_num + 1
|
|
entity = rfxtrx_core.RFX_DEVICES[id]['Temperature']
|
|
self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
|
|
self.assertEqual(13.3, entity.state)
|
|
self.assertEqual({'Humidity status': 'normal',
|
|
'Temperature': 13.3,
|
|
'Rssi numeric': 6, 'Humidity': 34,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('Test', entity.__str__())
|
|
|
|
self.assertEqual(2, device_num)
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|