mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 12:59:34 +00:00
Add new Dyson sensors (#8199)
* Add new Dyson sensors * Add unit of measurement for dust and air quality * Code review
This commit is contained in:
Charles Blonde
committed by
Paulus Schoutsen
Paulus Schoutsen
parent
5ae2bcdbb7
commit
222ad3ab6d
@@ -2,7 +2,7 @@
|
||||
import unittest
|
||||
from unittest import mock
|
||||
|
||||
from homeassistant.const import STATE_UNKNOWN
|
||||
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
|
||||
from homeassistant.components.sensor import dyson
|
||||
from tests.common import get_test_home_assistant
|
||||
|
||||
@@ -12,6 +12,7 @@ def _get_device_without_state():
|
||||
device = mock.Mock()
|
||||
device.name = "Device_name"
|
||||
device.state = None
|
||||
device.environmental_state = None
|
||||
return device
|
||||
|
||||
|
||||
@@ -21,6 +22,12 @@ def _get_with_state():
|
||||
device.name = "Device_name"
|
||||
device.state = mock.Mock()
|
||||
device.state.filter_life = 100
|
||||
device.environmental_state = mock.Mock()
|
||||
device.environmental_state.dust = 5
|
||||
device.environmental_state.humidity = 45
|
||||
device.environmental_state.temperature = 295
|
||||
device.environmental_state.volatil_organic_compounds = 2
|
||||
|
||||
return device
|
||||
|
||||
|
||||
@@ -45,27 +52,31 @@ class DysonTest(unittest.TestCase):
|
||||
def test_setup_component(self):
|
||||
"""Test setup component with devices."""
|
||||
def _add_device(devices):
|
||||
assert len(devices) == 1
|
||||
assert len(devices) == 5
|
||||
assert devices[0].name == "Device_name filter life"
|
||||
assert devices[1].name == "Device_name dust"
|
||||
assert devices[2].name == "Device_name humidity"
|
||||
assert devices[3].name == "Device_name temperature"
|
||||
assert devices[4].name == "Device_name air quality"
|
||||
|
||||
device = _get_device_without_state()
|
||||
self.hass.data[dyson.DYSON_DEVICES] = [device]
|
||||
dyson.setup_platform(self.hass, None, _add_device)
|
||||
|
||||
def test_dyson_filter_life_sensor(self):
|
||||
"""Test sensor with no value."""
|
||||
"""Test filter life sensor with no value."""
|
||||
sensor = dyson.DysonFilterLifeSensor(self.hass,
|
||||
_get_device_without_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, STATE_UNKNOWN)
|
||||
self.assertIsNone(sensor.state)
|
||||
self.assertEqual(sensor.unit_of_measurement, "hours")
|
||||
self.assertEqual(sensor.name, "Device_name filter life")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
sensor.on_message('message')
|
||||
|
||||
def test_dyson_filter_life_sensor_with_values(self):
|
||||
"""Test sensor with values."""
|
||||
"""Test filter sensor with values."""
|
||||
sensor = dyson.DysonFilterLifeSensor(self.hass, _get_with_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
@@ -74,3 +85,100 @@ class DysonTest(unittest.TestCase):
|
||||
self.assertEqual(sensor.name, "Device_name filter life")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
sensor.on_message('message')
|
||||
|
||||
def test_dyson_dust_sensor(self):
|
||||
"""Test dust sensor with no value."""
|
||||
sensor = dyson.DysonDustSensor(self.hass,
|
||||
_get_device_without_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertIsNone(sensor.state)
|
||||
self.assertEqual(sensor.unit_of_measurement, 'level')
|
||||
self.assertEqual(sensor.name, "Device_name dust")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_dust_sensor_with_values(self):
|
||||
"""Test dust sensor with values."""
|
||||
sensor = dyson.DysonDustSensor(self.hass, _get_with_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, 5)
|
||||
self.assertEqual(sensor.unit_of_measurement, 'level')
|
||||
self.assertEqual(sensor.name, "Device_name dust")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_humidity_sensor(self):
|
||||
"""Test humidity sensor with no value."""
|
||||
sensor = dyson.DysonHumiditySensor(self.hass,
|
||||
_get_device_without_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertIsNone(sensor.state)
|
||||
self.assertEqual(sensor.unit_of_measurement, '%')
|
||||
self.assertEqual(sensor.name, "Device_name humidity")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_humidity_sensor_with_values(self):
|
||||
"""Test humidity sensor with values."""
|
||||
sensor = dyson.DysonHumiditySensor(self.hass, _get_with_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, 45)
|
||||
self.assertEqual(sensor.unit_of_measurement, '%')
|
||||
self.assertEqual(sensor.name, "Device_name humidity")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_temperature_sensor(self):
|
||||
"""Test temperature sensor with no value."""
|
||||
sensor = dyson.DysonTemperatureSensor(self.hass,
|
||||
_get_device_without_state(),
|
||||
TEMP_CELSIUS)
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertIsNone(sensor.state)
|
||||
self.assertEqual(sensor.unit_of_measurement, '°C')
|
||||
self.assertEqual(sensor.name, "Device_name temperature")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_temperature_sensor_with_values(self):
|
||||
"""Test temperature sensor with values."""
|
||||
sensor = dyson.DysonTemperatureSensor(self.hass,
|
||||
_get_with_state(),
|
||||
TEMP_CELSIUS)
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, 21.9)
|
||||
self.assertEqual(sensor.unit_of_measurement, '°C')
|
||||
self.assertEqual(sensor.name, "Device_name temperature")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
sensor = dyson.DysonTemperatureSensor(self.hass,
|
||||
_get_with_state(),
|
||||
TEMP_FAHRENHEIT)
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, 71.3)
|
||||
self.assertEqual(sensor.unit_of_measurement, '°F')
|
||||
self.assertEqual(sensor.name, "Device_name temperature")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_air_quality_sensor(self):
|
||||
"""Test air quality sensor with no value."""
|
||||
sensor = dyson.DysonAirQualitySensor(self.hass,
|
||||
_get_device_without_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertIsNone(sensor.state)
|
||||
self.assertEqual(sensor.unit_of_measurement, 'level')
|
||||
self.assertEqual(sensor.name, "Device_name air quality")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
def test_dyson_air_quality_sensor_with_values(self):
|
||||
"""Test air quality sensor with values."""
|
||||
sensor = dyson.DysonAirQualitySensor(self.hass, _get_with_state())
|
||||
sensor.entity_id = "sensor.dyson_1"
|
||||
self.assertFalse(sensor.should_poll)
|
||||
self.assertEqual(sensor.state, 2)
|
||||
self.assertEqual(sensor.unit_of_measurement, 'level')
|
||||
self.assertEqual(sensor.name, "Device_name air quality")
|
||||
self.assertEqual(sensor.entity_id, "sensor.dyson_1")
|
||||
|
||||
Reference in New Issue
Block a user