diff options
| author |   Yury Selivanov <yury@magic.io> | 2017-12-23 15:42:27 -0500 |
|---|---|---|
| committer |   GitHub <noreply@github.com> | 2017-12-23 15:42:27 -0500 |
| commit | a8fb07978953d3f55cfce836e3669d8b8e82b4c1 (patch) | |
| tree | 9adcf1b5fa2bb6423b880722eac9c8b671df257e | |
| parent | bpo-32357: Use PySet_GET_SIZE macro in _is_coroutine() from _asynciomodule.c ... (diff) | |
| download | cpython-a8fb07978953d3f55cfce836e3669d8b8e82b4c1.tar.gz cpython-a8fb07978953d3f55cfce836e3669d8b8e82b4c1.tar.bz2 cpython-a8fb07978953d3f55cfce836e3669d8b8e82b4c1.zip | |
bpo-32415: Add more tests (#4995)
| -rw-r--r-- | Lib/test/test_asyncio/test_tasks.py | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/Lib/test/test_asyncio/test_tasks.py b/Lib/test/test_asyncio/test_tasks.py index 84669cd6c7e..7bb43058c80 100644 --- a/Lib/test/test_asyncio/test_tasks.py +++ b/Lib/test/test_asyncio/test_tasks.py @@ -2308,10 +2308,28 @@ class BaseTaskIntrospectionTests: _enter_task = None _leave_task = None - def test__register_task(self): - task = mock.Mock() + def test__register_task_1(self): + class TaskLike: + @property + def _loop(self): + return loop + + task = TaskLike() 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._register_task(task) self.assertEqual(asyncio.all_tasks(loop), {task}) |