mirror of https://github.com/python/cpython
Merge 3.5 (issue #25647)
This commit is contained in:
commit
578bc5b80b
|
@ -204,7 +204,8 @@ def coroutine(func):
|
|||
@functools.wraps(func)
|
||||
def coro(*args, **kw):
|
||||
res = func(*args, **kw)
|
||||
if isinstance(res, futures.Future) or inspect.isgenerator(res):
|
||||
if isinstance(res, futures.Future) or inspect.isgenerator(res) or \
|
||||
isinstance(res, CoroWrapper):
|
||||
res = yield from res
|
||||
elif _AwaitableABC is not None:
|
||||
# If 'func' returns an Awaitable (new in 3.5) we
|
||||
|
|
|
@ -1794,6 +1794,30 @@ class TaskTests(test_utils.TestCase):
|
|||
|
||||
self.assertRegex(message, re.compile(regex, re.DOTALL))
|
||||
|
||||
def test_return_coroutine_from_coroutine(self):
|
||||
"""Return of @asyncio.coroutine()-wrapped function generator object
|
||||
from @asyncio.coroutine()-wrapped function should have same effect as
|
||||
returning generator object or Future."""
|
||||
def check():
|
||||
@asyncio.coroutine
|
||||
def outer_coro():
|
||||
@asyncio.coroutine
|
||||
def inner_coro():
|
||||
return 1
|
||||
|
||||
return inner_coro()
|
||||
|
||||
result = self.loop.run_until_complete(outer_coro())
|
||||
self.assertEqual(result, 1)
|
||||
|
||||
# Test with debug flag cleared.
|
||||
with set_coroutine_debug(False):
|
||||
check()
|
||||
|
||||
# Test with debug flag set.
|
||||
with set_coroutine_debug(True):
|
||||
check()
|
||||
|
||||
def test_task_source_traceback(self):
|
||||
self.loop.set_debug(True)
|
||||
|
||||
|
|
Loading…
Reference in New Issue