From dfa1e1c586c5db61cc3d91c0337aa90d0d7ca0a2 Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Sat, 13 Dec 2014 22:49:49 -0800 Subject: [PATCH] Fix Chromecast tests --- ha_test/test_component_chromecast.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/ha_test/test_component_chromecast.py b/ha_test/test_component_chromecast.py index 3d9733648c9..c653c84b74f 100644 --- a/ha_test/test_component_chromecast.py +++ b/ha_test/test_component_chromecast.py @@ -67,19 +67,18 @@ class TestChromecast(unittest.TestCase): self.assertEqual(1, len(calls)) call = calls[-1] - self.assertEqual(call.domain, chromecast.DOMAIN) - self.assertEqual(call.service, service_name) - self.assertEqual(call.data, {}) + self.assertEqual(chromecast.DOMAIN, call.domain) + self.assertEqual(service_name, call.service) 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, - {ATTR_ENTITY_ID: self.test_entity}) + self.assertEqual(chromecast.DOMAIN, call.domain) + self.assertEqual(service_name, call.service) + self.assertEqual(self.test_entity, + call.data.get(ATTR_ENTITY_ID)) def test_setup(self): """