From a8fb07978953d3f55cfce836e3669d8b8e82b4c1 Mon Sep 17 00:00:00 2001 From: Yury Selivanov Date: Sat, 23 Dec 2017 15:42:27 -0500 Subject: bpo-32415: Add more tests (#4995) --- 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 84669cd..7bb4305 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}) -- cgit v0.12