From 244e2a0e7ee2efaddedfb663502cb804985c1de9 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <Paulus@PaulusSchoutsen.nl>
Date: Sun, 23 Nov 2014 22:18:40 -0800
Subject: [PATCH] Added initial Chromecast test coverage

---
 homeassistant/components/chromecast.py |  7 ++-
 test/helper.py                         | 15 +++++
 test/test_component_chromecast.py      | 87 ++++++++++++++++++++++++++
 3 files changed, 106 insertions(+), 3 deletions(-)
 create mode 100644 test/helper.py
 create mode 100644 test/test_component_chromecast.py

diff --git a/homeassistant/components/chromecast.py b/homeassistant/components/chromecast.py
index 1604e72f950..a72de83ce64 100644
--- a/homeassistant/components/chromecast.py
+++ b/homeassistant/components/chromecast.py
@@ -6,6 +6,7 @@ Provides functionality to interact with Chromecasts.
 """
 import logging
 
+import homeassistant as ha
 import homeassistant.util as util
 import homeassistant.components as components
 
@@ -113,8 +114,8 @@ def setup(hass, config):
 
         return False
 
-    if 'hosts' in config[DOMAIN]:
-        hosts = config[DOMAIN]['hosts'].split(",")
+    if ha.CONF_HOSTS in config[DOMAIN]:
+        hosts = config[DOMAIN][ha.CONF_HOSTS].split(",")
 
     # If no hosts given, scan for chromecasts
     else:
@@ -134,7 +135,7 @@ def setup(hass, config):
 
             casts[entity_id] = cast
 
-        except pychromecast.ChromecastConnectionError:
+        except ConnectionError:
             pass
 
     if not casts:
diff --git a/test/helper.py b/test/helper.py
new file mode 100644
index 00000000000..850510acdc1
--- /dev/null
+++ b/test/helper.py
@@ -0,0 +1,15 @@
+"""
+test.helper
+~~~~~~~~~~~
+
+Helper method for writing tests.
+"""
+
+
+def mock_service(hass, domain, service):
+    calls = []
+
+    hass.services.register(
+        domain, service, lambda call: calls.append(call))
+
+    return calls
diff --git a/test/test_component_chromecast.py b/test/test_component_chromecast.py
new file mode 100644
index 00000000000..27e5bbb0370
--- /dev/null
+++ b/test/test_component_chromecast.py
@@ -0,0 +1,87 @@
+"""
+test.test_component_chromecast
+~~~~~~~~~~~
+
+Tests Chromecast component.
+"""
+# pylint: disable=too-many-public-methods,protected-access
+import logging
+import unittest
+
+import homeassistant as ha
+import homeassistant.components as components
+import homeassistant.components.chromecast as chromecast
+from helper import mock_service
+
+
+def setUpModule():   # pylint: disable=invalid-name
+    """ Setup to ignore chromecast errors. """
+    logging.disable(logging.CRITICAL)
+
+
+class TestChromecast(unittest.TestCase):
+    """ Test the chromecast module. """
+
+    def setUp(self):  # pylint: disable=invalid-name
+        self.hass = ha.HomeAssistant()
+
+        self.test_entity = chromecast.ENTITY_ID_FORMAT.format('living_room')
+        self.hass.states.set(self.test_entity, chromecast.STATE_NO_APP)
+
+        self.test_entity2 = chromecast.ENTITY_ID_FORMAT.format('bedroom')
+        self.hass.states.set(self.test_entity2, "Youtube")
+
+    def tearDown(self):  # pylint: disable=invalid-name
+        """ Stop down stuff we started. """
+        self.hass._pool.stop()
+
+    def test_is_on(self):
+        """ Test is_on method. """
+        self.assertFalse(chromecast.is_on(self.hass, self.test_entity))
+        self.assertTrue(chromecast.is_on(self.hass, self.test_entity2))
+
+    def test_services(self):
+        """
+        Test if the call service methods conver to correct service calls.
+        """
+        services = {
+            components.SERVICE_TURN_OFF: chromecast.turn_off,
+            components.SERVICE_VOLUME_UP: chromecast.volume_up,
+            components.SERVICE_VOLUME_DOWN: chromecast.volume_down,
+            components.SERVICE_MEDIA_PLAY_PAUSE: chromecast.media_play_pause,
+            components.SERVICE_MEDIA_PLAY: chromecast.media_play,
+            components.SERVICE_MEDIA_PAUSE: chromecast.media_pause,
+            components.SERVICE_MEDIA_NEXT_TRACK: chromecast.media_next_track,
+            components.SERVICE_MEDIA_PREV_TRACK: chromecast.media_prev_track
+        }
+
+        for service_name, service_method in services.items():
+            calls = mock_service(self.hass, chromecast.DOMAIN, service_name)
+
+            service_method(self.hass)
+            self.hass._pool.block_till_done()
+
+            self.assertEqual(1, len(calls))
+            call = calls[-1]
+            self.assertEqual(call.domain, chromecast.DOMAIN)
+            self.assertEqual(call.service, service_name)
+            self.assertEqual(call.data, {})
+
+            service_method(self.hass, self.test_entity)
+            self.hass._pool.block_till_done()
+
+            self.assertEqual(2, len(calls))
+            call = calls[-1]
+            self.assertEqual(call.domain, chromecast.DOMAIN)
+            self.assertEqual(call.service, service_name)
+            self.assertEqual(call.data,
+                             {components.ATTR_ENTITY_ID: self.test_entity})
+
+    def test_setup(self):
+        """
+        Test Chromecast setup.
+        We do not have access to a Chromecast while testing so test errors.
+        In an ideal world we would create a mock pychromecast API..
+        """
+        self.assertFalse(chromecast.setup(
+            self.hass, {chromecast.DOMAIN: {ha.CONF_HOSTS: '127.0.0.1'}}))