Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Track tasks only during shutdown and tests (#4428)

Browse Source 
* Track tasks only when needed

* Tweak async_block_till_done
This commit is contained in:
Paulus Schoutsen
2016-11-24 14:49:29 -08:00
committed by GitHub
parent 42c99b0ccb
commit eacdce9ed9
4 changed files with 42 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_core.py
View File

@@ -9,8 +9,7 @@ import pytz
import homeassistant.core as ha
from homeassistant.exceptions import InvalidEntityFormatError
from homeassistant.util.async import (
run_callback_threadsafe, run_coroutine_threadsafe)
from homeassistant.util.async import run_coroutine_threadsafe
import homeassistant.util.dt as dt_util
from homeassistant.util.unit_system import (METRIC_SYSTEM)
from homeassistant.const import (
@@ -129,7 +128,7 @@ class TestHomeAssistant(unittest.TestCase):
"""Test Coro."""
call_count.append('call')
for i in range(50):
for i in range(3):
self.hass.add_job(test_coro())
run_coroutine_threadsafe(
@@ -137,13 +136,8 @@ class TestHomeAssistant(unittest.TestCase):
loop=self.hass.loop
).result()
with patch.object(self.hass.loop, 'call_later') as mock_later:
run_callback_threadsafe(
self.hass.loop, self.hass._async_tasks_cleanup).result()
assert mock_later.called
assert len(self.hass._pending_tasks) == 0
assert len(call_count) == 50
assert len(self.hass._pending_tasks) == 3
assert len(call_count) == 3
def test_async_add_job_pending_tasks_coro(self):
"""Add a coro to pending tasks."""