evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ensure recorder shuts down when its startup future is canceled out from under it (#72866)

Browse source
This commit is contained in:
J. Nick Koston 2022-06-01 19:13:09 -10:00 • committed by GitHub
parent d368b9e24f
commit f79e5e002b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 31 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/components/recorder/test_init.py
View file

@ -118,6 +118,26 @@ async def test_shutdown_before_startup_finishes(
assert run_info.end is not None
async def test_canceled_before_startup_finishes(
hass: HomeAssistant,
async_setup_recorder_instance: SetupRecorderInstanceT,
caplog: pytest.LogCaptureFixture,
):
"""Test recorder shuts down when its startup future is canceled out from under it."""
hass.state = CoreState.not_running
await async_setup_recorder_instance(hass)
instance = get_instance(hass)
await instance.async_db_ready
instance._hass_started.cancel()
with patch.object(instance, "engine"):
await hass.async_block_till_done()
await hass.async_add_executor_job(instance.join)
assert (
"Recorder startup was externally canceled before it could complete"
in caplog.text
)
async def test_shutdown_closes_connections(hass, recorder_mock):
"""Test shutdown closes connections."""