File: //usr/local/lib/python3.10/test/__pycache__/test_socketserver.cpython-310.opt-2.pyc
o
�i�G � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z d dl
m Z m
Z
d dl
mZ d dl
mZ d dl
mZ e j�d� dZejZeed�Ze�ed �Zeed
�Ze�ed �Zd d
� ZejZ e jj fdd�Z er�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 d d � d e j+�Z,G d d � d e j-e*�Z.er�G d d!� d!e j e*�Z/G d"d#� d#ej&�Z0G d$d%� d%ej&�Z1e2d&kr�e�3� dS dS )'� N)�
reap_children�verbose)� os_helper)�
socket_helper)�threading_helperZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t td�r
t�| � d S d S )N�alarm)�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 s\ � t �� }|dkrt �d� zzd V W n � W tjj|dd� d S tjj|dd� w )Nr �H )�exitcode)�osr �_exit�test�supportZ wait_process)Ztestcase�pidr
r
r �simple_subprocess; s �
�&r( c @ s� e Zd Z dd� Zdd� Zdd� Zdd� Zejd d
� �Z d d � Z
d
d� Z dd� Z dd� Z
edd� �Zedd� �Zedd� �Zeedd� ��Zdd � Zd d � Zed d � �Zed!d"� �Zed#d$� �Zeed%d&� ��Zejd'd(� �Zd)d*� Zd+d,� Z d-d.� Z d/S )0�SocketServerTestc 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 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 @ sV e Zd Z dd� Zdd� Zdd� Zdd� Zd d
� Zed d � �Z ed
d� �Z
dd� Z dS )�ErrorHandlerTestc 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 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� |�
� d S )Nc @ r )z2MiscTestCase.test_threads_reaped.<locals>.MyServerNr r
r
r
r rD � r� rD r �
)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__)4�
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 �