B 18™\<%ã@sddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ej Z dZdZGdd„de jƒZGdd„dejƒZd d „ZGd d „d e jƒZGd d„deƒZGdd„de jƒZGdd„de jƒZGdd„de jƒZedkrþe  ¡dS)é)ÚsupportNsQUIT g@c@s eZdZdZdd„Zdd„ZdS)Ú echo_serverécCs<tj |¡||_t tjtj¡|_t  |j¡|_ d|_ dS)N) Ú threadingÚThreadÚ__init__ÚeventÚsocketÚAF_INETÚ SOCK_STREAMÚsockrÚ bind_portÚportÚstart_resend_event)Úselfr©rúV/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/test/test_asynchat.pyrs  zecho_server.__init__cCsØ|j ¡|j ¡|j ¡\}}d|_x*t|jkrR| d¡}|sDP|j||_q*W|j td¡|_|j rt|j   ¡y@x:|jr°|  |jd|j …¡}t  d¡|j|d…|_qxWWn YnX| ¡|j ¡dS)Nórgü©ñÒMbP?)r ÚlistenrÚsetÚacceptÚbufferÚ SERVER_QUITÚrecvÚreplacerÚwaitÚsendÚ chunk_sizeÚtimeÚsleepÚclose)rÚconnÚclientÚdataÚnrrrÚrun#s*      zecho_server.runN)Ú__name__Ú __module__Ú __qualname__rrr%rrrrrs rc@s$eZdZdd„Zdd„Zdd„ZdS)Ú echo_clientcCs^tj |¡g|_| tjtj¡| t |f¡|  |¡d|_ dd„}t j dkrZdd„}dS)NrcSsdS)Nr)rrrrÚhandle_connectMsz,echo_client.__init__..handle_connectÚdarwincSsdS)Nr)rrrrÚ handle_exptSsz)echo_client.__init__..handle_expt)ÚasynchatÚ async_chatrÚcontentsZ create_socketr r r ÚconnectÚHOSTÚset_terminatorrÚsysÚplatform)rÚ terminatorZ server_portr*r,rrrrEs   zecho_client.__init__cCs|j|7_dS)N)r)rr#rrrÚcollect_incoming_dataVsz!echo_client.collect_incoming_datacCs|j |j¡d|_dS)Nr)r/Úappendr)rrrrÚfound_terminatorYszecho_client.found_terminatorN)r&r'r(rr6r8rrrrr)Csr)cCs:t ¡}t|ƒ}| ¡| ¡| ¡t d¡||fS)Ng{®Gáz„?)rÚEventrÚstartrÚclearrr)rÚsrrrÚstart_echo_server]s r=c@sˆeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd S)!Ú TestAsynchatFcCst ¡|_dS)N)rÚthreading_setupÚ_threads)rrrrÚsetUpjszTestAsynchat.setUpcCstj|jŽdS)N)rÚthreading_cleanupr@)rrrrÚtearDownmszTestAsynchat.tearDowncCsªt ¡}t|ƒ}||_| ¡| ¡| ¡t d¡t ||j ƒ}|  d¡|  d|¡|  d|¡|  t ¡t j|jdddtj|td| |jddg¡dS) Ng{®Gáz„?shello sworldsI'm not dead yet!i,)Úuse_pollÚcountÚtimeout)rFs hello world)rr9rrr:rr;rrr)rÚpushrÚasyncoreÚloopÚusepollrÚ join_threadÚTIMEOUTÚ assertEqualr/)rZtermZ server_chunkrr<ÚcrrrÚline_terminator_checkps    z"TestAsynchat.line_terminator_checkcCsxdD]}| d|¡qWdS)N)rééó )rO)rÚlrrrÚtest_line_terminator1†s z"TestAsynchat.test_line_terminator1cCsxdD]}| d|¡qWdS)N)rrPrQs )rO)rrSrrrÚtest_line_terminator2‹s z"TestAsynchat.test_line_terminator2cCsxdD]}| d|¡qWdS)N)rrPrQsqqq)rO)rrSrrrÚtest_line_terminator3s z"TestAsynchat.test_line_terminator3cCsjtƒ\}}t||jƒ}d}| |¡| t¡tj|jdddtj |t d|  |j |d|…g¡dS)Nshello world, I'm not dead yet! i,g{®Gáz„?)rDrErF)rF) r=r)rrGrrHrIrJrrKrLrMr/)rZtermlenr<rrNr#rrrÚnumeric_terminator_check•s    z%TestAsynchat.numeric_terminator_checkcCs| d¡dS)Nr)rW)rrrrÚtest_numeric_terminator1¡sz%TestAsynchat.test_numeric_terminator1cCs| d¡dS)Né)rW)rrrrÚtest_numeric_terminator2¦sz%TestAsynchat.test_numeric_terminator2cCsntƒ\}}td|jƒ}d}| |¡| t¡tj|jdddtj |t d|  |j g¡|  |j |¡dS)Nshello world, I'm not dead yet! i,g{®Gáz„?)rDrErF)rF)r=r)rrGrrHrIrJrrKrLrMr/r)rr<rrNr#rrrÚtest_none_terminator©s    z!TestAsynchat.test_none_terminatorcCsltƒ\}}td|jƒ}d}tj|tdd}| |¡tj|j dddt j |t d|  |jd d g¡dS) NrRshello world I'm not dead yet! é)Ú buffer_sizei,g{®Gáz„?)rDrErF)rFs hello worldsI'm not dead yet!)r=r)rr-Zsimple_producerrÚpush_with_producerrHrIrJrrKrLrMr/)rr<rrNr#ÚprrrÚtest_simple_producer¶s   z!TestAsynchat.test_simple_producercCs^tƒ\}}td|jƒ}d}| |t¡tj|jdddtj |t d|  |j ddg¡dS) NrRshello world I'm not dead yet! i,g{®Gáz„?)rDrErF)rFs hello worldsI'm not dead yet!) r=r)rr^rrHrIrJrrKrLrMr/)rr<rrNr#rrrÚtest_string_producerÁs  z!TestAsynchat.test_string_producercCsbtƒ\}}td|jƒ}| d¡| t¡tj|jdddtj |t d|  |j ddd g¡dS) NrRshello world I'm not dead yet! i,g{®Gáz„?)rDrErF)rFs hello worldrsI'm not dead yet!) r=r)rrGrrHrIrJrrKrLrMr/)rr<rrNrrrÚtest_empty_lineËs    zTestAsynchat.test_empty_linecCsŠtƒ\}}t ¡|_td|jƒ}| d¡| t¡| ¡t j |j ddd|j  ¡t j|td| |jg¡| t|jƒd¡dS)NrRshello world I'm not dead yet! i,g{®Gáz„?)rDrErF)rFr)r=rr9rr)rrGrZclose_when_donerHrIrJrrrKrLrMr/Ú assertGreaterÚlenr)rr<rrNrrrÚtest_close_when_done×s      z!TestAsynchat.test_close_when_donecCs¢tƒ\}}td|jƒ}d}| |¡| t|ƒ¡| t|ƒ¡| t|jd¡| t|jd¡| t¡t j |j dddt j |td| |jd d d g¡dS) NrRsbytes é Úunicodei,g{®Gáz„?)rDrErF)rFsbytes)r=r)rrGÚ bytearrayÚ memoryviewÚ assertRaisesÚ TypeErrorrrHrIrJrrKrLrMr/)rr<rrNr#rrrÚ test_pushís    zTestAsynchat.test_pushN)r&r'r(rJrArCrOrTrUrVrWrXrZr[r`rarbrerlrrrrr>gs      r>c@seZdZdZdS)ÚTestAsynchat_WithPollTN)r&r'r(rJrrrrrmþsrmc@seZdZdd„ZdS)ÚTestAsynchatMockedc Csjtj ¡}ttjƒ|j_t  ¡}|  |¡|  |j ¡tjj  |d¡}| ¡WdQRX| |j¡dS)NZ handle_error)ÚunittestZmockZMockÚBlockingIOErrorÚerrnoÚEAGAINrZ side_effectr-r.Z set_socketÚ addCleanupZ del_channelÚpatchÚobjectZ handle_readÚ assertFalseÚcalled)rr Z dispatcherÚerrorrrrÚtest_blockingioerrors   z'TestAsynchatMocked.test_blockingioerrorN)r&r'r(ryrrrrrnsrnc@seZdZdd„ZdS)ÚTestHelperFunctionscCs,| t dd¡d¡| t dd¡d¡dS)Nzqwerty z rZ qwertydkjfr)rMr-Zfind_prefix_at_end)rrrrÚtest_find_prefix_at_endsz+TestHelperFunctions.test_find_prefix_at_endN)r&r'r(r{rrrrrzsrzc@seZdZdd„ZdS)ÚTestNotConnectedcCst ¡}| t|jd¡dS)Néÿÿÿÿ)r-r.rjÚ ValueErrorr2)rr"rrrÚ!test_disallow_negative_terminatorsz2TestNotConnected.test_disallow_negative_terminatorN)r&r'r(rrrrrr|sr|Ú__main__)Útestrr-rHrqr r3Ú_threadÚthreadrrroZ unittest.mockr1rrLrrr.r)r=ÚTestCaser>rmrnrzr|r&ÚmainrrrrÚs0 .