o Í°“iÖã@svddlmZddlZddlZddlZddlmZddlmZGdd„dejƒZ Gdd„dejƒZ Gd d „d ejƒZ dS) é)ÚBytesION)ÚStringIO)Ú as_stringc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚChildUtilsTestscCs*ddlm}|ddiƒ}| |du¡dS)Nr)ÚgetRPCInterfaceÚSUPERVISOR_SERVER_URLúhttp://localhost:9001)Úsupervisor.childutilsrÚ assertTrue)ÚselfrÚrpc©r úT/home/ubuntu/.local/lib/python3.10/site-packages/supervisor/tests/test_childutils.pyÚtest_getRPCInterface s  z$ChildUtilsTests.test_getRPCInterfacecCsFddlm}|ddiƒ}| |jd¡| |jd¡| |jd¡dS)Nr©ÚgetRPCTransportrrÚ©r rÚ assertEqualÚusernameÚpasswordÚ serverurl)r rÚtr r rÚ"test_getRPCTransport_no_uname_passs  z2ChildUtilsTests.test_getRPCTransport_no_uname_passcCsNddlm}ddddœ}||ƒ}| |jd¡| |jd¡| |jd¡dS)NrrrÚchrismÚabc123)rÚSUPERVISOR_USERNAMEÚSUPERVISOR_PASSWORDr)r rÚenvrr r rÚ$test_getRPCTransport_with_uname_passs þz4ChildUtilsTests.test_getRPCTransport_with_uname_passcCs.ddlm}d}||ƒ}| |dddœ¡dS)Nr)Ú get_headersza:1 b:2Ú1Ú2©ÚaÚb)r r r)r r ÚlineÚresultr r rÚtest_get_headers s z ChildUtilsTests.test_get_headerscCs>ddlm}d}||ƒ\}}| |dddœ¡| |d¡dS)Nr)Ú eventdataza:1 b:2 thedata r!r"r#zthedata )r r)r)r r)ÚpayloadÚheadersÚdatar r rÚtest_eventdata&s  zChildUtilsTests.test_eventdatacCs.ddlm}t d¡}||ƒ}| |d¡dS)Nr)Ú get_asctime) iÙééééérréÿÿÿÿz2009-01-18 22:14:07,000)r r.ÚtimeÚmktimer)r r.Ú timestampr'r r rÚtest_get_asctime-s  z ChildUtilsTests.test_get_asctimeN) Ú__name__Ú __module__Ú __qualname__rrrr(r-r8r r r rrs  rc@s$eZdZdd„Zdd„Zdd„ZdS)Ú!TestProcessCommunicationsProtocolcCsRddlm}tƒ}| d|¡ddlm}|j}|j}| |  ¡|d|¡dS©Nr)Úpcommshello)ÚProcessCommunicationEvent) r r>rÚsendÚsupervisor.eventsr?Ú BEGIN_TOKENÚ END_TOKENrÚgetvalue)r r>Ústdoutr?ÚbeginÚendr r rÚ test_send4s   z+TestProcessCommunicationsProtocol.test_sendcCónddlm}tj}z)tƒ}t_| d¡ddlm}|j}|j}|  |  ¡|d|¡W|t_dS|t_wr=) r r>ÚsysrErrAr?rBrCrrD©r r>ÚoldÚior?rFrGr r rÚ test_stdout=ó    z-TestProcessCommunicationsProtocol.test_stdoutcCrIr=) r r>rJÚstderrrrAr?rBrCrrDrKr r rÚ test_stderrJrOz-TestProcessCommunicationsProtocol.test_stderrN)r9r:r;rHrNrQr r r rr<3s r<c@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚTestEventListenerProtocolcCsfddlm}Gdd„dƒ}|ƒ}tƒ}| ||¡\}}| |ddi¡| |d¡| | ¡d¡dS) Nr©Úlistenerc@seZdZdd„Zdd„ZdS)z2TestEventListenerProtocol.test_wait..DummycSódS)Nzlen:5r )r r r rÚreadline[óz;TestEventListenerProtocol.test_wait..Dummy.readlinecWrU)NÚhellor )r Úignoredr r rÚread]rWz7TestEventListenerProtocol.test_wait..Dummy.readN)r9r:r;rVrZr r r rÚDummyZs r[ÚlenÚ5rXzREADY )r rTrÚwaitrrD)r rTr[ÚstdinrEr+r*r r rÚ test_waitXs  z#TestEventListenerProtocol.test_waitcCsFddlm}ddlm}t|jƒ}tƒ}| |¡| |  ¡|¡dS)NrrS©ÚPEventListenerDispatcher) r rTÚsupervisor.dispatchersrbrÚREADY_FOR_EVENTS_TOKENrÚreadyrrD)r rTrbÚtokenrEr r rÚ test_tokenfs    z$TestEventListenerProtocol.test_tokencCóJddlm}ddlm}t|jƒ}tƒ}| |¡| |  ¡|d¡dS)NrrSraz2 OK) r rTrcrbrÚRESULT_TOKEN_STARTrÚokrrD©r rTrbrFrEr r rÚtest_oknó    z!TestEventListenerProtocol.test_okcCrh)NrrSraz4 FAIL) r rTrcrbrrirÚfailrrDrkr r rÚ test_failvrmz#TestEventListenerProtocol.test_failcCs^ddlm}ddlm}t|jƒ}tƒ}d}| ||¡d|t|ƒ|f}|  |  ¡|¡dS)NrrSrazthe body data ya fool z%s%s %s) r rTrcrbrrirr@r\rrD)r rTrbrFrEÚmsgÚexpectedr r rrH~s    z#TestEventListenerProtocol.test_sendN)r9r:r;r`rgrlrorHr r r rrRWs  rR) rMrrJr5ÚunittestÚsupervisor.compatrrÚTestCaserr<rRr r r rÚs   +$