Switch tests to use hass objects instead of direct (#37530)

* Switch tests to use hass objects instead of direct

* Make sure sensor update state

* Add some initial binary sensor tests

* Add initial binary sensor tests

* Add tests for pt2262

* Add test for off delay
This commit is contained in:
Joakim Plate
2020-07-05 22:41:11 +02:00
committed by GitHub
parent 3ad59f877c
commit 01fd33f173
11 changed files with 640 additions and 548 deletions

View File

@@ -1,5 +1,4 @@
"""The tests for the Rfxtrx sensor platform."""
from homeassistant.components import rfxtrx as rfxtrx_core
from homeassistant.const import TEMP_CELSIUS, UNIT_PERCENTAGE
from homeassistant.setup import async_setup_component
@@ -13,7 +12,7 @@ async def test_default_config(hass, rfxtrx):
)
await hass.async_block_till_done()
assert 0 == len(rfxtrx_core.RFX_DEVICES)
assert len(hass.states.async_all()) == 0
async def test_one_sensor(hass, rfxtrx):
@@ -35,11 +34,11 @@ async def test_one_sensor(hass, rfxtrx):
)
await hass.async_block_till_done()
assert 1 == len(rfxtrx_core.RFX_DEVICES)
entity = rfxtrx_core.RFX_DEVICES["sensor_05_02"]["Temperature"]
assert "Test Temperature" == entity.name
assert TEMP_CELSIUS == entity.unit_of_measurement
assert entity.state is None
state = hass.states.get("sensor.test_temperature")
assert state
assert state.state == "unknown"
assert state.attributes.get("friendly_name") == "Test Temperature"
assert state.attributes.get("unit_of_measurement") == TEMP_CELSIUS
async def test_one_sensor_no_datatype(hass, rfxtrx):
@@ -56,11 +55,11 @@ async def test_one_sensor_no_datatype(hass, rfxtrx):
)
await hass.async_block_till_done()
assert 1 == len(rfxtrx_core.RFX_DEVICES)
entity = rfxtrx_core.RFX_DEVICES["sensor_05_02"]["Temperature"]
assert "Test Temperature" == entity.name
assert TEMP_CELSIUS == entity.unit_of_measurement
assert entity.state is None
state = hass.states.get("sensor.test_temperature")
assert state
assert state.state == "unknown"
assert state.attributes.get("friendly_name") == "Test Temperature"
assert state.attributes.get("unit_of_measurement") == TEMP_CELSIUS
async def test_several_sensors(hass, rfxtrx):
@@ -86,27 +85,23 @@ async def test_several_sensors(hass, rfxtrx):
)
await hass.async_block_till_done()
assert 2 == len(rfxtrx_core.RFX_DEVICES)
device_num = 0
for id in rfxtrx_core.RFX_DEVICES:
if id == "sensor_06_01":
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 UNIT_PERCENTAGE == _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_05_02":
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__()
state = hass.states.get("sensor.test_temperature")
assert state
assert state.state == "unknown"
assert state.attributes.get("friendly_name") == "Test Temperature"
assert state.attributes.get("unit_of_measurement") == TEMP_CELSIUS
assert 2 == device_num
state = hass.states.get("sensor.bath_temperature")
assert state
assert state.state == "unknown"
assert state.attributes.get("friendly_name") == "Bath Temperature"
assert state.attributes.get("unit_of_measurement") == TEMP_CELSIUS
state = hass.states.get("sensor.bath_humidity")
assert state
assert state.state == "unknown"
assert state.attributes.get("friendly_name") == "Bath Humidity"
assert state.attributes.get("unit_of_measurement") == UNIT_PERCENTAGE
async def test_discover_sensor(hass, rfxtrx):
@@ -118,59 +113,61 @@ async def test_discover_sensor(hass, rfxtrx):
)
await hass.async_block_till_done()
event = rfxtrx_core.get_rfx_object("0a520801070100b81b0279")
event.data = bytearray(b"\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y")
await _signal_event(hass, event)
await _signal_event(hass, "0a520801070100b81b0279")
state = hass.states.get("sensor.0a520801070100b81b0279_temperature")
assert state
assert state.state == "18.4"
assert (
state.attributes.items()
>= {
"friendly_name": "0a520801070100b81b0279 Temperature",
"unit_of_measurement": TEMP_CELSIUS,
"Humidity status": "normal",
"Temperature": 18.4,
"Rssi numeric": 7,
"Humidity": 27,
"Battery numeric": 9,
"Humidity status numeric": 2,
}.items()
)
entity = rfxtrx_core.RFX_DEVICES["sensor_07_01"]["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__()
await _signal_event(hass, "0a52080405020095240279")
state = hass.states.get("sensor.0a52080405020095240279_temperature")
assert state
assert state.state == "14.9"
assert (
state.attributes.items()
>= {
"friendly_name": "0a52080405020095240279 Temperature",
"unit_of_measurement": TEMP_CELSIUS,
"Humidity status": "normal",
"Temperature": 14.9,
"Rssi numeric": 7,
"Humidity": 36,
"Battery numeric": 9,
"Humidity status numeric": 2,
}.items()
)
await _signal_event(hass, event)
assert 1 == len(rfxtrx_core.RFX_DEVICES)
await _signal_event(hass, "0a52085e070100b31b0279")
state = hass.states.get("sensor.0a520801070100b81b0279_temperature")
assert state
assert state.state == "17.9"
assert (
state.attributes.items()
>= {
"friendly_name": "0a520801070100b81b0279 Temperature",
"unit_of_measurement": TEMP_CELSIUS,
"Humidity status": "normal",
"Temperature": 17.9,
"Rssi numeric": 7,
"Humidity": 27,
"Battery numeric": 9,
"Humidity status numeric": 2,
}.items()
)
event = rfxtrx_core.get_rfx_object("0a52080405020095240279")
event.data = bytearray(b"\nR\x08\x04\x05\x02\x00\x95$\x02y")
await _signal_event(hass, event)
entity = rfxtrx_core.RFX_DEVICES["sensor_05_02"]["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")
await _signal_event(hass, event)
entity = rfxtrx_core.RFX_DEVICES["sensor_07_01"]["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")
await _signal_event(hass, event)
assert 2 == len(rfxtrx_core.RFX_DEVICES)
assert len(hass.states.async_all()) == 2
async def test_discover_sensor_noautoadd(hass, rfxtrx):
@@ -182,25 +179,14 @@ async def test_discover_sensor_noautoadd(hass, rfxtrx):
)
await hass.async_block_till_done()
event = rfxtrx_core.get_rfx_object("0a520801070100b81b0279")
event.data = bytearray(b"\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y")
await _signal_event(hass, "0a520801070100b81b0279")
assert len(hass.states.async_all()) == 0
assert 0 == len(rfxtrx_core.RFX_DEVICES)
await _signal_event(hass, event)
assert 0 == len(rfxtrx_core.RFX_DEVICES)
await _signal_event(hass, "0a52080405020095240279")
assert len(hass.states.async_all()) == 0
await _signal_event(hass, 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")
await _signal_event(hass, 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")
await _signal_event(hass, event)
assert 0 == len(rfxtrx_core.RFX_DEVICES)
await _signal_event(hass, "0a52085e070100b31b0279")
assert len(hass.states.async_all()) == 0
async def test_update_of_sensors(hass, rfxtrx):
@@ -226,83 +212,93 @@ async def test_update_of_sensors(hass, rfxtrx):
)
await hass.async_block_till_done()
assert 2 == len(rfxtrx_core.RFX_DEVICES)
device_num = 0
for id in rfxtrx_core.RFX_DEVICES:
if id == "sensor_06_01":
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 UNIT_PERCENTAGE == _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_05_02":
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__()
state = hass.states.get("sensor.test_temperature")
assert state
assert state.state == "unknown"
assert (
state.attributes.items()
>= {
"friendly_name": "Test Temperature",
"unit_of_measurement": TEMP_CELSIUS,
}.items()
)
assert 2 == device_num
state = hass.states.get("sensor.bath_temperature")
assert state
assert state.state == "unknown"
assert (
state.attributes.items()
>= {
"friendly_name": "Bath Temperature",
"unit_of_measurement": TEMP_CELSIUS,
}.items()
)
event = rfxtrx_core.get_rfx_object("0a520802060101ff0f0269")
event.data = bytearray(b"\nR\x08\x01\x07\x01\x00\xb8\x1b\x02y")
await _signal_event(hass, event)
state = hass.states.get("sensor.bath_humidity")
assert state
assert state.state == "unknown"
assert (
state.attributes.items()
>= {
"friendly_name": "Bath Humidity",
"unit_of_measurement": UNIT_PERCENTAGE,
}.items()
)
await _signal_event(hass, event)
event = rfxtrx_core.get_rfx_object("0a52080705020085220269")
event.data = bytearray(b"\nR\x08\x04\x05\x02\x00\x95$\x02y")
await _signal_event(hass, event)
assert len(hass.states.async_all()) == 3
assert 2 == len(rfxtrx_core.RFX_DEVICES)
await _signal_event(hass, "0a520802060101ff0f0269")
await _signal_event(hass, "0a52080705020085220269")
device_num = 0
for id in rfxtrx_core.RFX_DEVICES:
if id == "sensor_06_01":
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 UNIT_PERCENTAGE == _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__()
state = hass.states.get("sensor.test_temperature")
assert state
assert state.state == "13.3"
assert (
state.attributes.items()
>= {
"friendly_name": "Test Temperature",
"unit_of_measurement": TEMP_CELSIUS,
"Battery numeric": 9,
"Temperature": 13.3,
"Humidity": 34,
"Humidity status": "normal",
"Humidity status numeric": 2,
"Rssi numeric": 6,
}.items()
)
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_05_02":
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__()
state = hass.states.get("sensor.bath_temperature")
assert state
assert state.state == "51.1"
assert (
state.attributes.items()
>= {
"friendly_name": "Bath Temperature",
"unit_of_measurement": TEMP_CELSIUS,
"Battery numeric": 9,
"Temperature": 51.1,
"Humidity": 15,
"Humidity status": "normal",
"Humidity status numeric": 2,
"Rssi numeric": 6,
}.items()
)
assert 2 == device_num
assert 2 == len(rfxtrx_core.RFX_DEVICES)
state = hass.states.get("sensor.bath_humidity")
assert state
assert state.state == "15"
assert (
state.attributes.items()
>= {
"friendly_name": "Bath Humidity",
"unit_of_measurement": UNIT_PERCENTAGE,
"Battery numeric": 9,
"Temperature": 51.1,
"Humidity": 15,
"Humidity status": "normal",
"Humidity status numeric": 2,
"Rssi numeric": 6,
}.items()
)
assert len(hass.states.async_all()) == 3