(original) (raw)

changeset: 96037:0d80d46adfdb user: Yury Selivanov yselivanov@sprymix.com date: Wed May 13 16:49:35 2015 -0400 files: Lib/test/test_coroutines.py description: Issue 24017: More tests for 'async for' and 'async with'. diff -r 94eddf8ec0e5 -r 0d80d46adfdb Lib/test/test_coroutines.py --- a/Lib/test/test_coroutines.py Wed May 13 15:37:48 2015 -0400 +++ b/Lib/test/test_coroutines.py Wed May 13 16:49:35 2015 -0400 @@ -623,6 +623,27 @@ 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 @@ 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): /yselivanov@sprymix.com