Your IP : 216.73.216.196
o
0�i�G � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
mZ ddl mZ ddl mZ ddl mZ e j�d� dZejZeed �Ze �ed
�Zeed �Ze �ed �Zd
d� Z ejZ e jj fdd�Z er�er�G 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 � d e
j*�Z+G d d � d e
j,�Z-G d d � d e
j.e+�Z/er�G d!d"� d"e
j e+�Z0G d#d$� d$e j'�Z1G d%d&� d&e j'�Z2e3d'kr�e �4� dS dS )(z
Test suite for socketserver.
� N)�
reap_children�verbose)� os_helper)�
socket_helper)�threading_helperZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�r t�| � dS dS )z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr )�n� r
�=/opt/alt/python310/lib64/python3.10/test/test_socketserver.py� signal_alarm" s
�r c C s6 t | gg g |�\}}}| |v r| �|�S td| f ��)Nztimed out on %r)� _real_selectZrecv� RuntimeError)Zsockr �timeout�r�w�xr
r
r �receive* s
r c @ � e Zd ZdS )�ForkingUnixStreamServerN��__name__�
__module__� __qualname__r
r
r
r r 2 � r c @ r )�ForkingUnixDatagramServerNr r
r
r
r r 6 r r c
c sZ � t �� }|dkrt �d� zzdV W n � W tjj|dd� dS tjj|dd� w )zBTests that a custom child process is not waited on (Issue 1540386)r �H N)Zexitcode)�osr �_exit�test�supportZ wait_process)Ztestcase�pidr
r
r �simple_subprocess; s �
�&r% c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zej d
d � �Z
d d
� Z dd� Z dd� Z
dd� Zedd� �Zedd� �Zedd� �Zeedd� ��Zd d � Zd d � Zed d!� �Zed"d#� �Zed$d%� �Zeed&d'� ��Zej d(d)� �Zd*d+� Z d,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 �setUpM s
zSocketServerTest.setUpc C sL t d� t� | jD ]}zt�|� W q
ty Y q
w g | jd d �< d S �Nr )r r r( r �remove�OSError)r* �fnr
r
r �tearDownR s
�zSocketServerTest.tearDownc C s4 |t jkr tdfS d }tjd|d�}| j�|� |S )Nr z unix_socket.)�prefix�dir)�socketZAF_INET�HOST�tempfileZmktempr( �append)r* �protor2 r/ r
r
r �pickaddr] s
zSocketServerTest.pickaddrc
C s� G dd� d|�}G dd� d|�}t rtd� z|||�}W n ty9 } z| �d|||f � W Y d }~nd }~ww | �|j|j�� � |S )Nc @ � e Zd Zdd� ZdS )z.SocketServerTest.make_server.<locals>.MyServerc S s | � |� � �N)Z
close_request�r* �request�client_addressr
r
r � handle_errorj s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r> r
r
r
r �MyServeri � r? c @ r9 )z/SocketServerTest.make_server.<locals>.MyHandlerc S s | j �� }| j�|� d S r: )�rfile�readline�wfile�write)r* �liner
r
r �handleo s
z6SocketServerTest.make_server.<locals>.MyHandler.handleN�r r r rF r
r
r
r � MyHandlern r@ rH zcreating serverz!Cannot create server (%s, %s): %s)r �print�PermissionErrorZskipTest� assertEqual�server_addressr3 Z getsockname)r* �addr�svrcls�hdlrbaser? rH �server�er
r
r � make_serverh s ���z SocketServerTest.make_serverc C s� | � | �|j�||�}|j}trtd|� td|� tjd| |jddid�}d|_ |�
� tr4td� t d �D ]}trAtd
|� ||j|� q8trNtd � |� � |�
� |�� | �d |j�� � trqt|tj�rq| �|j� trytd
� d S d S )NzADDR =zCLASS =z
%s serving�
poll_interval�{�G�z�?��name�target�kwargsTzserver running� z test clientzwaiting for server����done)rR r8 �address_familyrL r rI � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closerK r3 �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalseZactive_children)r* rN rO ZtestfuncrP rM �t�ir
r
r �
run_server} s4 �
� zSocketServerTest.run_serverc C s� t � |t j��5}|�|� |�t� t|d� }}|r.d|vr.t|d�}||7 }|r.d|vs | �|t� W d � d S 1 s?w Y d S �N�d �
)r3 � SOCK_STREAM�connect�sendall�TEST_STRr rK �r* r7 rM �sZbuf�datar
r
r �stream_examine� s
�"�z SocketServerTest.stream_examinec C s� t � |t j��@}tr|t jkr|�| �|�� |�t|� t|d� }}|r9d|vr9t|d�}||7 }|r9d|vs*| � |t� W d � d S 1 sJw Y d S rn )
r3 Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr8 Zsendtort r rK ru r
r
r �
dgram_examine� s
�"�z SocketServerTest.dgram_examinec C � | � tjtj| j� d S r: )rm ri � TCPServer�StreamRequestHandlerrx r) r
r
r �test_TCPServer� � �z SocketServerTest.test_TCPServerc C r{ r: )rm ri ZThreadingTCPServerr} rx r) r
r
r �test_ThreadingTCPServer� r z(SocketServerTest.test_ThreadingTCPServerc C �B t | �� | �tjtj| j� W d � d S 1 sw Y d S r: )r% rm ri ZForkingTCPServerr} rx r) r
r
r �test_ForkingTCPServer� �
�"�z&SocketServerTest.test_ForkingTCPServerc C r{ r: )rm ri �UnixStreamServerr} rx r) r
r
r �test_UnixStreamServer� � �z&SocketServerTest.test_UnixStreamServerc C r{ r: )rm ri ZThreadingUnixStreamServerr} rx r) r
r
r � test_ThreadingUnixStreamServer� r� z/SocketServerTest.test_ThreadingUnixStreamServerc C s@ t | �� | �ttj| j� W d � d S 1 sw Y d S r: )r% rm r ri r} rx r) r
r
r � test_ForkingUnixStreamServer� s
�"�z-SocketServerTest.test_ForkingUnixStreamServerc C r{ r: )rm ri Z UDPServer�DatagramRequestHandlerrz r) r
r
r �test_UDPServer� r z SocketServerTest.test_UDPServerc C r{ r: )rm ri ZThreadingUDPServerr� rz r) r
r
r �test_ThreadingUDPServer� r z(SocketServerTest.test_ThreadingUDPServerc C r� r: )r% rm ri ZForkingUDPServerr� rz r) r
r
r �test_ForkingUDPServer� r� z&SocketServerTest.test_ForkingUDPServerc C r{ r: )rm ri �UnixDatagramServerr� rz r) r
r
r �test_UnixDatagramServer� r� z(SocketServerTest.test_UnixDatagramServerc C r{ r: )rm ri ZThreadingUnixDatagramServerr� rz r) r
r
r � test_ThreadingUnixDatagramServer� r� z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | � ttj| j� d S r: )rm r ri r� rz 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� q|D ] \}}|�
� |� � q9|D ] \}}|� � |�
� qHd S ) Nc @ r )z0SocketServerTest.test_shutdown.<locals>.MyServerNr r
r
r
r r? � r? c @ r )z1SocketServerTest.test_shutdown.<locals>.MyHandlerNr r
r
r
r rH r� rH � r zMyServer servingrS rT rU T)ri r| r} rb r4 r] r^ r_ r` r6 ra rc rd re )r* r? rH Zthreadsrl rv rk r
r
r �
test_shutdown� s& �
�z SocketServerTest.test_shutdownc C s4 G dd� dt jt j�}|tdfdd� �}|�� d S )Nc @ r )z9SocketServerTest.test_close_immediately.<locals>.MyServerNr r
r
r
r r? r� r? r c S � d S r: r
r
r
r
r �<lambda> s z9SocketServerTest.test_close_immediately.<locals>.<lambda>)ri �ThreadingMixInr| r4 re )r* r? rP r
r
r �test_close_immediately s z'SocketServerTest.test_close_immediatelyc C sN t d�D ] }| �t�� t�tdftj� W d � n1 s w Y qd S )Ni rZ )rb � assertRaises�
OverflowErrorri r| r4 r} )r* rl r
r
r �test_tcpserver_bind_leak s
� ���z)SocketServerTest.test_tcpserver_bind_leakc C sH t �tdft j��}W d � n1 sw Y | �d|j�� � d S )Nr rZ )ri r| r4 r} rK r3 rf )r* rP r
r
r �test_context_manager) s
� �z%SocketServerTest.test_context_managerN) r r r �__doc__r+ r0 r8 rR r Z reap_threadsrm rx rz r~ r� �requires_forkingr� �requires_unix_socketsr� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r& J sJ
"
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 �t j� d S r: )r �unlink�TESTFNr) r
r
r r0 5 s zErrorHandlerTest.tearDownc C � t t� | jdd� d S �NT��handled)�BaseErrorTestServer�
ValueError� check_resultr) r
r
r �test_sync_handled8 � z"ErrorHandlerTest.test_sync_handledc C sB | � t�� tt� W d � n1 sw Y | jdd� d S �NFr� )r� �
SystemExitr� r� r) r
r
r �test_sync_not_handled< s
�z&ErrorHandlerTest.test_sync_not_handledc C r� r� )�ThreadingErrorTestServerr� r� r) r
r
r �test_threading_handledA r� z'ErrorHandlerTest.test_threading_handledc C sP t �� �}tt� | jdd� | �|jt� W d � d S 1 s!w Y d S r� )r Zcatch_threading_exceptionr� r� r� ZassertIs�exc_type)r* �cmr
r
r �test_threading_not_handledE s
"�z+ErrorHandlerTest.test_threading_not_handledc C r� r� )�ForkingErrorTestServerr� r� r) r
r
r �test_forking_handledL � z%ErrorHandlerTest.test_forking_handledc C r� r� )r� r� r� r) r
r
r �test_forking_not_handledQ r� z)ErrorHandlerTest.test_forking_not_handledc C sL t tj��}dd| }| �|�� |� W d � d S 1 s w Y d S )N�Handler called
�Error handled
)�openr r� rK �read)r* r� �log�expectedr
r
r r� V s "�z ErrorHandlerTest.check_resultN)
r r r r� r0 r� r� r� r� r� r� r� r� r
r
r
r r� 0 s
r� c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )r� c sp || _ t� �tdft� t�| j�� W d � n1 s w Y z
| �� W | � � n| � � w | �
� d S r, ) � exception�super�__init__r4 �
BadHandlerr3 �create_connectionrL �handle_requestre � wait_done)r* r� �� __class__r
r r� ] s �
z BaseErrorTestServer.__init__c C s< t tjd��}|�d� W d � d S 1 sw Y d S )N�ar� )r� r r� rD )r* r< r= r� r
r
r r> h s "�z BaseErrorTestServer.handle_errorc C r� r: r
r) r
r
r r� l � z BaseErrorTestServer.wait_done)r r r r� r> r� �
__classcell__r
r
r� r r� \ s r� c @ r9 )r� c C sB t tjd��
}|�d� W d � n1 sw Y | j�d��)Nr� r� z
Test error)r� r r� rD rP r� )r* r� r
r
r rF q s � zBadHandler.handleNrG r
r
r
r r� p r@ 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� y s
z!ThreadingErrorTestServer.__init__c s t � j|i |�� | j�� d S r: )r� �shutdown_requestr[ �setr� r� r
r r� } 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� w s r� c @ r )r� Nr r
r
r
r r� � r� r� c @ s e Zd Zdd� Zdd� ZdS )�SocketWriterTestc C s� G dd� dt j�}t �tdf|�}| �|j� t�|jtjtj �}|� |�
|j � W d � n1 s4w Y |� � | �
|jtj� | �|j|j� d S )Nc @ r9 )z-SocketWriterTest.test_basics.<locals>.Handlerc S s* | j | j_ | j �� | j_| j�� | j_d S r: )rC rP rf � wfile_filenor< �request_filenor) r
r
r rF � s
z4SocketWriterTest.test_basics.<locals>.Handler.handleNrG r
r
r
r �Handler� r@ r� r )ri r} r| r4 �
addCleanupre r3 r\ rq � IPPROTO_TCPrr rL r� ZassertIsInstancerC �io�BufferedIOBaserK r� r� )r* r� rP rv 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 @ r9 )z,SocketWriterTest.test_write.<locals>.Handlerc S s> | j �d�| j_| j�� | j_dtjj }| j �|�| j_
d S )N� write data
� ) rC rD rP �sent1rA rB �receivedr" r# �
SOCK_MAX_SIZE�sent2)r* Z big_chunkr
r
r rF � s z3SocketWriterTest.test_write.<locals>.Handler.handleNrG r
r
r
r r� � r@ r� r c s � � � d S r: )r� )Zsignum�frame)� interruptedr
r �signal_handler� s z3SocketWriterTest.test_write.<locals>.signal_handlerc s� t � �jt jt j�} | �Q | �d��3}| ��j� |�� �| �d� |� d� ��t
j � � j t
d�d�r7nq(t|� � ��W d � n1 sHw Y W d � d S W d � d S 1 s`w Y d S )N�rb� client response
ro T� )r )r3 r\ rq r� Zmakefilerr rL rB rs r� r �SIGUSR1r� �float�len)rv �reader�r� Z main_threadr� Z received2Z response1rP r
r �
run_client� s �
�P�z/SocketWriterTest.test_write.<locals>.run_client)rW r� r� ro )r" r# Z
get_attributer ri r} r| r4 r� re r] r� r� � get_identr^ ra r� rd rK 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 ]}|�d�s tt|�}t|dd �dkr |�|� q| �tj|� d S )N�_r ri )r2 ri �
startswith�getattrr6 ZassertCountEqual�__all__)r* r� rV 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� r� zbMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.verify_requestr c S s | j d7 _ tj�| |� d S )Nr� )�shutdown_calledri 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 r? � s r? r r� )ri r| r4 r} r3 r\ rq rr rL �closer� rK r� re )r* r? rP rv 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 ] }t�|j�� |� � W d � n1 s.w Y 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 @ r )z2MiscTestCase.test_threads_reaped.<locals>.MyServerNr r
r
r
r r? � r� r? r �
N)ri r� r| r4 r} rb r3 r� rL r� Z
assertLessr� Z_threadsre )r* r? rP 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__)5r� �
contextlibr� r Zselectr r3 r5 r] Zunittestri Z test.supportr" r r r r r r# Zrequiresrt r4 r
ry Z
skipUnlessr� rg r� r r Z
SHORT_TIMEOUTr rj r� r r� r �contextmanagerr% ZTestCaser& r� r| r� ZBaseRequestHandlerr� r� r� r� r� r� r �mainr
r
r
r �<module> sh
�
��
g,�M2 �