Add more unittest for async_add_job (#4320)
* Add more unittest for async_add_job * fix test * lint
This commit is contained in:
parent
8cbb8f6527
commit
8fc853ba11
1 changed files with 34 additions and 3 deletions
|
@ -118,7 +118,7 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
|
||||
# self.assertEqual(1, len(calls))
|
||||
|
||||
def test_async_add_job_pending_taks_add(self):
|
||||
def test_async_add_job_pending_tasks_add(self):
|
||||
"""Add a coro to pending tasks."""
|
||||
call_count = []
|
||||
|
||||
|
@ -133,8 +133,8 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
self.hass.block_till_done()
|
||||
assert len(call_count) == 1
|
||||
|
||||
def test_async_add_job_pending_taks_cleanup(self):
|
||||
"""Add a coro to pending tasks."""
|
||||
def test_async_add_job_pending_tasks_cleanup(self):
|
||||
"""Add a coro to pending tasks and test cleanup."""
|
||||
call_count = []
|
||||
|
||||
@asyncio.coroutine
|
||||
|
@ -155,6 +155,37 @@ class TestHomeAssistant(unittest.TestCase):
|
|||
self.hass.block_till_done()
|
||||
assert len(call_count) == 51
|
||||
|
||||
def test_async_add_job_pending_tasks_executor(self):
|
||||
"""Run a executor in pending tasks."""
|
||||
call_count = []
|
||||
|
||||
def test_executor():
|
||||
"""Test executor."""
|
||||
call_count.append('call')
|
||||
|
||||
for i in range(40):
|
||||
self.hass.add_job(test_executor)
|
||||
|
||||
assert len(self.hass._pending_tasks) == 40
|
||||
self.hass.block_till_done()
|
||||
assert len(call_count) == 40
|
||||
|
||||
def test_async_add_job_pending_tasks_callback(self):
|
||||
"""Run a callback in pending tasks."""
|
||||
call_count = []
|
||||
|
||||
@ha.callback
|
||||
def test_callback():
|
||||
"""Test callback."""
|
||||
call_count.append('call')
|
||||
|
||||
for i in range(40):
|
||||
self.hass.add_job(test_callback)
|
||||
|
||||
assert len(self.hass._pending_tasks) == 0
|
||||
self.hass.block_till_done()
|
||||
assert len(call_count) == 40
|
||||
|
||||
|
||||
class TestEvent(unittest.TestCase):
|
||||
"""A Test Event class."""
|
||||
|
|
Loading…
Add table
Reference in a new issue