jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats 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()