Your IP : 216.73.217.129
U
i�f�F � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
mZmZ e j�d� dZe jjZeed�Ze �ed�Zeed�Ze �ed �Zd
d � ZejZd&d
d�Z e�re�rG dd� de
j e
j �Z G dd� de
j e
j �Z!ej"dd� �Z#G dd� de j$�Z%G dd� de j$�Z&G dd� de
j'�Z(G dd � d e
j)�Z*G d d � d e
j+e(�Z,e�r�G d d � d e
j e(�Z-G d!d"� d"e j$�Z.G d#d$� d$e j$�Z/e0d%k�r�e �1� dS )'z
Test suite for socketserver.
� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�rt�| � dS )z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr )�n� r �;/opt/alt/python38/lib64/python3.8/test/test_socketserver.py� signal_alarm s
r
� c C s: t | gg g |�\}}}| |kr(| �|�S td| f ��d S )Nztimed out on %r)� _real_selectZrecv� RuntimeError)Zsockr
�timeout�r�w�xr r r �receive' s
r c @ s e Zd ZdS )�ForkingUnixStreamServerN��__name__�
__module__� __qualname__r r r r r / s r c @ s e Zd ZdS )�ForkingUnixDatagramServerNr r r r r r 3 s r c
c sh t �� }|dkrt �d� z z
dV W n � Y nX W 5 t �|d�\}}| �||� | �d|� X dS )zBTests that a custom child process is not waited on (Issue 1540386)r �H i H N)�osr �_exit�waitpid� assertEqual)Ztestcase�pidZpid2Zstatusr r r �simple_subprocess8 s
r" c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z d d
� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zedd� �Zeedd� ��Zd d � Zd d � Zed d!� �Zed"d#� �Zed$d%� �Zeed&d'� ��Zed(d)� �Zd*d+� Zd,d-� Z d.d/� Z d0S )1�SocketServerTestzTest all socket servers.c C s t d� d| _g | _d S )N�< r )r
Z port_seed�
test_files��selfr r r �setUpL s zSocketServerTest.setUpc C sP t d� t� | jD ](}zt�|� W q tk
r: Y qX qg | jd d �< d S �Nr )r
r r% r �remove�OSError)r' �fnr r r �tearDownQ s
zSocketServerTest.tearDownc C s8 |t jkrtdfS d }tjd|d�}| j�|� |S d S )Nr z unix_socket.)�prefix�dir)�socketZAF_INET�HOST�tempfileZmktempr% �append)r' �protor/ r, r r r �pickaddr\ s
zSocketServerTest.pickaddrc
C s� G dd� d|�}G dd� d|�}t r,td� z|||�}W n6 tk
rp } z| �d|||f � W 5 d }~X Y nX | �|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_errori s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r: r r r r �MyServerh s r; c @ s e Zd Zdd� ZdS )z/SocketServerTest.make_server.<locals>.MyHandlerc S s | j �� }| j�|� d S r6 )�rfile�readline�wfile�write)r' �liner r r �handlen s
z6SocketServerTest.make_server.<locals>.MyHandler.handleN�r r r rA r r r r � MyHandlerm s rC zcreating serverz!Cannot create server (%s, %s): %s)r �print�PermissionErrorZskipTestr �server_addressr0 Z getsockname)r' �addr�svrcls�hdlrbaser; rC �server�er r r � make_serverg 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)rL r5 �address_familyrF r rD � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closer r0 �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalseZactive_children)r' rH rI ZtestfuncrJ rG �t�ir r r �
run_server| s> �
�
zSocketServerTest.run_serverc C sl t � |t j��T}|�|� |�t� t|d� }}|rRd|krRt|d�}||7 }q2| �|t� W 5 Q R X d S �N�d �
)r0 � SOCK_STREAM�connect�sendall�TEST_STRr r �r' r4 rG �sZbuf�datar r r �stream_examine� s
z SocketServerTest.stream_examinec C s� t � |t j��j}tr.|t jkr.|�| �|�� |�t|� t|d� }}|rhd|krht|d�}||7 }qH| � |t� W 5 Q R X d S rh )
r0 Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr5 Zsendtorn r r ro r r r �
dgram_examine� s
z SocketServerTest.dgram_examinec C s | � tjtj| j� d S r6 )rg rc � TCPServer�StreamRequestHandlerrr r& r r r �test_TCPServer� s �z SocketServerTest.test_TCPServerc C s | � tjtj| j� d S r6 )rg rc ZThreadingTCPServerrv rr r& r r r �test_ThreadingTCPServer� s �z(SocketServerTest.test_ThreadingTCPServerc C s, t | �� | �tjtj| j� W 5 Q R X d S r6 )r" rg rc ZForkingTCPServerrv rr r& r r r �test_ForkingTCPServer� s
�z&SocketServerTest.test_ForkingTCPServerc C s | � tjtj| j� d S r6 )rg rc �UnixStreamServerrv rr r& r r r �test_UnixStreamServer� s �z&SocketServerTest.test_UnixStreamServerc C s | � tjtj| j� d S r6 )rg rc ZThreadingUnixStreamServerrv rr r& r r r � test_ThreadingUnixStreamServer� s �z/SocketServerTest.test_ThreadingUnixStreamServerc C s* t | �� | �ttj| j� W 5 Q R X d S r6 )r" rg r rc rv rr r& r r r � test_ForkingUnixStreamServer� s
�z-SocketServerTest.test_ForkingUnixStreamServerc C s | � tjtj| j� d S r6 )rg rc Z UDPServer�DatagramRequestHandlerrt r& r r r �test_UDPServer� s �z SocketServerTest.test_UDPServerc C s | � tjtj| j� d S r6 )rg rc ZThreadingUDPServerr~ rt r& r r r �test_ThreadingUDPServer� s �z(SocketServerTest.test_ThreadingUDPServerc C s, t | �� | �tjtj| j� W 5 Q R X d S r6 )r" rg rc ZForkingUDPServerr~ rt r& r r r �test_ForkingUDPServer� s
�z&SocketServerTest.test_ForkingUDPServerc C s | � tjtj| j� d S r6 )rg rc �UnixDatagramServerr~ rt r& r r r �test_UnixDatagramServer� s �z(SocketServerTest.test_UnixDatagramServerc C s | � tjtj| j� d S r6 )rg rc ZThreadingUnixDatagramServerr~ rt r& r r r � test_ThreadingUnixDatagramServer� s �z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | � ttj| j� d S r6 )rg r rc r~ rt 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 r; s r; c @ s e Zd ZdS )z1SocketServerTest.test_shutdown.<locals>.MyHandlerNr r r r r rC s rC r r zMyServer servingrM rN rO T)rc ru rv r\ r1 rW rX rY rZ r3 r[ r] r^ r_ )r' r; rC Zthreadsrf rp re 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 r; s r; r c S s d S r6 r r r r r �<lambda> � z9SocketServerTest.test_close_immediately.<locals>.<lambda>)rc �ThreadingMixInru r1 r_ )r' r; rJ r r r �test_close_immediately s z'SocketServerTest.test_close_immediatelyc
C s: t d�D ],}| �t�� t�tdftj� W 5 Q R X qd S )Ni rT )r\ � assertRaises�
OverflowErrorrc ru r1 rv )r' rf r r r �test_tcpserver_bind_leak s
�z)SocketServerTest.test_tcpserver_bind_leakc C s4 t �tdft j��}W 5 Q R X | �d|j�� � d S )Nr rT )rc ru r1 rv r r0 r` )r' rJ r r r �test_context_manager( s
�
z%SocketServerTest.test_context_managerN) r r r �__doc__r( r- r5 rL r rg rr rt rw rx �requires_forkingry �requires_unix_socketsr{ r| r} r r� r� r� r� r� r� r� r� r� r r r r r# I sH
"
r# c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze d d
� �Z
e dd� �Z dd� Z dS )�ErrorHandlerTestz�Test that the servers pass normal exceptions from the handler to
handle_error(), and that exiting exceptions like SystemExit and
KeyboardInterrupt are not passed.c C s t j�t jj� d S r6 )�test�support�unlink�TESTFNr& r r r r- 4 s zErrorHandlerTest.tearDownc C s t t� | jdd� d S �NT��handled)�BaseErrorTestServer�
ValueError� check_resultr& r r r �test_sync_handled7 s z"ErrorHandlerTest.test_sync_handledc C s. | � t�� tt� W 5 Q R X | 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_handledD s z+ErrorHandlerTest.test_threading_not_handledc C s t t� | jdd� d S r� )�ForkingErrorTestServerr� r� r& r r r �test_forking_handledH s z%ErrorHandlerTest.test_forking_handledc C s t t� | jdd� d S r� )r� r� r� r& r r r �test_forking_not_handledM s z)ErrorHandlerTest.test_forking_not_handledc C s8 t tjj��"}dd| }| �|�� |� W 5 Q R X d S )N�Handler called
�Error handled
)�openr� r� r� r �read)r' r� �log�expectedr r r r� R s z ErrorHandlerTest.check_resultN)
r r r r� r- 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 sT || _ t� �tdft� t�| j�� W 5 Q R X z | � � W 5 | �� X | �
� d S r) ) � exception�super�__init__r1 �
BadHandlerr0 �create_connectionrF r_ �handle_request� wait_done)r' r� �� __class__r r r� Y s
z BaseErrorTestServer.__init__c C s( t tjjd��}|�d� W 5 Q R X d S )N�ar� )r� r� r� r� r? )r' r8 r9 r� r r r r: d s z BaseErrorTestServer.handle_errorc C s d S r6 r r& r r r r� h s z BaseErrorTestServer.wait_done)r r r r� r: r� �
__classcell__r r r� r r� X s r� c @ s e Zd Zdd� ZdS )r� c C s4 t tjjd��}|�d� W 5 Q R X | j�d��d S )Nr� r� z
Test error)r� r� r� r� r? rJ r� )r' r� r r r rA m s zBadHandler.handleNrB r r r r r� l s r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )r� c s t �� | _t� j||� d S r6 )rW �EventrU r� r� �r' �pos�kwr� r r r� u s
z!ThreadingErrorTestServer.__init__c s t � j||� | j�� d S r6 )r� �shutdown_requestrU �setr� r� r r r� y s z)ThreadingErrorTestServer.shutdown_requestc C s | j �� d S r6 )rU �waitr& r r r r� } s z"ThreadingErrorTestServer.wait_done)r r r r� r� r� r� r r r� r r� s 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 5 Q R X |� � | �
|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 r6 )r> rJ r` � wfile_filenor8 �request_filenor& r r r rA � s
z4SocketWriterTest.test_basics.<locals>.Handler.handleNrB r r r r �Handler� s r� r )rc rv ru r1 �
addCleanupr_ r0 rV rk � IPPROTO_TCPrl rF r� ZassertIsInstancer> �io�BufferedIOBaser r� r� )r' r� rJ rp 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
� ) r> r? rJ �sent1r< r= �receivedr� r� �
SOCK_MAX_SIZE�sent2)r' Z big_chunkr r r rA � s z3SocketWriterTest.test_write.<locals>.Handler.handleNrB r r r r r� � s r� r c s � � � d S r6 )r� )Zsignum�frame)� interruptedr r �signal_handler� s z3SocketWriterTest.test_write.<locals>.signal_handlerc
s� t � �jt jt j�} | �p | �d��Z}| ��j� |�� �| �d� |� d� ��t
j � � j t
d�d�rNqnqNt|� � ��W 5 Q R X W 5 Q R X d S )N�rb� client response
ri � )r )r0 rV rk r� Zmakefilerl rF r= rm r� r �SIGUSR1r� �float�len)rp �reader�r� Z main_threadr� Z received2Z response1rJ r r �
run_client� s �
z/SocketWriterTest.test_write.<locals>.run_client)rQ r� r� ri )r� r� Z
get_attributer rc rv ru r1 r� r_ rW r� r� � get_identrX r[ r� r^ r 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 rc )r/ rc �
startswith�getattrr3 ZassertCountEqual�__all__)r' r� rP 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 r7 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_calledrc ru r� )r' r8 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 r; � s r; r r� )rc ru r1 rv r0 rV rk rl rF �closer� r r� r_ )r' r; rJ rp r r r �4test_shutdown_request_called_if_verify_request_false� s zAMiscTestCase.test_shutdown_request_called_if_verify_request_falsec
C sr G dd� dt jt j�}|tdft j�}td�D ]$}t�|j�� |� � W 5 Q R X q.| �
t |j �d� |�
� dS )z�
In #37193, users reported a memory leak
due to the saving of every request thread. Ensure that
not all threads are kept forever.
c @ s e Zd ZdS )z2MiscTestCase.test_threads_reaped.<locals>.MyServerNr r r r r r; � s r; r �
N)rc r� ru r1 rv r\ r0 r� rF r� Z
assertLessr� Z_threadsr_ )r' r; rJ 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__)r )2r� �
contextlibr� r Zselectr r0 r2 rW Zunittestrc Z test.supportr� r r r r� Zrequiresrn r1 r rs Z
skipUnlessr� ra r� r
r r rd rz r r� r �contextmanagerr" ZTestCaser# r� ru r� ZBaseRequestHandlerr� r� r� r� r� r� r �mainr r r r �<module> s^
�
��
g)�M2