File: //lib64/python3.6/test/__pycache__/test_socketserver.cpython-36.pyc
3
\vD �
@ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z ddl
m Z m
Z
mZ y ddlZW n ek
r� dZY nX e jjd� dZe jjZeed�Zejed�Zeed�Zejed �Zd
d � ZejZ d'd
d�Z eo�e�r G dd� de j e j �Z G dd� de j e j!�Z"ej#dd� �Z$ejed�G dd� dej%��Z&G dd� dej%�Z'G dd� de j(�Z)G d d � 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#ej%�Z/G d$d%� d%ej%�Z0e1d&k�r�ej2� dS )(z
Test suite for socketserver.
� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�rtj| � dS )z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr )�n� r �)/usr/lib64/python3.6/test_socketserver.py� signal_alarm! s
r
� c C s: t | gg g |�\}}}| |kr(| j|�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 ZdZdS )�ForkingUnixStreamServerTN)�__name__�
__module__� __qualname__�_block_on_closer r r r r 1 s r c @ s e Zd ZdZdS )�ForkingUnixDatagramServerTN)r r r r r r r r r 5 s r c c sh t j� }|dkrt jd� z y
dV W n � Y nX W dt j|d�\}}| j||� | jd|� X dS )zBTests that a custom child process is not waited on (Issue 1540386)r �H N� i H )�osr �_exit�waitpid� assertEqual)Ztestcase�pidZpid2Zstatusr r r �simple_subprocess: s
r# z!Threading required for this test.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.S )/�SocketServerTestzTest all socket servers.c C s t d� d| _g | _d S )N�< r )r
Z port_seed�
test_files)�selfr r r �setUpO s zSocketServerTest.setUpc C sT t d� t� x2| jD ](}ytj|� W q tk
r< Y qX qW g | jd d �< d S )Nr )r
r r& r �remove�OSError)r' �fnr r r �tearDownT s
zSocketServerTest.tearDownc C s8 |t jkrtdfS d }tjd|d�}| jj|� |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 sN G dd� d|�}G dd� d|�}t r,td� |||�}| j|j|jj� � |S )Nc @ s e Zd ZdZdd� ZdS )z.SocketServerTest.make_server.<locals>.MyServerTc S s | j |� � d S )N)Z
close_request)r' �request�client_addressr r r � handle_errorn s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r r7 r r r r �MyServerk s r8 c @ s e Zd Zdd� ZdS )z/SocketServerTest.make_server.<locals>.MyHandlerc S s | j j� }| jj|� d S )N)�rfile�readline�wfile�write)r' �liner r r �handles s
z6SocketServerTest.make_server.<locals>.MyHandler.handleN)r r r r> r r r r � MyHandlerr s r? zcreating server)r �printr! �server_addressr/ Z getsockname)r' �addr�svrcls�hdlrbaser8 r? �serverr r r � make_serverj s
z SocketServerTest.make_serverc C s� | j | j|j�||�}|j}tr4td|� td|� tjd| |jddid�}d|_ |j
� trhtd� x*t d �D ] }tr�td
|� ||j|� qrW tr�td � |j � |j
� |j� | jd|jj� � tr�t|tj�r�| j|j� tr�td
� d S )NzADDR =zCLASS =z
%s serving�
poll_intervalg{�G�z�?)�name�target�kwargsTzserver running� z test clientzwaiting for server� �done���)rF r4 �address_familyrA r r@ � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closer! r/ �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalseZactive_children)r' rC rD ZtestfuncrE rB �t�ir r r �
run_server| s8
zSocketServerTest.run_serverc C sl t j |t j�}|j|� |jt� t|d� }}x"|rRd|krRt|d�}||7 }q2W | j|t� |j� d S )N�d �
)r/ � SOCK_STREAM�connect�sendall�TEST_STRr r! �close)r' r3 rB �s�buf�datar r r �stream_examine� s
z SocketServerTest.stream_examinec C s� t j |t j�}tr,|t jkr,|j| j|�� |jt|� t|d� }}x"|rhd|krht|d�}||7 }qHW | j |t� |j
� d S )Nra rb ) r/ Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr4 Zsendtorf r r! rg )r' r3 rB rh ri rj r r r �
dgram_examine� s
z SocketServerTest.dgram_examinec C s | j tjtj| j� d S )N)r` r\ � TCPServer�StreamRequestHandlerrk )r' r r r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | j tjtj| j� d S )N)r` r\ ZThreadingTCPServerro rk )r' r r r �test_ThreadingTCPServer� s z(SocketServerTest.test_ThreadingTCPServerc C s, t | �� | jtjtj| j� W d Q R X d S )N)r# r` r\ ZForkingTCPServerro rk )r' r r r �test_ForkingTCPServer� s
z&SocketServerTest.test_ForkingTCPServerc C s | j tjtj| j� d S )N)r` r\ �UnixStreamServerro rk )r' r r r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | j tjtj| j� d S )N)r` r\ ZThreadingUnixStreamServerro rk )r' r r r � test_ThreadingUnixStreamServer� s z/SocketServerTest.test_ThreadingUnixStreamServerc C s* t | �� | jttj| j� W d Q R X d S )N)r# r` r r\ ro rk )r' r r r � test_ForkingUnixStreamServer� s
z-SocketServerTest.test_ForkingUnixStreamServerc C s | j tjtj| j� d S )N)r` r\ Z UDPServer�DatagramRequestHandlerrm )r' r r r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | j tjtj| j� d S )N)r` r\ ZThreadingUDPServerrw rm )r' r r r �test_ThreadingUDPServer� s z(SocketServerTest.test_ThreadingUDPServerc C s, t | �� | jtjtj| j� W d Q R X d S )N)r# r` r\ ZForkingUDPServerrw rm )r' r r r �test_ForkingUDPServer� s
z&SocketServerTest.test_ForkingUDPServerc C s | j tjtj| j� d S )N)r` r\ �UnixDatagramServerrw rm )r' r r r �test_UnixDatagramServer� s z(SocketServerTest.test_UnixDatagramServerc C s | j tjtj| j� d S )N)r` r\ ZThreadingUnixDatagramServerrw rm )r' r r r � test_ThreadingUnixDatagramServer� s z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | j ttj| j� d S )N)r` r r\ rw rm )r' r r r � test_ForkingUnixDatagramServer� s z/SocketServerTest.test_ForkingUnixDatagramServerc C s� G dd� dt j�}G dd� dt j�}g }xHtd�D ]<}|tdf|�}tjd|jdd id
�}d |_|j ||f� q2W x |D ]\}}|j
� |j � qxW x |D ]\}}|j � |j
� q�W d S ) Nc @ s e Zd ZdS )z0SocketServerTest.test_shutdown.<locals>.MyServerN)r r r r r r r r8 s r8 c @ s e Zd ZdS )z1SocketServerTest.test_shutdown.<locals>.MyHandlerN)r r r r r r r r? s r? r r zMyServer servingrG g{�G�z�?)rH rI rJ T)r\ rn ro rU r0 rP rQ rR rS r2 rT rV rW rX )r' r8 r? Zthreadsr_ rh r^ r r r �
test_shutdown� s" z SocketServerTest.test_shutdownc C s> x8t d�D ],}| jt�� tjtdftj� W d Q R X q
W d S )Ni rL rN )rU � assertRaises�
OverflowErrorr\ rn r0 ro )r' r_ r r r �test_tcpserver_bind_leak s
z)SocketServerTest.test_tcpserver_bind_leakc
C s4 t jtdft j��}W d Q R X | jd|jj� � d S )Nr rL rN )r\ rn r0 ro r! r/ rY )r' rE r r r �test_context_manager# s
z%SocketServerTest.test_context_managerN) r r r �__doc__r( r, r4 rF r r` rk rm rp rq �requires_forkingrr �requires_unix_socketsrt ru rv rx ry rz r| r} r~ r r� r� r r r r r$ K s2 #
r$ c @ sp e Zd ZdZdd� Zdd� Zdd� Zeje d�d d
� �Z
eje d�d d � �Z e d
d� �Z
e dd� �Zdd� ZdS )�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 jjt jj� t� d S )N)�test�support�unlink�TESTFNr )r' r r r r, / s zErrorHandlerTest.tearDownc C s t t� | jdd� d S )NT)�handled)�BaseErrorTestServer�
ValueError� check_result)r' r r r �test_sync_handled3 s z"ErrorHandlerTest.test_sync_handledc
C s. | j t�� tt� W d Q R X | jdd� d S )NF)r� )r� �
SystemExitr� r� )r' r r r �test_sync_not_handled7 s z&ErrorHandlerTest.test_sync_not_handledz!Threading required for this test.c C s t t� | jdd� d S )NT)r� )�ThreadingErrorTestServerr� r� )r' r r r �test_threading_handled< s z'ErrorHandlerTest.test_threading_handledc C s t t� | jdd� d S )NF)r� )r� r� r� )r' r r r �test_threading_not_handledA s z+ErrorHandlerTest.test_threading_not_handledc C s t t� | jdd� d S )NT)r� )�ForkingErrorTestServerr� r� )r' r r r �test_forking_handledF s z%ErrorHandlerTest.test_forking_handledc C s t t� | jdd� d S )NF)r� )r� r� r� )r' r r r �test_forking_not_handledK s z)ErrorHandlerTest.test_forking_not_handledc
C s8 t tjj��"}dd| }| j|j� |� W d Q R X d S )NzHandler called
zError handled
)�openr� r� r� r! �read)r' r� �log�expectedr r r r� P s z ErrorHandlerTest.check_resultN)r r r r� r, r� r� �unittest�
skipUnlessrP r� r� r� r� r� r� r r r r r� * s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )r� Tc sT || _ t� jtdft� tj| j�� W d Q R X z | j� W d | j � X | j
� d S )Nr ) � exception�super�__init__r0 �
BadHandlerr/ Zcreate_connectionrA �handle_requestrX � wait_done)r' r� )� __class__r r r� Y s
z BaseErrorTestServer.__init__c C s( t tjjd��}|jd� W d Q R X d S )N�azError handled
)r� r� r� r� r< )r' r5 r6 r� r r r r7 d s z BaseErrorTestServer.handle_errorc C s d S )Nr )r' r r r r� h s z BaseErrorTestServer.wait_done)r r r r r� r7 r� �
__classcell__r r )r� r r� V s r� c @ s e Zd Zdd� ZdS )r� c C s4 t tjjd��}|jd� W d Q R X | jjd��d S )Nr� zHandler called
z
Test error)r� r� r� r� r< rE r� )r' r� r r r r> m s zBadHandler.handleN)r r r r> r r r r r� l s r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )r� c s t j� | _t� j||� d S )N)rP �EventrM r� r� )r' �pos�kw)r� r r r� u s
z!ThreadingErrorTestServer.__init__c s t � j||� | jj� d S )N)r� �shutdown_requestrM �set)r' r� r� )r� r r r� y s z)ThreadingErrorTestServer.shutdown_requestc C s | j j� d S )N)rM �wait)r' 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 ZdZdS )r� TN)r r r r r r r r r� � s r� c @ s( e Zd Zdd� Zejed�dd� �ZdS )�SocketWriterTestc
C s� G dd� dt j�}t jtdf|�}| j|j� tj|jtjtj �}|� |j
|j � W d Q R X |j � | j
|jtj� | j|j|j� d S )Nc @ s e Zd Zdd� ZdS )z-SocketWriterTest.test_basics.<locals>.Handlerc S s* | j | j_ | j j� | j_| jj� | j_d S )N)r; rE rY � wfile_filenor5 �request_fileno)r' r r r r> � s
z4SocketWriterTest.test_basics.<locals>.Handler.handleN)r r r r> r r r r �Handler� s r� r )r\ ro rn r0 �
addCleanuprX r/ rO rc � IPPROTO_TCPrd rA r� ZassertIsInstancer; �io�BufferedIOBaser! r� r� )r' r� rE rh r r r � test_basics� s z SocketWriterTest.test_basicsz!Threading required for this test.c s t jjtd��G dd� dtj�}tjtdf|��| j�j � t
j � � � fdd�}tjtj |�}| jtjtj |� d �d �t
j
� �� �����fdd�}t
j|d �}|j� �j� |j� | j�jt��� | j�d
� | j�jd � | j�jt jj� | j�t jjd � d S )
N� pthread_killc @ s e Zd Zdd� ZdS )z,SocketWriterTest.test_write.<locals>.Handlerc S s> | j jd�| j_| jj� | j_dtjj }| j j|�| j_
d S )Ns write data
� ) r; r< rE �sent1r9 r: �receivedr� r� �
SOCK_MAX_SIZE�sent2)r' Z big_chunkr r r r> � s z3SocketWriterTest.test_write.<locals>.Handler.handleN)r r r r> r r r r r� � s r� r c s � j � d S )N)r� )Zsignum�frame)� interruptedr r �signal_handler� s z3SocketWriterTest.test_write.<locals>.signal_handlerc s� t j �jt jt j�} | �t | jd��^}| j�j� |j� �| jd� |j d� x"��t
j � � j t
d�d�rPP qPW t|j � ��W d Q R X W d Q R X d S )N�rbs client response
ra rL )r )r/ rO rc r� �makefilerd rA r: re r� r �SIGUSR1r� �float�len)rh �reader)r� � main_threadr� � received2� response1rE r r �
run_client� s
z/SocketWriterTest.test_write.<locals>.run_client)rI s write data
s client response
ra )r� r� Z
get_attributer r\ ro rn r0 r� rX rP r� r� � get_identrQ rT r� rW r! r� r� r� r� r� )r' r� r� Zoriginalr� Z
backgroundr )r� r� r� r� r� rE r �
test_write� s* zSocketWriterTest.test_writeN)r r r r� r� r� rP r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� MiscTestCasec C sV g }x>t t�D ]2}|jd�stt|�}t|dd �dkr|j|� qW | jtj|� d S )N�_r r\ )r. r\ �
startswith�getattrr2 ZassertCountEqual�__all__)r' r� rH Z
mod_objectr r r �test_all� s
zMiscTestCase.test_allc C sh G dd� dt j�}|tdft j�}tj|jtj�}|j|j� |j � |j
� | j |j d� |j
� 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' r5 r6 r r r �verify_request� s zbMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServer.verify_requestr c S s | j d7 _ tjj| |� d S )NrL )�shutdown_calledr\ rn r� )r' r5 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 r8 � s r8 r rL )r\ rn r0 ro r/ rO rc rd rA rg r� r! r� rX )r' r8 rE rh r r r �4test_shutdown_request_called_if_verify_request_false� s zAMiscTestCase.test_shutdown_request_called_if_verify_request_falseN)r r r r� r� r r r r r� � s
r� �__main__)r )3r� �
contextlibr� r Zselectr r/ r1 r� r\ Z test.supportr� r r r rP � ImportErrorr� Zrequiresrf r0 r rl r� r� rZ r� r
r r r] rs r r{ r �contextmanagerr# ZTestCaser$ r� rn r� ZBaseRequestHandlerr� ZThreadingMixInr� r� r� r� r �mainr r r r �<module> s\
_,
N"