File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/functional.cpython-310.pyc
o
�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 �9/usr/local/lib/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! �tempfile�NamedTemporaryFile�name�
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% � 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 r3 r4 )r$ r% r5 r2 r! r6 r r r � unix_clientY r: 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& �TemporaryDirectory�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% �AF_INETr ZLOOPBACK_TIMEOUTr/ r2 r9 r; �
contextlib�contextmanagerrC rG 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� )�len�recv�ConnectionAbortedError)r �n�buf�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)rZ r[ Zdo_handshake_on_connect)Z wrap_socketrO Z do_handshaker )r Z ssl_contextrZ r[ Zssl_sockr r r � start_tls� s � �
zTestSocketWrapper.start_tlsc C s t | j|�S r )�getattrrO )r r( r r r � __getattr__� � z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typerH rO r r r r �__repr__� s zTestSocketWrapper.__repr__)rH rI rJ rQ rY r\ r^ rb r r r r rN v s
� rN c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SocketThreadc C s d| _ | �� d S �NF)�_activer? r r r r rD � s zSocketThread.stopc C s | � � | S r )�startr r r r � __enter__� s zSocketThread.__enter__c G s | � � d S r )rD )r �excr r r �__exit__� r_ zSocketThread.__exit__N)rH rI rJ rD rg ri r r r r rc � s rc c @ s e Zd Zdd� Zdd� ZdS )r1 c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrQ �daemon�_timeout�_sockre �_prog�_test)r �testr. �progr r r r rQ � s
zTestThreadedClient.__init__c
C sJ z | � t| j�� W d S ty$ } z
| j�|� W Y d }~d S d }~ww r )ro rN rn � Exceptionrp rG rE r r r �run� s ��zTestThreadedClient.runN)rH rI rJ rQ rt r r r r r1 � s
r1 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)rj rk rQ rl �_clientsZ_finished_clients� _max_clientsrm rn re ro r% �
socketpair�_s1�_s2� setblockingrp )r rq r. rr r r r r r rQ � 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)ry �fileno�sendrA �superrD r �� __class__r r rD � 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 rd )rn rz �_runrx r ry r r r r rt � 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)re ru rv �selectrn rx rm �accept�BlockingIOError� TimeoutErrorr+ �_handle_clientrs rp rG )r �r�w�x�connr rF r r r r� � sB
�
� �����zTestThreadedServer._runc C s | � t|�� d S r )ro rN rP r r r r� s z!TestThreadedServer._handle_clientc C s
| j �� S r )rn � getsocknamer r r r r s
zTestThreadedServer.addr) rH rI rJ rQ rD rt r� r� �propertyr �
__classcell__r r r r r, � s
"r, )r Zasyncio.eventsrL r= r r� r% r& rj rq r r rN rk rc r1 r, r r r r �<module> s i(