fix incompatability with pythongh-124392

Author: graingert

Lib/test/test_asyncio/test_taskgroups.py

@@ -1,6 +1,7 @@
 # Adapted with permission from the EdgeDB project;
 # license: PSFL.
 
+import sys
 import gc
 import asyncio
 import contextvars
@@ -28,6 +29,15 @@ def get_error_types(eg):
     return {type(exc) for exc in eg.exceptions}
 
 
+def no_other_refs():
+    # due to gh-124392 coroutines now refer to their locals
+    coro = asyncio.current_task().get_coro()
+    frame = sys._getframe(1)
+    while coro.cr_frame != frame:
+        coro = coro.cr_await
+    return [coro]
+
+
 class TestTaskGroup(unittest.IsolatedAsyncioTestCase):
 
     async def test_taskgroup_01(self):
@@ -913,7 +923,7 @@ class _Done(Exception):
             exc = e
 
         self.assertIsNotNone(exc)
-        self.assertListEqual(gc.get_referrers(exc), [])
+        self.assertListEqual(gc.get_referrers(exc), no_other_refs())
 
 
     async def test_exception_refcycles_errors(self):
@@ -931,7 +941,7 @@ class _Done(Exception):
             exc = excs.exceptions[0]
 
         self.assertIsInstance(exc, _Done)
-        self.assertListEqual(gc.get_referrers(exc), [])
+        self.assertListEqual(gc.get_referrers(exc), no_other_refs())
 
 
     async def test_exception_refcycles_parent_task(self):
@@ -953,7 +963,7 @@ async def coro_fn():
             exc = excs.exceptions[0].exceptions[0]
 
         self.assertIsInstance(exc, _Done)
-        self.assertListEqual(gc.get_referrers(exc), [])
+        self.assertListEqual(gc.get_referrers(exc), no_other_refs())
 
     async def test_exception_refcycles_propagate_cancellation_error(self):
         """Test that TaskGroup deletes propagate_cancellation_error"""
@@ -968,7 +978,7 @@ async def test_exception_refcycles_propagate_cancellation_error(self):
             exc = e.__cause__
 
         self.assertIsInstance(exc, asyncio.CancelledError)
-        self.assertListEqual(gc.get_referrers(exc), [])
+        self.assertListEqual(gc.get_referrers(exc), no_other_refs())
 
     async def test_exception_refcycles_base_error(self):
         """Test that TaskGroup deletes self._base_error"""
@@ -985,7 +995,7 @@ class MyKeyboardInterrupt(KeyboardInterrupt):
             exc = e
 
         self.assertIsNotNone(exc)
-        self.assertListEqual(gc.get_referrers(exc), [])
+        self.assertListEqual(gc.get_referrers(exc), no_other_refs())
 
 
 if __name__ == "__main__":