From ce23ffd34090ecaea2be6bcf8aac79a9ef0d4de2 Mon Sep 17 00:00:00 2001
From: sobolevn <mail@sobolevn.me>
Date: Wed, 10 Sep 2025 16:49:40 +0300
Subject: [PATCH] gh-138750: Increase coverage of `inspect.getcoroutinelocals`

---
 Lib/test/test_inspect/test_inspect.py | 58 +++++++++++++++++++++++++--
 1 file changed, 54 insertions(+), 4 deletions(-)

diff --git a/Lib/test/test_inspect/test_inspect.py b/Lib/test/test_inspect/test_inspect.py
index 555efb78dcc6aa..4d5e620b449542 100644
--- a/Lib/test/test_inspect/test_inspect.py
+++ b/Lib/test/test_inspect/test_inspect.py
@@ -2859,21 +2859,71 @@ def test_easy_debugging(self):
             self.assertIn(name, str(state))
 
     def test_getcoroutinelocals(self):
+        from collections.abc import Coroutine
+
         @types.coroutine
         def gencoro():
             yield
 
-        gencoro = gencoro()
+        genc = gencoro()
         async def func(a=None):
             b = 'spam'
-            await gencoro
+            await genc
 
         coro = func()
         self.assertEqual(inspect.getcoroutinelocals(coro),
-                         {'a': None, 'gencoro': gencoro})
+                         {'a': None, 'genc': genc})
         coro.send(None)
         self.assertEqual(inspect.getcoroutinelocals(coro),
-                         {'a': None, 'gencoro': gencoro, 'b': 'spam'})
+                         {'a': None, 'genc': genc, 'b': 'spam'})
+
+        def make_coro():
+            genc = gencoro()
+            async def coro():
+                local1 = 1
+                await genc
+                local2 = 2
+            return genc, coro()
+
+        # Does not have `cr_frame`:
+
+        class CustomCoro(Coroutine):
+            def __init__(self, coro):
+                self._coro = coro
+
+            def __await__(self):
+                return self._coro.__await__
+
+            def send(self, value):
+                return self._coro.send(value)
+
+            def throw(self, *args):
+                return super().throw(*args)
+
+        genc, cr = make_coro()
+        self.assertEqual(inspect.getcoroutinelocals(CustomCoro(cr)), {})
+
+        cr.send(None)
+        self.assertEqual(inspect.getcoroutinelocals(CustomCoro(cr)), {})
+
+        # Has custom `cr_frame` implementation:
+
+        class CustomCoroWithCRFrame(CustomCoro):
+            @property
+            def cr_frame(self):
+                return self._coro.cr_frame
+
+        genc, cr = make_coro()
+        self.assertEqual(
+            inspect.getcoroutinelocals(CustomCoroWithCRFrame(cr)),
+            {'genc': genc},
+        )
+
+        cr.send(None)
+        self.assertEqual(
+            inspect.getcoroutinelocals(CustomCoroWithCRFrame(cr)),
+            {'genc': genc, 'local1': 1},
+        )
 
 
 @support.requires_working_socket()