Commit a8fb0797 authored by Yury Selivanov's avatar Yury Selivanov Committed by GitHub

bpo-32415: Add more tests (#4995)

parent 0f47fa2c
......@@ -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})
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment