Add device_class information to onewire sensors (#42308)
This commit is contained in:
epenet
GitHub
parent
d13402108b
commit
5db1474099
3 changed files with 87 additions and 33 deletions
|
|
@ -9,6 +9,12 @@ from homeassistant.components.onewire.const import (
|
|||
)
|
||||
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
||||
from homeassistant.const import (
|
||||
DEVICE_CLASS_CURRENT,
|
||||
DEVICE_CLASS_HUMIDITY,
|
||||
DEVICE_CLASS_ILLUMINANCE,
|
||||
DEVICE_CLASS_PRESSURE,
|
||||
DEVICE_CLASS_TEMPERATURE,
|
||||
DEVICE_CLASS_VOLTAGE,
|
||||
ELECTRICAL_CURRENT_AMPERE,
|
||||
LIGHT_LUX,
|
||||
PERCENTAGE,
|
||||
|
|
@ -22,7 +28,7 @@ from tests.async_mock import patch
|
|||
from tests.common import mock_registry
|
||||
|
||||
MOCK_CONFIG = {
|
||||
"sensor": {
|
||||
SENSOR_DOMAIN: {
|
||||
"platform": DOMAIN,
|
||||
"host": "localhost",
|
||||
"port": DEFAULT_OWSERVER_PORT,
|
||||
|
|
@ -42,6 +48,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 25.123",
|
||||
"result": "25.1",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -53,6 +60,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 25.123",
|
||||
"result": "25.1",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.12_111111111111_pressure",
|
||||
|
|
@ -60,6 +68,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 1025.123",
|
||||
"result": "1025.1",
|
||||
"unit": PRESSURE_MBAR,
|
||||
"class": DEVICE_CLASS_PRESSURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -71,6 +80,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 251123",
|
||||
"result": "251123",
|
||||
"unit": "count",
|
||||
"class": None,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.1d_111111111111_counter_b",
|
||||
|
|
@ -78,6 +88,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 248125",
|
||||
"result": "248125",
|
||||
"unit": "count",
|
||||
"class": None,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -89,6 +100,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": ProtocolError,
|
||||
"result": "unknown",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -100,6 +112,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 25.123",
|
||||
"result": "25.1",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_humidity",
|
||||
|
|
@ -107,6 +120,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 72.7563",
|
||||
"result": "72.8",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_humidity_hih3600",
|
||||
|
|
@ -114,6 +128,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 73.7563",
|
||||
"result": "73.8",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_humidity_hih4000",
|
||||
|
|
@ -121,6 +136,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 74.7563",
|
||||
"result": "74.8",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_humidity_hih5030",
|
||||
|
|
@ -128,6 +144,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 75.7563",
|
||||
"result": "75.8",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_humidity_htm1735",
|
||||
|
|
@ -135,6 +152,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": ProtocolError,
|
||||
"result": "unknown",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_pressure",
|
||||
|
|
@ -142,6 +160,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 969.265",
|
||||
"result": "969.3",
|
||||
"unit": PRESSURE_MBAR,
|
||||
"class": DEVICE_CLASS_PRESSURE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_illuminance",
|
||||
|
|
@ -149,6 +168,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 65.8839",
|
||||
"result": "65.9",
|
||||
"unit": LIGHT_LUX,
|
||||
"class": DEVICE_CLASS_ILLUMINANCE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_voltage_vad",
|
||||
|
|
@ -156,6 +176,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 2.97",
|
||||
"result": "3.0",
|
||||
"unit": VOLT,
|
||||
"class": DEVICE_CLASS_VOLTAGE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_voltage_vdd",
|
||||
|
|
@ -163,6 +184,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 4.74",
|
||||
"result": "4.7",
|
||||
"unit": VOLT,
|
||||
"class": DEVICE_CLASS_VOLTAGE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.26_111111111111_current",
|
||||
|
|
@ -170,6 +192,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 1",
|
||||
"result": "1.0",
|
||||
"unit": ELECTRICAL_CURRENT_AMPERE,
|
||||
"class": DEVICE_CLASS_CURRENT,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -181,6 +204,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 26.984",
|
||||
"result": "27.0",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -192,6 +216,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 28.243",
|
||||
"result": "28.2",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -203,6 +228,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 29.123",
|
||||
"result": "29.1",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
]
|
||||
},
|
||||
|
|
@ -217,6 +243,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 67.745",
|
||||
"result": "67.7",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.ef_111111111111_humidity_raw",
|
||||
|
|
@ -224,6 +251,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 65.541",
|
||||
"result": "65.5",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.ef_111111111111_temperature",
|
||||
|
|
@ -231,6 +259,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 25.123",
|
||||
"result": "25.1",
|
||||
"unit": TEMP_CELSIUS,
|
||||
"class": DEVICE_CLASS_TEMPERATURE,
|
||||
},
|
||||
],
|
||||
},
|
||||
|
|
@ -249,6 +278,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 41.745",
|
||||
"result": "41.7",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.ef_111111111112_wetness_1",
|
||||
|
|
@ -256,6 +286,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 42.541",
|
||||
"result": "42.5",
|
||||
"unit": PERCENTAGE,
|
||||
"class": DEVICE_CLASS_HUMIDITY,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.ef_111111111112_moisture_2",
|
||||
|
|
@ -263,6 +294,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 43.123",
|
||||
"result": "43.1",
|
||||
"unit": PRESSURE_CBAR,
|
||||
"class": DEVICE_CLASS_PRESSURE,
|
||||
},
|
||||
{
|
||||
"entity_id": "sensor.ef_111111111112_moisture_3",
|
||||
|
|
@ -270,6 +302,7 @@ MOCK_DEVICE_SENSORS = {
|
|||
"injected_value": b" 44.123",
|
||||
"result": "44.1",
|
||||
"unit": PRESSURE_CBAR,
|
||||
"class": DEVICE_CLASS_PRESSURE,
|
||||
},
|
||||
],
|
||||
},
|
||||
|
|
@ -308,5 +341,6 @@ async def test_owserver_setup_valid_device(hass, device_id):
|
|||
assert registry_entry is not None
|
||||
assert registry_entry.unique_id == expected_sensor["unique_id"]
|
||||
assert registry_entry.unit_of_measurement == expected_sensor["unit"]
|
||||
assert registry_entry.device_class == expected_sensor["class"]
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == expected_sensor["result"]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue