evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Updated entity tests to work with new attribute overwritting logic.

Browse source
This commit is contained in:
Ryan Kraus 2015-04-25 01:43:03 -04:00
parent a95aad324f
commit 04a98f99b7
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_helper_entity.py
View file

@ -25,7 +25,8 @@ class TestHelpersEntity(unittest.TestCase):
def tearDown(self): # pylint: disable=invalid-name
""" Stop down stuff we started. """
self.hass.stop()
entity.Entity.overwrite_hidden(self.entity.entity_id, None)
entity.Entity.overwrite_attribute(self.entity.entity_id,
ATTR_HIDDEN, None)
def test_default_hidden_not_in_attributes(self):
""" Test that the default hidden property is set to False. """
@ -43,7 +44,8 @@ class TestHelpersEntity(unittest.TestCase):
def test_overwriting_hidden_property_to_true(self):
""" Test we can overwrite hidden property to True. """
entity.Entity.overwrite_hidden(self.entity.entity_id, True)
entity.Entity.overwrite_attribute(self.entity.entity_id,
ATTR_HIDDEN, True)
self.entity.update_ha_state()
state = self.hass.states.get(self.entity.entity_id)
@ -51,7 +53,8 @@ class TestHelpersEntity(unittest.TestCase):
def test_overwriting_hidden_property_to_false(self):
""" Test we can overwrite hidden property to True. """
entity.Entity.overwrite_hidden(self.entity.entity_id, False)
entity.Entity.overwrite_attribute(self.entity.entity_id,
ATTR_HIDDEN, False)
self.entity.hidden = True
self.entity.update_ha_state()