File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_threads.cpython-310.opt-2.pyc
o
�iZ � @ s\ d dl Z d dlZd dlmZ d dlmZ dd� ZG dd� dej�Zedkr,e� � dS dS ) � N)�
ContextVar)�mockc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �;/usr/local/lib/python3.10/test/test_asyncio/test_threads.py�tearDownModule
s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
ToThreadTestsc � s( �t �tddg�I d H }| �|d� d S )N�( � �* )r � to_thread�sum� assertEqual)�self�resultr r r �test_to_thread s �z ToThreadTests.test_to_threadc � sL �dd� }| � td�� t�|�I d H W d � d S 1 s w Y d S )Nc S s t d��)N�test)� RuntimeErrorr r r r �
raise_runtime � z=ToThreadTests.test_to_thread_exception.<locals>.raise_runtimer )ZassertRaisesRegexr r r
)r r r r r �test_to_thread_exception s
�"�z&ToThreadTests.test_to_thread_exceptionc � s& �t �� }t�|�I d H |�� d S r )r �Mockr r
Zassert_called_once�r �funcr r r �test_to_thread_once s � z!ToThreadTests.test_to_thread_oncec � sR �t �� }g }td�D ] }t�|�}|�|� q tj|� I d H | �|jd� d S )N�
) r r �ranger r
�appendZgatherr Z
call_count)r r Zfuts�_�futr r r �test_to_thread_concurrent s �
z'ToThreadTests.test_to_thread_concurrentc � s2 �t �� }tj|ddd�I d H |jddd� d S )Nr T)Z something)r r r r
Zassert_called_once_withr r r r �test_to_thread_args_kwargs+ s �z(ToThreadTests.test_to_thread_args_kwargsc � s@ �t d�� � fdd�}� �d� t�|�I d H }| �|d� d S )N�test_ctxc s � � � S r )�getr �r# r r �get_ctx6 r z9ToThreadTests.test_to_thread_contextvars.<locals>.get_ctxZparrot)r �setr r
r )r r& r r r% r �test_to_thread_contextvars3 s �
z(ToThreadTests.test_to_thread_contextvarsN) �__name__�
__module__� __qualname__r r r r! r" r( r r r r r s r �__main__)
r ZunittestZ contextvarsr r r ZIsolatedAsyncioTestCaser r) �mainr r r r �<module> s 1 �