From e09e78347b9eee6e579c51935ac518a21c184dba Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Fri, 21 Aug 2015 21:51:54 -0700 Subject: [PATCH] Upgrade remote test --- tests/test_remote.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test_remote.py b/tests/test_remote.py index 0f45091d598..e5bfd71199f 100644 --- a/tests/test_remote.py +++ b/tests/test_remote.py @@ -130,9 +130,12 @@ class TestRemoteMethods(unittest.TestCase): def test_set_state(self): """ Test Python API set_state. """ - self.assertTrue(remote.set_state(master_api, 'test.test', 'set_test')) + hass.states.set('test.test', 'set_test') - self.assertEqual('set_test', hass.states.get('test.test').state) + state = hass.states.get('test.test') + + self.assertIsNotNone(state) + self.assertEqual('set_test', state.state) self.assertFalse(remote.set_state(broken_api, 'test.test', 'set_test'))