Your IP : 216.73.216.213
3
��2avD �
@ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl m Z m Z m
Z
y d dlZW n ek
r� dZY nX e
jjd� dZe
jjZeed�Zejed�Zeed�Zejed�Zd d
� ZejZd&d d
�Z eo�e�r G dd� dej ej �Z G dd� dej ej �Z!e j"dd� �Z#ejed�G dd� dej$��Z%G dd� dej$�Z&G dd� dej'�Z(G dd � d ej)�Z*G d d � d ej+e(�Z,e�r�G d d � d ej e(�Z-G d!d"� d"ej$�Z.G d#d$� d$ej$�Z/e0d%k�r�ej1� dS )'� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�rtj| � d S )N�alarm)�hasattr�signalr )�n� r �6/opt/alt/python36/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
d V W n � Y nX W d t j|d�\}}| j||� | jd|� X d S )Nr �H � i H )�osr �_exit�waitpid� assertEqual)Ztestcase�pidZpid2�statusr r r �simple_subprocess: s
r$ z!Threading required for this test.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-S ).�SocketServerTestc 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 r8 r r r r �MyServerk s r9 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_addressr0 Z getsockname)r( �addr�svrcls�hdlrbaser9 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���)rG r5 �address_familyrB r rA � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closer! r0 �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalseZactive_children)r( rD rE ZtestfuncrF rC �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 �
)r0 � SOCK_STREAM�connect�sendall�TEST_STRr r! �close)r( r4 rC �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 )Nrb rc ) r0 Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr5 Zsendtorg r r! rh )r( r4 rC ri rj rk r r r �
dgram_examine� s
z SocketServerTest.dgram_examinec C s | j tjtj| j� d S )N)ra r] � TCPServer�StreamRequestHandlerrl )r( r r r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | j tjtj| j� d S )N)ra r] ZThreadingTCPServerrp rl )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$ ra r] ZForkingTCPServerrp rl )r( r r r �test_ForkingTCPServer� s
z&SocketServerTest.test_ForkingTCPServerc C s | j tjtj| j� d S )N)ra r] �UnixStreamServerrp rl )r( r r r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | j tjtj| j� d S )N)ra r] ZThreadingUnixStreamServerrp rl )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$ ra r r] rp rl )r( r r r � test_ForkingUnixStreamServer� s
z-SocketServerTest.test_ForkingUnixStreamServerc C s | j tjtj| j� d S )N)ra r] Z UDPServer�DatagramRequestHandlerrn )r( r r r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | j tjtj| j� d S )N)ra r] ZThreadingUDPServerrx rn )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$ ra r] ZForkingUDPServerrx rn )r( r r r �test_ForkingUDPServer� s
z&SocketServerTest.test_ForkingUDPServerc C s | j tjtj| j� d S )N)ra r] �UnixDatagramServerrx rn )r( r r r �test_UnixDatagramServer� s z(SocketServerTest.test_UnixDatagramServerc C s | j tjtj| j� d S )N)ra r] ZThreadingUnixDatagramServerrx rn )r( r r r � test_ThreadingUnixDatagramServer� s z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | j ttj| j� d S )N)ra r r] rx rn )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 r9 s r9 c @ s e Zd ZdS )z1SocketServerTest.test_shutdown.<locals>.MyHandlerN)r r r r r r r r@ s r@ r r zMyServer servingrH g{�G�z�?)rI rJ rK T)r] ro rp rV r1 rQ rR rS rT r3 rU rW rX rY )r( r9 r@ Zthreadsr` ri 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 rM rO )rV � assertRaises�
OverflowErrorr] ro r1 rp )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 rM rO )r] ro r1 rp r! r0 rZ )r( rF r r r �test_context_manager# s
z%SocketServerTest.test_context_managerN) r r r r) r- r5 rG r ra rl rn rq rr �requires_forkingrs �requires_unix_socketsru rv rw ry rz r{ r} r~ r r� r� r� r r r r r% K s0 #
r% c @ sl e Zd Zdd� Zdd� Zdd� Zejed�dd � �Z ejed�d
d � �Z
e d d
� �Z e dd� �Z
dd� ZdS )�ErrorHandlerTestc 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� �unittest�
skipUnlessrQ 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__r1 �
BadHandlerr0 Zcreate_connectionrB �handle_requestrY � 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( r6 r7 r� r r r r8 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� r8 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= rF 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)rQ �EventrN r� r� )r( �pos�kw)r� r r r� u s
z!ThreadingErrorTestServer.__init__c s t � j||� | jj� d S )N)r� �shutdown_requestrN �set)r( r� r� )r� r r r� y s z)ThreadingErrorTestServer.shutdown_requestc C s | j j� d S )N)rN �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< rF rZ � wfile_filenor6 �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] rp ro r1 �
addCleanuprY r0 rP rd � IPPROTO_TCPre rB r� ZassertIsInstancer< �io�BufferedIOBaser! r� r� )r( r� rF ri 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= rF �sent1r: 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
rb rM )r )r0 rP rd r� �makefilere rB r; rf r� r �SIGUSR1r� �float�len)ri �reader)r� � main_threadr� � received2� response1rF r r �
run_client� s
z/SocketWriterTest.test_write.<locals>.run_client)rJ s write data
s client response
rb )r� r� Z
get_attributer r] rp ro r1 r� rY rQ r� r� � get_identrR rU r� rX r! r� r� r� r� r� )r( r� r� Zoriginalr� Z
backgroundr )r� r� r� r� r� rF r �
test_write� s* zSocketWriterTest.test_writeN)r r r r� r� r� rQ 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�getattrr3 ZassertCountEqual�__all__)r( r� rI 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( r6 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 _ tjj| |� d S )NrM )�shutdown_calledr] ro r� )r( r6 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 r9 � s r9 r rM )r] ro r1 rp r0 rP rd re rB rh r� r! r� rY )r( r9 rF ri 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 )2�
contextlibr� r Zselectr r0 r2 r� r] Z test.supportr� r r r rQ � ImportErrorr� Zrequiresrg r1 r rm r� r� r[ r� r
r r r^ rt r r| r �contextmanagerr$ ZTestCaser% r� ro r� ZBaseRequestHandlerr� ZThreadingMixInr� r� r� r� r �mainr r r r �<module> sZ
_,
N"