mirror of
https://github.com/python/cpython.git
synced 2025-07-15 23:35:23 +00:00
bpo-32415: Add more tests (#4995)
This commit is contained in:
parent
0f47fa2c89
commit
a8fb079789
1 changed files with 21 additions and 3 deletions
|
@ -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})
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue