Pressure entity should have valid unit_of_measurement to be supported

Author: dext0r

custom_components/yandex_smart_home/prop_float.py

@@ -181,9 +181,10 @@ class PressureProperty(FloatProperty):
     instance = const.FLOAT_INSTANCE_PRESSURE
 
     def supported(self) -> bool:
-        # TODO: check pressure unit
         if self.state.domain == sensor.DOMAIN:
-            return self.state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_PRESSURE
+            if self.state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_PRESSURE:
+                if self.state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) in PRESSURE_TO_PASCAL:
+                    return True
 
         return False
 

tests/test_prop_float.py

@@ -145,8 +145,16 @@ def test_property_float_pressure(hass, yandex_pressure_unit, v):
     state = State('sensor.test', '740', {
         ATTR_DEVICE_CLASS: DEVICE_CLASS_PRESSURE
     })
+    assert_no_properties(hass, config, state, PROPERTY_FLOAT, const.FLOAT_INSTANCE_PRESSURE)
 
+    state = State('sensor.test', '740', {
+        ATTR_DEVICE_CLASS: DEVICE_CLASS_PRESSURE,
+        ATTR_UNIT_OF_MEASUREMENT: const.PRESSURE_UNIT_MMHG
+    })
     prop = get_exact_one_property(hass, config, state, PROPERTY_FLOAT, const.FLOAT_INSTANCE_PRESSURE)
+    prop.state = State('sensor.test', '740', {
+        ATTR_DEVICE_CLASS: DEVICE_CLASS_PRESSURE
+    })
     with pytest.raises(SmartHomeError):
         prop.get_value()