| @@ -2308,10 +2308,28 @@ class BaseTaskIntrospectionTests: |
|
|
| 2308 |
2308 |
_enter_task = None |
| 2309 |
2309 |
_leave_task = None |
| 2310 |
2310 |
|
| 2311 |
|
-def test__register_task(self): |
| 2312 |
|
-task = mock.Mock() |
|
2311 |
+def test__register_task_1(self): |
|
2312 |
+class TaskLike: |
|
2313 |
+@property |
|
2314 |
+def _loop(self): |
|
2315 |
+return loop |
|
2316 |
+ |
|
2317 |
+task = TaskLike() |
| 2313 |
2318 |
loop = mock.Mock() |
| 2314 |
|
-task.get_loop = lambda: loop |
|
2319 |
+ |
|
2320 |
+self.assertEqual(asyncio.all_tasks(loop), set()) |
|
2321 |
+self._register_task(task) |
|
2322 |
+self.assertEqual(asyncio.all_tasks(loop), {task}) |
|
2323 |
+self._unregister_task(task) |
|
2324 |
+ |
|
2325 |
+def test__register_task_2(self): |
|
2326 |
+class TaskLike: |
|
2327 |
+def get_loop(self): |
|
2328 |
+return loop |
|
2329 |
+ |
|
2330 |
+task = TaskLike() |
|
2331 |
+loop = mock.Mock() |
|
2332 |
+ |
| 2315 |
2333 |
self.assertEqual(asyncio.all_tasks(loop), set()) |
| 2316 |
2334 |
self._register_task(task) |
| 2317 |
2335 |
self.assertEqual(asyncio.all_tasks(loop), {task}) |