add tests that check already running aclose/athrow get closed, and don't warn unawaited

graingert · graingert · commit 1220aee3d7a1 · 2024-04-26T10:26:57.000+01:00
diff --git a/Lib/test/test_asyncgen.py b/Lib/test/test_asyncgen.py
@@ -393,6 +393,10 @@ async def gen():
                 r'anext\(\): asynchronous generator is already running'):
             an.__next__()
 
+        with self.assertRaisesRegex(RuntimeError,
+                r"cannot reuse already awaited __anext__\(\)/asend\(\)"):
+            an.send(None)
+
     def test_async_gen_asend_throw_concurrent_with_send(self):
         import types
 
@@ -422,6 +426,10 @@ async def agenfn():
                 r'anext\(\): asynchronous generator is already running'):
             gen2.throw(MyExc)
 
+        with self.assertRaisesRegex(RuntimeError,
+                r"cannot reuse already awaited __anext__\(\)/asend\(\)"):
+            gen2.send(None)
+
     def test_async_gen_athrow_throw_concurrent_with_send(self):
         import types
 
@@ -451,6 +459,10 @@ async def agenfn():
                 r'athrow\(\): asynchronous generator is already running'):
             gen2.throw(MyExc)
 
+        with self.assertRaisesRegex(RuntimeError,
+            r"cannot reuse already awaited aclose\(\)/athrow\(\)"):
+            gen2.send(None)
+
     def test_async_gen_asend_throw_concurrent_with_throw(self):
         import types
 
@@ -485,6 +497,10 @@ async def agenfn():
                 r'anext\(\): asynchronous generator is already running'):
             gen2.throw(MyExc)
 
+        with self.assertRaisesRegex(RuntimeError,
+                r"cannot reuse already awaited __anext__\(\)/asend\(\)"):
+            gen2.send(None)
+
     def test_async_gen_athrow_throw_concurrent_with_throw(self):
         import types
 
@@ -518,6 +534,10 @@ async def agenfn():
                 r'athrow\(\): asynchronous generator is already running'):
             gen2.throw(MyExc)
 
+        with self.assertRaisesRegex(RuntimeError,
+                r"cannot reuse already awaited aclose\(\)/athrow\(\)"):
+            gen2.send(None)
+
     def test_async_gen_3_arg_deprecation_warning(self):
         async def gen():
             yield 123
@@ -1941,5 +1961,52 @@ class MyException(Exception):
         del g
         gc_collect()  # does not warn unawaited
 
+    def test_asend_send_already_running(self):
+        @types.coroutine
+        def _async_yield(v):
+            return (yield v)
+
+        async def agenfn():
+            while True:
+                await _async_yield(1)
+            return
+            yield
+
+        agen = agenfn()
+        gen = agen.asend(None)
+        gen.send(None)
+        gen2 = agen.asend(None)
+
+        with self.assertRaisesRegex(RuntimeError,
+                r'anext\(\): asynchronous generator is already running'):
+            gen2.send(None)
+
+        del gen2
+        gc_collect()  # does not warn unawaited
+
+
+    def test_athrow_send_already_running(self):
+        @types.coroutine
+        def _async_yield(v):
+            return (yield v)
+
+        async def agenfn():
+            while True:
+                await _async_yield(1)
+            return
+            yield
+
+        agen = agenfn()
+        gen = agen.asend(None)
+        gen.send(None)
+        gen2 = agen.athrow(Exception)
+
+        with self.assertRaisesRegex(RuntimeError,
+                r'athrow\(\): asynchronous generator is already running'):
+            gen2.send(None)
+
+        del gen2
+        gc_collect()  # does not warn unawaited
+
 if __name__ == "__main__":
     unittest.main()
