Your IP : 216.73.216.52
��Yf�) �
@ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z d d l
m Z m
Z
m Z y d d l Z Wn e k
r� d Z Yn Xe j j d � d Z e j j Z e e d � Z e j e d � Z e e d � Z e j e d � Z d
d � Z e j Z d d
d � Z e r�Gd d � d e j e j � Z Gd d � d e j e j! � Z" e j# d d � � Z$ e j e d � Gd d � d e j% � � Z& Gd d � d e j% � Z' e( d k r e j) � d S)z
Test suite for socketserver.
� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t t d � r t j | � d S)z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr )�n� r �;/opt/alt/python35/lib64/python3.5/test/test_socketserver.py� signal_alarm" s r
� c C sQ t | g g g | � \ } } } | | k r: | j | � St d | f � � d S)Nztimed out on %r)� _real_selectZrecv� RuntimeError)Zsockr
Ztimeout�r�w�xr r r �receive* s !
r c @ s e Z d Z d S)�ForkingUnixStreamServerN)�__name__�
__module__� __qualname__r r r r r 2 s r c @ s e Z d Z d S)�ForkingUnixDatagramServerN)r r r r r r r r 6 s r c c sf t j � } | d k r% t j d � d Vt j | d � \ } } | j | | � | j d | � d S)Nr �H � i H )�osr �_exit�waitpid� assertEqual)Ztestcase�pidZpid2Zstatusr r r �simple_subprocess; s
r! z!Threading required for this test.c @ sZ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z e d
d � � Z d d
� Z
d d � Z d d � Z d d � Z
e d d � � Z e d d � � Z e d d � � Z e e d d � � � Z d d � Z d d � Z e d d! � � Z e d" d# � � Z e d$ d% � � Z e e d& d' � � � Z e d( d) � � Z d* 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 �setUpK s
zSocketServerTest.setUpc C sb t d � t � x7 | j D], } y t j | � Wq t k
rF Yq Xq Wg | j d d � <d S)Nr )r
r r$ r �remove�OSError)r% �fnr r r �tearDownP s
zSocketServerTest.tearDownc C sO | t j k r t d f Sd } t j d d d | � } | j j | � | Sd S)Nr �prefixz unix_socket.�dir)�socketZAF_INET�HOST�tempfileZmktempr$ �append)r% �protor, r) r r r �pickaddr[ s
zSocketServerTest.pickaddrc C sk Gd d � d | � } Gd d � d | � } t r< t d � | | | � } | j | j | j j � � | S)Nc @ s e Z d Z d d � Z d S)z.SocketServerTest.make_server.<locals>.MyServerc S s | j | � | j � � d S)N)Z
close_request� server_close)r% �request�client_addressr r r � handle_errorh s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r6 r r r r �MyServerg s r7 c @ s e Z d Z d d � Z d S)z/SocketServerTest.make_server.<locals>.MyHandlerc S s# | j j � } | j j | � d S)N)Zrfile�readlineZwfile�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 server)r �printr �server_addressr- Z getsockname)r% �addr�svrcls�hdlrbaser7 r< �serverr r r � make_serverf s
z SocketServerTest.make_serverc C s- | j | j | j � | | � } | j } t rJ t d | � t d | � t j d d | d | j d d d i � } d | _ | j
� t r� t d
� x7 t d � D]) } t r� t d | � | | j | � q� Wt r� t d
� | j � | j
� | j � | j d | j j � � t r)t d � d S)NzADDR =zCLASS =�namez
%s serving�target�kwargs�
poll_intervalg{�G�z�?Tzserver running� z test clientzwaiting for server� �done���)rC r2 �address_familyr> r r= � threading�Thread�
serve_forever�daemon�start�range�shutdown�joinr3 r r- �fileno)r% r@ rA ZtestfuncrB r? �t�ir r r �
run_serverw s4
zSocketServerTest.run_serverc C s� t j | t j � } | j | � | j t � t | d � } } x/ | rs d | k rs t | d � } | | 7} qE W| j | t � | j � d S)N�d s
)r- � SOCK_STREAM�connectZsendall�TEST_STRr r �close)r% r1 r? �s�buf�datar r r �stream_examine� s
z SocketServerTest.stream_examinec C s� t j | t j � } t r@ | t j k r@ | j | j | � � | j t | � t | d � } } x/ | r� d | k r� t | d � } | | 7} qf W| j | t � | j
� d S)NrY s
) r- Z
SOCK_DGRAM�HAVE_UNIX_SOCKETSr Zbindr2 Zsendtor\ r r r] )r% r1 r? r^ r_ r` r r r �
dgram_examine� s z SocketServerTest.dgram_examinec C s | j t j t j | j � d S)N)rX � socketserver� TCPServer�StreamRequestHandlerra )r% r r r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | j t j t j | j � d S)N)rX rd ZThreadingTCPServerrf ra )r% r r r �test_ThreadingTCPServer� s z(SocketServerTest.test_ThreadingTCPServerc C s4 t | � �! | j t j t j | j � Wd QRXd S)N)r! rX rd ZForkingTCPServerrf ra )r% r r r �test_ForkingTCPServer� s
z&SocketServerTest.test_ForkingTCPServerc C s | j t j t j | j � d S)N)rX rd �UnixStreamServerrf ra )r% r r r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | j t j t j | j � d S)N)rX rd ZThreadingUnixStreamServerrf ra )r% r r r � test_ThreadingUnixStreamServer� s z/SocketServerTest.test_ThreadingUnixStreamServerc C s1 t | � � | j t t j | j � Wd QRXd S)N)r! rX r rd rf ra )r% r r r � test_ForkingUnixStreamServer� s
z-SocketServerTest.test_ForkingUnixStreamServerc C s | j t j t j | j � d S)N)rX rd Z UDPServer�DatagramRequestHandlerrc )r% r r r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | j t j t j | j � d S)N)rX rd ZThreadingUDPServerrn rc )r% r r r �test_ThreadingUDPServer� s z(SocketServerTest.test_ThreadingUDPServerc C s4 t | � �! | j t j t j | j � Wd QRXd S)N)r! rX rd ZForkingUDPServerrn rc )r% r r r �test_ForkingUDPServer� s
z&SocketServerTest.test_ForkingUDPServerc C s | j t j t j | j � d S)N)rX rd �UnixDatagramServerrn rc )r% r r r �test_UnixDatagramServer� s z(SocketServerTest.test_UnixDatagramServerc C s | j t j t j | j � d S)N)rX rd ZThreadingUnixDatagramServerrn rc )r% r r r � test_ThreadingUnixDatagramServer� s z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | j t t j | j � d S)N)rX r rd rn rc )r% r r r � test_ForkingUnixDatagramServer� s z/SocketServerTest.test_ForkingUnixDatagramServerc C s Gd d � d t j � } Gd d � d t j � } g } xl t d � D]^ } | t d f | � } t j d d d | j d
d d i � } d
| _ | j | | f � qE Wx( | D] \ } } | j
� | j � q� Wx( | D] \ } } | j � | j
� q� Wd S)Nc @ s e Z d Z d S)z0SocketServerTest.test_shutdown.<locals>.MyServerN)r r r r r r r r7 � s r7 c @ s e Z d Z d S)z1SocketServerTest.test_shutdown.<locals>.MyHandlerN)r r r r r r r r< � s r< r r rD zMyServer servingrE rF rG g{�G�z�?T)rd re rf rR r. rM rN rO rP r0 rQ rS rT r3 )r% r7 r< ZthreadsrW r^ rV r r r �
test_shutdown� s"
z SocketServerTest.test_shutdownc C sK xD t d � D]6 } | j t � � t j t d f t j � Wd QRXq
Wd S)Ni rI rK )rR Z assertRaises�
OverflowErrorrd re r. rf )r% rW r r r �test_tcpserver_bind_leak s z)SocketServerTest.test_tcpserver_bind_leakN) r r r �__doc__r& r* r2 rC r rX ra rc rg rh �requires_forkingri �requires_unix_socketsrk rl rm ro rp rq rs rt ru rv rx r r r r r" G s0 r" c @ s( e Z d Z d d � Z d d � Z d S)� MiscTestCasec C sw g } xW t t � D]I } | j d � s t t | � } t | d d � d k r | j | � q W| j t j | � d S)N�_r rd )r, rd �
startswith�getattrr0 ZassertCountEqual�__all__)r% ZexpectedrD Z
mod_objectr r r �test_all s zMiscTestCase.test_allc C s� Gd d � d t j � } | t d f t j � } t j | j t j � } | j | j � | j � | j
� | j | j d � | j
� d S)Nc @ s. e Z d Z d d � Z d Z d d � Z d S)zSMiscTestCase.test_shutdown_request_called_if_verify_request_false.<locals>.MyServerc S s d S)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 d 7_ t j j | | � d S)NrI )�shutdown_calledrd re �shutdown_request)r% r4 r r r r� 0 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 rI )rd re r. rf r- rL rZ r[ r> r] Zhandle_requestr r� r3 )r% r7 rB r^ 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__)*ry �
contextlibr Zselectr r- �errnor/ Zunittestrd Z test.supportZtestr r r rM � ImportErrorZsupportZrequiresr\ r. r rb Z
skipUnlessr{ Z HAVE_FORKINGrz r
r r Z ForkingMixInrj r rr r �contextmanagerr! ZTestCaser" r| r �mainr r r r �<module> sL
�"