File: //usr/local/lib/python3.10/test/__pycache__/test_socketserver.cpython-310.pyc
o
�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
�3/usr/local/lib/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_select�recv� RuntimeError)�sockr �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)�exitcode)�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)�socket�AF_INET�HOST�tempfile�mktempr+ �append)r- �protor5 r2 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 rC r
r
r
r �MyServeri � rD c @ r> )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 rK r
r
r
r � MyHandlern rE rM zcreating serverz!Cannot create server (%s, %s): %s)r �print�PermissionErrorZskipTest� assertEqual�server_addressr6 � getsockname)r- �addr�svrcls�hdlrbaserD rM �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)rX r= �address_familyrQ r rN � threading�Thread�
serve_forever�daemon�start�range�shutdown�join� server_closerP r6 �fileno� HAVE_FORKING�
isinstance� socketserver� ForkingMixInZ assertFalse�active_children)r- rT rU ZtestfuncrV rS �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 �
)r6 � SOCK_STREAM�connect�sendall�TEST_STRr rP �r- r< rS �s�buf�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 ru )
r6 �
SOCK_DGRAM�HAVE_UNIX_SOCKETSr �bindr= �sendtor{ r rP r| r
r
r �
dgram_examine� s
�"�z SocketServerTest.dgram_examinec C � | � tjtj| j� d S r? )rt ro � TCPServer�StreamRequestHandlerr� r, r
r
r �test_TCPServer� � �z SocketServerTest.test_TCPServerc C r� r? )rt ro ZThreadingTCPServerr� r� 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( rt ro ZForkingTCPServerr� r� r, r
r
r �test_ForkingTCPServer� �
�"�z&SocketServerTest.test_ForkingTCPServerc C r� r? )rt ro �UnixStreamServerr� r� r, r
r
r �test_UnixStreamServer� � �z&SocketServerTest.test_UnixStreamServerc C r� r? )rt ro ZThreadingUnixStreamServerr� r� 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( rt r ro r� r� r, r
r
r � test_ForkingUnixStreamServer� s
�"�z-SocketServerTest.test_ForkingUnixStreamServerc C r� r? )rt ro Z UDPServer�DatagramRequestHandlerr� r, r
r
r �test_UDPServer� r� z SocketServerTest.test_UDPServerc C r� r? )rt ro ZThreadingUDPServerr� r� r, r
r
r �test_ThreadingUDPServer� r� z(SocketServerTest.test_ThreadingUDPServerc C r� r? )r( rt ro ZForkingUDPServerr� r� r, r
r
r �test_ForkingUDPServer� r� z&SocketServerTest.test_ForkingUDPServerc C r� r? )rt ro �UnixDatagramServerr� r� r, r
r
r �test_UnixDatagramServer� r� z(SocketServerTest.test_UnixDatagramServerc C r� r? )rt ro ZThreadingUnixDatagramServerr� r� r, r
r
r � test_ThreadingUnixDatagramServer� r� z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | � ttj| j� d S r? )rt r ro r� r� 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 rD � rD c @ r )z1SocketServerTest.test_shutdown.<locals>.MyHandlerNr r
r
r
r rM r� rM � r zMyServer servingrY rZ r[ T)ro r� r� rh r8 rc rd re rf r; rg ri rj rk )r- rD rM �threadsrs r} rr 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 rD r� rD r c S � d S r? r
r
r
r
r �<lambda> s z9SocketServerTest.test_close_immediately.<locals>.<lambda>)ro �ThreadingMixInr� r8 rk )r- rD rV 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 r` )rh � assertRaises�
OverflowErrorro r� r8 r� )r- rs 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 r` )ro r� r8 r� rP r6 rl )r- rV r
r
r �test_context_manager) s
� �z%SocketServerTest.test_context_managerN) r r r �__doc__r. r3 r= rX r Z reap_threadsrt r� r� 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 r3 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� rP �read)r- r� �log�expectedr
r
r r� V s "�z ErrorHandlerTest.check_resultN)
r r r r� r3 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__r8 �
BadHandlerr6 �create_connectionrQ �handle_requestrk � 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� rI )r- rA rB r� r
r
r rC h s "�z BaseErrorTestServer.handle_errorc C r� r? r
r, r
r
r r� l � z BaseErrorTestServer.wait_done)r r r r� rC r� �
__classcell__r
r
r� r r� \ s r� c @ r> )r� c C sB t tjd��
}|�d� W d � n1 sw Y | j�d��)Nr� r� z
Test error)r� r r� rI rV r� )r- r� r
r
r rK q s � zBadHandler.handleNrL r
r
r
r r� p rE r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )r� c s t �� | _t� j|i |�� d S r? )rc �Eventra r� r� �r- �pos�kwr� r
r r� y s
z!ThreadingErrorTestServer.__init__c s t � j|i |�� | j�� d S r? )r� �shutdown_requestra �setr� r� r
r r� } s z)ThreadingErrorTestServer.shutdown_requestc C s | j �� d S r? )ra �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 @ r> )z-SocketWriterTest.test_basics.<locals>.Handlerc S s* | j | j_ | j �� | j_| j�� | j_d S r? )rH rV rl � wfile_filenorA �request_filenor, r
r
r rK � s
z4SocketWriterTest.test_basics.<locals>.Handler.handleNrL r
r
r
r �Handler� rE r� r )ro r� r� r8 �
addCleanuprk r6 rb rx � IPPROTO_TCPry rQ r� ZassertIsInstancerH �io�BufferedIOBaserP r� r� )r- r� rV r} 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 @ r> )z,SocketWriterTest.test_write.<locals>.Handlerc S s> | j �d�| j_| j�� | j_dtjj }| j �|�| j_
d S )N� write data
� ) rH rI rV �sent1rF rG �receivedr% r& �
SOCK_MAX_SIZE�sent2)r- Z big_chunkr
r
r rK � s z3SocketWriterTest.test_write.<locals>.Handler.handleNrL r
r
r
r r� � rE r� r c s � � � d S r? )r� )�signum�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
rv T� )r )r6 rb rx r� �makefilery rQ rG rz r� r �SIGUSR1r� �float�len)r} �reader�r� � main_threadr� Z received2Z response1rV r
r �
run_client� s �
�P�z/SocketWriterTest.test_write.<locals>.run_client)r] r� r� rv )r% r& Z
get_attributer ro r� r� r8 r� rk rc r� r� � get_identrd rg r� rj rP 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 ro )r5 ro �
startswith�getattrr; ZassertCountEqual�__all__)r- r� r\ 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_calledro r� r� )r- rA 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 rD � s rD r r� )ro r� r8 r� r6 rb rx ry rQ �closer� rP r rk )r- rD rV r} 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 rD � r� rD r �
N)ro r� r� r8 r� rh r6 r� rQ r� Z
assertLessr� Z_threadsrk )r- rD rV 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# �selectr r6 r9 rc Zunittestro Z test.supportr% r r r r r r& Zrequiresr{ r8 r
r� Z
skipUnlessr� rm r� r r Z
SHORT_TIMEOUTr rp r� r r� r �contextmanagerr( ZTestCaser) r� r� r� ZBaseRequestHandlerr� r� r� r� r� r� r �mainr
r
r
r �<module> sh
�
��
g,�M2 �