B tZV#ã@sddlmZmZddlZddlTddlZddlZddlmZddl m Z ddlZddlZddl Z ddlZddlZddlZdZdejd<d ZGd d„deƒZGdd „d ejjƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZedkrþe ¡dS)é)Úabsolute_importÚprint_functionN)Ú*)Ú HTTPError)ÚIOLoopgð?Z20ZASYNC_TEST_TIMEOUTéc@sxeZdZdZdd„Zejjdd„ƒZejje fdd„ƒZ dd „Zejje fd d„ƒZdd „Z ejjdd„ƒZdd„ZdS)ÚTestTermClientz%Test connection to a terminal managercCs||_d|_dS)N)ÚwsÚpending_read)ÚselfÚ websocket©r ú9lib/python3.7/site-packages/terminado/tests/basic_test.pyÚ__init__"szTestTermClient.__init__ccsB|jdkr|j ¡|_|jV}d|_|r2t |¡}tj |¡‚dS)N)r r Zread_messageÚjsonÚloadsÚtornadoÚgenÚReturn)rÚresponser r rÚread_msg&s zTestTermClient.read_msgccsfg}tj|d}xPy| ¡}tj ||¡V}Wn$tjjk rRtj |¡‚YnX| |¡qWdS)z"Read messages until read times out)ZsecondsN) ÚdatetimeZ timedeltarrrZwith_timeoutÚTimeoutErrorrÚappend)rÚtimeoutÚmsglistZdeltaZmfÚmsgr r rÚread_all_msg4szTestTermClient.read_all_msgcCs|j t |¡¡dS)N)r Z write_messagerÚdumps)rrr r rÚ write_msgBszTestTermClient.write_msgccsB| |¡V}d dd„|Dƒ¡}dd„|Dƒ}tj ||f¡‚dS)zkRead standard output until timeout read reached, return stdout and any non-stdout msgs received.ÚcSs g|]}|ddkr|d‘qS)rÚstdoutér )Ú.0rr r rú Jsz.TestTermClient.read_stdout..cSsg|]}|ddkr|‘qS)rr!r )r#rr r rr$KsN)rÚjoinrrr)rrrr!Zothermsgr r rÚread_stdoutEszTestTermClient.read_stdoutcCs| d|g¡dS)zWrite to terminal stdinÚstdinN)r)rÚdatar r rÚwrite_stdinNszTestTermClient.write_stdinccsl| ¡V| d¡| ¡V\}}tjdkrJt d|¡}t| ¡dƒ}nt| d¡dƒ}t j |¡‚dS)z(Get process ID of terminal shell processzecho $$ Úntzecho \$\$\x1b\[0K\r\n(\d+)rÚ r"N)r&r)ÚosÚnameÚreÚsearchÚintÚgroupsÚsplitrrr)rr!ZextraÚmatchÚpidr r rÚget_pidRs zTestTermClient.get_pidcCs|j ¡dS)N)r Úclose)rr r rr6_szTestTermClient.closeN)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrÚ coroutinerÚDONE_TIMEOUTrrr&r)r5r6r r r rr s rc@sHeZdZejjdd„ƒZejjdd„ƒZejjdd„ƒZdd„Z d Z d S)ÚTermTestCaseccsN| ¡}d||f}tjj|dd|id}tj |¡V}tj t|ƒ¡‚dS)Nzws://127.0.0.1:%d%sZOriginzhttp://127.0.0.1:%d)Úheaders) Z get_http_portrZ httpclientZHTTPRequestrZwebsocket_connectrrr)rÚpathZportÚurlZrequestr r r rÚget_term_clientfszTermTestCase.get_term_clientc#s$‡fdd„|DƒV}tj |¡‚dS)Ncsg|]}ˆ |¡‘qSr )rA)r#r?)rr rr$rsz1TermTestCase.get_term_clients..)rrr)rÚpathsÚtmsr )rrÚget_term_clientspszTermTestCase.get_term_clientsccs6g}x |D]}| ¡V}| |¡q Wtj |¡‚dS)N)r5rrrr)rZtm_listÚpidsÚtmr4r r rÚget_pidsus zTermTestCase.get_pidscs˜tdgt|jd|_tdg|jd|_tdgt|jd|_|j‰G‡fdd„dtj j ƒ}tj jd|fdtd|jifd td|jifd td|jifgddS) NZbash)Ú shell_commandZ max_terminalsÚioloop)rHrIcseZdZdZ‡fdd„ZdS)z0TermTestCase.get_app..NewTerminalHandlerz,Create a new named terminal, return redirectcs"ˆ ¡\}}|jd|dddS)Nz/named/F)Z permanent)Znew_named_terminalZredirect)rr-Zterminal)Únamed_tmr rÚget‹sz4TermTestCase.get_app..NewTerminalHandler.getN)r7r8r9r:rKr )rJr rÚNewTerminalHandler‰srLz/newz/named/(\w+)Zterm_managerz/singlez/uniqueT)Údebug) ZNamedTermManagerÚ MAX_TERMSZio_looprJZSingleTermManagerZ single_tmZUniqueTermManagerZ unique_tmrZwebZRequestHandlerZApplicationZ TermSocket)rrLr )rJrÚget_app~s zTermTestCase.get_app)z/named/term1z/uniquez/singleN)r7r8r9rrr;rArDrGrOÚ test_urlsr r r rr=bs r=c@s,eZdZejjdd„ƒZejjdd„ƒZdS)ÚCommonTestsccspxj|jD]`}| |¡V}| ¡V}| |dig¡| ¡V}| |dd¡| t|dƒd¡| ¡qWdS)NÚsetuprr!r")rPrArÚassertEqualZ assertGreaterÚlenr6)rr@rFrr r rÚ test_basic™s zCommonTests.test_basicccs|xv|jD]l}| |¡V}| ¡V| d¡| ¡V\}}tjdkrRd|ks`t‚n| d¡s`t‚|gkslt‚| ¡qWdS)Nzwhoami r*ZwhoamiZwho) rPrArr)r&r,r-ÚAssertionErrorÚ startswithr6)rr@rFr!Úotherr r rÚtest_basic_command¦s zCommonTests.test_basic_commandN)r7r8r9rÚtestingÚgen_testrUrYr r r rrQ˜s rQc@s4eZdZdd„Zejjdd„ƒZejjdd„ƒZdS)ÚNamedTermTestscCsH|jddd}| |jd¡|jd}| d¡d}| ||jj¡dS)Nz/newF)Zfollow_redirectsi.ZLocationú/é)ZfetchrSÚcoder>r2ZassertInrJZ terminals)rrr@r-r r rÚtest_newµs zNamedTermTests.test_newccsldgddgd}| |¡V}| |¡V}| |d|d¡| |d|d¡| |d|d¡dS)Nz/named/1r^z/named/2rr"r)rDrGrSÚassertNotEqual)rÚnamesrCrEr r rÚtest_namespace¾szNamedTermTests.test_namespaceccs`dd„ttdƒDƒ}| |dt…¡V}| |¡V}| |t¡V}| ¡V}| |d¡dS)NcSsg|]}d|‘qS)z /named/%dr )r#Úir r rr$Êsz5NamedTermTests.test_max_terminals..r")ÚrangerNrDrGrArrS)rZurlsrCrErFrr r rÚtest_max_terminalsÈs z!NamedTermTests.test_max_terminalsN) r7r8r9r`rrZr[rcrfr r r rr\´s r\c@seZdZejjdd„ƒZdS)ÚSingleTermTestsccs4| ddg¡V}| |¡V}| |d|d¡dS)Nz/singlerr")rDrGrS)rrCrEr r rÚtest_single_processÔsz#SingleTermTests.test_single_processN)r7r8r9rrZr[rhr r r rrgÓsrgc@s,eZdZejjdd„ƒZejjdd„ƒZdS)ÚUniqueTermTestsccs4| ddg¡V}| |¡V}| |d|d¡dS)Nz/uniquerr")rDrGra)rrCrEr r rÚtest_unique_processesÛsz%UniqueTermTests.test_unique_processesccs¤| dgt¡V}| |¡V}| tt|ƒƒt¡| d¡V}| ¡V}| |d¡|d ¡|d ¡V}| |d¡| d¡V}| ¡V}| |dd¡dS)Nz/uniquerrR) rDrNrGrSrTÚsetrArr6ZassertEquals)rrCrErFrr r rrfás z"UniqueTermTests.test_max_terminalsN)r7r8r9rrZr[rjrfr r r rriÚsriÚ__main__)Z __future__rrZunittestZ terminadorZtornado.httpserverZtornado.httpclientrZtornado.iolooprZtornado.testingrZloggingrr,r.r<ÚenvironrNÚobjectrrZZAsyncHTTPTestCaser=rQr\rgrir7Úmainr r r rÚs. B6