Sindbad~EG File Manager
a
��g7 � @ 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 �8/usr/local/lib/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! �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 r td��|dkr.td��|�|� 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 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 r3 )r$ r% r4 r2 r! r5 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& �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 | �|� 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% �AF_INETr ZLOOPBACK_TIMEOUTr/ r2 r8 r9 �
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 r. r r r �__init__x s zTestSocketWrapper.__init__c C s< d}t |�|k r8| �|t |� �}|dkr.t�||7 }q|S )N� )�len�recv�ConnectionAbortedError)r �n�buf�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)rX rY Zdo_handshake_on_connect)Z wrap_socketrM Z do_handshaker )r Z ssl_contextrX rY Zssl_sockr r r � start_tls� s �
zTestSocketWrapper.start_tlsc C s t | j|�S r )�getattrrM )r r( r r r � __getattr__� s z TestSocketWrapper.__getattr__c C s d� t| �j| j�S )Nz <{} {!r}>)�format�typerF rM r r r r �__repr__� s zTestSocketWrapper.__repr__)rF rG rH rO rW rZ r\ r_ r r r r rL v s
� rL 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 rd rf r r r r r` � s r` c @ s e Zd Zdd� Zdd� ZdS )r1 c C s: t j�| d d d� d| _|| _|| _d| _|| _|| _d S )Nz test-clientT) � threading�ThreadrO �daemon�_timeout�_sockrb �_prog�_test)r �testr. �progr r r r rO � s zTestThreadedClient.__init__c
C sJ z| � t| j�� W n0 tyD } z| j�|� W Y d }~n
d }~0 0 d S r )rl rL rk � Exceptionrm rE rC r r r �run� s zTestThreadedClient.runN)rF rG rH rO rq 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)rg rh rO ri �_clientsZ_finished_clients� _max_clientsrj rk rb rl r% �
socketpair�_s1�_s2� setblockingrm )r rn r. ro r r r r r rO � 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)rv �fileno�sendr? �superrB r �� __class__r r rB � s zTestThreadedServer.stopc C sn zR| j �$ | j �d� | �� W d � n1 s20 Y W | j�� | j�� n| j�� | j�� 0 d S ra )rk rw �_runru r rv r r r r rq � 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)rb rr rs �selectrk ru rj �accept�BlockingIOErrorr% r r+ �_handle_clientrp rm rE )r �r�w�x�connr rD r r r r~ � s6 �
,zTestThreadedServer._runc C s | � t|�� d S r )rl rL rN r r r r� s z!TestThreadedServer._handle_clientc C s
| j �� S r )rk � getsocknamer r r r r s zTestThreadedServer.addr) rF rG rH rO rB rq r~ r� �propertyr �
__classcell__r r r| r r, � s
"r, )r Zasyncio.eventsrJ r; r r r% r& rg rn r r rL rh r` r1 r, r r r r �<module> s i(
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists