jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-20 11:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Include unit_of_measurement in sensor device trigger capabilities (#27265)

Browse Source 
* Expose unit_of_measurement in sensor device trigger

* Update test
This commit is contained in:
Erik Montnemery 2019-10-08 21:52:25 +02:00 committed by Paulus Schoutsen
parent 071476343c
commit 7f20210e93
2 changed files with 41 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
homeassistant/components/sensor/device_trigger.py
View File

@ -11,6 +11,7 @@ from homeassistant.const import (
CONF_ENTITY_ID, CONF_ENTITY_ID,
CONF_FOR, CONF_FOR,
CONF_TYPE, CONF_TYPE,
CONF_UNIT_OF_MEASUREMENT,
DEVICE_CLASS_BATTERY, DEVICE_CLASS_BATTERY,
DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_HUMIDITY,
DEVICE_CLASS_ILLUMINANCE, DEVICE_CLASS_ILLUMINANCE,
@ -149,11 +150,22 @@ async def async_get_triggers(hass, device_id):
async def async_get_trigger_capabilities(hass, config): async def async_get_trigger_capabilities(hass, config):
"""List trigger capabilities.""" """List trigger capabilities."""
state = hass.states.get(config[CONF_ENTITY_ID])
unit_of_measurement = (
state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) if state else ""
)
return { return {
"extra_fields": vol.Schema( "extra_fields": vol.Schema(
{ {
vol.Optional(CONF_ABOVE): vol.Coerce(float), vol.Optional(
vol.Optional(CONF_BELOW): vol.Coerce(float), CONF_ABOVE,
description={CONF_UNIT_OF_MEASUREMENT: unit_of_measurement},
): vol.Coerce(float),
vol.Optional(
CONF_BELOW,
description={CONF_UNIT_OF_MEASUREMENT: unit_of_measurement},
): vol.Coerce(float),
vol.Optional(CONF_FOR): cv.positive_time_period_dict, vol.Optional(CONF_FOR): cv.positive_time_period_dict,
} }
) )

31
tests/components/sensor/test_device_trigger.py
View File

@ -74,26 +74,49 @@ async def test_get_triggers(hass, device_reg, entity_reg):
if device_class != "none" if device_class != "none"
] ]
triggers = await async_get_device_automations(hass, "trigger", device_entry.id) triggers = await async_get_device_automations(hass, "trigger", device_entry.id)
assert len(triggers) == 8
assert triggers == expected_triggers assert triggers == expected_triggers
async def test_get_trigger_capabilities(hass, device_reg, entity_reg): async def test_get_trigger_capabilities(hass, device_reg, entity_reg):
"""Test we get the expected capabilities from a binary_sensor trigger.""" """Test we get the expected capabilities from a sensor trigger."""
platform = getattr(hass.components, f"test.{DOMAIN}")
platform.init()
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
device_entry = device_reg.async_get_or_create( device_entry = device_reg.async_get_or_create(
config_entry_id=config_entry.entry_id, config_entry_id=config_entry.entry_id,
connections={(device_registry.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(device_registry.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
) )
entity_reg.async_get_or_create(DOMAIN, "test", "5678", device_id=device_entry.id) entity_reg.async_get_or_create(
DOMAIN,
"test",
platform.ENTITIES["battery"].unique_id,
device_id=device_entry.id,
)
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {CONF_PLATFORM: "test"}})
expected_capabilities = { expected_capabilities = {
"extra_fields": [ "extra_fields": [
{"name": "above", "optional": True, "type": "float"}, {
{"name": "below", "optional": True, "type": "float"}, "description": {"unit_of_measurement": "%"},
"name": "above",
"optional": True,
"type": "float",
},
{
"description": {"unit_of_measurement": "%"},
"name": "below",
"optional": True,
"type": "float",
},
{"name": "for", "optional": True, "type": "positive_time_period_dict"}, {"name": "for", "optional": True, "type": "positive_time_period_dict"},
] ]
} }
triggers = await async_get_device_automations(hass, "trigger", device_entry.id) triggers = await async_get_device_automations(hass, "trigger", device_entry.id)
assert len(triggers) == 1
for trigger in triggers: for trigger in triggers:
capabilities = await async_get_device_automation_capabilities( capabilities = await async_get_device_automation_capabilities(
hass, "trigger", trigger hass, "trigger", trigger