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

Manual updates (#17278)

Browse source
This commit is contained in:
Paulus Schoutsen 2018-10-09 16:54:38 +02:00 • committed by Pascal Vizeli
parent cf249e3e5e
commit e903f7ffda
5 changed files with 72 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/helpers/test_entity_component.py
View file

@ -415,3 +415,19 @@ async def test_unload_entry_fails_if_never_loaded(hass):
with pytest.raises(ValueError):
await component.async_unload_entry(entry)
async def test_update_entity(hass):
"""Test that we can update an entity with the helper."""
component = EntityComponent(_LOGGER, DOMAIN, hass)
entity = MockEntity()
entity.async_update_ha_state = Mock(return_value=mock_coro())
await component.async_add_entities([entity])
# Called as part of async_add_entities
assert len(entity.async_update_ha_state.mock_calls) == 1
await hass.helpers.entity_component.async_update_entity(entity.entity_id)
assert len(entity.async_update_ha_state.mock_calls) == 2
assert entity.async_update_ha_state.mock_calls[-1][1][0] is True