From c59d45caa35afd7a69165c6c7563dd4ecbfb8417 Mon Sep 17 00:00:00 2001
From: Nick Horvath <nhorvath@gmail.com>
Date: Sun, 24 Mar 2019 21:13:06 -0400
Subject: [PATCH] Expose detailed Ecobee equipment status (#20767)

* ecobee: expose detailed equipment status

* Fix #18244 for ecobee by moving current_operation property to current_operation_mode which is more accurate and defining current_operation properly, thanks @ZetaPhoenix

* fix docstring and lint issue

* Revert "fix docstring and lint issue"

This reverts commit d3a645f075f8a4017756f5eae05f00f05e2556cf.

* Revert "Fix #18244 for ecobee by moving current_operation property to current_operation_mode which is more accurate and defining current_operation properly, thanks @ZetaPhoenix"

This reverts commit bfd90551ef759b2c160d69da0778df89251e156b.
---
 homeassistant/components/ecobee/climate.py |  1 +
 tests/components/ecobee/test_climate.py    | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/homeassistant/components/ecobee/climate.py b/homeassistant/components/ecobee/climate.py
index bfc67e7cfaf..44a3800afa9 100644
--- a/homeassistant/components/ecobee/climate.py
+++ b/homeassistant/components/ecobee/climate.py
@@ -279,6 +279,7 @@ class Thermostat(ClimateDevice):
             "fan": self.fan,
             "climate_mode": self.mode,
             "operation": operation,
+            "equipment_running": status,
             "climate_list": self.climate_list,
             "fan_min_on_time": self.fan_min_on_time
         }
diff --git a/tests/components/ecobee/test_climate.py b/tests/components/ecobee/test_climate.py
index f4412b5d564..3215a9d5b4c 100644
--- a/tests/components/ecobee/test_climate.py
+++ b/tests/components/ecobee/test_climate.py
@@ -183,7 +183,8 @@ class TestEcobee(unittest.TestCase):
                 'fan': 'off',
                 'fan_min_on_time': 10,
                 'climate_mode': 'Climate1',
-                'operation': 'heat'} == \
+                'operation': 'heat',
+                'equipment_running': 'heatPump2'} == \
             self.thermostat.device_state_attributes
 
         self.ecobee['equipmentStatus'] = 'auxHeat2'
@@ -192,7 +193,8 @@ class TestEcobee(unittest.TestCase):
                 'fan': 'off',
                 'fan_min_on_time': 10,
                 'climate_mode': 'Climate1',
-                'operation': 'heat'} == \
+                'operation': 'heat',
+                'equipment_running': 'auxHeat2'} == \
             self.thermostat.device_state_attributes
         self.ecobee['equipmentStatus'] = 'compCool1'
         assert {'actual_humidity': 15,
@@ -200,7 +202,8 @@ class TestEcobee(unittest.TestCase):
                 'fan': 'off',
                 'fan_min_on_time': 10,
                 'climate_mode': 'Climate1',
-                'operation': 'cool'} == \
+                'operation': 'cool',
+                'equipment_running': 'compCool1'} == \
             self.thermostat.device_state_attributes
         self.ecobee['equipmentStatus'] = ''
         assert {'actual_humidity': 15,
@@ -208,7 +211,8 @@ class TestEcobee(unittest.TestCase):
                 'fan': 'off',
                 'fan_min_on_time': 10,
                 'climate_mode': 'Climate1',
-                'operation': 'idle'} == \
+                'operation': 'idle',
+                'equipment_running': ''} == \
             self.thermostat.device_state_attributes
 
         self.ecobee['equipmentStatus'] = 'Unknown'
@@ -217,7 +221,8 @@ class TestEcobee(unittest.TestCase):
                 'fan': 'off',
                 'fan_min_on_time': 10,
                 'climate_mode': 'Climate1',
-                'operation': 'Unknown'} == \
+                'operation': 'Unknown',
+                'equipment_running': 'Unknown'} == \
             self.thermostat.device_state_attributes
 
     def test_is_away_mode_on(self):