Your IP : 216.73.216.213
U
i�f� � @ s� d dl Z d dlZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZG dd� d�Z G dd� d�Z
G dd� dej �Z G dd � d e �Z
G d
d � d e �ZdS ) � Nc @ s� e Zd Zdd� Zdd�dd�Zdd� Zd d
� Zd d � Zej d
dddd�dd�Z
ej dfdd�Z dd� Z dd� Z
ejdd� �Zd d � Zd
S ) �FunctionalTestCaseMixinc C s t �� S �N)�asyncioZnew_event_loop��self� r �A/opt/alt/python38/lib64/python3.8/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 sB | � � | _t�d � | j�| j� g | _tjj| _ dd� tj_d S )Nc S s d S r r r r r r �<lambda>! � z/FunctionalTestCaseMixin.setUp.<locals>.<lambda>)
r r r �set_event_loopZset_exception_handlerr r �events�_get_running_loop�_old_get_running_loopr r r r �setUp s
z FunctionalTestCaseMixin.setUpc C sT z2| j�� | jr0td� t � | j� | �
d� W 5 | j tj_t�d � d | _X d S )Nz2Unexpected calls to loop.call_exception_handler():z1unexpected calls to loop.call_exception_handler()) r r r r r r �closer �print�pprint�failr r r r �tearDown# s
z FunctionalTestCaseMixin.tearDownN� � �
)�family�addr�timeout�backlog� max_clientsc C s� |d kr<t td�r8|tjkr8t�� � }|j}W 5 Q R X nd}tj|||d�}|d kr\td��|dkrltd��|�|� 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% Ztmp�sockr r r �
tcp_server1 s$
�z"FunctionalTestCaseMixin.tcp_serverc C sF t � |t j�}|d kr td��|dkr.td��|�|� t| |||�S )Nr' r r( )r* Z SOCK_STREAMr- r. �TestThreadedClient)r Z client_progr! r# r0 r r r �
tcp_clientI s
�z"FunctionalTestCaseMixin.tcp_clientc O s$ t td�st�| j|dtji|��S �Nr& r! )r) r* �NotImplementedErrorr1 r& �r �args�kwargsr r r � unix_serverX s
z#FunctionalTestCaseMixin.unix_serverc O s$ t td�st�| j|dtji|��S r4 )r) r* r5 r3 r& r6 r r r � unix_client] s
z#FunctionalTestCaseMixin.unix_clientc c sX t �� �F}tj�|d�}z
|V W 5 zt�|� W n tk
rF Y nX X W 5 Q R X d S )Nr0 )r+ ZTemporaryDirectory�os�path�join�unlink�OSError)r Ztd�fnr r r �unix_sock_nameb s
z&FunctionalTestCaseMixin.unix_sock_namec C s z| j�� W 5 | � |� X d S r )r r �stop�r �exr r r �_abort_socket_testn s z*FunctionalTestCaseMixin._abort_socket_test)�__name__�
__module__� __qualname__r r
r r r r* ZAF_INETr1 r3 r9 r: �
contextlib�contextmanagerrA rE 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 r0 r r r �__init__| s zTestSocketWrapper.__init__c C s< d}t |�|k r8| �|t |� �}|dkr.t�||7 }q|S )Nr )�lenZrecv�ConnectionAbortedError)r �nZbuf�datar r r �recv_all s
zTestSocketWrapper.recv_allFN)� server_side�server_hostnamec C sT |j | j||dd�}z(z |�� W n |�� � Y nX W 5 | j�� X || _d S )NF)rT rU Zdo_handshake_on_connect)Z wrap_socketrL r Z do_handshake)r Z ssl_contextrT rU Zssl_sockr r r � start_tls� s � zTestSocketWrapper.start_tlsc C s t | j|�S r )�getattrrL )r r, r r r � __getattr__� s z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typerF rL r r r r �__repr__� s zTestSocketWrapper.__repr__)rF rG rH rN rS rV rX r[ r r r r rK z s
� rK c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SocketThreadc C s d| _ | �� d S )NF)�_activer= r r r r rB � s zSocketThread.stopc C s | � � | S r )�startr r r r � __enter__� s zSocketThread.__enter__c G s | � � d S r )rB )r �excr r r �__exit__� s zSocketThread.__exit__N)rF rG rH rB r_ ra r r r r r\ � s r\ c @ s e Zd Zdd� Zdd� ZdS )r2 c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrN �daemon�_timeout�_sockr] �_prog�_test)r �testr0 �progr# r r r rN � s zTestThreadedClient.__init__c
C sH z| � t| j�� W n. tk
rB } z| j�|� W 5 d }~X Y nX d S r )rg rK rf � Exceptionrh rE rC r r r �run� s zTestThreadedClient.runN)rF rG rH rN rl r r r r r2 � s
r2 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)rb rc rN rd �_clientsZ_finished_clients� _max_clientsre rf r] rg r* Z
socketpair�_s1�_s2� setblockingrh )r ri r0 rj r# r% r r r rN � s zTestThreadedServer.__init__c sP z>| jr<| j�� dkr<z| j�d� W n tk
r: Y nX W 5 t � �� X d S )N���s stop)�superrB rp �fileno�sendr? r �� __class__r r rB � s
zTestThreadedServer.stopc C sF z*| j� | j�d� | �� W 5 Q R X W 5 | j �� | j�� X d S )Nr )ro r rp rf rq �_runr r r r rl � s
zTestThreadedServer.runc C s | j �r| j| jkrd S t�| j| jgg g | j�\}}}| j|krFd S | j|kr z| j�� \}}W n: tk
rz Y q Y q t j
k
r� | j s�Y d S � Y q X | jd7 _|� | j� z |� | � |� W 5 Q R X W q t
k
�r } z d| _ z� W 5 | j�|� X W 5 d }~X Y q X q d S )Nr F)r] rm rn �selectrf ro re Zaccept�BlockingIOErrorr* r# r. �_handle_clientrk rh rE )r �r�w�xZconnr" rD r r r rx � s<
�
zTestThreadedServer._runc C s | � t|�� d S r )rg rK rM r r r r{ s z!TestThreadedServer._handle_clientc C s
| j �� S r )rf Z getsocknamer r r r r" s zTestThreadedServer.addr) rF rG rH rN rB rl rx r{ �propertyr" �
__classcell__r r rv r r/ � s
"r/ )r Zasyncio.eventsrI r; r ry r* r+ rb r rK rc r\ r2 r/ r r r r �<module> s n(