add tests

home-assistant · Oct 22, 2017 · e233573 · e233573
1 parent 4732a0c
commit e233573
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/tests/components/google_assistant/test_smart_home.py b/tests/components/google_assistant/test_smart_home.py
@@ -15,6 +15,57 @@
         const.SERVICE_TURN_ON,
         {'entity_id': 'light.test', 'brightness': 242}
     )
+}, {  # Test light color temperature
+    'entity_id': 'light.test',
+    'command': ga.const.COMMAND_COLOR,
+    'params': {
+        'color': {
+            'temperature': 2300,
+            'name': 'warm white'
+        }
+    },
+    'expected': (
+        const.SERVICE_TURN_ON,
+        {'entity_id': 'light.test', 'kelvin': 2300}
+    )
+}, {  # Test light color blue
+    'entity_id': 'light.test',
+    'command': ga.const.COMMAND_COLOR,
+    'params': {
+        'color': {
+            'spectrumRGB': 255,
+            'name': 'blue'
+        }
+    },
+    'expected': (
+        const.SERVICE_TURN_ON,
+        {'entity_id': 'light.test', 'rgb_color': [0, 0, 255]}
+    )
+}, {  # Test light color yellow
+    'entity_id': 'light.test',
+    'command': ga.const.COMMAND_COLOR,
+    'params': {
+        'color': {
+            'spectrumRGB': 16776960,
+            'name': 'yellow'
+        }
+    },
+    'expected': (
+        const.SERVICE_TURN_ON,
+        {'entity_id': 'light.test', 'rgb_color': [255, 255, 0]}
+    )
+}, {  # Test unhandled action/service
+    'entity_id': 'light.test',
+    'command': ga.const.COMMAND_COLOR,
+    'params': {
+        'color': {
+            'unhandled': 2300
+        }
+    },
+    'expected': (
+        None,
+        {'entity_id': 'light.test'}
+    )
 }, {  # Test light on / off
     'entity_id': 'light.test',
     'command': ga.const.COMMAND_ONOFF,