Proper fix for Python 3.5

Author: elprans

tests/test_pool.py

@@ -12,6 +12,7 @@
 import platform
 import random
 import sys
+import textwrap
 import time
 import unittest
 
@@ -676,35 +677,41 @@ async def test_pool_handles_transaction_exit_in_asyncgen_1(self):
         pool = await self.create_pool(database='postgres',
                                       min_size=1, max_size=1)
 
-        async def iterate(con):
-            async with con.transaction():
-                for record in await con.fetch("SELECT 1"):
-                    yield record
+        locals_ = {}
+        exec(textwrap.dedent('''\
+            async def iterate(con):
+                async with con.transaction():
+                    for record in await con.fetch("SELECT 1"):
+                        yield record
+        '''), globals(), locals_)
 
         class MyException(Exception):
             pass
 
         with self.assertRaises(MyException):
             async with pool.acquire() as con:
-                async for _ in iterate(con):  # noqa
+                async for _ in locals_['iterate'](con):  # noqa
                     raise MyException()
 
     @unittest.skipIf(sys.version_info[:2] < (3, 6), 'no asyncgen support')
     async def test_pool_handles_transaction_exit_in_asyncgen_2(self):
         pool = await self.create_pool(database='postgres',
                                       min_size=1, max_size=1)
 
-        async def iterate(con):
-            async with con.transaction():
-                for record in await con.fetch("SELECT 1"):
-                    yield record
+        locals_ = {}
+        exec(textwrap.dedent('''\
+            async def iterate(con):
+                async with con.transaction():
+                    for record in await con.fetch("SELECT 1"):
+                        yield record
+        '''), globals(), locals_)
 
         class MyException(Exception):
             pass
 
         with self.assertRaises(MyException):
             async with pool.acquire() as con:
-                iterator = iterate(con)
+                iterator = locals_['iterate'](con)
                 async for _ in iterator:  # noqa
                     raise MyException()
 
@@ -715,17 +722,20 @@ async def test_pool_handles_asyncgen_finalization(self):
         pool = await self.create_pool(database='postgres',
                                       min_size=1, max_size=1)
 
-        async def iterate(con):
-            for record in await con.fetch("SELECT 1"):
-                yield record
+        locals_ = {}
+        exec(textwrap.dedent('''\
+            async def iterate(con):
+                for record in await con.fetch("SELECT 1"):
+                    yield record
+        '''), globals(), locals_)
 
         class MyException(Exception):
             pass
 
         with self.assertRaises(MyException):
             async with pool.acquire() as con:
                 async with con.transaction():
-                    async for _ in iterate(con):  # noqa
+                    async for _ in locals_['iterate'](con):  # noqa
                         raise MyException()