From: Yury Selivanov <yselivanov@sprymix.com>
Date: Wed, 13 May 2015 20:49:35 +0000 (-0400)
Subject: Issue 24017: More tests for 'async for' and 'async with'.
X-Git-Tag: v3.5.0b1~146
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=9113dc7c97c649e2490c65157be1fb9b3fe7751b;p=python

Issue 24017: More tests for 'async for' and 'async with'.
---

diff --git a/Lib/test/test_coroutines.py b/Lib/test/test_coroutines.py
index b1105cb771..7efd5c90a6 100644
--- a/Lib/test/test_coroutines.py
+++ b/Lib/test/test_coroutines.py
@@ -623,6 +623,27 @@ class CoroutineTest(unittest.TestCase):
 
         run_async(foo())
 
+    def test_with_13(self):
+        CNT = 0
+
+        class CM:
+            async def __aenter__(self):
+                1/0
+
+            async def __aexit__(self, *e):
+                return True
+
+        async def foo():
+            nonlocal CNT
+            CNT += 1
+            async with CM():
+                CNT += 1000
+            CNT += 10000
+
+        with self.assertRaises(ZeroDivisionError):
+            run_async(foo())
+        self.assertEqual(CNT, 1)
+
     def test_for_1(self):
         aiter_calls = 0
 
@@ -859,6 +880,20 @@ class CoroutineTest(unittest.TestCase):
         run_async(main())
         self.assertEqual(I, 20555255)
 
+    def test_for_7(self):
+        CNT = 0
+        class AI:
+            async def __aiter__(self):
+                1/0
+        async def foo():
+            nonlocal CNT
+            async for i in AI():
+                CNT += 1
+            CNT += 10
+        with self.assertRaises(ZeroDivisionError):
+            run_async(foo())
+        self.assertEqual(CNT, 0)
+
 
 class CoroAsyncIOCompatTest(unittest.TestCase):