add supported_features functionality

homeassistant/components/climate/nuheat.py

@@ -9,6 +9,10 @@ from datetime import timedelta
 
 from homeassistant.components.climate import (
     ClimateDevice,
+    SUPPORT_AWAY_MODE,
+    SUPPORT_HOLD_MODE,
+    SUPPORT_OPERATION_MODE,
+    SUPPORT_TARGET_TEMPERATURE,
     STATE_HEAT,
     STATE_IDLE)
 from homeassistant.components.nuheat import DATA_NUHEAT
@@ -39,6 +43,9 @@ SCHEDULE_HOLD = 3
 SCHEDULE_RUN = 1
 SCHEDULE_TEMPORARY_HOLD = 2
 
+SUPPORT_FLAGS = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_HOLD_MODE |
+                 SUPPORT_AWAY_MODE | SUPPORT_OPERATION_MODE)
+
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
     """Set up the NuHeat thermostat(s)."""
@@ -74,6 +81,11 @@ class NuHeatThermostat(ClimateDevice):
         """Return the icon to use in the frontend."""
         return ICON
 
+    @property
+    def supported_features(self):
+        """Return the list of supported features."""
+        return SUPPORT_FLAGS
+
     @property
     def temperature_unit(self):
         """Return the unit of measurement."""

tests/components/climate/test_nuheat.py

@@ -2,7 +2,13 @@
 import unittest
 from unittest.mock import PropertyMock, Mock, patch
 
-from homeassistant.components.climate import STATE_HEAT, STATE_IDLE
+from homeassistant.components.climate import (
+    SUPPORT_AWAY_MODE,
+    SUPPORT_HOLD_MODE,
+    SUPPORT_OPERATION_MODE,
+    SUPPORT_TARGET_TEMPERATURE,
+    STATE_HEAT,
+    STATE_IDLE)
 import homeassistant.components.climate.nuheat as nuheat
 from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
 
@@ -73,6 +79,12 @@ class TestNuHeat(unittest.TestCase):
         """Test name property."""
         self.assertEqual(self.thermostat.icon, "mdi:thermometer")
 
+    def test_supported_features(self):
+        """Test name property."""
+        features = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_HOLD_MODE |
+                 SUPPORT_AWAY_MODE | SUPPORT_OPERATION_MODE)
+        self.assertEqual(self.thermostat.supported_features, features)
+
     def test_temperature_unit(self):
         """Test temperature unit."""
         self.assertEqual(self.thermostat.temperature_unit, TEMP_FAHRENHEIT)