Your IP : 216.73.216.213
�
g fQ+ �
@ 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" n 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) � n d S)z
Test suite for socketserver.
� N)�
reap_children� reap_threads�verboseZnetworks hello world
ZAF_UNIXzrequires Unix sockets�forkzrequires forkingc C s# t t d � r t j | � n d S)z7Call signal.alarm when it exists (i.e. not on Windows).�alarmN)�hasattr�signalr )�n� r
�;/opt/alt/python34/lib64/python3.4/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 si t j � } | d k r( t j d � n 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 @ s? 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 j d" d# � � Z 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 sn Gd d � d | � } Gd d � d | � } t r? t d � n | | | � } | 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$ ZrequestZclient_addressr
r
r � handle_errorh s
z;SocketServerTest.make_server.<locals>.MyServer.handle_errorN)r r r r3 r
r
r
r �MyServerg s r4 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 r8 r
r
r
r � MyHandlerm s r9 zcreating server)r �printr �server_addressr, Z getsockname)r$ �addr�svrcls�hdlrbaser4 r9 �serverr
r
r � make_serverf s
z SocketServerTest.make_serverc C s= | j | j | j � | | � } | j } t rM t d | � t d | � n t j d d | d | j d i d d 6� } d | _ | j
� t r� t d
� n x: t d � D], } t r� t d | � n | | j | � q� Wt r� t d
� n | j � | j
� | j � | j d | j j � � t r9t d � n d S)NzADDR =zCLASS =�namez
%s serving�target�kwargsg{�G�z�?�
poll_intervalTzserver running� z test clientzwaiting for server� �done���)r@ r1 Zaddress_familyr; r r: � threading�Thread�
serve_forever�daemon�start�range�shutdown�joinr2 r r, �fileno)r$ r= r> Ztestfuncr? 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, Z SOCK_STREAMZconnectZsendall�TEST_STRr r �close)r$ r0 r< �s�buf�datar
r
r �stream_examine� s
z SocketServerTest.stream_examinec C s� t j | t j � } | j t | � t | d � } } x/ | ri d | k ri t | d � } | | 7} q; W| j | t � | j � d S)NrU s
)r, Z
SOCK_DGRAMZsendtorV r r rW )r$ r0 r< rX rY rZ r
r
r �
dgram_examine� s z SocketServerTest.dgram_examinec C s | j t j t j | j � d S)N)rT � socketserver� TCPServer�StreamRequestHandlerr[ )r$ r
r
r �test_TCPServer� s z SocketServerTest.test_TCPServerc C s | j t j t j | j � d S)N)rT r] ZThreadingTCPServerr_ r[ )r$ r
r
r �test_ThreadingTCPServer� s z(SocketServerTest.test_ThreadingTCPServerc C s3 t | � �! | j t j t j | j � Wd QXd S)N)r rT r] ZForkingTCPServerr_ r[ )r$ r
r
r �test_ForkingTCPServer� s
z&SocketServerTest.test_ForkingTCPServerc C s | j t j t j | j � d S)N)rT r] �UnixStreamServerr_ r[ )r$ r
r
r �test_UnixStreamServer� s z&SocketServerTest.test_UnixStreamServerc C s | j t j t j | j � d S)N)rT r] ZThreadingUnixStreamServerr_ r[ )r$ r
r
r � test_ThreadingUnixStreamServer� s z/SocketServerTest.test_ThreadingUnixStreamServerc C s0 t | � � | j t t j | j � Wd QXd S)N)r rT r r] r_ r[ )r$ r
r
r � test_ForkingUnixStreamServer� s
z-SocketServerTest.test_ForkingUnixStreamServerc C s | j t j t j | j � d S)N)rT r] Z UDPServer�DatagramRequestHandlerr\ )r$ r
r
r �test_UDPServer� s z SocketServerTest.test_UDPServerc C s | j t j t j | j � d S)N)rT r] ZThreadingUDPServerrg r\ )r$ r
r
r �test_ThreadingUDPServer� s z(SocketServerTest.test_ThreadingUDPServerc C s3 t | � �! | j t j t j | j � Wd QXd S)N)r rT r] ZForkingUDPServerrg r\ )r$ r
r
r �test_ForkingUDPServer� s
z&SocketServerTest.test_ForkingUDPServerc # sK t j � G� f d d � d � } | � t _ z t j VWd � t _ Xd S)z<Mocks the select.select() call to raise EINTR for first callc s. e Z d Z d d � Z � f d d � Z d S)z9SocketServerTest.mocked_select_module.<locals>.MockSelectc S s
d | _ d S)Nr )�called)r$ r
r
r �__init__� s zBSocketServerTest.mocked_select_module.<locals>.MockSelect.__init__c sM | j d 7_ | j d k r? t t j t j t j � � � n
� | � Sd S)NrF )rk r' �errnoZEINTRr �strerror)r$ �args)�
old_selectr
r �__call__� s !zBSocketServerTest.mocked_select_module.<locals>.MockSelect.__call__N)r r r rl rq r
)rp r
r �
MockSelect� s rr N)�select)r$ rr r
)rp r �mocked_select_module� s
z%SocketServerTest.mocked_select_modulec C sJ | j � �8 } | j t j t j | j � } | j | j d � Wd QXd S)NrF )rt rT r] r^ r_ r[ Z
assertGreaterrk )r$ Z mock_selectr r
r
r � test_InterruptServerSelectCall� s
z/SocketServerTest.test_InterruptServerSelectCallc
C s Gd d � d t j � } Gd d � d t j � } g } xm t d � D]_ } | t d f | � } t j d d d | j d
i d d 6� } 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 r4 s r4 c @ s e Z d Z d S)z1SocketServerTest.test_shutdown.<locals>.MyHandlerN)r r r r
r
r
r r9 s r9 r
r rA zMyServer servingrB rC g{�G�z�?rD T)r] r^ r_ rN r- rI rJ rK rL r/ rM rO rP r2 )r$ r4 r9 ZthreadsrS rX rR r
r
r �
test_shutdown s"
z SocketServerTest.test_shutdownc C sJ xC t d � D]5 } | j t � � t j t d f t j � Wd QXq
Wd S)Ni rF rH )rN Z assertRaises�
OverflowErrorr] r^ r- r_ )r$ rS r
r
r �test_tcpserver_bind_leak1 s z)SocketServerTest.test_tcpserver_bind_leakN) r r r �__doc__r% r) r1 r@ r rT r[ r\ r` ra �requires_forkingrb �requires_unix_socketsrd re rf rh ri rj �
contextlib�contextmanagerrt ru rv rx r
r
r
r r! G s,
r! c @ s e Z d Z d d � Z d S)� MiscTestCasec C s} g } x] t t � D]O } | j d � s t t | � } t | d d � d k rb | j | � qb q q W| j t j | � d S)N�_r r] )r+ r] �
startswith�getattrr/ ZassertCountEqual�__all__)r$ ZexpectedrA Z
mod_objectr
r
r �test_all> s zMiscTestCase.test_allN)r r r r� r
r
r
r r~ < s r~ �__main__)*ry r| r rs r r, rm r. Zunittestr] Z test.supportZtestr r r rI � ImportErrorZsupportZrequiresrV r- r ZHAVE_UNIX_SOCKETSZ
skipUnlessr{ Z HAVE_FORKINGrz r r r Z ForkingMixInrc r ZUnixDatagramServerr r} r ZTestCaser! r~ r �mainr
r
r
r �<module> sL
�