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/20143#discussion_r427808225) in GH-20143 for context.

Automerge-Triggered-By: @aeros
This commit is contained in:
Kyle Stanley 2020-05-21 01:20:43 -04:00 committed by GitHub
parent 7efb826c3e
commit 0f56263e62
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 25 additions and 3 deletions

View file

@ -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()