jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Allow removing a state via API + remote StateMachine

Browse source
This commit is contained in:
Paulus Schoutsen 2016-02-13 23:00:38 -08:00
parent ada2fb4ec0
commit 06de73ff80
3 changed files with 83 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/test_remote.py
View file

@ -135,9 +135,17 @@ class TestRemoteMethods(unittest.TestCase):
self.assertEqual(hass.states.all(), remote.get_states(master_api))
self.assertEqual([], remote.get_states(broken_api))
def test_remove_state(self):
""" Test Python API set_state. """
hass.states.set('test.remove_state', 'set_test')
self.assertIn('test.remove_state', hass.states.entity_ids())
remote.remove_state(master_api, 'test.remove_state')
self.assertNotIn('test.remove_state', hass.states.entity_ids())
def test_set_state(self):
""" Test Python API set_state. """
hass.states.set('test.test', 'set_test')
remote.set_state(master_api, 'test.test', 'set_test')
state = hass.states.get('test.test')
@ -225,6 +233,29 @@ class TestRemoteClasses(unittest.TestCase):
self.assertEqual("remote.statemachine test",
slave.states.get("remote.test").state)
def test_statemachine_remove_from_master(self):
hass.states.set("remote.master_remove", "remove me!")
hass.pool.block_till_done()
self.assertIn('remote.master_remove', slave.states.entity_ids())
hass.states.remove("remote.master_remove")
hass.pool.block_till_done()
self.assertNotIn('remote.master_remove', slave.states.entity_ids())
def test_statemachine_remove_from_slave(self):
hass.states.set("remote.slave_remove", "remove me!")
hass.pool.block_till_done()
self.assertIn('remote.slave_remove', slave.states.entity_ids())
self.assertTrue(slave.states.remove("remote.slave_remove"))
slave.pool.block_till_done()
hass.pool.block_till_done()
self.assertNotIn('remote.slave_remove', slave.states.entity_ids())
def test_eventbus_fire(self):
""" Test if events fired from the eventbus get fired. """
test_value = []