mirror of
https://github.com/python/cpython.git
synced 2025-08-03 16:39:00 +00:00
bpo-32309: Add support for contextvars in asyncio.to_thread() (GH-20278)
Allows contextvars from the main thread to be accessed in the separate thread used in `asyncio.to_thread()`. See the [discussion](https://github.com/python/cpython/pull/20143GH-discussion_r427808225) in GH-20143 for context.
Automerge-Triggered-By: @aeros
(cherry picked from commit 0f56263e62
)
Co-authored-by: Kyle Stanley <aeros167@gmail.com>
This commit is contained in:
parent
6387358939
commit
3e650545bf
3 changed files with 25 additions and 3 deletions
|
@ -3,6 +3,7 @@
|
|||
import asyncio
|
||||
import unittest
|
||||
|
||||
from contextvars import ContextVar
|
||||
from unittest import mock
|
||||
from test.test_asyncio import utils as test_utils
|
||||
|
||||
|
@ -74,6 +75,19 @@ class ToThreadTests(test_utils.TestCase):
|
|||
self.loop.run_until_complete(main())
|
||||
func.assert_called_once_with('test', something=True)
|
||||
|
||||
def test_to_thread_contextvars(self):
|
||||
test_ctx = ContextVar('test_ctx')
|
||||
|
||||
def get_ctx():
|
||||
return test_ctx.get()
|
||||
|
||||
async def main():
|
||||
test_ctx.set('parrot')
|
||||
return await asyncio.to_thread(get_ctx)
|
||||
|
||||
result = self.loop.run_until_complete(main())
|
||||
self.assertEqual(result, 'parrot')
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue