core/tests/components/binary_sensor/common.py

20 lines
533 B
Python

"""Common test utilities for binary_sensor entity component tests."""
from homeassistant.components.binary_sensor import BinarySensorEntity
from tests.common import MockEntity
class MockBinarySensor(MockEntity, BinarySensorEntity):
"""Mock Binary Sensor class."""
@property
def is_on(self):
"""Return true if the binary sensor is on."""
return self._handle("is_on")
@property
def device_class(self):
"""Return the class of this sensor."""
return self._handle("device_class")