jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-09 02:49:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restore states when removing/adding entities (#18890)

Browse Source
This commit is contained in:
Adam Mills
2018-12-02 04:51:15 -05:00
committed by Paulus Schoutsen
parent bbb40fde84
commit a10cbadb57
2 changed files with 30 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/helpers/test_restore_state.py
View File

@@ -198,3 +198,23 @@ async def test_load_error(hass):
state = await entity.async_get_last_state()
assert state is None
async def test_state_saved_on_remove(hass):
"""Test that we save entity state on removal."""
entity = RestoreEntity()
entity.hass = hass
entity.entity_id = 'input_boolean.b0'
await entity.async_added_to_hass()
hass.states.async_set('input_boolean.b0', 'on')
data = await RestoreStateData.async_get_instance(hass)
# No last states should currently be saved
assert not data.last_states
await entity.async_will_remove_from_hass()
# We should store the input boolean state when it is removed
assert data.last_states['input_boolean.b0'].state.state == 'on'