Your IP : 216.73.216.213
a
XC?h7 � @ s� d dl Z d dlZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
G dd� d�Z G dd� d�Z G dd� dej
�ZG d d
� d
e�ZG d d � d e�ZdS )
� N)�supportc @ s� e Zd Zdd� Zdd�dd�Zdd� Zd d
� Zd d � Zej d
e
j ddd�dd�Z ej e
j fdd�Z
dd� Zdd� Zejdd� �Zdd � Zd
S ) �FunctionalTestCaseMixinc C s t �� S �N)�asyncioZnew_event_loop��self� r �A/opt/alt/python39/lib64/python3.9/test/test_asyncio/functional.py�new_loop s z FunctionalTestCaseMixin.new_loopg{�G�z�?)�delayc C s | j �t�|�� d S r )�loopZrun_until_completer �sleep)r r r r r �run_loop_briefly s z(FunctionalTestCaseMixin.run_loop_brieflyc C s | j �|� | j�|� d S r )�._FunctionalTestCaseMixin__unhandled_exceptions�appendr Zdefault_exception_handler)r r �contextr r r �loop_exception_handler s z.FunctionalTestCaseMixin.loop_exception_handlerc C s, | � � | _t�d � | j�| j� g | _d S r )r
r r �set_event_loopZset_exception_handlerr r r r r r �setUp s
z FunctionalTestCaseMixin.setUpc C sZ zB| j �� | jr0td� t�| j� | �d� W t�d � d | _ nt�d � d | _ 0 d S )Nz2Unexpected calls to loop.call_exception_handler():z1unexpected calls to loop.call_exception_handler())r �closer �print�pprint�failr r r r r r �tearDown s
�
z FunctionalTestCaseMixin.tearDownN� �
)�family�addr�timeout�backlog� max_clientsc C s� |d u rPt td�rL|tjkrLt�� �}|j}W d � qP1 s@0 Y nd}tj|||d�}|d u rptd��|dkr�td��|�|� t | ||||�S )N�AF_UNIX)z 127.0.0.1r )r r �timeout is requiredr �#only blocking sockets are supported)
�hasattr�socketr! �tempfileZNamedTemporaryFile�nameZ
create_server� RuntimeError�
settimeout�TestThreadedServer) r Z server_progr r r r r �tmp�sockr r r �
tcp_server- s
&
�z"FunctionalTestCaseMixin.tcp_serverc C sF t � |t j�}|d u r td��|dkr.td��|�|� t| |||�S )Nr" r r# )r% Z SOCK_STREAMr( r) �TestThreadedClient)r Z client_progr r r, r r r �
tcp_clientE s
�z"FunctionalTestCaseMixin.tcp_clientc O s$ t td�st�| j|dtji|��S �Nr! r )r$ r% �NotImplementedErrorr- r! �r �args�kwargsr r r � unix_serverT s
z#FunctionalTestCaseMixin.unix_serverc O s$ t td�st�| j|dtji|��S r0 )r$ r% r1 r/ r! r2 r r r � unix_clientY s
z#FunctionalTestCaseMixin.unix_clientc c s� t �� �p}tj�|d�}z,|V W zt�|� W qj tyB Y qj0 n$zt�|� W n tyf Y n0 0 W d � n1 s~0 Y d S )Nr, )r& ZTemporaryDirectory�os�path�join�unlink�OSError)r Ztd�fnr r r �unix_sock_name^ s
� z&FunctionalTestCaseMixin.unix_sock_namec C s* z| j �� W | �|� n | �|� 0 d S r )r �stopr �r �exr r r �_abort_socket_testj s z*FunctionalTestCaseMixin._abort_socket_test)�__name__�
__module__� __qualname__r
r r r r r% ZAF_INETr ZLOOPBACK_TIMEOUTr- r/ r5 r6 �
contextlib�contextmanagerr= rA r r r r r
s& � �
r c @ s< e Zd Zdd� Zdd� Zddd�dd �Zd
d � Zd d
� ZdS )�TestSocketWrapperc C s
|| _ d S r )�_TestSocketWrapper__sock�r r, r r r �__init__x s zTestSocketWrapper.__init__c C s< d}t |�|k r8| �|t |� �}|dkr.t�||7 }q|S )N� )�lenZrecv�ConnectionAbortedError)r �nZbuf�datar r r �recv_all{ s
zTestSocketWrapper.recv_allFN)� server_side�server_hostnamec C s^ |j | j||dd�}z2z |�� W n |�� � Y n0 W | j�� n | j�� 0 || _d S )NF)rQ rR Zdo_handshake_on_connect)Z wrap_socketrH Z do_handshaker )r Z ssl_contextrQ rR Zssl_sockr r r � start_tls� s �
zTestSocketWrapper.start_tlsc C s t | j|�S r )�getattrrH )r r' r r r � __getattr__� s z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typerB rH r r r r �__repr__� s zTestSocketWrapper.__repr__)rB rC rD rJ rP rS rU rX r r r r rG v s
� rG c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SocketThreadc C s d| _ | �� d S �NF)�_activer9 r r r r r> � s zSocketThread.stopc C s | � � | S r )�startr r r r � __enter__� s zSocketThread.__enter__c G s | � � d S r )r> )r �excr r r �__exit__� s zSocketThread.__exit__N)rB rC rD r> r] r_ r r r r rY � s rY c @ s e Zd Zdd� Zdd� ZdS )r. c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrJ �daemon�_timeout�_sockr[ �_prog�_test)r �testr, �progr r r r rJ � s zTestThreadedClient.__init__c
C sJ z| � t| j�� W n0 tyD } z| j�|� W Y d }~n
d }~0 0 d S r )re rG rd � Exceptionrf rA r? r r r �run� s zTestThreadedClient.runN)rB rC rD rJ rj r r r r r. � s
r. c sH e Zd Zdd� Z� fdd�Zdd� Zdd� Zd d
� Zed d � �Z � Z
S )
r* c C sh t j�| d d d� d| _d| _d| _|| _|| _|| _d| _ || _
t � � \| _
| _| j
�d� || _d S )Nz test-serverTr F)r` ra rJ rb �_clientsZ_finished_clients� _max_clientsrc rd r[ re r% Z
socketpair�_s1�_s2� setblockingrf )r rg r, rh r r r r r rJ � s zTestThreadedServer.__init__c sX zF| j r:| j �� dkr:z| j �d� W n ty8 Y n0 W t� �� n t� �� 0 d S )N���s stop)rn �fileno�sendr; �superr> r �� __class__r r r> � s zTestThreadedServer.stopc C sn zR| j �$ | j �d� | �� W d � n1 s20 Y W | j�� | j�� n| j�� | j�� 0 d S rZ )rd ro �_runrm r rn r r r r rj � s (
�
zTestThreadedServer.runc C s< | j �r8| j| jkrd S t�| j| jgg g | j�\}}}| j|v rFd S | j|v r z| j�� \}}W n6 tyx Y q Y q t j
y� | j s�Y d S � Y q 0 | jd7 _|� | j� z2|� | � |� W d � n1 s�0 Y W q t
�y4 } z4d| _ z� W | j�|� n| j�|� 0 W Y d }~q d }~0 0 q d S )Nr F)r[ rk rl �selectrd rm rc Zaccept�BlockingIOErrorr% r r) �_handle_clientri rf rA )r �r�w�xZconnr r@ r r r rv � s6 �
,zTestThreadedServer._runc C s | � t|�� d S r )re rG rI r r r ry s z!TestThreadedServer._handle_clientc C s
| j �� S r )rd Z getsocknamer r r r r s zTestThreadedServer.addr) rB rC rD rJ r> rj rv ry �propertyr �
__classcell__r r rt r r* � s
"r* )r Zasyncio.eventsrE r7 r rw r% r& r` rg r r rG ra rY r. r* r r r r �<module> s i(