jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-24 17:27:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix threading error in stream tests (#62221)

Browse Source
This commit is contained in:
J. Nick Koston 2021-12-17 11:26:30 -06:00 committed by GitHub
parent 3b97c544b1
commit a0cd29bbcf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/components/stream/conftest.py
View File

@ -78,8 +78,9 @@ class SaveRecordWorkerSync:
to avoid thread leaks in tests.
"""
def __init__(self):
def __init__(self, hass):
"""Initialize SaveRecordWorkerSync."""
self._hass = hass
self._save_event = None
self._segments = None
self._save_thread = None
@ -91,7 +92,7 @@ class SaveRecordWorkerSync:
assert self._save_thread is None
self._segments = segments
self._save_thread = threading.current_thread()
self._save_event.set()
self._hass.loop.call_soon_threadsafe(self._save_event.set)
async def get_segments(self):
"""Return the recorded video segments."""
@ -115,7 +116,7 @@ class SaveRecordWorkerSync:
@pytest.fixture()
def record_worker_sync(hass):
"""Patch recorder_save_worker for clean thread shutdown for test."""
sync = SaveRecordWorkerSync()
sync = SaveRecordWorkerSync(hass)
with patch(
"homeassistant.components.stream.recorder.recorder_save_worker",
side_effect=sync.recorder_save_worker,