226 lines
9.3 KiB
Python
226 lines
9.3 KiB
Python
"""The tests for the Rfxtrx sensor platform."""
|
|
import unittest
|
|
|
|
|
|
from homeassistant.components.sensor import rfxtrx
|
|
from homeassistant.components import rfxtrx as rfxtrx_core
|
|
from homeassistant.const import TEMP_CELCIUS
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
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(0)
|
|
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS = []
|
|
rfxtrx_core.RFX_DEVICES = {}
|
|
self.hass.stop()
|
|
|
|
def test_default_config(self):
|
|
"""Test with 0 sensor."""
|
|
config = {'devices': {}}
|
|
devices = []
|
|
|
|
def add_dev_callback(devs):
|
|
"""Add a callback to add devices."""
|
|
for dev in devs:
|
|
devices.append(dev)
|
|
|
|
rfxtrx.setup_platform(self.hass, config, add_dev_callback)
|
|
self.assertEqual(0, len(devices))
|
|
|
|
def test_one_sensor(self):
|
|
"""Test with 1 sensor."""
|
|
config = {'devices':
|
|
{'sensor_0502': {
|
|
'name': 'Test',
|
|
'packetid': '0a52080705020095220269',
|
|
'data_type': 'Temperature'}}}
|
|
devices = []
|
|
|
|
def add_dev_callback(devs):
|
|
"""Add a callback to add devices."""
|
|
for dev in devs:
|
|
devices.append(dev)
|
|
|
|
rfxtrx.setup_platform(self.hass, config, add_dev_callback)
|
|
|
|
self.assertEqual(1, len(devices))
|
|
entity = devices[0]
|
|
self.assertEqual('Test', entity.name)
|
|
self.assertEqual(TEMP_CELCIUS, entity.unit_of_measurement)
|
|
self.assertEqual(14.9, entity.state)
|
|
self.assertEqual({'Humidity status': 'normal', 'Temperature': 14.9,
|
|
'Rssi numeric': 6, 'Humidity': 34,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
|
|
def test_several_sensors(self):
|
|
"""Test with 3 sensors."""
|
|
config = {'devices':
|
|
{'sensor_0502': {
|
|
'name': 'Test',
|
|
'packetid': '0a52080705020095220269',
|
|
'data_type': 'Temperature'},
|
|
'sensor_0601': {
|
|
'name': 'Bath_Humidity',
|
|
'packetid': '0a520802060100ff0e0269',
|
|
'data_type': 'Humidity'},
|
|
'sensor_0601 2': {
|
|
'name': 'Bath',
|
|
'packetid': '0a520802060100ff0e0269'}}}
|
|
devices = []
|
|
|
|
def add_dev_callback(devs):
|
|
"""Add a callback to add devices."""
|
|
for dev in devs:
|
|
devices.append(dev)
|
|
|
|
rfxtrx.setup_platform(self.hass, config, add_dev_callback)
|
|
|
|
self.assertEqual(3, len(devices))
|
|
device_num = 0
|
|
for entity in devices:
|
|
if entity.name == 'Bath_Humidity':
|
|
device_num = device_num + 1
|
|
self.assertEqual('%', entity.unit_of_measurement)
|
|
self.assertEqual(14, entity.state)
|
|
self.assertEqual({'Battery numeric': 9, 'Temperature': 25.5,
|
|
'Humidity': 14, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('Bath_Humidity', entity.__str__())
|
|
elif entity.name == 'Bath':
|
|
device_num = device_num + 1
|
|
self.assertEqual(TEMP_CELCIUS, entity.unit_of_measurement)
|
|
self.assertEqual(25.5, entity.state)
|
|
self.assertEqual({'Battery numeric': 9, 'Temperature': 25.5,
|
|
'Humidity': 14, 'Humidity status': 'normal',
|
|
'Humidity status numeric': 2,
|
|
'Rssi numeric': 6},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('Bath', entity.__str__())
|
|
elif entity.name == 'Test':
|
|
device_num = device_num + 1
|
|
self.assertEqual(TEMP_CELCIUS, entity.unit_of_measurement)
|
|
self.assertEqual(14.9, entity.state)
|
|
self.assertEqual({'Humidity status': 'normal',
|
|
'Temperature': 14.9,
|
|
'Rssi numeric': 6, 'Humidity': 34,
|
|
'Battery numeric': 9,
|
|
'Humidity status numeric': 2},
|
|
entity.device_state_attributes)
|
|
self.assertEqual('Test', entity.__str__())
|
|
|
|
self.assertEqual(3, device_num)
|
|
|
|
def test_discover_sensor(self):
|
|
"""Test with discovery of sensor."""
|
|
config = {'devices': {}}
|
|
devices = []
|
|
|
|
def add_dev_callback(devs):
|
|
"""Add a callback to add devices."""
|
|
for dev in devs:
|
|
devices.append(dev)
|
|
|
|
rfxtrx.setup_platform(self.hass, config, add_dev_callback)
|
|
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 = devices[0]
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual(1, len(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('sensor_0701 : 0a520801070100b81b0279',
|
|
entity.__str__())
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(1, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual(1, len(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 = devices[1]
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual(2, len(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('sensor_0502 : 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 = devices[0]
|
|
self.assertEqual(2, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual(2, len(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('sensor_0701 : 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))
|
|
self.assertEqual(2, len(devices))
|
|
|
|
def test_discover_sensor_noautoadd(self):
|
|
"""Test with discover of sensor when auto add is False."""
|
|
config = {'automatic_add': False, 'devices': {}}
|
|
devices = []
|
|
|
|
def add_dev_callback(devs):
|
|
"""Add a callback to add devices."""
|
|
for dev in devs:
|
|
devices.append(dev)
|
|
|
|
rfxtrx.setup_platform(self.hass, config, add_dev_callback)
|
|
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))
|
|
self.assertEqual(0, len(devices))
|
|
|
|
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS[0](event)
|
|
self.assertEqual(0, len(rfxtrx_core.RFX_DEVICES))
|
|
self.assertEqual(0, len(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))
|
|
self.assertEqual(0, len(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))
|
|
self.assertEqual(0, len(devices))
|