File: //usr/local/lib/python3.7/test/__pycache__/test_socketserver.cpython-37.opt-2.pyc
B
��g}C � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z d dl
m Z m
Z
mZ e j�d� dZe jjZeed�Ze�ed�Zeed�Ze�ed�Zd d
� ZejZd%d d
�Zer�er�G dd� de j e j �Z G dd� de j e j �Z e j!dd� �Z"G dd� dej#�Z$G dd� dej#�Z%G dd� de j&�Z'G dd� de 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#�Z.e/d$k�r�e�0� dS )&� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�rt�| � d S )N�alarm)�hasattr�signalr )�n� r �2/usr/local/lib/python3.7/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 )�ForkingUnixDatagramServerN)r r r r r r r r 3 s r c
c sh t �� }|dkrt �d� z y
d V W n � Y nX W d t �|d�\}}| �||� | �d|� X d S )Nr �H i H )�osr �_exit�waitpid� assertEqual)Ztestcase�pidZpid2�statusr r r �simple_subprocess8 s
r" 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 �setUpL s zSocketServerTest.setUpc C sT t d� t� x2| jD ](}yt�|� W q tk
r< Y qX qW g | 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� y|||�}W n6 tk
rp } z| �d|||f � W d 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 r6 r r r r �MyServerh s r7 c @ s e Zd Zdd� ZdS )z/SocketServerTest.make_server.<locals>.MyHandlerc S s | j �� }| j�|� d S )N)�rfile�readline�wfile�write)r& �liner r r �handlen s
z6SocketServerTest.make_server.<locals>.MyHandler.handleN)r r r r= r r r r � MyHandlerm s r> zcreating serverz!Cannot create server (%s, %s): %s)r �print�PermissionErrorZskipTestr �server_addressr. Z getsockname)r& �addr�svrcls�hdlrbaser7 r> �server�er r r � make_serverg s z SocketServerTest.make_serverc C s� | � | �|j�||�}|j}tr4td|� td|� tjd| |jddid�}d|_ |�
� trhtd� x*t d �D ] }tr�td
|� ||j|� qrW tr�td � |� � |�
� |�� | �d |j�� � tr�t|tj�r�| �|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 r3 �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 � |t j�}|�|� |�t� t|d� }}x"|rRd|krRt|d�}||7 }q2W | �|t� |�� d S )N�d �
)r. � SOCK_STREAM�connect�sendall�TEST_STRr r �close)r& r2 rB �s�buf�datar r r �stream_examine� s
z SocketServerTest.stream_examinec C s� t � |t j�}tr,|t jkr,|�| �|�� |�t|� t|d� }}x"|rhd|krht|d�}||7 }qHW | � |t� |�
� d S )Nra rb ) r. Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr3 Zsendtorf r r rg )r& r2 rB rh ri rj r r r �
dgram_examine� s
z SocketServerTest.dgram_examinec C s | � tjtj| j� d S )N)r` r\ � TCPServer�StreamRequestHandlerrk )r& r r r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | � tjtj| j� d S )N)r` r\ ZThreadingTCPServerro rk )r& r r r �test_ThreadingTCPServer� s z(SocketServerTest.test_ThreadingTCPServerc C s, t | �� | �tjtj| 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 | � tjtj| j� d S )N)r` r\ �UnixStreamServerro rk )r& r r r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | � tjtj| j� d S )N)r` r\ ZThreadingUnixStreamServerro rk )r& r r r � test_ThreadingUnixStreamServer� s z/SocketServerTest.test_ThreadingUnixStreamServerc C s* t | �� | �ttj| 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 | � tjtj| j� d S )N)r` r\ Z UDPServer�DatagramRequestHandlerrm )r& r r r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | � tjtj| j� d S )N)r` r\ ZThreadingUDPServerrw rm )r& r r r �test_ThreadingUDPServer� s z(SocketServerTest.test_ThreadingUDPServerc C s, t | �� | �tjtj| 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 | � tjtj| j� d S )N)r` r\ �UnixDatagramServerrw rm )r& r r r �test_UnixDatagramServer� s z(SocketServerTest.test_UnixDatagramServerc C s | � tjtj| j� d S )N)r` r\ ZThreadingUnixDatagramServerrw rm )r& r r r � test_ThreadingUnixDatagramServer� s z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | � 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 |_|� ||f� q2W x |D ]\}}|�
� |� � qxW x |D ]\}}|� � |�
� q�W d S ) Nc @ s e Zd ZdS )z0SocketServerTest.test_shutdown.<locals>.MyServerN)r r r r r r r r7 s r7 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\ rn ro rU r/ rP rQ rR rS r1 rT rV rW rX )r& r7 r> Zthreadsr_ rh r^ r r r �
test_shutdown� s" z SocketServerTest.test_shutdownc
C s> x8t d�D ],}| �t�� t�tdftj� W d Q R X q
W d S )Ni rM )rU � assertRaises�
OverflowErrorr\ rn r/ ro )r& r_ r r r �test_tcpserver_bind_leak s
z)SocketServerTest.test_tcpserver_bind_leakc C s4 t �tdft j��}W d Q R X | �d|j�� � d S )Nr rM )r\ rn r/ ro r r. rY )r& rE r r r �test_context_manager# s
z%SocketServerTest.test_context_managerN) r r r r' r+ r3 rG 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# I s0 #
r# c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zed d � �Z ed
d� �Z
dd� Z dS )�ErrorHandlerTestc C s t j�t jj� d S )N)�test�support�unlink�TESTFN)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_handled2 s z"ErrorHandlerTest.test_sync_handledc C s. | � t�� tt� W d Q R X | jdd� d S )NF)r� )r� �
SystemExitr� r� )r& r r r �test_sync_not_handled6 s z&ErrorHandlerTest.test_sync_not_handledc 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_handled? s z+ErrorHandlerTest.test_threading_not_handledc C s t t� | jdd� d S )NT)r� )�ForkingErrorTestServerr� r� )r& r r r �test_forking_handledC 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_handledH s z)ErrorHandlerTest.test_forking_not_handledc C s8 t tjj��"}dd| }| �|�� |� W d Q R X d S )NzHandler called
zError handled
)�openr� r� r� r �read)r& r� �log�expectedr r r r� M s z ErrorHandlerTest.check_resultN) 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 d Q R X z | �� W d | � � X | �
� d S )Nr ) � exception�super�__init__r/ �
BadHandlerr. Zcreate_connectionrA �handle_requestrX � wait_done)r& r� )� __class__r r r� T s
z BaseErrorTestServer.__init__c C s( t tjjd��}|�d� W d Q R X d S )N�azError handled
)r� r� r� r� r; )r& r4 r5 r� r r r r6 _ s z BaseErrorTestServer.handle_errorc C s d S )Nr )r& r r r r� c s z BaseErrorTestServer.wait_done)r r r r� r6 r� �
__classcell__r r )r� r r� S s r� c @ s e Zd Zdd� ZdS )r� c C s4 t tjjd��}|�d� W d Q R X | j�d��d S )Nr� zHandler called
z
Test error)r� r� r� r� r; rE r� )r& r� r r r r= h s zBadHandler.handleN)r r r r= r r r r r� g s r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )r� c s t �� | _t� j||� d S )N)rP �EventrN r� r� )r& �pos�kw)r� r r r� p s
z!ThreadingErrorTestServer.__init__c s t � j||� | j�� d S )N)r� �shutdown_requestrN �set)r& r� r� )r� r r r� t s z)ThreadingErrorTestServer.shutdown_requestc C s | j �� d S )N)rN �wait)r& r r r r� x s z"ThreadingErrorTestServer.wait_done)r r r r� r� r� r� r r )r� r r� n s r� c @ s e Zd ZdS )r� N)r r r 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 d 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 )N)r: rE rY � wfile_filenor4 �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 r/ �
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_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 )Ns write data
� ) r: r; rE �sent1r8 r9 �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 � � � d S )N)r� )Zsignum�frame)� interruptedr r �signal_handler� s z3SocketWriterTest.test_write.<locals>.signal_handlerc
s� t � �jt jt j�} | �t | �d��^}| ��j� |�� �| �d� |� d� x"��t
j � � j t
d�d�rPP qPW t|� � ��W d Q R X W d Q R X d S )N�rbs client response
ra � )r )r. rO rc r� Zmakefilerd rA r9 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)rJ s write data
s client response
ra )r� r� Z
get_attributer r\ ro rn r/ 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 r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� MiscTestCasec C sV g }x>t t�D ]2}|�d�stt|�}t|dd �dkr|�|� qW | �tj|� d S )N�_r r\ )r- r\ �
startswith�getattrr1 ZassertCountEqual�__all__)r& r� rI 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& r4 r5 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_calledr\ rn r� )r& r4 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 r7 � s r7 r r� )r\ rn r/ ro r. rO rc rd rA rg r� r r� rX )r& r7 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 )1�
contextlibr� r Zselectr r. r0 rP Zunittestr\ Z test.supportr� r r r r� Zrequiresrf r/ r rl Z
skipUnlessr� 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> sR
b)
M"