Your IP : 216.73.216.213
a
XC?h� � @ sf d Z ddlZddlZddlmZ ddlmZ ddlmZ dd� Z G dd � d ej
�Z e d
krbe�
� dS ) z Tests for asyncio/threads.py� N)�
ContextVar)�mock)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �C/opt/alt/python39/lib64/python3.9/test/test_asyncio/test_threads.py�tearDownModule s r c sX e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�
ToThreadTestsc s$ t � �� t�� | _t�| j� d S r )�super�setUpr Znew_event_loop�loop�set_event_loop��self�� __class__r r r s
zToThreadTests.setUpc s: | j �| j �� � | j �� t�d � d | _ t� �� d S r )r
�run_until_completeZshutdown_default_executor�closer r r �tearDownr r r r r s �
zToThreadTests.tearDownc C s& dd� }| j �|� �}| �|d� d S )Nc � s t �tddg�I d H S )N�( � )r � to_thread�sumr r r r �main s z*ToThreadTests.test_to_thread.<locals>.main�* )r
r � assertEqual�r r �resultr r r �test_to_thread s z ToThreadTests.test_to_threadc sR dd� � � fdd�}| � td�� | j�|� � W d � n1 sD0 Y d S )Nc S s t d��d S )N�test)� RuntimeErrorr r r r �
raise_runtime% s z=ToThreadTests.test_to_thread_exception.<locals>.raise_runtimec � s t �� �I d H d S r �r r r �r" r r r ( s z4ToThreadTests.test_to_thread_exception.<locals>.mainr )ZassertRaisesRegexr! r
r �r r r r$ r �test_to_thread_exception$ s z&ToThreadTests.test_to_thread_exceptionc s. t �� � � fdd�}| j�|� � � �� d S )Nc � s t �� �I d H d S r r# r ��funcr r r 1 s z/ToThreadTests.test_to_thread_once.<locals>.main)r �Mockr
r Zassert_called_oncer% r r' r �test_to_thread_once. s z!ToThreadTests.test_to_thread_oncec s4 t �� � � fdd�}| j�|� � | �� jd� d S )Nc � s: g } t d�D ]}t�� �}| �|� q tj| � I d H d S )N�
)�ranger r �appendZgather)Zfuts�_Zfutr' r r r : s
z5ToThreadTests.test_to_thread_concurrent.<locals>.mainr+ )r r) r
r r Z
call_countr% r r' r �test_to_thread_concurrent7 s z'ToThreadTests.test_to_thread_concurrentc s4 t �� � � fdd�}| j�|� � � jddd� d S )Nc � s t j� ddd�I d H d S )Nr T�Z somethingr# r r' r r r H s z6ToThreadTests.test_to_thread_args_kwargs.<locals>.mainr Tr0 )r r) r
r Zassert_called_once_withr% r r' r �test_to_thread_args_kwargsD s z(ToThreadTests.test_to_thread_args_kwargsc s@ t d���fdd�� � �fdd�}| j�|� �}| �|d� d S )N�test_ctxc s � � � S r )�getr )r2 r r �get_ctxQ s z9ToThreadTests.test_to_thread_contextvars.<locals>.get_ctxc � s �� d� t�� �I d H S )N�parrot)�setr r r �r4 r2 r r r T s
z6ToThreadTests.test_to_thread_contextvars.<locals>.mainr5 )r r
r r r r r7 r �test_to_thread_contextvarsN s
z(ToThreadTests.test_to_thread_contextvars) �__name__�
__module__� __qualname__r r r r& r* r/ r1 r8 �
__classcell__r r r r r
s
r
�__main__)�__doc__r ZunittestZ contextvarsr r Ztest.test_asyncior Z
test_utilsr ZTestCaser
r9 r r r r r �<module> s M