Your IP : 216.73.216.213
U
i�f * � @ 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� de�ZG dd� de�ZG dd� de�ZG dd� de�Zedk�re�� dS )� N)�mockZwin32z Windows only)�windows_events)�utilsc C s t �d � d S �N)�asyncio�set_event_loop_policy� r r �J/opt/alt/python38/lib64/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� Zdd� Zdd� ZdS )�WindowsEventsTestCasec C s || _ t|� d S r )� _unraisable�print)r Z
unraisabler r r �_unraisablehook( s z%WindowsEventsTestCase._unraisablehookc C s t j| _d | _| jt _d S r )�sys�unraisablehook�_prev_unraisablehookr r! r
r r r �setUp. s zWindowsEventsTestCase.setUpc C s | j t_| �| j� d S r )r$ r" r# Z assertIsNoner r
r r r �tearDown3 s z WindowsEventsTestCase.tearDownN)r r r 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_loop� call_soon�start� run_foreverZfail�KeyboardInterruptr )r r. �thread�loopr r r � test_ctrl_c9 s
z ProactorLoopCtrlC.test_ctrl_cN)r r r r: r r r r r' 7 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 �coroQ s zAProactorMultithreading.test_run_from_nonmain_thread.<locals>.coroc s$ t �� } | �� � � | �� d�d S )NT)r Znew_event_loop�run_until_completer �r9 �r< Zfinishedr r �funcT s zAProactorMultithreading.test_run_from_nonmain_thread.<locals>.funcr/ )r1 r2 r5 r �
assertTrue)r r@ r8 r r? r � test_run_from_nonmain_threadN s z3ProactorMultithreading.test_run_from_nonmain_threadN)r r r rB r r r r r; M s r; c sd e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z � Z
S )�
ProactorTestsc s$ t � �� t�� | _| �| j� d S r )�superr% r �ProactorEventLoopr9 Zset_event_loopr
�� __class__r r r% d s
zProactorTests.setUpc C sh t �� \}}| j�|t�� �}tj| j�|d�| jd�}|�� | j� |� | �
|� � d� |�� d S )N�d r> r ) �socketZ
socketpairr9 Z_make_socket_transportr �ProtocolZ
ensure_futureZ sock_recvr r= � assertEqual�result)r �a�br �fr r r �
test_closei 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_bindr s
z ProactorTests.test_double_bindc C s | j �| �� �}| �|d� d S )N�done)r9 r= �
_test_piperK )r �resr r r � test_pipey 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� r> c s � S r r r �Zprotocolr r �<lambda>� r z*ProactorTests._test_pipe.<locals>.<lambda>z lower-{}
z LOWER-{}
rX )rQ rR rT �FileNotFoundErrorr9 Zcreate_pipe_connectionr rJ Zstart_serving_piper �assertIsInstancer rS �range� StreamReaderZStreamReaderProtocolZ TransportrK r � enumerater �format�encode�readliner ) r rV ZserverZclients�iZ
stream_readerr �proto�r�wZresponser r] r rY } 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_BUSYZwinerrorr Zpatch�objectr9 � _proactorZ connect_pipeZ create_task�cancelrT r �CancelledErrorr= )r �excZconnectr< 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�?)rl � CreateEvent�
addCleanup�_winapi� CloseHandler9 rn �wait_for_handler) r= rK � assertFalserL rA ZSetEventro )r �event�futr5 rX �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 )NTFrs r r( )rl rt ru rv rw r9 rn rx ro r) rT r rp r= rA )r rz r{ r5 r| 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 r^ � r z;ProactorTests.test_read_self_pipe_restart.<locals>.<lambda>) r ZMockr9 Zcall_exception_handlerZrun_in_executor�stopr6 r= r3 ry Zcalled)r rO r r r �test_read_self_pipe_restart� s
z)ProactorTests.test_read_self_pipe_restartc sz t �� �� �fdd�}t j|d�}t�� d }� j�|j� � jsf� j�� � j� � t�� |kr<qfq<��
� |� � d S )Nc s( �� � s$� j�dd� � t�d� q d S )Nc S s d S r r r r r r r^ r zEProactorTests.test_loop_restart.<locals>.threadMain.<locals>.<lambda>g{�G�z�?)Zis_setr9 Zcall_soon_threadsafer) r* r �r r r r �
threadMain s z3ProactorTests.test_loop_restart.<locals>.threadMainr/ rs ) r1 ZEventr2 r) r9 r4 r5 r r r6 �setr )r r� ZthrZend_timer r� r �test_loop_restart s
z ProactorTests.test_loop_restart)r r r r% rP rW r[ rY rr r} r~ r� r� �
__classcell__r r rF r rC b s &
" rC 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� rE 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__) rQ r+ rI r" r) r1 Zunittestr �platformZSkipTestrl rv r r Ztest.test_asyncior Z
test_utilsr
rJ r ZTestCaser r' r; rC r� r r� r r r r �<module> s0
: