Skip to content

Commit 0d7ace8

Browse files
fix tests
1 parent 40909d9 commit 0d7ace8

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

Lib/test/test_asyncio/test_tasks.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2254,7 +2254,7 @@ def test_wait_invalid_args(self):
22542254
asyncio.wait([]))
22552255

22562256
def test_log_destroyed_pending_task(self):
2257-
2257+
loop = asyncio.new_event_loop()
22582258
async def kill_me(loop):
22592259
future = self.new_future(loop)
22602260
await future
@@ -2263,20 +2263,20 @@ async def kill_me(loop):
22632263
raise Exception("code never reached")
22642264

22652265
mock_handler = mock.Mock()
2266-
self.loop.set_debug(True)
2267-
self.loop.set_exception_handler(mock_handler)
2266+
loop.set_debug(True)
2267+
loop.set_exception_handler(mock_handler)
22682268

22692269
# schedule the task
2270-
coro = kill_me(self.loop)
2271-
task = self.new_task(self.loop, coro)
2270+
coro = kill_me(loop)
2271+
task = self.new_task(loop, coro)
22722272

2273-
self.assertEqual(self.all_tasks(loop=self.loop), {task})
2273+
self.assertEqual(self.all_tasks(loop=loop), {task})
22742274

22752275
asyncio._set_event_loop(None)
22762276

22772277
# execute the task so it waits for future
2278-
self.loop._run_once()
2279-
self.assertEqual(len(self.loop._ready), 0)
2278+
loop.run_until_complete(asyncio.sleep(0))
2279+
self.assertEqual(len(loop._ready), 0)
22802280

22812281
coro = None
22822282
source_traceback = task._source_traceback
@@ -2285,7 +2285,7 @@ async def kill_me(loop):
22852285
# no more reference to kill_me() task: the task is destroyed by the GC
22862286
support.gc_collect()
22872287

2288-
mock_handler.assert_called_with(self.loop, {
2288+
mock_handler.assert_called_with(loop, {
22892289
'message': 'Task was destroyed but it is pending!',
22902290
'task': mock.ANY,
22912291
'source_traceback': source_traceback,
@@ -2294,7 +2294,9 @@ async def kill_me(loop):
22942294
# task got resurrected by the exception handler
22952295
support.gc_collect()
22962296

2297-
self.assertEqual(self.all_tasks(loop=self.loop), set())
2297+
self.assertEqual(self.all_tasks(loop=loop), set())
2298+
2299+
loop.close()
22982300

22992301

23002302
@mock.patch('asyncio.base_events.logger')

0 commit comments

Comments
 (0)