summaryrefslogtreecommitdiff
path: root/Lib/test/test_asyncio/test_threads.py
diff options
context:
space:
mode:
authorMiss Islington (bot) <31488909+miss-islington@users.noreply.github.com>2020-05-20 22:38:00 -0700
committerGitHub <noreply@github.com>2020-05-20 22:38:00 -0700
commit3e650545bfe949fa435b0d41e54986f615891ec8 (patch)
tree6555a3f8d81983ca14ee94ae4a6cc2c36fb96b11 /Lib/test/test_asyncio/test_threads.py
parent63873589392aa63a97d15c793a60b2f3f6ab9bc1 (diff)
downloadcpython-git-3e650545bfe949fa435b0d41e54986f615891ec8.tar.gz
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 0f56263e62ba91d0baae40fb98947a3a98034a73) Co-authored-by: Kyle Stanley <aeros167@gmail.com>
Diffstat (limited to 'Lib/test/test_asyncio/test_threads.py')
-rw-r--r--Lib/test/test_asyncio/test_threads.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/Lib/test/test_asyncio/test_threads.py b/Lib/test/test_asyncio/test_threads.py
index 99a00f2183..2af322421d 100644
--- a/Lib/test/test_asyncio/test_threads.py
+++ b/Lib/test/test_asyncio/test_threads.py
@@ -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()