a zeF@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZddl mZe jddZejZeedZeedZeedZeed Zd d ZejZe jjfd d ZererGddde je jZ Gddde je j!Z"ej#ddZ$Gdddej%Z&Gdddej%Z'Gddde j(Z)Gddde j*Z+Gddde j,e)Z-erGddde je)Z.Gd d!d!ej%Z/Gd"d#d#ej%Z0e1d$kre2dS)%N) reap_children reap_threadsverbose) socket_helperZnetworks hello world AF_UNIXzrequires Unix socketsforkzrequires forkingcCsttdrt|dS)Nalarm)hasattrsignalr)nr J/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_socketserver.py signal_alarm s rcCs:t|ggg|\}}}||vr(||Std|fdS)Nztimed out on %r) _real_selectrecv RuntimeError)sockr timeoutrwxr r r receive(s rc@s eZdZdS)ForkingUnixStreamServerN__name__ __module__ __qualname__r r r r r0src@s eZdZdS)ForkingUnixDatagramServerNrr r r r r4src cs`t}|dkrtdz.z dVWnYn0Wtjj|ddntjj|dd0dS)NrH)exitcode)osr_exittestsupportZ wait_process)Ztestcasepidr r r simple_subprocess9s   r%c@seZdZddZddZddZddZed d Zd d Z d dZ ddZ ddZ e ddZeddZeddZee ddZddZddZe dd Zed!d"Zed#d$Zee d%d&Zed'd(Zd)d*Zd+d,Zd-d.Zd/S)0SocketServerTestcCstdd|_g|_dS)N<r)rZ port_seed test_filesselfr r r setUpKszSocketServerTest.setUpc CsNtdt|jD]&}zt|Wqty8Yq0qg|jdd<dSNr)rrr(r removeOSError)r*fnr r r tearDownPs  zSocketServerTest.tearDowncCs8|tjkrtdfSd}tjd|d}|j||SdS)Nrz unix_socket.)prefixdir)socketAF_INETHOSTtempfilemktempr(append)r*protor2r/r r r pickaddr[s   zSocketServerTest.pickaddrc CsGddd|}Gddd|}tr,tdz|||}Wn8tyr}z |d|||fWYd}~n d}~00||j|j|S)Nc@seZdZddZdS)z.SocketServerTest.make_server..MyServercSs||dSN)Z close_requestr*requestclient_addressr r r handle_errorhs z;SocketServerTest.make_server..MyServer.handle_errorN)rrrr?r r r r MyServergsr@c@seZdZddZdS)z/SocketServerTest.make_server..MyHandlercSs|j}|j|dSr;)rfilereadlinewfilewrite)r*liner r r handlems z6SocketServerTest.make_server..MyHandler.handleNrrrrFr r r r MyHandlerlsrHzcreating serverz!Cannot create server (%s, %s): %s)rprintPermissionErrorZskipTest assertEqualserver_addressr3 getsockname)r*addrsvrclshdlrbaser@rHserverer r r make_serverfs zSocketServerTest.make_servercCs|||j||}|j}tr4td|td|tjd||jddid}d|_ | trhtdt d D]}trtd |||j|qptrtd | | ||d |jtrt|tjr||jtrtd dS)NzADDR =zCLASS =z %s serving poll_interval{Gz?nametargetkwargsTzserver runningz test clientzwaiting for serverdone)rSr:address_familyrLrrI threadingThread serve_foreverdaemonstartrangeshutdownjoin server_closerKr3fileno HAVE_FORKING isinstance socketserver ForkingMixInZ assertFalseactive_children)r*rOrPZtestfuncrQrNtir r r run_server{s4      zSocketServerTest.run_servercCst|tj^}|||tt|d}}|rRd|vrRt|d}||7}q2||tWdn1sr0YdSNd )r3 SOCK_STREAMconnectsendallTEST_STRrrKr*r9rNsbufdatar r r stream_examines     zSocketServerTest.stream_examinecCst|tjt}tr.|tjkr.||||t|t|d}}|rhd|vrht|d}||7}qH| |tWdn1s0YdSrp) r3 SOCK_DGRAMHAVE_UNIX_SOCKETSrbindr:sendtorvrrKrwr r r dgram_examines    zSocketServerTest.dgram_examinecCs|tjtj|jdSr;)rorj TCPServerStreamRequestHandlerr{r)r r r test_TCPServerszSocketServerTest.test_TCPServercCs|tjtj|jdSr;)rorjZThreadingTCPServerrr{r)r r r test_ThreadingTCPServersz(SocketServerTest.test_ThreadingTCPServercCs@t|$|tjtj|jWdn1s20YdSr;)r%rorjZForkingTCPServerrr{r)r r r test_ForkingTCPServers  z&SocketServerTest.test_ForkingTCPServercCs|tjtj|jdSr;)rorjUnixStreamServerrr{r)r r r test_UnixStreamServersz&SocketServerTest.test_UnixStreamServercCs|tjtj|jdSr;)rorjZThreadingUnixStreamServerrr{r)r r r test_ThreadingUnixStreamServersz/SocketServerTest.test_ThreadingUnixStreamServercCs>t|"|ttj|jWdn1s00YdSr;)r%rorrjrr{r)r r r test_ForkingUnixStreamServers  z-SocketServerTest.test_ForkingUnixStreamServercCs|tjtj|jdSr;)rorjZ UDPServerDatagramRequestHandlerrr)r r r test_UDPServerszSocketServerTest.test_UDPServercCs|tjtj|jdSr;)rorjZThreadingUDPServerrrr)r r r test_ThreadingUDPServersz(SocketServerTest.test_ThreadingUDPServercCs@t|$|tjtj|jWdn1s20YdSr;)r%rorjZForkingUDPServerrrr)r r r test_ForkingUDPServers  z&SocketServerTest.test_ForkingUDPServercCs|tjtj|jdSr;)rorjUnixDatagramServerrrr)r r r test_UnixDatagramServersz(SocketServerTest.test_UnixDatagramServercCs|tjtj|jdSr;)rorjZThreadingUnixDatagramServerrrr)r r r test_ThreadingUnixDatagramServersz1SocketServerTest.test_ThreadingUnixDatagramServercCs|ttj|jdSr;)rorrjrrr)r r r test_ForkingUnixDatagramServersz/SocketServerTest.test_ForkingUnixDatagramServercCsGdddtj}Gdddtj}g}tdD]<}|tdf|}tjd|jdd id }d |_| ||fq0|D]\}}| | qr|D]\}}| | qdS) Nc@s eZdZdS)z0SocketServerTest.test_shutdown..MyServerNrr r r r r@sr@c@s eZdZdS)z1SocketServerTest.test_shutdown..MyHandlerNrr r r r rHsrHrzMyServer servingrTrUrVT)rjrrrcr5r^r_r`rar8rbrdrerf)r*r@rHthreadsrnrxrmr r r test_shutdowns$    zSocketServerTest.test_shutdowncCs4Gdddtjtj}|tdfdd}|dS)Nc@s eZdZdS)z9SocketServerTest.test_close_immediately..MyServerNrr r r r r@sr@rcSsdSr;r r r r r z9SocketServerTest.test_close_immediately..)rjThreadingMixInrr5rf)r*r@rQr r r test_close_immediatelysz'SocketServerTest.test_close_immediatelyc CsNtdD]@}|t"ttdftjWdq1s>0YqdS)Nir[)rc assertRaises OverflowErrorrjrr5r)r*rnr r r test_tcpserver_bind_leaks    z)SocketServerTest.test_tcpserver_bind_leakcCsHttdftj}Wdn1s(0Y|d|jdS)Nrr[)rjrr5rrKr3rg)r*rQr r r test_context_manager's  z%SocketServerTest.test_context_managerN)rrrr+r0r:rSrror{rrrrequires_forkingrrequires_unix_socketsrrrrrrrrrrrrrr r r r r&HsF   "            r&c@sTeZdZddZddZddZddZd d Zed d Z ed dZ ddZ dS)ErrorHandlerTestcCstjtjjdSr;)r"r#unlinkTESTFNr)r r r r03szErrorHandlerTest.tearDowncCstt|jdddSNThandled)BaseErrorTestServer ValueError check_resultr)r r r test_sync_handled6sz"ErrorHandlerTest.test_sync_handledcCsB|tttWdn1s(0Y|jdddSNFr)r SystemExitrrr)r r r test_sync_not_handled:s &z&ErrorHandlerTest.test_sync_not_handledcCstt|jdddSr)ThreadingErrorTestServerrrr)r r r test_threading_handled?sz'ErrorHandlerTest.test_threading_handledcCstt|jdddSr)rrrr)r r r test_threading_not_handledCsz+ErrorHandlerTest.test_threading_not_handledcCstt|jdddSr)ForkingErrorTestServerrrr)r r r test_forking_handledGsz%ErrorHandlerTest.test_forking_handledcCstt|jdddSr)rrrr)r r r test_forking_not_handledLsz)ErrorHandlerTest.test_forking_not_handledcCsLttjj,}dd|}|||Wdn1s>0YdS)NHandler called Error handled )openr"r#rrKread)r*rlogexpectedr r r rQs zErrorHandlerTest.check_resultN) rrrr0rrrrrrrrr r r r r.s  rcs,eZdZfddZddZddZZS)rcsp||_ttdftt|jWdn1s:0Yz|W| n | 0| dSr,) exceptionsuper__init__r5 BadHandlerr3create_connectionrLhandle_requestrf wait_done)r*r __class__r r rXs zBaseErrorTestServer.__init__cCs<ttjjd}|dWdn1s.0YdS)Nar)rr"r#rrD)r*r=r>rr r r r?csz BaseErrorTestServer.handle_errorcCsdSr;r r)r r r rgszBaseErrorTestServer.wait_done)rrrrr?r __classcell__r r rr rWs rc@seZdZddZdS)rcCsHttjjd}|dWdn1s.0Y|jddS)Nrrz Test error)rr"r#rrDrQr)r*rr r r rFls(zBadHandler.handleNrGr r r r rksrcs0eZdZfddZfddZddZZS)rcs t|_tj|i|dSr;)r^Eventr\rrr*poskwrr r rts z!ThreadingErrorTestServer.__init__cs tj|i||jdSr;)rshutdown_requestr\setrrr r rxsz)ThreadingErrorTestServer.shutdown_requestcCs|jdSr;)r\waitr)r r r r|sz"ThreadingErrorTestServer.wait_done)rrrrrrrr r rr rrs  rc@s eZdZdS)rNrr r r r rsrc@seZdZddZddZdS)SocketWriterTestcCsGdddtj}ttdf|}||jt|jtjtj }|| |j Wdn1sh0Y| | |jtj||j|jdS)Nc@seZdZddZdS)z-SocketWriterTest.test_basics..HandlercSs*|j|j_|j|j_|j|j_dSr;)rCrQrg wfile_filenor=request_filenor)r r r rFs z4SocketWriterTest.test_basics..Handler.handleNrGr r r r Handlersrr)rjrrr5 addCleanuprfr3r]rs IPPROTO_TCPrtrLrZassertIsInstancerCioBufferedIOBaserKrr)r*rrQrxr r r test_basicss  *zSocketWriterTest.test_basicscstjtdGdddtj}ttdf||j t fdd}ttj |}|tjtj |ddt fdd}t j|d }|||jt|d |jd |jtjj|tjjd dS) N pthread_killc@seZdZddZdS)z,SocketWriterTest.test_write..HandlercSs>|jd|j_|j|j_dtjj }|j||j_ dS)N write data ) rCrDrQsent1rArBreceivedr"r# SOCK_MAX_SIZEsent2)r*Z big_chunkr r r rFs z3SocketWriterTest.test_write..Handler.handleNrGr r r r rsrrcs dSr;)r)signumframe) interruptedr r signal_handlersz3SocketWriterTest.test_write..signal_handlerc stjtjtj}||dd}|j||d| dt j j t ddrNqnqNt| Wdn1s0YWdn1s0YdS)Nrbclient response rq)r)r3r]rsrmakefilertrLrBrurr SIGUSR1rfloatlen)rxreaderr main_threadrZ received2Z response1rQr r run_clients     z/SocketWriterTest.test_write..run_client)rXrrrq)r"r#Z get_attributer rjrrr5rrfr^rr get_identr_rbrrerKrrrrr)r*rrZoriginalrZ backgroundr rr test_writes*    zSocketWriterTest.test_writeN)rrrrrr r r r rsrc@s$eZdZddZddZddZdS) MiscTestCasecCsRg}ttD]2}|ds tt|}t|dddkr ||q |tj|dS)N_rrj)r2rj startswithgetattrr8ZassertCountEqual__all__)r*rrWZ mod_objectr r r test_alls    zMiscTestCase.test_allcCshGdddtj}|tdftj}t|jtj}||j| | | |j d| dS)Nc@s eZdZddZdZddZdS)zSMiscTestCase.test_shutdown_request_called_if_verify_request_false..MyServercSsdS)NFr r<r r r verify_requestszbMiscTestCase.test_shutdown_request_called_if_verify_request_false..MyServer.verify_requestrcSs |jd7_tj||dS)Nr)shutdown_calledrjrr)r*r=r r r rszdMiscTestCase.test_shutdown_request_called_if_verify_request_false..MyServer.shutdown_requestN)rrrrrrr r r r r@sr@rr)rjrr5rr3r]rsrtrLcloserrKrrf)r*r@rQrxr r r 4test_shutdown_request_called_if_verify_request_falses  zAMiscTestCase.test_shutdown_request_called_if_verify_request_falsec CsGdddtjtj}|tdftj}tdD]8}t|j| Wdq.1s\0Yq.| t |j d| dS)Nc@s eZdZdS)z2MiscTestCase.test_threads_reaped..MyServerNrr r r r r@sr@r )rjrrr5rrcr3rrLrZ assertLessrZ_threadsrf)r*r@rQr r r r test_threads_reapeds (z MiscTestCase.test_threads_reapedN)rrrrrrr r r r rs r__main__)3 contextlibrr selectr r3r6r^ZunittestrjZ test.supportr"rrrrr#Zrequiresrvr5r r}Z skipUnlessrrhrrrZ SHORT_TIMEOUTrrkrrrrcontextmanagerr%ZTestCaser&rrrZBaseRequestHandlerrrrrrrrmainr r r r s^       g)M2