mirrors
/
cpython
mirror of https://github.com/python/cpython.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bpo-32415: Add more tests (#4995)

This commit is contained in:
Yury Selivanov 2017-12-23 15:42:27 -05:00 committed by GitHub
parent 0f47fa2c89
commit a8fb079789
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
Lib/test/test_asyncio/test_tasks.py
View File

@ -2308,10 +2308,28 @@ class BaseTaskIntrospectionTests:
_enter_task = None _enter_task = None
_leave_task = None _leave_task = None
def test__register_task(self): def test__register_task_1(self):
task = mock.Mock() class TaskLike:
@property
def _loop(self):
return loop
task = TaskLike()
loop = mock.Mock() loop = mock.Mock()
task.get_loop = lambda: loop
self.assertEqual(asyncio.all_tasks(loop), set())
self._register_task(task)
self.assertEqual(asyncio.all_tasks(loop), {task})
self._unregister_task(task)
def test__register_task_2(self):
class TaskLike:
def get_loop(self):
return loop
task = TaskLike()
loop = mock.Mock()
self.assertEqual(asyncio.all_tasks(loop), set()) self.assertEqual(asyncio.all_tasks(loop), set())
self._register_task(task) self._register_task(task)
self.assertEqual(asyncio.all_tasks(loop), {task}) self.assertEqual(asyncio.all_tasks(loop), {task})