1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gh-103186: Suppress RuntimeWarning about unclosed async iterator in test_sys_settrace (GH-109075)

Browse Source
This commit is contained in:
Serhiy Storchaka 2023-09-07 16:34:44 +03:00 committed by GitHub
parent e4bb0026b9
commit d485551c9d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
Lib/test/test_sys_settrace.py
View File

@ -41,6 +41,20 @@ async def asynciter(iterable):
for x in iterable: for x in iterable:
yield x yield x
def clean_asynciter(test):
@wraps(test)
async def wrapper(*args, **kwargs):
cleanups = []
def wrapped_asynciter(iterable):
it = asynciter(iterable)
cleanups.append(it.aclose)
return it
try:
return await test(*args, **kwargs, asynciter=wrapped_asynciter)
finally:
while cleanups:
await cleanups.pop()()
return wrapper
# A very basic example. If this fails, we're in deep trouble. # A very basic example. If this fails, we're in deep trouble.
def basic(): def basic():
@ -1936,7 +1950,11 @@ class JumpTestCase(unittest.TestCase):
def run_test(self, func, jumpFrom, jumpTo, expected, error=None, def run_test(self, func, jumpFrom, jumpTo, expected, error=None,
event='line', decorated=False, warning=None): event='line', decorated=False, warning=None):
tracer = JumpTracer(func, jumpFrom, jumpTo, event, decorated) wrapped = func
while hasattr(wrapped, '__wrapped__'):
wrapped = wrapped.__wrapped__
tracer = JumpTracer(wrapped, jumpFrom, jumpTo, event, decorated)
sys.settrace(tracer.trace) sys.settrace(tracer.trace)
output = [] output = []
@ -1952,7 +1970,11 @@ class JumpTestCase(unittest.TestCase):
def run_async_test(self, func, jumpFrom, jumpTo, expected, error=None, def run_async_test(self, func, jumpFrom, jumpTo, expected, error=None,
event='line', decorated=False, warning=None): event='line', decorated=False, warning=None):
tracer = JumpTracer(func, jumpFrom, jumpTo, event, decorated) wrapped = func
while hasattr(wrapped, '__wrapped__'):
wrapped = wrapped.__wrapped__
tracer = JumpTracer(wrapped, jumpFrom, jumpTo, event, decorated)
sys.settrace(tracer.trace) sys.settrace(tracer.trace)
output = [] output = []
@ -2023,7 +2045,8 @@ class JumpTestCase(unittest.TestCase):
output.append(7) output.append(7)
@async_jump_test(4, 5, [3, 5]) @async_jump_test(4, 5, [3, 5])
async def test_jump_out_of_async_for_block_forwards(output): @clean_asynciter
async def test_jump_out_of_async_for_block_forwards(output, asynciter):
for i in [1]: for i in [1]:
async for i in asynciter([1, 2]): async for i in asynciter([1, 2]):
output.append(3) output.append(3)
@ -2031,7 +2054,8 @@ class JumpTestCase(unittest.TestCase):
output.append(5) output.append(5)
@async_jump_test(5, 2, [2, 4, 2, 4, 5, 6]) @async_jump_test(5, 2, [2, 4, 2, 4, 5, 6])
async def test_jump_out_of_async_for_block_backwards(output): @clean_asynciter
async def test_jump_out_of_async_for_block_backwards(output, asynciter):
for i in [1]: for i in [1]:
output.append(2) output.append(2)
async for i in asynciter([1]): async for i in asynciter([1]):