Your IP : 216.73.216.52
o
0�i5 � @ 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 �C/opt/alt/python310/lib64/python3.10/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 sn z-| j �� | jr#td� t�| j� | �d� W t�d � d | _ d S W t�d � d | _ d S t�d � d | _ w )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 r(t td�r&|tjkr&t�� � }|j}W d � n1 s w Y nd}tj|||d�}|d u r8td��|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 rtd��|dkrtd��|�|� 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 �$ t td�st�| j|dtji|��S �Nr! r )r$ r% �NotImplementedErrorr- r! �r �args�kwargsr r r � unix_serverT �
z#FunctionalTestCaseMixin.unix_serverc O r0 r1 )r$ r% r2 r/ r! r3 r r r � unix_clientY r7 z#FunctionalTestCaseMixin.unix_clientc c s� � t �� �?}tj�|d�}z|V W zt�|� W n ty" Y nw zt�|� W w ty3 Y w w W d � d S W d � d S 1 sGw 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 | �|� d S | �|� w 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/ r6 r8 �
contextlib�contextmanagerr? rC 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 sF d}t |�|k r!| �|t |� �}|dkrt�||7 }t |�|k s|S )N� )�lenZrecv�ConnectionAbortedError)r �nZbuf�datar r r �recv_all{ s �zTestSocketWrapper.recv_allFN)� server_side�server_hostnamec C sX |j | j||dd�}zz|�� W n |�� � W | j�� n| j�� w || _d S )NF)rS rT Zdo_handshake_on_connect)Z wrap_socketrJ Z do_handshaker )r Z ssl_contextrS rT Zssl_sockr r r � start_tls� s � �
zTestSocketWrapper.start_tlsc C s t | j|�S r )�getattrrJ )r r' r r r � __getattr__� � z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typerD rJ r r r r �__repr__� s zTestSocketWrapper.__repr__)rD rE rF rL rR rU rW r[ r r r r rI v s
� rI c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SocketThreadc C s d| _ | �� d S �NF)�_activer; 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__� rX zSocketThread.__exit__N)rD rE rF r@ r` rb r r r r r\ � s r\ c @ s e Zd Zdd� Zdd� ZdS )r. c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrL �daemon�_timeout�_sockr^ �_prog�_test)r �testr, �progr r r r rL � s
zTestThreadedClient.__init__c
C sJ z | � t| j�� W d S ty$ } z
| j�|� W Y d }~d S d }~ww r )rh rI rg � Exceptionri rC rA r r r �run� s ��zTestThreadedClient.runN)rD rE rF rL rm 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)rc rd rL re �_clientsZ_finished_clients� _max_clientsrf rg r^ rh r% Z
socketpair�_s1�_s2� setblockingri )r rj r, rk r r r r r rL � s
zTestThreadedServer.__init__c s� z<| j r | j �� dkr%z| j �d� W n ty Y nw W t� �� d S W t� �� d S W t� �� d S W t� �� d S t� �� w )N���s stop)rq �fileno�sendr= �superr@ r �� __class__r r r@ � s �����zTestThreadedServer.stopc C s� z7| j � | j �d� | �� W d � n1 sw Y W | j�� | j�� d S W | j�� | j�� d S | j�� | j�� w r] )rg rr �_runrp r rq r r r r rm � s
�
�
� zTestThreadedServer.runc C s | j r�| j| jkr d S t�| j| jgg g | j�\}}}| j|v r"d S | j|v r�z | j�� \}}W n ty9 Y q t yF | j sEY d S � w | jd7 _|�
| j� z|�
| � |� W d � n1 sgw Y W n t y� } z d| _ z� | j
�|� w d }~ww | j sd S d S )Nr F)r^ rn ro �selectrg rp rf Zaccept�BlockingIOError� TimeoutErrorr) �_handle_clientrl ri rC )r �r�w�xZconnr rB r r r ry � sB
�
� �����zTestThreadedServer._runc C s | � t|�� d S r )rh rI rK r r r r} s z!TestThreadedServer._handle_clientc C s
| j �� S r )rg Z getsocknamer r r r r s
zTestThreadedServer.addr) rD rE rF rL r@ rm ry r} �propertyr �
__classcell__r r rw r r* � s
"r* )r Zasyncio.eventsrG r9 r rz r% r& rc rj r r rI rd r\ r. r* r r r r �<module> s i(