Sindbad~EG File Manager
a
��g�F � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z d dl
m Z m
Z
mZ d dl
mZ e j�d� dZejZeed�Ze�ed�Zeed�Ze�ed �Zd
d � ZejZe jj fd d
�Z e�re�rG dd� de j e j �Z G dd� de j e j!�Z"e j#dd� �Z$G dd� dej%�Z&G dd� dej%�Z'G dd� de j(�Z)G dd� de j*�Z+G d d � d e j,e)�Z-e�r�G d d � d e j e)�Z.G d d!� d!ej%�Z/G d"d#� d#ej%�Z0e1d$k�r�e�2� dS )%� N)�
reap_children� reap_threads�verbose)�
socket_helper�networks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�rt�| � d S )N�alarm)�hasattr�signalr )�n� r
�2/usr/local/lib/python3.9/test/test_socketserver.py� signal_alarm s
r c C s: t | gg g |�\}}}| |v r(| �|�S td| f ��d S )Nztimed out on %r)� _real_select�recv� RuntimeError)�sockr �timeout�r�w�xr
r
r �receive( s
r c @ s e Zd ZdS )�ForkingUnixStreamServerN��__name__�
__module__� __qualname__r
r
r
r r 0 s r c @ s e Zd ZdS )�ForkingUnixDatagramServerNr r
r
r
r r 4 s r c
c s` t �� }|dkrt �d� z.z
d V W n � Y n0 W tjj|dd� ntjj|dd� 0 d S )Nr �H )�exitcode)�osr �_exit�test�supportZ wait_process)Ztestcase�pidr
r
r �simple_subprocess9 s
r& c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zed d
� �Zd d � Z d
d� Z
dd� Z dd� Z e
dd� �Zedd� �Zedd� �Zee
dd� ��Zdd � Zd d � Ze
d d � �Zed!d"� �Zed#d$� �Zee
d%d&� ��Zed'd(� �Zd)d*� Zd+d,� Zd-d.� Z d/S )0�SocketServerTestc C s t d� d| _g | _d S )N�< r )r Z port_seed�
test_files��selfr
r
r �setUpK s zSocketServerTest.setUpc C sN t d� t� | jD ]&}zt�|� W q ty8 Y q0 qg | jd d �< d S �Nr )r r r) r! �remove�OSError)r+ �fnr
r
r �tearDownP s
zSocketServerTest.tearDownc C s8 |t jkrtdfS d }tjd|d�}| j�|� |S d S )Nr z unix_socket.)�prefix�dir)�socket�AF_INET�HOST�tempfile�mktempr) �append)r+ �protor3 r0 r
r
r �pickaddr[ s
zSocketServerTest.pickaddrc
C s� G dd� d|�}G dd� d|�}t r,td� z|||�}W n8 tyr } z | �d|||f � W Y d }~n
d }~0 0 | �|j|j�� � |S )Nc @ s e Zd Zdd� ZdS )z.SocketServerTest.make_server.<locals>.MyServerc S s | � |� � d S �N)Z
close_request�r+ �request�client_addressr
r
r � handle_errorh s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r@ r
r
r
r �MyServerg s rA c @ s e Zd Zdd� ZdS )z/SocketServerTest.make_server.<locals>.MyHandlerc S s | j �� }| j�|� d S r< )�rfile�readline�wfile�write)r+ �liner
r
r �handlem s
z6SocketServerTest.make_server.<locals>.MyHandler.handleN�r r r rG r
r
r
r � MyHandlerl s rI zcreating serverz!Cannot create server (%s, %s): %s)r �print�PermissionErrorZskipTest� assertEqual�server_addressr4 � getsockname)r+ �addr�svrcls�hdlrbaserA rI �server�er
r
r � make_serverf s � z SocketServerTest.make_serverc C s� | � | �|j�||�}|j}tr4td|� td|� tjd| |jddid�}d|_ |�
� trhtd� t d �D ] }tr�td
|� ||j|� qptr�td � |� � |�
� |�� | �d |j�� � tr�t|tj�r�| �|j� tr�td
� d S )NzADDR =zCLASS =z
%s serving�
poll_interval�{�G�z�?��name�target�kwargsTzserver running� z test clientzwaiting for server����done)rT r; �address_familyrM r rJ � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closerL r4 �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalse�active_children)r+ rP rQ ZtestfuncrR rO �t�ir
r
r �
run_server{ s4 �
� zSocketServerTest.run_serverc C s� t � |t j��^}|�|� |�t� t|d� }}|rRd|vrRt|d�}||7 }q2| �|t� W d � n1 sr0 Y d S �N�d �
)r4 � SOCK_STREAM�connect�sendall�TEST_STRr rL �r+ r: rO �s�buf�datar
r
r �stream_examine� s
z SocketServerTest.stream_examinec C s� t � |t j��t}tr.|t jkr.|�| �|�� |�t|� t|d� }}|rhd|vrht|d�}||7 }qH| � |t� W d � n1 s�0 Y d S rq )
r4 �
SOCK_DGRAM�HAVE_UNIX_SOCKETSr �bindr; �sendtorw r rL rx r
r
r �
dgram_examine� s
z SocketServerTest.dgram_examinec C s | � tjtj| j� d S r< )rp rk � TCPServer�StreamRequestHandlerr| r* r
r
r �test_TCPServer� s �z SocketServerTest.test_TCPServerc C s | � tjtj| j� d S r< )rp rk ZThreadingTCPServerr� r| r* r
r
r �test_ThreadingTCPServer� s �z(SocketServerTest.test_ThreadingTCPServerc C s@ t | ��$ | �tjtj| j� W d � n1 s20 Y d S r< )r& rp rk ZForkingTCPServerr� r| r* r
r
r �test_ForkingTCPServer� s
�z&SocketServerTest.test_ForkingTCPServerc C s | � tjtj| j� d S r< )rp rk �UnixStreamServerr� r| r* r
r
r �test_UnixStreamServer� s �z&SocketServerTest.test_UnixStreamServerc C s | � tjtj| j� d S r< )rp rk ZThreadingUnixStreamServerr� r| r* r
r
r � test_ThreadingUnixStreamServer� s �z/SocketServerTest.test_ThreadingUnixStreamServerc C s> t | ��" | �ttj| j� W d � n1 s00 Y d S r< )r& rp r rk r� r| r* r
r
r � test_ForkingUnixStreamServer� s
�z-SocketServerTest.test_ForkingUnixStreamServerc C s | � tjtj| j� d S r< )rp rk Z UDPServer�DatagramRequestHandlerr� r* r
r
r �test_UDPServer� s �z SocketServerTest.test_UDPServerc C s | � tjtj| j� d S r< )rp rk ZThreadingUDPServerr� r� r* r
r
r �test_ThreadingUDPServer� s �z(SocketServerTest.test_ThreadingUDPServerc C s@ t | ��$ | �tjtj| j� W d � n1 s20 Y d S r< )r& rp rk ZForkingUDPServerr� r� r* r
r
r �test_ForkingUDPServer� s
�z&SocketServerTest.test_ForkingUDPServerc C s | � tjtj| j� d S r< )rp rk �UnixDatagramServerr� r� r* r
r
r �test_UnixDatagramServer� s �z(SocketServerTest.test_UnixDatagramServerc C s | � tjtj| j� d S r< )rp rk ZThreadingUnixDatagramServerr� r� r* r
r
r � test_ThreadingUnixDatagramServer� s �z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | � ttj| j� d S r< )rp r rk r� r� r* r
r
r � test_ForkingUnixDatagramServer� s �z/SocketServerTest.test_ForkingUnixDatagramServerc C s� G dd� dt j�}G dd� dt j�}g }td�D ]<}|tdf|�}tjd|jdd id
�}d |_|� ||f� q0|D ]\}}|�
� |� � qr|D ]\}}|� � |�
� q�d S ) Nc @ s e Zd ZdS )z0SocketServerTest.test_shutdown.<locals>.MyServerNr r
r
r
r rA s rA c @ s e Zd ZdS )z1SocketServerTest.test_shutdown.<locals>.MyHandlerNr r
r
r
r rI s rI � r zMyServer servingrU rV rW T)rk r� r� rd r6 r_ r` ra rb r9 rc re rf rg )r+ rA rI �threadsro ry rn r
r
r �
test_shutdown� s$ �
z SocketServerTest.test_shutdownc C s4 G dd� dt jt j�}|tdfdd� �}|�� d S )Nc @ s e Zd ZdS )z9SocketServerTest.test_close_immediately.<locals>.MyServerNr r
r
r
r rA s rA r c S s d S r< r
r
r
r
r �<lambda> � z9SocketServerTest.test_close_immediately.<locals>.<lambda>)rk �ThreadingMixInr� r6 rg )r+ rA rR r
r
r �test_close_immediately s z'SocketServerTest.test_close_immediatelyc C sN t d�D ]@}| �t��" t�tdftj� W d � q1 s>0 Y qd S )Ni r\ )rd � assertRaises�
OverflowErrorrk r� r6 r� )r+ ro r
r
r �test_tcpserver_bind_leak s
�z)SocketServerTest.test_tcpserver_bind_leakc C sH t �tdft j��}W d � n1 s(0 Y | �d|j�� � d S )Nr r\ )rk r� r6 r� rL r4 rh )r+ rR r
r
r �test_context_manager' s
� z%SocketServerTest.test_context_managerN) r r r r, r1 r; rT r rp r| r� r� r� �requires_forkingr� �requires_unix_socketsr� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r' H sF
"
r' c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zed d � �Z ed
d� �Z
dd� Z dS )�ErrorHandlerTestc C s t j�t jj� d S r< )r# r$ �unlink�TESTFNr* r
r
r r1 3 s zErrorHandlerTest.tearDownc C s t t� | jdd� d S �NT��handled)�BaseErrorTestServer�
ValueError� check_resultr* r
r
r �test_sync_handled6 s z"ErrorHandlerTest.test_sync_handledc C sB | � t�� tt� W d � n1 s(0 Y | jdd� d S �NFr� )r� �
SystemExitr� r� r* r
r
r �test_sync_not_handled: s &z&ErrorHandlerTest.test_sync_not_handledc C s t t� | jdd� d S r� )�ThreadingErrorTestServerr� r� r* r
r
r �test_threading_handled? s z'ErrorHandlerTest.test_threading_handledc C s t t� | jdd� d S r� )r� r� r� r* r
r
r �test_threading_not_handledC s z+ErrorHandlerTest.test_threading_not_handledc C s t t� | jdd� d S r� )�ForkingErrorTestServerr� r� r* r
r
r �test_forking_handledG s z%ErrorHandlerTest.test_forking_handledc C s t t� | jdd� d S r� )r� r� r� r* r
r
r �test_forking_not_handledL s z)ErrorHandlerTest.test_forking_not_handledc C sL t tjj��,}dd| }| �|�� |� W d � n1 s>0 Y d S )N�Handler called
�Error handled
)�openr# r$ r� rL �read)r+ r� �log�expectedr
r
r r� Q s z ErrorHandlerTest.check_resultN) r r r r1 r� r� r� r� r� r� r� r� r
r
r
r r� . s
r� c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )r� c sp || _ t� �tdft� t�| j�� W d � n1 s:0 Y z| �� W | � � n
| � � 0 | �
� d S r- ) � exception�super�__init__r6 �
BadHandlerr4 �create_connectionrM �handle_requestrg � wait_done)r+ r� �� __class__r
r r� X s
z BaseErrorTestServer.__init__c C s< t tjjd��}|�d� W d � n1 s.0 Y d S )N�ar� )r� r# r$ r� rE )r+ r> r? r� r
r
r r@ c s z BaseErrorTestServer.handle_errorc C s d S r< r
r* r
r
r r� g s z BaseErrorTestServer.wait_done)r r r r� r@ r� �
__classcell__r
r
r� r r� W s r� c @ s e Zd Zdd� ZdS )r� c C sH t tjjd��}|�d� W d � n1 s.0 Y | j�d��d S )Nr� r� z
Test error)r� r# r$ r� rE rR r� )r+ r� r
r
r rG l s (zBadHandler.handleNrH r
r
r
r r� k s r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )r� c s t �� | _t� j|i |�� d S r< )r_ �Eventr] r� r� �r+ �pos�kwr� r
r r� t s
z!ThreadingErrorTestServer.__init__c s t � j|i |�� | j�� d S r< )r� �shutdown_requestr] �setr� r� r
r r� x s z)ThreadingErrorTestServer.shutdown_requestc C s | j �� d S r< )r] �waitr* r
r
r r� | s z"ThreadingErrorTestServer.wait_done)r r r r� r� r� r� r
r
r� r r� r s r� c @ s e Zd ZdS )r� Nr r
r
r
r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�SocketWriterTestc C s� G dd� dt j�}t �tdf|�}| �|j� t�|jtjtj �}|� |�
|j � W d � n1 sh0 Y |� � | �
|jtj� | �|j|j� d S )Nc @ s e Zd Zdd� ZdS )z-SocketWriterTest.test_basics.<locals>.Handlerc S s* | j | j_ | j �� | j_| j�� | j_d S r< )rD rR rh � wfile_filenor> �request_filenor* r
r
r rG � s
z4SocketWriterTest.test_basics.<locals>.Handler.handleNrH r
r
r
r �Handler� s r� r )rk r� r� r6 �
addCleanuprg r4 r^ rt � IPPROTO_TCPru rM r� ZassertIsInstancerD �io�BufferedIOBaserL r� r� )r+ r� rR ry r
r
r � test_basics� s �*z SocketWriterTest.test_basicsc s t j�td��G dd� dtj�}t�tdf|��| ��j � t
� � � � fdd�}t�tj |�}| �tjtj |� d �d �t
�
� �� �����fdd�}t
j|d �}|�� ��� |�� | ��jt��� | ��d
� | ��jd � | ��jt jj� | ��t jjd � d S )
N� pthread_killc @ s e Zd Zdd� ZdS )z,SocketWriterTest.test_write.<locals>.Handlerc S s> | j �d�| j_| j�� | j_dtjj }| j �|�| j_
d S )N� write data
� ) rD rE rR �sent1rB rC �receivedr# r$ �
SOCK_MAX_SIZE�sent2)r+ Z big_chunkr
r
r rG � s z3SocketWriterTest.test_write.<locals>.Handler.handleNrH r
r
r
r r� � s r� r c s � � � d S r< )r� )�signum�frame)� interruptedr
r �signal_handler� s z3SocketWriterTest.test_write.<locals>.signal_handlerc s� t � �jt jt j�} | �� | �d��d}| ��j� |�� �| �d� |� d� ��t
j � � j t
d�d�rNqnqNt|� � ��W d � n1 s�0 Y W d � n1 s�0 Y d S )N�rb� client response
rr � )r )r4 r^ rt r� �makefileru rM rC rv r� r �SIGUSR1r� �float�len)ry �reader�r� � main_threadr� Z received2Z response1rR r
r �
run_client� s �
z/SocketWriterTest.test_write.<locals>.run_client)rY r� r� rr )r# r$ Z
get_attributer rk r� r� r6 r� rg r_ r� r� � get_identr` rc r� rf rL r� r� r� r� r� )r+ r� r� Zoriginalr� Z
backgroundr
r� r �
test_write� s* zSocketWriterTest.test_writeN)r r r r� r� r
r
r
r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� MiscTestCasec C sR g }t t�D ]2}|�d�s tt|�}t|dd �dkr |�|� q | �tj|� d S )N�_r rk )r3 rk �
startswith�getattrr9 ZassertCountEqual�__all__)r+ r� rX Z
mod_objectr
r
r �test_all� s
zMiscTestCase.test_allc C sh G dd� dt j�}|tdft j�}t�|jtj�}|�|j� |� � |�
� | � |j d� |�
� d S )Nc @ s e Zd Zdd� ZdZdd� ZdS )zSMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServerc S s dS )NFr
r= r
r
r �verify_request� s zbMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.verify_requestr c S s | j d7 _ tj�| |� d S )Nr� )�shutdown_calledrk r� r� )r+ r> r
r
r r� � s zdMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.shutdown_requestN)r r r r� r� r� r
r
r
r rA � s rA r r� )rk r� r6 r� r4 r^ rt ru rM �closer� rL r� rg )r+ rA rR ry r
r
r �4test_shutdown_request_called_if_verify_request_false� s zAMiscTestCase.test_shutdown_request_called_if_verify_request_falsec C s� G dd� dt jt j�}|tdft j�}td�D ]8}t�|j�� |� � W d � q.1 s\0 Y q.| �
t |j �d� |�
� d S )Nc @ s e Zd ZdS )z2MiscTestCase.test_threads_reaped.<locals>.MyServerNr r
r
r
r rA � s rA r �
)rk r� r� r6 r� rd r4 r� rM r� Z
assertLessr� Z_threadsrg )r+ rA rR r r
r
r �test_threads_reaped� s (z MiscTestCase.test_threads_reapedN)r r r r� r� r� r
r
r
r r� � s
r� �__main__)3�
contextlibr� r! �selectr r4 r7 r_ Zunittestrk Z test.supportr# r r r r r$ Zrequiresrw r6 r
r~ Z
skipUnlessr� ri r� r r Z
SHORT_TIMEOUTr rl r� r r� r �contextmanagerr& ZTestCaser' r� r� r� ZBaseRequestHandlerr� r� r� r� r� r� r �mainr
r
r
r �<module> s^
�
��
g)�M2
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists