Skip to content

Commit 51482c0

Browse files
committed
remove potential flake in test
1 parent 03b736b commit 51482c0

1 file changed

Lines changed: 12 additions & 9 deletions

File tree

‎Lib/test/test_external_inspection.py‎

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -515,15 +515,18 @@ async def main():
515515
loop.run_in_executor(pool, _asyncio_in_subinterpreter)
516516
task = asyncio.create_task(main_worker(), name="main_worker")
517517
self.addCleanup(task.cancel)
518-
await asyncio.sleep(1)
519-
return [
520-
[frame.funcname.rpartition(".")[2]
521-
for frame in coro.call_stack]
522-
for info in RemoteUnwinder(
523-
os.getpid()).get_all_awaited_by()
524-
for task in info.awaited_by
525-
for coro in task.coroutine_stack
526-
]
518+
for _ in busy_retry(SHORT_TIMEOUT):
519+
await asyncio.sleep(0)
520+
stacks = [
521+
[frame.funcname.rpartition(".")[2]
522+
for frame in coro.call_stack]
523+
for info in RemoteUnwinder(
524+
os.getpid()).get_all_awaited_by()
525+
for task in info.awaited_by
526+
for coro in task.coroutine_stack
527+
]
528+
if ["sleep", "sub_worker"] in stacks:
529+
return stacks
527530

528531
stacks = asyncio.run(main())
529532
self.assertIn(["sleep", "sub_worker"], stacks)

0 commit comments

Comments
 (0)