From 8ac763c6f6f9b9fae2c9a849bd07cd2835d912de Mon Sep 17 00:00:00 2001
From: Ryan Kraus <rmkraus@gmail.com>
Date: Sat, 30 Jan 2016 12:26:28 -0500
Subject: [PATCH] Added test for universal mp service routing.

Added tests to ensure that the Universal Media Player is routing
service calls correctly.
---
 .../components/media_player/test_universal.py | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/tests/components/media_player/test_universal.py b/tests/components/media_player/test_universal.py
index 6600f9091ed..9ed0193959e 100644
--- a/tests/components/media_player/test_universal.py
+++ b/tests/components/media_player/test_universal.py
@@ -14,6 +14,8 @@ import homeassistant.components.switch as switch
 import homeassistant.components.media_player as media_player
 import homeassistant.components.media_player.universal as universal
 
+from tests.common import mock_service
+
 
 class MockMediaPlayer(media_player.MediaPlayerDevice):
     """ Mock media player for testing """
@@ -28,6 +30,9 @@ class MockMediaPlayer(media_player.MediaPlayerDevice):
         self._media_title = None
         self._supported_media_commands = 0
 
+        self.turn_off_service_calls = mock_service(
+            hass, media_player.DOMAIN, media_player.SERVICE_TURN_OFF)
+
     @property
     def name(self):
         """ name of player """
@@ -399,3 +404,38 @@ class TestMediaPlayer(unittest.TestCase):
             | universal.SUPPORT_VOLUME_STEP | universal.SUPPORT_VOLUME_MUTE
 
         self.assertEqual(check_flags, ump.supported_media_commands)
+
+    def test_service_call_to_child(self):
+        """ test a service call that should be routed to a child """
+        config = self.config_children_only
+        universal.validate_config(config)
+
+        ump = universal.UniversalMediaPlayer(self.hass, **config)
+        ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name'])
+        ump.update()
+
+        self.mock_mp_2._state = STATE_PLAYING
+        self.mock_mp_2.update_ha_state()
+        ump.update()
+
+        ump.turn_off()
+        self.assertEqual(len(self.mock_mp_2.turn_off_service_calls), 1)
+
+    def test_service_call_to_command(self):
+        config = self.config_children_only
+        config['commands'] = \
+            {'turn_off': {'service': 'test.turn_off', 'data': {}}}
+        universal.validate_config(config)
+
+        service = mock_service(self.hass, 'test', 'turn_off')
+
+        ump = universal.UniversalMediaPlayer(self.hass, **config)
+        ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name'])
+        ump.update()
+
+        self.mock_mp_2._state = STATE_PLAYING
+        self.mock_mp_2.update_ha_state()
+        ump.update()
+
+        ump.turn_off()
+        self.assertEqual(len(service), 1)