feedc0de/homeassistant-core
1
0
Fork 0
forked from home-assistant/core
Code Pull Requests Activity

Fix flapping reload tests (#34837)

Browse Source
This commit is contained in:
J. Nick Koston
2020-04-29 00:58:55 -05:00
committed by GitHub
parent 942cd7caa5
commit 13f4393042
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/helpers/entity.py
View File

@@ -493,13 +493,14 @@ class Entity(ABC):
async def async_remove(self) -> None:
"""Remove entity from Home Assistant."""
assert self.hass is not None
await self.async_internal_will_remove_from_hass()
await self.async_will_remove_from_hass()
if self._on_remove is not None:
while self._on_remove:
self._on_remove.pop()()
await self.async_internal_will_remove_from_hass()
await self.async_will_remove_from_hass()
self.hass.states.async_remove(self.entity_id, context=self._context)
async def async_added_to_hass(self) -> None: