evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Restore missing device_class to template binary_sensor (#39805)

Browse source
This commit is contained in:
J. Nick Koston 2020-09-08 10:12:23 -05:00 • committed by GitHub
parent 0507ec9d8b
commit 4bbd69955b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
homeassistant/components/template/binary_sensor.py
View file

@ -194,3 +194,8 @@ class BinarySensorTemplate(TemplateEntity, BinarySensorEntity):
def is_on(self):
"""Return true if sensor is on."""
return self._state
@property
def device_class(self):
"""Return the sensor class of the binary sensor."""
return self._device_class

11
tests/components/template/test_binary_sensor.py
View file

@ -6,6 +6,7 @@ from unittest import mock
from homeassistant import setup
from homeassistant.const import (
ATTR_DEVICE_CLASS,
EVENT_HOMEASSISTANT_START,
STATE_OFF,
STATE_ON,
@ -411,7 +412,10 @@ async def test_available_without_availability_template(hass):
await hass.async_start()
await hass.async_block_till_done()
assert hass.states.get("binary_sensor.test").state != STATE_UNAVAILABLE
state = hass.states.get("binary_sensor.test")
assert state.state != STATE_UNAVAILABLE
assert state.attributes[ATTR_DEVICE_CLASS] == "motion"
async def test_availability_template(hass):
@ -443,7 +447,10 @@ async def test_availability_template(hass):
hass.states.async_set("sensor.test_state", STATE_ON)
await hass.async_block_till_done()
assert hass.states.get("binary_sensor.test").state != STATE_UNAVAILABLE
state = hass.states.get("binary_sensor.test")
assert state.state != STATE_UNAVAILABLE
assert state.attributes[ATTR_DEVICE_CLASS] == "motion"
async def test_invalid_attribute_template(hass, caplog):