File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_windows_events.cpython-38.pyc
U
/D�g�# � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ejdkrXe� d��d dl
Z
d dl Z d dl Z d dl m
Z
d dlmZ dd� ZG d d
� d
e j�ZG d d � d ej�ZG d
d� dej�ZG dd� dej�ZG dd� dej�Zedkr�e�� dS )� N)�mock�win32z Windows only)�windows_events)�utilsc C s t �d � d S �N)�asyncio�set_event_loop_policy� r r �D/opt/cppython/lib/python3.8/test/test_asyncio/test_windows_events.py�tearDownModule s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
UpperProtoc C s
g | _ d S r )�buf��selfr r r
�__init__ s zUpperProto.__init__c C s
|| _ d S r )�trans)r r r r r
�connection_made s zUpperProto.connection_madec C s: | j �|� d|kr6| j�d�| j ��� � | j�� d S )N�
� )r
�appendr �write�join�upper�close)r �datar r r
�
data_received s zUpperProto.data_receivedN)�__name__�
__module__� __qualname__r r r r r r r
r s r c @ s e Zd Zdd� ZdS )�ProactorLoopCtrlCc C sr dd� }t j|d�}t�� }z<z"|�|j� |�� | �d� W n t k
rT Y nX W 5 | �|� X |�
� d S )Nc S s t �d� t�tj� d S )N皙�����?)�time�sleep�signal� raise_signal�SIGINTr r r r
�SIGINT_after_delay+ s
z9ProactorLoopCtrlC.test_ctrl_c.<locals>.SIGINT_after_delay��targetz%should not fall through 'run_forever') � threading�Threadr Zget_event_loop�
close_loopZ call_soon�start� run_foreverZfail�KeyboardInterruptr )r r& �thread�loopr r r
� test_ctrl_c) s
z ProactorLoopCtrlC.test_ctrl_cN)r r r r1 r r r r
r ' s r c @ s e Zd Zdd� ZdS )�ProactorMultithreadingc sD d�dd� � � �fdd�}t j|d�}|�� |�� | ��� d S )NFc � s t �d�I d H d S )Nr )r r" r r r r
�coroA s zAProactorMultithreading.test_run_from_nonmain_thread.<locals>.coroc s$ t �� } | �� � � | �� d�d S )NT)r Znew_event_loop�run_until_completer �r0 �r3 �finishedr r
�funcD s zAProactorMultithreading.test_run_from_nonmain_thread.<locals>.funcr' )r) r* r, r �
assertTrue)r r8 r/ r r6 r
� test_run_from_nonmain_thread> s z3ProactorMultithreading.test_run_from_nonmain_threadN)r r r r: r r r r
r2 = s r2 c s\ e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )�
ProactorTestsc s$ t � �� t�� | _| �| j� d S r )�super�setUpr �ProactorEventLoopr0 Zset_event_loopr �� __class__r r
r= T s
zProactorTests.setUpc C sh t �� \}}| j�|t�� �}tj| j�|d�| jd�}|�� | j� |� | �
|� � d� |�� d S )N�d r5 r ) �socket�
socketpairr0 Z_make_socket_transportr �ProtocolZ
ensure_futureZ sock_recvr r4 � assertEqual�result)r �a�br �fr r r
�
test_closeY s zProactorTests.test_closec C sB dt �� }t�|�}| �t�� t�|� W 5 Q R X |�� d S )Nz \\.\pipe\test_double_bind-%s)�os�getpidr �
PipeServer� assertRaises�PermissionErrorr )r �ADDRESSZserver1r r r
�test_double_bindb s
z ProactorTests.test_double_bindc C s | j �| �� �}| �|d� d S )N�done)r0 r4 �
_test_piperE )r �resr r r
� test_pipei s zProactorTests.test_pipec � sz dt �� }| �t�� | j�tj|�I d H W 5 Q R X | j�t |�I d H \}| �
|t j � g }t
d�D ]j}tj| jd�}tj|| jd�� | j�� fdd�|�I d H \}}| �
|tj� | �� |� |�||f� qht|�D ] \}\}} | �d�|��� � q�t|�D ]:\}\}} |�� I d H }
| �|
d�|��� � | �� �q|�� | �t�� | j�tj|�I d H W 5 Q R X dS ) Nz\\.\pipe\_test_pipe-%s� r5 c s � S r r r ��protocolr r
�<lambda>~ r z*ProactorTests._test_pipe.<locals>.<lambda>z lower-{}
z LOWER-{}
rR )rK rL rN �FileNotFoundErrorr0 Zcreate_pipe_connectionr rD Zstart_serving_piper �assertIsInstancer rM �range� StreamReaderZStreamReaderProtocolZ TransportrE r � enumerater �format�encode�readliner ) r rP ZserverZclients�iZ
stream_readerr �proto�r�w�responser rW r
rS m sJ � � �
� �zProactorTests._test_pipec
C sv t � }tj|_tjjtd|d��L}| jj� d�}| j�
|�}|� � | � t
j�� | j�|� W 5 Q R X W 5 Q R X d S )NZ ConnectPipe)Z side_effectZ pipe_address)�OSError� _overlappedZERROR_PIPE_BUSY�winerrorr Zpatch�objectr0 � _proactorZ connect_pipeZ create_task�cancelrN r �CancelledErrorr4 )r �exc�connectr3 Ztaskr r r
�test_connect_pipe_cancel� s
� z&ProactorTests.test_connect_pipe_cancelc C s t �d ddd �}| �tj|� | jj�|d�}| j�� }| j� |�}| j�� | }| �
|d� | � |� � � | �
d| ko�dkn |� t �|� | jj�|d�}| j�� }| j� |�}| j�� | }| �
|d� | �
|� � � | �
d| ko�dk n |� |�� d S ) NTFg �?g�������?g�������?�
r g333333�?)rh � CreateEvent�
addCleanup�_winapi� CloseHandler0 rk �wait_for_handler! r4 rE � assertFalserF r9 ZSetEventrl )r �event�futr, rR �elapsedr r r
�test_wait_for_handle� s$
z"ProactorTests.test_wait_for_handlec C s� t �d ddd �}| �tj|� | jj�|d�}|�� | j� � }| �
t j �� | j�
|� W 5 Q R X | j� � | }| �d| ko�dk n |� | jj�|�}|�� |�� d S )NTFrq r r )rh rr rs rt ru r0 rk rv rl r! rN r rm r4 r9 )r rx ry r, rz r r r
�test_wait_for_handle_cancel� s
z)ProactorTests.test_wait_for_handle_cancelc C sr t �� | j_| j�d dd� �}| j�� | j�� | j�� | j�� | j�|� | �| j� | � | jjj
� d S )Nc S s d S r r r r r r
rY � r z;ProactorTests.test_read_self_pipe_restart.<locals>.<lambda>) r ZMockr0 Zcall_exception_handlerZrun_in_executor�stopr- r4 r+ rw Zcalled)r rI r r r
�test_read_self_pipe_restart� s
z)ProactorTests.test_read_self_pipe_restart)
r r r r= rJ rQ rU rS rp r{ r| r~ �
__classcell__r r r? r
r; R s &
"r; c @ s e Zd Zdd� Zdd� ZdS )�WinPolicyTestsc sD � fdd�}t �� }z t �t �� � t �|� � W 5 t �|� X d S )Nc � s � � t�� tj� d S r )r[ r �get_running_loopZSelectorEventLoopr r r r
�main� s �z5WinPolicyTests.test_selector_win_policy.<locals>.main)r �get_event_loop_policyr Z WindowsSelectorEventLoopPolicy�run�r r� Z
old_policyr r r
�test_selector_win_policy� s �z'WinPolicyTests.test_selector_win_policyc sD � fdd�}t �� }z t �t �� � t �|� � W 5 t �|� X d S )Nc � s � � t�� tj� d S r )r[ r r� r> r r r r
r� s �z5WinPolicyTests.test_proactor_win_policy.<locals>.main)r r� r Z WindowsProactorEventLoopPolicyr� r� r r r
�test_proactor_win_policy s �z'WinPolicyTests.test_proactor_win_policyN)r r r r� r� r r r r
r� � s r� �__main__)rK r# rB �sysr! r) Zunittestr �platformZSkipTestrh rt r r Ztest.test_asyncior Z
test_utilsr rD r ZTestCaser r2 r; r� r r� r r r r
�<module> s.
"