From 37b3f838815d80f0b6510da4306a9d139d49d512 Mon Sep 17 00:00:00 2001 From: Yury Selivanov Date: Sat, 23 Dec 2017 15:18:19 -0500 Subject: [PATCH] bpo-32415: Add more tests --- Lib/test/test_asyncio/test_tasks.py | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/Lib/test/test_asyncio/test_tasks.py b/Lib/test/test_asyncio/test_tasks.py index 84669cd6c7ec87..7bb43058c80529 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})