Add context arg to Runner.run()

asvetlov · asvetlov · commit 98e7a22febe1 · 2022-03-14T15:14:56.000+02:00
diff --git a/Lib/asyncio/runners.py b/Lib/asyncio/runners.py
@@ -46,26 +46,27 @@ def close(self):
         if self._loop is None:
             return
         try:
-            _cancel_all_tasks(self._loop)
-            self._loop.run_until_complete(self._loop.shutdown_asyncgens())
-            self._loop.run_until_complete(self._loop.shutdown_default_executor())
+            loop = self._loop
+            _cancel_all_tasks(loop)
+            loop.run_until_complete(loop.shutdown_asyncgens())
+            loop.run_until_complete(loop.shutdown_default_executor())
         finally:
-            self._loop.close()
+            loop.close()
             self._loop = None
 
-    def run(self, coro):
+    def run(self, coro, *, context=None):
         """Run a coroutine inside the embedded event loop."""
         if not coroutines.iscoroutine(coro):
             raise ValueError("a coroutine was expected, got {!r}".format(coro))
 
-        return self._loop.run_until_complete(coro)
+        task = self._loop.create_task(coro, context=context)
+        return self._loop.run_until_complete(task)
 
     def get_loop(self):
         """Returnb embedded event loop."""
         return self._loop
 
 
-
 def run(main, *, debug=None):
     """Execute the coroutine and return the result.
 
