Your IP : 216.73.216.85

Current Path : /opt/alt/python39/lib64/python3.9/test/__pycache__/
Upload File :
Current File : //opt/alt/python39/lib64/python3.9/test/__pycache__/test_socketserver.cpython-39.pyc

a

XC?h�F�@s�dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlm
Z
mZmZddlmZej�d�dZejZeed�Ze	�ed�Zeed	�Ze	�ed
�Zdd�ZejZejjfd
d�Ze�re�rGdd�de
je
j �Z!Gdd�de
je
j"�Z#ej$dd��Z%Gdd�de	j&�Z'Gdd�de	j&�Z(Gdd�de
j)�Z*Gdd�de
j+�Z,Gdd�de
j-e*�Z.e�r�Gdd �d e
je*�Z/Gd!d"�d"e	j&�Z0Gd#d$�d$e	j&�Z1e2d%k�r�e	�3�dS)&z
Test suite for socketserver.
�N)�
reap_children�reap_threads�verbose)�
socket_helperZnetworkshello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingcCsttd�rt�|�dS)z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr)�n�r�;/opt/alt/python39/lib64/python3.9/test/test_socketserver.py�signal_alarm s
rcCs:t|ggg|�\}}}||vr(|�|�Std|f��dS)Nztimed out on %r)�_real_selectZrecv�RuntimeError)Zsockr�timeout�r�w�xrrr
�receive(s
rc@seZdZdS)�ForkingUnixStreamServerN��__name__�
__module__�__qualname__rrrr
r0src@seZdZdS)�ForkingUnixDatagramServerNrrrrr
r4src
cs`t��}|dkrt�d�z.z
dVWn�Yn0Wtjj|dd�ntjj|dd�0dS)zBTests that a custom child process is not waited on (Issue 1540386)r�HN)Zexitcode)�osr�_exit�test�supportZwait_process)Ztestcase�pidrrr
�simple_subprocess9s


r"c@s�eZdZdZdd�Zdd�Zdd�Zdd	�Zed
d��Z	dd
�Z
dd�Zdd�Zdd�Z
edd��Zedd��Zedd��Zeedd���Zdd�Zdd�Zed d!��Zed"d#��Zed$d%��Zeed&d'���Zed(d)��Zd*d+�Zd,d-�Zd.d/�Zd0S)1�SocketServerTestzTest all socket servers.cCstd�d|_g|_dS)N�<r)rZ	port_seed�
test_files��selfrrr
�setUpKszSocketServerTest.setUpc	CsNtd�t�|jD]&}zt�|�Wqty8Yq0qg|jdd�<dS�Nr)rrr%r�remove�OSError)r'�fnrrr
�tearDownPs
zSocketServerTest.tearDowncCs8|tjkrtdfSd}tjd|d�}|j�|�|SdS)Nrzunix_socket.)�prefix�dir)�socketZAF_INET�HOST�tempfileZmktempr%�append)r'�protor/r,rrr
�pickaddr[s
zSocketServerTest.pickaddrc
Cs�Gdd�d|�}Gdd�d|�}tr,td�z|||�}Wn8tyr}z |�d|||f�WYd}~n
d}~00|�|j|j���|S)Nc@seZdZdd�ZdS)z.SocketServerTest.make_server.<locals>.MyServercSs|�|��dS�N)Z
close_request�r'�request�client_addressrrr
�handle_errorhs
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)rrrr:rrrr
�MyServergsr;c@seZdZdd�ZdS)z/SocketServerTest.make_server.<locals>.MyHandlercSs|j��}|j�|�dSr6)�rfile�readline�wfile�write)r'�linerrr
�handlems
z6SocketServerTest.make_server.<locals>.MyHandler.handleN�rrrrArrrr
�	MyHandlerlsrCzcreating serverz!Cannot create server (%s, %s): %s)r�print�PermissionErrorZskipTest�assertEqual�server_addressr0Zgetsockname)r'�addr�svrcls�hdlrbaser;rC�server�errr
�make_serverfs�zSocketServerTest.make_servercCs�|�|�|j�||�}|j}tr4td|�td|�tjd||jddid�}d|_	|�
�trhtd�td	�D]}tr�td
|�||j|�qptr�td�|��|�
�|��|�d|j���tr�t|tj�r�|�|j�tr�td
�dS)NzADDR =zCLASS =z
%s serving�
poll_interval�{�G�z�?��name�target�kwargsTzserver running�ztest clientzwaiting for server����done)rMr5�address_familyrGrrD�	threading�Thread�
serve_forever�daemon�start�range�shutdown�join�server_closerFr0�fileno�HAVE_FORKING�
isinstance�socketserver�ForkingMixInZassertFalseZactive_children)r'rIrJZtestfuncrKrH�t�irrr
�
run_server{s4�

�zSocketServerTest.run_servercCs�t�|tj��^}|�|�|�t�t|d�}}|rRd|vrRt|d�}||7}q2|�|t�Wd�n1sr0YdS�N�d�
)r0�SOCK_STREAM�connect�sendall�TEST_STRrrF�r'r4rH�sZbuf�datarrr
�stream_examine�s



zSocketServerTest.stream_examinecCs�t�|tj��t}tr.|tjkr.|�|�|��|�t|�t|d�}}|rhd|vrht|d�}||7}qH|�	|t�Wd�n1s�0YdSri)
r0Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSrZbindr5ZsendtororrFrprrr
�
dgram_examine�s

zSocketServerTest.dgram_examinecCs|�tjtj|j�dSr6)rhrd�	TCPServer�StreamRequestHandlerrsr&rrr
�test_TCPServer�s�zSocketServerTest.test_TCPServercCs|�tjtj|j�dSr6)rhrdZThreadingTCPServerrwrsr&rrr
�test_ThreadingTCPServer�s�z(SocketServerTest.test_ThreadingTCPServercCs@t|��$|�tjtj|j�Wd�n1s20YdSr6)r"rhrdZForkingTCPServerrwrsr&rrr
�test_ForkingTCPServer�s

�z&SocketServerTest.test_ForkingTCPServercCs|�tjtj|j�dSr6)rhrd�UnixStreamServerrwrsr&rrr
�test_UnixStreamServer�s�z&SocketServerTest.test_UnixStreamServercCs|�tjtj|j�dSr6)rhrdZThreadingUnixStreamServerrwrsr&rrr
�test_ThreadingUnixStreamServer�s�z/SocketServerTest.test_ThreadingUnixStreamServercCs>t|��"|�ttj|j�Wd�n1s00YdSr6)r"rhrrdrwrsr&rrr
�test_ForkingUnixStreamServer�s

�z-SocketServerTest.test_ForkingUnixStreamServercCs|�tjtj|j�dSr6)rhrdZ	UDPServer�DatagramRequestHandlerrur&rrr
�test_UDPServer�s�zSocketServerTest.test_UDPServercCs|�tjtj|j�dSr6)rhrdZThreadingUDPServerrrur&rrr
�test_ThreadingUDPServer�s�z(SocketServerTest.test_ThreadingUDPServercCs@t|��$|�tjtj|j�Wd�n1s20YdSr6)r"rhrdZForkingUDPServerrrur&rrr
�test_ForkingUDPServer�s

�z&SocketServerTest.test_ForkingUDPServercCs|�tjtj|j�dSr6)rhrd�UnixDatagramServerrrur&rrr
�test_UnixDatagramServer�s�z(SocketServerTest.test_UnixDatagramServercCs|�tjtj|j�dSr6)rhrdZThreadingUnixDatagramServerrrur&rrr
� test_ThreadingUnixDatagramServer�s�z1SocketServerTest.test_ThreadingUnixDatagramServercCs|�ttj|j�dSr6)rhrrdrrur&rrr
�test_ForkingUnixDatagramServer�s�z/SocketServerTest.test_ForkingUnixDatagramServercCs�Gdd�dtj�}Gdd�dtj�}g}td�D]<}|tdf|�}tjd|jdd	id
�}d|_|�	||f�q0|D]\}}|�
�|��qr|D]\}}|��|�
�q�dS)Nc@seZdZdS)z0SocketServerTest.test_shutdown.<locals>.MyServerNrrrrr
r;sr;c@seZdZdS)z1SocketServerTest.test_shutdown.<locals>.MyHandlerNrrrrr
rCsrC�rzMyServer servingrNrOrPT)rdrvrwr]r1rXrYrZr[r3r\r^r_r`)r'r;rCZthreadsrgrqrfrrr
�
test_shutdown�s$�
zSocketServerTest.test_shutdowncCs4Gdd�dtjtj�}|tdfdd��}|��dS)Nc@seZdZdS)z9SocketServerTest.test_close_immediately.<locals>.MyServerNrrrrr
r;sr;rcSsdSr6rrrrr
�<lambda>�z9SocketServerTest.test_close_immediately.<locals>.<lambda>)rd�ThreadingMixInrvr1r`)r'r;rKrrr
�test_close_immediatelysz'SocketServerTest.test_close_immediatelyc	CsNtd�D]@}|�t��"t�tdftj�Wd�q1s>0YqdS)NirU)r]�assertRaises�
OverflowErrorrdrvr1rw)r'rgrrr
�test_tcpserver_bind_leaks

�z)SocketServerTest.test_tcpserver_bind_leakcCsHt�tdftj��}Wd�n1s(0Y|�d|j���dS)NrrU)rdrvr1rwrFr0ra)r'rKrrr
�test_context_manager's
�z%SocketServerTest.test_context_managerN)rrr�__doc__r(r-r5rMrrhrsrurxry�requires_forkingrz�requires_unix_socketsr|r}r~r�r�r�r�r�r�r�r�r�r�rrrr
r#HsH
"








r#c@sXeZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Ze	dd
��Z
e	dd��Zdd�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.cCstj�tjj�dSr6)rr �unlink�TESTFNr&rrr
r-3szErrorHandlerTest.tearDowncCstt�|jdd�dS�NT��handled)�BaseErrorTestServer�
ValueError�check_resultr&rrr
�test_sync_handled6sz"ErrorHandlerTest.test_sync_handledcCsB|�t��tt�Wd�n1s(0Y|jdd�dS�NFr�)r��
SystemExitr�r�r&rrr
�test_sync_not_handled:s&z&ErrorHandlerTest.test_sync_not_handledcCstt�|jdd�dSr�)�ThreadingErrorTestServerr�r�r&rrr
�test_threading_handled?sz'ErrorHandlerTest.test_threading_handledcCstt�|jdd�dSr�)r�r�r�r&rrr
�test_threading_not_handledCsz+ErrorHandlerTest.test_threading_not_handledcCstt�|jdd�dSr�)�ForkingErrorTestServerr�r�r&rrr
�test_forking_handledGsz%ErrorHandlerTest.test_forking_handledcCstt�|jdd�dSr�)r�r�r�r&rrr
�test_forking_not_handledLsz)ErrorHandlerTest.test_forking_not_handledcCsLttjj��,}dd|}|�|��|�Wd�n1s>0YdS)N�Handler called
�Error handled
)�openrr r�rF�read)r'r��log�expectedrrr
r�QszErrorHandlerTest.check_resultN)
rrrr�r-r�r�r�r�r�r�r�r�rrrr
r�.s

r�cs,eZdZ�fdd�Zdd�Zdd�Z�ZS)r�csp||_t��tdft�t�|j��Wd�n1s:0Yz|��W|�	�n
|�	�0|�
�dSr))�	exception�super�__init__r1�
BadHandlerr0�create_connectionrG�handle_requestr`�	wait_done)r'r���	__class__rr
r�Xs
zBaseErrorTestServer.__init__cCs<ttjjd��}|�d�Wd�n1s.0YdS)N�ar�)r�rr r�r?)r'r8r9r�rrr
r:csz BaseErrorTestServer.handle_errorcCsdSr6rr&rrr
r�gszBaseErrorTestServer.wait_done)rrrr�r:r��
__classcell__rrr�r
r�Wsr�c@seZdZdd�ZdS)r�cCsHttjjd��}|�d�Wd�n1s.0Y|j�d��dS)Nr�r�z
Test error)r�rr r�r?rKr�)r'r�rrr
rAls(zBadHandler.handleNrBrrrr
r�ksr�cs0eZdZ�fdd�Z�fdd�Zdd�Z�ZS)r�cs t��|_t�j|i|��dSr6)rX�EventrVr�r��r'�pos�kwr�rr
r�ts
z!ThreadingErrorTestServer.__init__cs t�j|i|��|j��dSr6)r��shutdown_requestrV�setr�r�rr
r�xsz)ThreadingErrorTestServer.shutdown_requestcCs|j��dSr6)rV�waitr&rrr
r�|sz"ThreadingErrorTestServer.wait_done)rrrr�r�r�r�rrr�r
r�rsr�c@seZdZdS)r�Nrrrrr
r��sr�c@seZdZdd�Zdd�ZdS)�SocketWriterTestcCs�Gdd�dtj�}t�tdf|�}|�|j�t�|jtjtj	�}|�|�
|j�Wd�n1sh0Y|��|�
|jtj�|�|j|j�dS)Nc@seZdZdd�ZdS)z-SocketWriterTest.test_basics.<locals>.HandlercSs*|j|j_|j��|j_|j��|j_dSr6)r>rKra�wfile_filenor8�request_filenor&rrr
rA�s
z4SocketWriterTest.test_basics.<locals>.Handler.handleNrBrrrr
�Handler�sr�r)rdrwrvr1�
addCleanupr`r0rWrl�IPPROTO_TCPrmrGr�ZassertIsInstancer>�io�BufferedIOBaserFr�r�)r'r�rKrqrrr
�test_basics�s�*zSocketWriterTest.test_basicscstj�td��Gdd�dtj�}t�tdf|��|��j	�t
����fdd�}t�tj|�}|�tjtj|�d�d�t
�
��������fdd�}t
j|d	�}|�����|��|��jt���|��d
�|��jd�|��jtjj�|��tjjd�dS)
N�pthread_killc@seZdZdd�ZdS)z,SocketWriterTest.test_write.<locals>.HandlercSs>|j�d�|j_|j��|j_dtjj	}|j�|�|j_
dS)N�write data
�)r>r?rK�sent1r<r=�receivedrr �
SOCK_MAX_SIZE�sent2)r'Z	big_chunkrrr
rA�sz3SocketWriterTest.test_write.<locals>.Handler.handleNrBrrrr
r��sr�rcs���dSr6)r�)Zsignum�frame)�interruptedrr
�signal_handler�sz3SocketWriterTest.test_write.<locals>.signal_handlerc	s�t��jtjtj�}|��|�d��d}|��j�|���|�d�|�	d���t
j��jt
d�d�rNqnqNt|�	���Wd�n1s�0YWd�n1s�0YdS)N�rb�client response
rj�)r)r0rWrlr�ZmakefilermrGr=rnr�r
�SIGUSR1r��float�len)rq�reader�r�Zmain_threadr�Z	received2Z	response1rKrr
�
run_client�s�

z/SocketWriterTest.test_write.<locals>.run_client)rRr�r�rj)rr Z
get_attributer
rdrwrvr1r�r`rXr�r��	get_identrYr\r�r_rFr�r�r�r�r�)r'r�r�Zoriginalr�Z
backgroundrr�r
�
test_write�s*zSocketWriterTest.test_writeN)rrrr�r�rrrr
r��sr�c@s$eZdZdd�Zdd�Zdd�ZdS)�MiscTestCasecCsRg}tt�D]2}|�d�stt|�}t|dd�dkr|�|�q|�tj|�dS)N�_rrd)r/rd�
startswith�getattrr3ZassertCountEqual�__all__)r'r�rQZ
mod_objectrrr
�test_all�s

zMiscTestCase.test_allcCshGdd�dtj�}|tdftj�}t�|jtj�}|�|j�|�	�|�
�|�|jd�|�
�dS)Nc@s eZdZdd�ZdZdd�ZdS)zSMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServercSsdS)NFrr7rrr
�verify_request�szbMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.verify_requestrcSs |jd7_tj�||�dS)Nr�)�shutdown_calledrdrvr�)r'r8rrr
r��szdMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.shutdown_requestN)rrrr�r�r�rrrr
r;�sr;rr�)rdrvr1rwr0rWrlrmrG�closer�rFr�r`)r'r;rKrqrrr
�4test_shutdown_request_called_if_verify_request_false�s	zAMiscTestCase.test_shutdown_request_called_if_verify_request_falsec	Cs�Gdd�dtjtj�}|tdftj�}td�D]8}t�|j��|�	�Wd�q.1s\0Yq.|�
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eZdZdS)z2MiscTestCase.test_threads_reaped.<locals>.MyServerNrrrrr
r;�sr;r�
N)rdr�rvr1rwr]r0r�rGr�Z
assertLessr�Z_threadsr`)r'r;rKrrrr
�test_threads_reaped�s(z MiscTestCase.test_threads_reapedN)rrrr�r�r�rrrr
r��s
r��__main__)4r��
contextlibr�rZselectr
r0r2rXZunittestrdZtest.supportrrrrrr Zrequiresror1r	rtZ
skipUnlessr�rbr�rrZ
SHORT_TIMEOUTrrer{rr�r�contextmanagerr"ZTestCaser#r�rvr�ZBaseRequestHandlerr�r�r�r�r�r�r�mainrrrr
�<module>s`
�
��
g)�M2