File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/functional.cpython-37.opt-1.pyc
B
��gt � @ 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 �8/usr/local/lib/python3.7/test/test_asyncio/functional.py�new_loop s z FunctionalTestCaseMixin.new_loopg{�G�z�?)�delayc C s | j �tj|| j d�� d S )N)�loop)r Zrun_until_completer Zsleep)r r r r r �run_loop_briefly s z(FunctionalTestCaseMixin.run_loop_brieflyc C s | j �|� | j�|� d S )N)�._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 )Nr 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_loop)r r r r �setUp s
z FunctionalTestCaseMixin.setUpc C sT z2| j �� | jr0td� t�| j� | �d� W d | jtj_ t�
d � d | _ X 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 r r r �tearDown# s
z FunctionalTestCaseMixin.tearDownN� � �
)�family�addr�timeout�backlog� max_clientsc
C s� |d kr<t td�r8|tjkr8t�� � }|j}W d Q R X nd}t�|tj�}|d krZtd��|dkrjtd��|�|� y|� |� |�
|� W n. t k
r� } z|� � | �W d d } ~ X Y nX t
| ||||�S )N�AF_UNIX)z 127.0.0.1r ztimeout is requiredr z#only blocking sockets are supported)�hasattr�socketr# �tempfileZNamedTemporaryFile�name� SOCK_STREAM� RuntimeError�
settimeoutZbindZlisten�OSErrorr �TestThreadedServer)
r Z server_progr r r r! r" Ztmp�sock�exr r r �
tcp_server1 s&
z"FunctionalTestCaseMixin.tcp_serverc C sF t � |t j�}|d kr td��|dkr.td��|�|� t| |||�S )Nztimeout is requiredr z#only blocking sockets are supported)r% r( r) r* �TestThreadedClient)r Z client_progr r r- r r r �
tcp_clientQ 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_server` s
z#FunctionalTestCaseMixin.unix_serverc O s$ t td�st�| j|dtji|��S )Nr# r )r$ r% r2 r1 r# )r r3 r4 r r r � unix_cliente s
z#FunctionalTestCaseMixin.unix_clientc c sX t �� �F}tj�|d�}z
|V W d yt�|� W n tk
rF Y nX X W d Q R X d S )Nr- )r& ZTemporaryDirectory�os�path�join�unlinkr+ )r Ztd�fnr r r �unix_sock_namej s
z&FunctionalTestCaseMixin.unix_sock_namec C s z| j �� W d | �|� X d S )N)r �stopr )r r. r r r �_abort_socket_testv s z*FunctionalTestCaseMixin._abort_socket_test)�__name__�
__module__� __qualname__r r
r r r r% ZAF_INETr/ r1 r5 r6 �
contextlib�contextmanagerr< r> 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 )N)�_TestSocketWrapper__sock)r r- r r r �__init__� s zTestSocketWrapper.__init__c C s@ d}x6t |�|k r:| �|t |� �}|dkr0t�||7 }qW |S )Nr )�lenZrecv�ConnectionAbortedError)r �n�buf�datar r r �recv_all� s zTestSocketWrapper.recv_allFN)� server_side�server_hostnamec C sT |j | j||dd�}z(y |�� W n |�� � Y nX W d | j�� X || _d S )NF)rM rN Zdo_handshake_on_connect)Z wrap_socketrE Z do_handshaker )r Z ssl_contextrM rN Zssl_sockr r r � start_tls� s zTestSocketWrapper.start_tlsc C s t | j|�S )N)�getattrrE )r r' r r r � __getattr__� s z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typer? rE )r r r r �__repr__� s zTestSocketWrapper.__repr__)r? r@ rA rF rL rO rQ rT r r r r rD � s
rD 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 )N)�start)r r r r � __enter__� s zSocketThread.__enter__c G s | � � d S )N)r= )r �excr r r �__exit__� s zSocketThread.__exit__N)r? r@ rA r= rX rZ r r r r rU � s rU c @ s e Zd Zdd� Zdd� ZdS )r0 c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrF �daemon�_timeout�_sockrV �_prog�_test)r �testr- �progr r r r rF � s zTestThreadedClient.__init__c
C sH y| � t| j�� W n. tk
rB } z| j�|� W d d }~X Y nX d S )N)r` rD r_ � Exceptionra r> )r r. r r r �run� s zTestThreadedClient.runN)r? r@ rA rF re r r r r r0 � s
r0 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[ r\ rF r] �_clientsZ_finished_clients� _max_clientsr^ r_ rV r` r% Z
socketpair�_s1�_s2� setblockingra )r rb r- rc r r" r r r rF � s zTestThreadedServer.__init__c sP z>| j r<| j �� dkr<y| j �d� W n tk
r: Y nX W d t� �� X d S )N���s stop)ri �fileno�sendr+ �superr= )r )� __class__r r r= � s
zTestThreadedServer.stopc C sF z*| j � | j �d� | �� W d Q R X W d | j�� | j�� X d S )Nr )r_ rj �_runrh r ri )r r r r re � s
zTestThreadedServer.runc C s �x| j �r| j| jkr d S t�| j| jgg g | j�\}}}| j|krJd S | j|kry| j�� \}}W n6 tk
r| wY q t j
k
r� | j s�d S � Y qX | jd7 _|� | j� y |� | � |� W d Q R X W q t
k
�r } z d| _ z� W d | j�|� X W d d }~X Y qX qW d S )Nr F)rV rf rg �selectr_ rh r^ Zaccept�BlockingIOErrorr% r r* �_handle_clientrd ra r> )r �r�w�xZconnr r. r r r rp � s4
zTestThreadedServer._runc C s | � t|�� d S )N)r` rD )r r- r r r rs s z!TestThreadedServer._handle_clientc C s
| j �� S )N)r_ Z getsockname)r r r r r s zTestThreadedServer.addr) r? r@ rA rF r= re rp rs �propertyr �
__classcell__r r )ro r r, � s
"r, )r Zasyncio.eventsrB r7 r rq r% r& r[ r rD r\ rU r0 r, r r r r �<module> s v(