changeset:   104589:79b9257f3386
parent:      104586:59838bede1de
parent:      104588:cfe2109ce2c0
user:        Yury Selivanov <yury@magic.io>
date:        Thu Oct 20 15:54:51 2016 -0400
files:       Misc/NEWS
description:
Merge 3.6 (issue #28492)


diff -r 59838bede1de -r 79b9257f3386 Lib/test/test_asyncio/test_futures.py
--- a/Lib/test/test_asyncio/test_futures.py	Thu Oct 20 15:40:56 2016 -0400
+++ b/Lib/test/test_asyncio/test_futures.py	Thu Oct 20 15:54:51 2016 -0400
@@ -464,6 +464,19 @@
         futures._set_result_unless_cancelled(fut, 2)
         self.assertTrue(fut.cancelled())
 
+    def test_future_stop_iteration_args(self):
+        fut = asyncio.Future(loop=self.loop)
+        fut.set_result((1, 2))
+        fi = fut.__iter__()
+        result = None
+        try:
+            fi.send(None)
+        except StopIteration as ex:
+            result = ex.args[0]
+        else:
+            self.fail('StopIteration was expected')
+        self.assertEqual(result, (1, 2))
+
 
 class FutureDoneCallbackTests(test_utils.TestCase):
 
diff -r 59838bede1de -r 79b9257f3386 Modules/_asynciomodule.c
--- a/Modules/_asynciomodule.c	Thu Oct 20 15:40:56 2016 -0400
+++ b/Modules/_asynciomodule.c	Thu Oct 20 15:54:51 2016 -0400
@@ -787,9 +787,26 @@
 
     res = FutureObj_result(fut, NULL);
     if (res != NULL) {
-        // normal result
-        PyErr_SetObject(PyExc_StopIteration, res);
+        /* The result of the Future is not an exception.
+
+           We cunstruct an exception instance manually with
+           PyObject_CallFunctionObjArgs and pass it to PyErr_SetObject
+           (similarly to what genobject.c does).
+
+           This is to handle a situation when "res" is a tuple, in which
+           case PyErr_SetObject would set the value of StopIteration to
+           the first element of the tuple.
+
+           (See PyErr_SetObject/_PyErr_CreateException code for details.)
+        */
+        PyObject *e = PyObject_CallFunctionObjArgs(
+            PyExc_StopIteration, res, NULL);
         Py_DECREF(res);
+        if (e == NULL) {
+            return NULL;
+        }
+        PyErr_SetObject(PyExc_StopIteration, e);
+        Py_DECREF(e);
     }
 
     it->future = NULL;