Your IP : 216.73.216.213
���]�) �
@ s
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 re j( � d S)� N)�
reap_children� reap_threads�verboseZnetworks hello world
�AF_UNIXzrequires Unix sockets�forkzrequires forkingc C s t t d � r t j | � d S)N�alarm)�hasattr�signalr )�n� r �6/opt/alt/python35/lib64/python3.5/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_select�recv� RuntimeError)�sockr
�timeout�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�pidZpid2�statusr r r �simple_subprocess; s
r% z!Threading required for this test.c @ sT e 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),�SocketServerTestc 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)�socket�AF_INET�HOST�tempfile�mktempr( �append)r) �protor0 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 r< r r r r �MyServerg s r= 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 rA r r r r � MyHandlerm s rB zcreating server)r �printr" �server_addressr1 � getsockname)r) �addr�svrcls�hdlrbaser= rB �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���)rJ r8 �address_familyrD r rC � threading�Thread�
serve_forever�daemon�start�range�shutdown�joinr9 r" r1 �fileno)r) rG rH ZtestfuncrI rF �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
)r1 � SOCK_STREAM�connect�sendall�TEST_STRr r" �close)r) r7 rF �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)Nr` s
) r1 �
SOCK_DGRAM�HAVE_UNIX_SOCKETSr �bindr8 �sendtord r r" re )r) r7 rF rf rg rh r r r �
dgram_examine� s z SocketServerTest.dgram_examinec C s | j t j t j | j � d S)N)r_ � socketserver� TCPServer�StreamRequestHandlerri )r) r r r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | j t j t j | j � d S)N)r_ ro ZThreadingTCPServerrq ri )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% r_ ro ZForkingTCPServerrq ri )r) r r r �test_ForkingTCPServer� s
z&SocketServerTest.test_ForkingTCPServerc C s | j t j t j | j � d S)N)r_ ro �UnixStreamServerrq ri )r) r r r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | j t j t j | j � d S)N)r_ ro ZThreadingUnixStreamServerrq ri )r) r r r � test_ThreadingUnixStreamServer� s z/SocketServerTest.test_ThreadingUnixStreamServerc C s1 t | � � | j t t j | j � Wd QRXd S)N)r% r_ r ro rq ri )r) r r r � test_ForkingUnixStreamServer� s
z-SocketServerTest.test_ForkingUnixStreamServerc C s | j t j t j | j � d S)N)r_ ro Z UDPServer�DatagramRequestHandlerrn )r) r r r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | j t j t j | j � d S)N)r_ ro ZThreadingUDPServerry rn )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% r_ ro ZForkingUDPServerry rn )r) r r r �test_ForkingUDPServer� s
z&SocketServerTest.test_ForkingUDPServerc C s | j t j t j | j � d S)N)r_ ro �UnixDatagramServerry rn )r) r r r �test_UnixDatagramServer� s z(SocketServerTest.test_UnixDatagramServerc C s | j t j t j | j � d S)N)r_ ro ZThreadingUnixDatagramServerry rn )r) r r r � test_ThreadingUnixDatagramServer� s z1SocketServerTest.test_ThreadingUnixDatagramServerc C s | j t t j | j � d S)N)r_ r ro ry rn )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 r= � s r= c @ s e Z d Z d S)z1SocketServerTest.test_shutdown.<locals>.MyHandlerN)r r r r r r r rB � s rB r r rK zMyServer servingrL rM rN g{�G�z�?T)ro rp rq rY r3 rT rU rV rW r6 rX rZ r[ r9 )r) r= rB �threadsr^ rf r] 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 rP rR )rY Z assertRaises�
OverflowErrorro rp r3 rq )r) r^ r r r �test_tcpserver_bind_leak s z)SocketServerTest.test_tcpserver_bind_leakN)r r r r* r. r8 rJ r r_ ri rn rr rs �requires_forkingrt �requires_unix_socketsrv rw rx rz r{ r| r~ r r� r� r� r r r r r& G s. 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 ro )r0 ro �
startswith�getattrr6 ZassertCountEqual�__all__)r) ZexpectedrK 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) r: r; 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)NrP )�shutdown_calledro rp �shutdown_request)r) r: 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 r= + s r= r rP )ro rp r3 rq r1 rS ra rb rD re Zhandle_requestr" r� r9 )r) r= rI rf 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__))�
contextlibr �selectr r1 �errnor4 Zunittestro Z test.supportZtestr r r rT � ImportErrorZsupportZrequiresrd r3 r rk Z
skipUnlessr� Z HAVE_FORKINGr� r
r r Z ForkingMixInru r r} r �contextmanagerr% ZTestCaser& r� r �mainr r r r �<module> sJ
�"