File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_windows_events.cpython-310.pyc
o
�i�* � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ejdkr,e� 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�Zedkre�� dS dS )� N)�mock�win32z Windows only)�windows_events)�utilsc C s t �d � d S �N)�asyncio�set_event_loop_policy� r r �B/usr/local/lib/python3.10/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__ �
zUpperProto.__init__c C s
|| _ d S r )�trans)r r r r r
�connection_made r zUpperProto.connection_madec C s> | j �|� d|v r | j�d�| j ��� � | j�� d S 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 @ � e Zd Zdd� ZdS )�ProactorLoopCtrlCc C sz dd� }t j|d�}t�� }z"z|�|j� |�� | �d� W n ty) Y nw W | � |� n| � |� w |�
� 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_loopZ call_soon�start� run_foreverZfail�KeyboardInterrupt�
close_loopr )r r( �thread�loopr r r
� test_ctrl_c) s �� z ProactorLoopCtrlC.test_ctrl_cN)r r r r3 r r r r
r! ' s r! c @ r )�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 �r2 �r5 �finishedr r
�funcD s zAProactorMultithreading.test_run_from_nonmain_thread.<locals>.funcr) )r+ r, r- r �
assertTrue)r r: r1 r r8 r
� test_run_from_nonmain_thread> s z3ProactorMultithreading.test_run_from_nonmain_threadN)r r r r< r r r r
r4 = s r4 c st e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�
ProactorTestsc s$ t � �� t�� | _| �| j� d S r )�super�setUpr �ProactorEventLoopr2 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 r7 r ) �socket�
socketpairr2 Z_make_socket_transportr �ProtocolZ
ensure_futureZ sock_recvr r6 � assertEqual�result)r �a�br �fr r r
�
test_closeY s zProactorTests.test_closec C sV dt �� }t�|�}| �t��
t�|� W d � n1 s w Y |�� 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 � | j �| �� �}| �|d� d S �N�done)r2 r6 �
_test_piperG �r �resr r r
� test_pipei � zProactorTests.test_pipec � s� �dt �� }| �t�� | j�tj|�I d H W d � n1 s"w Y | j�t |�I d H \}| �
|t j � g }t
d�D ]5}tj| jd�}tj|| jd�� | j�� fdd�|�I d H \}}| �
|tj� | �� |� |�||f� q?t|�D ]\}\}} | �d�|��� � qyt|�D ] \}\}} |�� I d H }
| �|
d�|��� � | �� q�|�� | �t�� | j�tj|�I d H W d � dS 1 s�w Y dS ) Nz\\.\pipe\_test_pipe-%s� r7 c s � S r r r ��protocolr r
�<lambda>~ � z*ProactorTests._test_pipe.<locals>.<lambda>z lower-{}
z LOWER-{}
rV )rM rN rP �FileNotFoundErrorr2 Zcreate_pipe_connectionr rF �start_serving_piper �assertIsInstancer rO �range� StreamReaderZStreamReaderProtocolZ TransportrG r � enumerater �format�encode�readliner ) r rR �serverZclients�iZ
stream_readerr �proto�r�w�responser r] r
rW m sL � � � � � �
�
��zProactorTests._test_pipec C s� t � }tj|_tjjtd|d��>}| jj� d�}| j�
|�}|� � | � t
j�� | j�|� W d � n1 s9w Y W d � d S W d � d S 1 sQw Y d S )N� ConnectPipe)Z side_effectZ pipe_address)�OSError� _overlapped�ERROR_PIPE_BUSY�winerrorr Zpatch�objectr2 � _proactorZ connect_pipe� create_task�cancelrP r �CancelledErrorr6 )r �exc�connectr5 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| koAdkn |� t �|� | jj�|d�}| j�� }| j� |�}| j�� | }| �
|d� | �
|� � � | �
d| ko}dk n |� |�� d S ) NTFg �?g�������?g�������?�
r g333333�?)rr � CreateEvent�
addCleanup�_winapi� CloseHandler2 rv �wait_for_handler# r6 rG � assertFalserH r; ZSetEventrx )r �event�futr- rV �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 d � n1 s7w Y | j� � | }| �d| koNdk n |� | jj�|�}|�� |�� d S )NTFr} r r" )rr r~ r r� r� r2 rv r� rx r# rP r ry r6 r; )r r� r� r- 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 ZMockr2 Zcall_exception_handlerZrun_in_executor�stopr. r6 r0 r� Zcalled)r rK r r r
�test_read_self_pipe_restart� s
z)ProactorTests.test_read_self_pipe_restartc C s� | j j}tjtjd�}d }| �t�� |�||� W d � n1 s#w Y | �t�� |j|d|d� W d � n1 s@w Y |�� d S )N)�types abc)�addr) r2 rv rD �
SOCK_DGRAMrP � TypeErrorr{ �sendtor )r Zproactor�sockZ bad_addressr r r
� test_address_argument_type_error� s � � z.ProactorTests.test_address_argument_type_errorc C rT rU )r2 r6 �_test_client_pipe_statrG rX r r r
�test_client_pipe_stat� r[ z#ProactorTests.test_client_pipe_statc � s �dt �� � � fdd�}| �t�� |� I d H W d � n1 s#w Y | j�tj� �I d H \}| �|t j
� g �| j� �fdd�� t d�D ] }| j�
|� �I d H qK| �t��d�� |�� | �t�� |� I d H W d � dS 1 s|w Y dS ) Nz!\\.\pipe\test_client_pipe_stat-%sc
� s� �t �� �} z4z
t�t �� �� W n ty* } z
|jt jkr � W Y d }~n
d }~ww W t�| � d S W t�| � d S t�| � w r )rr rp r� r� rq rt rs )�h�e)rR r r
�probe s �
�����z3ProactorTests._test_client_pipe_stat.<locals>.probec s
� � |�S r )r )�_r )�errorsr r
r_ s
z6ProactorTests._test_client_pipe_stat.<locals>.<lambda>r\ r rV )rM rN rP ra r2 rb r rF rc r rO Zset_exception_handlerrd rw rG �lenr )r r� rj rk r )rR r� r
r� s( � �
��z$ProactorTests._test_client_pipe_stat)r r r r? rL rS rZ rW r| r� r� r� r� r� r� �
__classcell__r r rA r
r= R s &
" r= c @ s e Zd Zdd� Zdd� ZdS )�WinPolicyTestsc �L � fdd�}t �� }zt �t �� � t �|� � W t �|� d S t �|� w )Nc � � �� � t�� tj� d S r )rc r �get_running_loopZSelectorEventLoopr r r r
�main) �
��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( � �z'WinPolicyTests.test_selector_win_policyc r� )Nc � r� r )rc r r� r@ r r r r
r� 7 r� z5WinPolicyTests.test_proactor_win_policy.<locals>.main)r r� r Z WindowsProactorEventLoopPolicyr� r� r r r
�test_proactor_win_policy6 r� z'WinPolicyTests.test_proactor_win_policyN)r r r r� r� r r r r
r� & s r� �__main__)rM r% rD �sysr# r+ Zunittestr �platformZSkipTestrr r� r r Ztest.test_asyncior Z
test_utilsr rF r ZTestCaser! r4 r= r� r r� r r r r
�<module> s2
U �