B u9a<%ã@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ÚsockrZ bind_portÚportÚstart_resend_event)Úselfr©rú#/usr/lib/python3.7/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 ZlistenrÚsetZacceptÚbufferÚ SERVER_QUITÚrecvÚreplacerÚwaitÚsendÚ chunk_sizeÚtimeÚsleepÚclose)rZconnÚ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 ZconnectÚHOSTÚset_terminatorrÚsysÚplatform)rÚ terminatorZ server_portr&r(rrrrEs   zecho_client.__init__cCs|j|7_dS)N)r)rrrrrÚcollect_incoming_dataVsz!echo_client.collect_incoming_datacCs|j |j¡d|_dS)Nr)r+Úappendr)rrrrÚfound_terminatorYszecho_client.found_terminatorN)r"r#r$rr1r3rrrrr%Csr%cCs:t ¡}t|ƒ}| ¡| ¡| ¡t d¡||fS)Ng{®Gáz„?)rÚEventrÚstartrÚclearrr)rÚsrrrÚstart_echo_server]s r8c@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)rZthreading_setupÚ_threads)rrrrÚsetUpjszTestAsynchat.setUpcCstj|jŽdS)N)rZthreading_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)r?s hello world)rr4rrr5rr6rrr%r ÚpushrÚasyncoreÚloopÚusepollrÚ join_threadÚTIMEOUTÚ assertEqualr+)rZtermZ server_chunkrr7ÚcrrrÚline_terminator_checkps    z"TestAsynchat.line_terminator_checkcCsxdD]}| d|¡qWdS)N)rééó )rH)rÚlrrrÚtest_line_terminator1†s z"TestAsynchat.test_line_terminator1cCsxdD]}| d|¡qWdS)N)rrIrJs )rH)rrLrrrÚtest_line_terminator2‹s z"TestAsynchat.test_line_terminator2cCsxdD]}| d|¡qWdS)N)rrIrJsqqq)rH)rrLrrrÚ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„?)r=r>r?)r?) r8r%r r@rrArBrCrrDrErFr+)rZtermlenr7rrGrrrrÚnumeric_terminator_check•s    z%TestAsynchat.numeric_terminator_checkcCs| d¡dS)Nr)rP)rrrrÚtest_numeric_terminator1¡sz%TestAsynchat.test_numeric_terminator1cCs| d¡dS)Né)rP)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„?)r=r>r?)r?)r8r%r r@rrArBrCrrDrErFr+r)rr7rrGrrrrÚtest_none_terminator©s    z!TestAsynchat.test_none_terminatorcCsltƒ\}}td|jƒ}d}tj|tdd}| |¡tj|j dddt j |t d|  |jd d g¡dS) NrKshello world I'm not dead yet! é)Ú buffer_sizei,g{®Gáz„?)r=r>r?)r?s hello worldsI'm not dead yet!)r8r%r r)Zsimple_producerrÚpush_with_producerrArBrCrrDrErFr+)rr7rrGrÚprrrÚtest_simple_producer¶s   z!TestAsynchat.test_simple_producercCs^tƒ\}}td|jƒ}d}| |t¡tj|jdddtj |t d|  |j ddg¡dS) NrKshello world I'm not dead yet! i,g{®Gáz„?)r=r>r?)r?s hello worldsI'm not dead yet!) r8r%r rWrrArBrCrrDrErFr+)rr7rrGrrrrÚtest_string_producerÁs  z!TestAsynchat.test_string_producercCsbtƒ\}}td|jƒ}| d¡| t¡tj|jdddtj |t d|  |j ddd g¡dS) NrKshello world I'm not dead yet! i,g{®Gáz„?)r=r>r?)r?s hello worldrsI'm not dead yet!) r8r%r r@rrArBrCrrDrErFr+)rr7rrGrrrÚ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)NrKshello world I'm not dead yet! i,g{®Gáz„?)r=r>r?)r?r)r8rr4rr%r r@rZclose_when_donerArBrCrrrDrErFr+Z assertGreaterÚlenr)rr7rrGrrrÚ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) NrKsbytes é Zunicodei,g{®Gáz„?)r=r>r?)r?sbytes)r8r%r r@Ú bytearrayÚ memoryviewÚ assertRaisesÚ TypeErrorrrArBrCrrDrErFr+)rr7rrGrrrrÚ test_pushís    zTestAsynchat.test_pushN)r"r#r$rCr;r<rHrMrNrOrPrQrSrTrYrZr[r]rcrrrrr9gs      r9c@seZdZdZdS)ÚTestAsynchat_WithPollTN)r"r#r$rCrrrrrdþsrdc@seZdZdd„ZdS)ÚTestAsynchatMockedc Csjtj ¡}ttjƒ|j_t  ¡}|  |¡|  |j ¡tjj  |d¡}| ¡WdQRX| |j¡dS)NZ handle_error)ÚunittestZmockZMockÚBlockingIOErrorÚerrnoZEAGAINrZ side_effectr)r*Z set_socketZ addCleanupZ del_channelZpatchÚobjectZ handle_readZ assertFalseZcalled)rr Z dispatcherÚerrorrrrÚtest_blockingioerrors   z'TestAsynchatMocked.test_blockingioerrorN)r"r#r$rkrrrrresrec@seZdZdd„ZdS)ÚTestHelperFunctionscCs,| t dd¡d¡| t dd¡d¡dS)Nzqwerty z rZ qwertydkjfr)rFr)Zfind_prefix_at_end)rrrrÚtest_find_prefix_at_endsz+TestHelperFunctions.test_find_prefix_at_endN)r"r#r$rmrrrrrlsrlc@seZdZdd„ZdS)ÚTestNotConnectedcCst ¡}| t|jd¡dS)Néÿÿÿÿ)r)r*raÚ ValueErrorr-)rrrrrÚ!test_disallow_negative_terminatorsz2TestNotConnected.test_disallow_negative_terminatorN)r"r#r$rqrrrrrnsrnÚ__main__)Ztestrr)rArhr r.Ú_threadZthreadrrrfZ unittest.mockr,rrErrr*r%r8ZTestCaser9rdrerlrnr"ÚmainrrrrÚs0 .