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

Fix contextmanager-generator-missing-cleanup warning in tests (#119478)

Browse Source
This commit is contained in:
epenet 2024-06-12 22:38:11 +02:00 committed by GitHub
parent f7326d3baf
commit e3e80c83b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/common.py
View File

@ -195,9 +195,11 @@ def get_test_home_assistant() -> Generator[HomeAssistant]:
threading.Thread(name="LoopThread", target=run_loop, daemon=False).start()
yield hass
loop.run_until_complete(context_manager.__aexit__(None, None, None))
loop.close()
try:
yield hass
finally:
loop.run_until_complete(context_manager.__aexit__(None, None, None))
loop.close()
class StoreWithoutWriteLoad[_T: (Mapping[str, Any] | Sequence[Any])](storage.Store[_T]):
@ -359,10 +361,11 @@ async def async_test_home_assistant(
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_CLOSE, clear_instance)
yield hass
# Restore timezone, it is set when creating the hass object
dt_util.set_default_time_zone(orig_tz)
try:
yield hass
finally:
# Restore timezone, it is set when creating the hass object
dt_util.set_default_time_zone(orig_tz)
def async_mock_service(