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

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 Paulus Schoutsen
parent fa07787007
commit f1de903fb5
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): def is_on(self):
"""Return true if sensor is on.""" """Return true if sensor is on."""
return self._state 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 import setup
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS,
EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_START,
STATE_OFF, STATE_OFF,
STATE_ON, STATE_ON,
@ -411,7 +412,10 @@ async def test_available_without_availability_template(hass):
await hass.async_start() await hass.async_start()
await hass.async_block_till_done() 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): 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) hass.states.async_set("sensor.test_state", STATE_ON)
await hass.async_block_till_done() 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): async def test_invalid_attribute_template(hass, caplog):