U ┬Улjbу@sfdZddlZddlZddlmZddlmZddlmZddl m Z ddlmZm Z ddlmZdd lmZmZmZdd lmZddlmZddlmZmZed ГZeejГZejа ej!бed<GddДdeГZ"ddlm#Z#ddlm$Z$ddl%m&Z&GddДdej'ГZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1ddl m2Z2z@dd l3m4Z4m5Z5m6Z6m7Z7m8Z8dd!l9m:Z:m;Z;mZ>m?Z?Wne@k РrРd"ZAYnОXd#ZAGd$d%Дd%e:jBГZCGd&d'Дd'e;jDГZEGd(d)Дd)e?jFГZGGd*d+Дd+e>jHГZIGd,d-Дd-e7ГZJGd.d/Дd/e6ГZKGd0d1Дd1e8ГZLeаMeKeLe=jNбGd2d3Дd3e$jOГZPGd4d5Дd5ГZQGd6d7Дd7eQГZRdd8lSmTZTGd9d:Дd:ejUeTjVГZWGd;d<ДdДd>eQГZZGd?d@Дd@eQГZ[GdAdBДdBГZ\GdCdDДdDeZee\ГZ]GdEdFДdFeRee\ГZ^eedGГGdHdIДdIe[ee\ГГZ_GdJdKДdKГZ`GdLdMДdMeZee`ГZaGdNdOДdOeRee`ГZbeedPГGdQdRДdRe[ee`ГГZcGdSdTДdTeГZddS)UzU Tests for L{twisted.conch.recvline} and fixtures for testing related functionality. щN)┌skipIf)┌recvline)┌insults)┌portal)┌defer┌error)┌StringTransport)┌ components┌filepath┌reflect)┌ iterbytes)┌ requireModule)┌SkipTest┌TestCaseztwisted.conch.stdio┌ PYTHONPATHc@sleZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZdS)┌ArrowsTestscsFtГИ_tабИ_tабИ_ЗfddДИj_ИИj_ Иjа ИjбdS)NcsИjSйN)┌pйй┌selfr·D/tmp/pip-unpacked-wheel-59kc99xd/twisted/conch/test/test_recvline.py┌"єz#ArrowsTests.setUp..)rZunderlyingTransportr┌ServerProtocol┌ptr┌HistoricRecvLiner┌protocolFactory┌factory┌makeConnectionrrrr┌setUps zArrowsTests.setUpcCs@|jаddб|jаddб|jаddб|а|jабdбdS)zy When L{HistoricRecvLine} receives a printable character, it adds it to the current line buffer. єxNєyєzйєxyzr)r┌keystrokeReceived┌assertEqual┌currentLineBufferrrrr┌test_printableCharacters&sz$ArrowsTests.test_printableCharacterscsFЗfddД}tdГD]}||ГqИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdб|ИjjГИаИjабdбdS) z╡ When L{HistoricRecvLine} receives a LEFT_ARROW or RIGHT_ARROW keystroke it moves the cursor left or right in the current line buffer, respectively. csИjа|dбSrйrr&й┌chrrrr7rz3ArrowsTests.test_horizontalArrows..r%r$)єxyr#)r!syz)rr%N)rr'rr(rZRIGHT_ARROW┌ LEFT_ARROWйr┌kRr,rrr┌test_horizontalArrows1s, z!ArrowsTests.test_horizontalArrowscs|ЗfddД}tdГD]}||ГqИаИjабdб|dГ|dГ|dГИаИjабdб|dГИаИjабd бd S)zЗ When {HistoricRecvLine} receives a newline, it adds the current line buffer to the end of its history buffer. csИjа|dбSrr*r+rrrr]rz*ArrowsTests.test_newline..єxyz abc 123 ййr%єabcє123rєcєbєaє ))r%r5r6scbarN)rr'r┌currentHistoryBufferr/rrr┌test_newlineXs zArrowsTests.test_newlinecs8ЗfddД}tdГD]}||ГqИаИjабdбИаИjабdб|ИjjГИаИjабdбИаИjабdб|ИjjГИаИjабdбИаИjабd б|ИjjГИаИjабd бИаИjабdб|ИjjГИаИjабd бИаИjабdбtdГD]}|ИjjГРqИаИjабdбd S)a When L{HistoricRecvLine} receives UP_ARROW or DOWN_ARROW keystrokes it move the current index in the current history buffer up or down, and resets the current line buffer to the previous or next line in history, respectively for each. csИjа|dбSrr*r+rrrrurz1ArrowsTests.test_verticalArrows..r2r3йrr))r%r5)r6)r6r))r%)r5r6)r5r)rr4r$щN) rr'rr;r(rZUP_ARROW┌rangeZ DOWN_ARROW)rr0r,┌irrr┌test_verticalArrowsns( zArrowsTests.test_verticalArrowscsVЗfddД}tdГD]}||ГqИаИjабdб|ИjjГИаИjабdбdS)zН When L{HistoricRecvLine} receives a HOME keystroke it moves the cursor to the beginning of the current line buffer. csИjа|dбSrr*r+rrrrЦrz'ArrowsTests.test_home..єhello, worldйrBr)rrBN)rr'rr(r┌HOMEr/rrr┌ test_homeСs zArrowsTests.test_homecsbЗfddД}tdГD]}||ГqИаИjабdб|ИjjГ|ИjjГИаИjабdбdS)zЗ When L{HistoricRecvLine} receives an END keystroke it moves the cursor to the end of the current line buffer. csИjа|dбSrr*r+rrrrдrz&ArrowsTests.test_end..rBrCN)rr'rr(rrDZENDr/rrr┌test_endЯs zArrowsTests.test_endcsЮЗfddД}tdГD]}||ГqИаИjабdб|ИjjГИаИjабdб|ИjjГ|ИjjГИаИjабdб|ИjjГИаИjабdбdS)zЙ When L{HistoricRecvLine} receives a BACKSPACE keystroke it deletes the character immediately before the cursor. csИjа|dбSrr*r+rrrr│rz,ArrowsTests.test_backspace..r%r$йr-r)rr"N)rr'rr(rZ BACKSPACEr.r/rrr┌test_backspaceоs zArrowsTests.test_backspacecsЄЗfddД}tdГD]}||ГqИаИjабdб|ИjjГИаИjабdб|ИjjГ|ИjjГИаИjабdб|ИjjГ|ИjjГИаИjабdб|ИjjГ|ИjjГИаИjабdб|ИjjГИаИjабdбdS) zЕ When L{HistoricRecvLine} receives a DELETE keystroke, it delets the character immediately after the cursor. csИjа|dбSrr*r+rrrr╚rz)ArrowsTests.test_delete..r%r$rG)r!rr=N)rr'rr(r┌DELETEr.r/rrr┌test_delete├s" zArrowsTests.test_deletecsrЗfddД}tdГD]}||Гq|ИjjГ|dГИаИjабdб|ИjjГ|dГИаИjабdбdS) zЛ When not in INSERT mode, L{HistoricRecvLine} inserts the typed character at the cursor before the next character. csИjа|dбSrr*r+rrrrхrz)ArrowsTests.test_insert..r%єA)єxyAr#єB)єxyBsAzN)rrr.r'rr(r/rrr┌test_insertрs zArrowsTests.test_insertcs~ЗfddД}tdГD]}||Гq|ИjjГ|ИjjГ|dГИаИjабdб|ИjjГ|dГИаИjабdбdS) a When in INSERT mode and upon receiving a keystroke with a printable character, L{HistoricRecvLine} replaces the character at the cursor with the typed character rather than inserting before. Ah, the ironies of INSERT mode. csИjа|dбSrr*r+rrrr∙rz+ArrowsTests.test_typeover..r%rK)rLrrM)rNrN)rrZINSERTr.r'rr(r/rrr┌ test_typeoverЄs zArrowsTests.test_typeovercsrЗfddД}Иj}|j|j|j|j|j|j|j|j|j |j |j|j|j |jfD]}||ГИаИjабdбqNdS)zб When L{HistoricRecvLine} receives a keystroke for an unprintable function key with no assigned behavior, the line buffer is unmodified. csИjа|dбSrr*r+rrrr rz8ArrowsTests.test_unprintableCharacters..r=N)rZF1ZF2ZF3ZF4ZF5ZF6ZF7ZF8ZF9ZF10ZF11ZF12ZPGUPZPGDNr'rr()rr0rr,rrr┌test_unprintableCharacterss&Єz&ArrowsTests.test_unprintableCharactersN)┌__name__┌ __module__┌__qualname__r r)r1r<rArErFrHrJrOrPrQrrrrrs'#r)┌telnet)┌helper)┌ LoopbackRelayc@seZdZddДZdS)┌ EchoServercCs |jа|d|j|jбdS)Nr:)Zterminal┌writeZpsZpn)r┌linerrr┌lineReceived*szEchoServer.lineReceivedN)rRrSrTr[rrrrrX)srXs[Ds[Cs[As[Bs[2~s[1~s[3~s[4~є)┌checkers)┌ConchFactory┌ TerminalRealm┌TerminalSession┌TerminalSessionTransport┌TerminalUser)┌channel┌ connection┌keys┌session┌ transport┌userauthFTc@s0eZdZdZddДZddДZddДZdd ДZd S)┌SessionChannelssessioncOs6tjj|f|Ю|О||_||_||_||_||_dSr)rc┌ SSHChannel┌__init__r┌protocolArgs┌protocolKwArgs┌width┌heightйrrrlrmrnro┌a┌kwrrrrkSszSessionChannel.__init__cCsdtаd|j|jddfdб}|jа|d|б|jа|ddб|j|j|jО|_ ||j _ |j а|бdS)Nsvt102rrspty-reqsshell)rfZpackRequest_pty_reqrorn┌connZsendRequestrrlrm┌_protocolInstancerr)r┌dataZtermrrr┌channelOpen_s zSessionChannel.channelOpencCs|jаtаббdSr)rt┌connectionLostrZConnectionDonerrrr┌closedlszSessionChannel.closedcCs|jа|бdSr)rt┌dataReceivedйrrurrrryoszSessionChannel.dataReceivedN)rRrSrT┌namerkrvrxryrrrrriPs ric@s$eZdZddДZddДZddДZdS)┌TestConnectioncOs6tjj|f|Ю|О||_||_||_||_||_dSr)rd┌ SSHConnectionrkrrlrmrnrorprrrrksszTestConnection.__init__cCs,t|j|j|j|j|jГ|_|а|jбdSr)rirrlrmrnro┌_TestConnection__channelZopenChannelrrrr┌serviceStarteds√zTestConnection.serviceStartedcCs|jа|бSr)r~rYrzrrrrYЙszTestConnection.writeN)rRrSrTrkrrYrrrrr|rs r|c@seZdZddДZddДZdS)┌TestAuthcOs tjj||f|Ю|О||_dSr)rh┌SSHUserAuthClientrk┌password)r┌usernamerВrqrrrrrrkНszTestAuth.__init__cCstа|jбSr)r┌succeedrВrrrr┌getPasswordСszTestAuth.getPasswordN)rRrSrTrkrЕrrrrrАМsrАc@s,eZdZddДZddДZddДZddДZd S) ┌ TestTransportc Os.||_||_||_||_||_||_||_dSr)rrlrmrГrВrnro) rrrlrmrГrВrnrorqrrrrrrkХszTestTransport.__init__cCs tаdбS)NT)rrД)rZhostKey┌fingerprintrrr┌ verifyHostKeyйszTestTransport.verifyHostKeycCs8t|j|j|j|j|jГ|_|аt|j |j |jГбdSr)r|rrlrmrnro┌_TestTransport__connectionZrequestServicerАrГrВrrrr┌connectionSecureмs√ zTestTransport.connectionSecurecCs|jа|бSr)rЙrYrzrrrrY╕szTestTransport.writeN)rRrSrTrkrИrКrYrrrrrЖФsrЖc@seZdZddДZdS)┌TestSessionTransportcCs|jjjjабSr)Zavatarrsrgr┌serverProtocolrrrrr╝sz$TestSessionTransport.protocolFactoryN)rRrSrTrrrrrrЛ╗srЛc@seZdZeZdS)┌TestSessionN)rRrSrTrЛZtransportFactoryrrrrrН┐srНc@seZdZdS)┌TestUserNйrRrSrTrrrrrО┬srОc@s$eZdZddДZddДZddДZdS)┌NotifyingExpectableBuffercCstаб|_tаб|_dSr)rZDeferred┌onConnection┌onDisconnectionrrrrrk╔s z"NotifyingExpectableBuffer.__init__cCstjа|б|jа|бdSr)rV┌ExpectableBuffer┌connectionMaderС┌callbackrrrrrФ═sz(NotifyingExpectableBuffer.connectionMadecCs|jа|бdSr)rТZerrback)r┌reasonrrrrw╤sz(NotifyingExpectableBuffer.connectionLostN)rRrSrTrkrФrwrrrrrР╚srРc@s$eZdZdZdZddДZddДZdS)┌ _BaseMixinщPщcCs:|jабаб}|dg|jt|Гd}|а||бdS)Nrщ)┌recvlineClient┌ __bytes__┌ splitlines┌HEIGHT┌lenr')r┌linesZ receivedLinesZ expectedLinesrrr┌ _assertBuffer┘sz_BaseMixin._assertBuffercs.Иjаdб}Иа|бЗЗfddД}|а|бS)NsdonecsИаИбdSr)rб)Zignй┌outputrrr┌finishedуsz)_BaseMixin._trivialTest..finished)rЫ┌expect┌ _testwriteZaddCallback)rZ inputLinerг┌donerдrrвr┌_trivialTest▐s z_BaseMixin._trivialTestN)rRrSrT┌WIDTHrЮrбrиrrrrrЧ╒srЧc@seZdZddДZddДZdS)┌ _SSHMixincs>tstdГВd\}}tГ}t|_ЗfddД|_tаб}|а||бt а |б}|а|бt|Г}t jtа|аббddН}||jd<||jd<|j|_|аб|абЙtаЗfddДбЙ|аdб}t|Г} tГЙtаЗfd dДбЙtЗfd dДdi|||j|jГ} t| Г}| а| б|а|бИ|_ | |_!||_"| |_#||_$Иj%S)NzMcryptography requirements missing, can't run historic recvline tests over ssh)stestuserstestpasscsИSrrrй┌ insultsServerrrrЇrz!_SSHMixin.setUp..i)ZkeySizesssh-rsacsИSrrrй┌recvlineServerrrrrcsИSrrrйrЫrrrrcsИSrrrй┌ insultsClientrrr rr)&┌sshrr_rОZuserFactoryZchainedProtocolFactoryr]Z'InMemoryUsernamePasswordDatabaseDontUseZaddUserrZPortalZregisterCheckerr^reZ_getPersistentRSAKeyr ZFilePath┌mktempZ publicKeysZprivateKeysrМZstartFactoryrrZ buildProtocolrWrР┌ClientProtocolrЖrйrЮrrЫ┌ sshClient┌ sshServer┌clientTransport┌serverTransportrС)r┌ur┌rlm┌checkerZptlZ sshFactoryZsshKeyr╢r╖r╡r╕rйr▒rмrЫrоrr ъsZ z_SSHMixin.setUpcCs|jа|бdSr)r╡rYrzrrrrжsz_SSHMixin._testwriteNйrRrSrTr rжrrrrrкщs2rк)┌test_telnetc@seZdZdS)┌TestInsultsClientProtocolNrПrrrrr┐#sr┐c@seZdZdS)┌TestInsultsServerProtocolNrПrrrrr└'sr└c@seZdZddДZddДZdS)┌_TelnetMixincsд|абЙtЗfddДГЙtаЗfddДб}t|Г}tГЙtЗfddДГЙtаЗfddДб}t|Г}|а|б|а|б|аб|абИ|_ ||_ ||_||_Иj S)NcsИSrrrrнrrr.rz$_TelnetMixin.setUp..csИSrrrrлrrr/rcsИSrrrrпrrr3rcsИSrrrr░rrr4r)rМr└rUZTelnetTransportrWrРr┐rZclearBufferrЫ┌telnetClientr╖r╕rС)rZtelnetServerr╖r┬r╕rr╝rr ,s" z_TelnetMixin.setUpcCs|jа|бdSr)r┬rYrzrrrrжDsz_TelnetMixin._testwriteNr╜rrrrr┴+sr┴c@s$eZdZddДZddДZddДZdS)┌_StdioMixincs┤tГЙtаЗfddДб}tа|б}tj}tj}|аdбsB|аdбrN|ddЕ}||t а |jбg}ddlm }|j|||tdd Н}И|_|_||_||_tаtd|jИаd бgГбS)NcsИSrrrй┌testTerminalrrrSrz#_StdioMixin.setUp..z.pycz.pyoщ r)┌reactorT)┌envZusePTYs>>> )rРrr┤┌stdioZTerminalProcessProtocol┌sys┌ executable┌__file__┌endswithrZqualrМ┌twisted.internetr╟ZspawnProcess┌ properEnvrЫr┼┌ processClientr╖rZ gatherResults┌filterrСrе)rr▒r╨Zexe┌module┌argsr╟r╖rr─rr Is, z_StdioMixin.setUpc sFzИjаdбWntjtfk r*YnXЗfddД}Иjjа|бS)NZKILLcs.|аtjбИа|jjбИа|jjdбdS)Nщ )┌traprZProcessTerminatedZassertIsNone┌valueZexitCoder'┌status)Zfailurerrrr╒{sz"_StdioMixin.tearDown..trap)r╖Z signalProcessrZProcessExitedAlready┌OSErrorr┼rТZ addErrback)rr╒rrr┌tearDowntsz_StdioMixin.tearDowncCs|jа|бdSr)r╖rYrzrrrrжВsz_StdioMixin._testwriteN)rRrSrTr r┘rжrrrrr├Hs+r├c@sXeZdZeZddДZddДZddДZddДZd d ДZ ddДZ d dДZddДZddДZ dS)┌RecvlineLoopbackMixincCs|аddddgбS)Nsfirst line doneє>>> first lineє first lineє>>> done)rиrrrr┌ testSimpleЙs z RecvlineLoopbackMixin.testSimplecCs"|аtdtdddddgбS)Nr▄r>є xxxx dones>>> first xxxxs first xxxxr▌)rи┌insert┌leftrrrr┌ testLeftArrowОs■z#RecvlineLoopbackMixin.testLeftArrowcCs*|аtdtdtdddddgбS)Ns right liner>щsxx dones>>> right lixxs right lixxr▌)rиrрrс┌rightrrrr┌testRightArrowФs■z$RecvlineLoopbackMixin.testRightArrowcCs|аdtdddddgбS)Nєsecond liner>r▀s>>> second xxxxssecond xxxxr▌)rи┌ backspacerrrr┌ testBackspaceЪs■z#RecvlineLoopbackMixin.testBackspacecCs&|аdtdtdddddgбS)Nsdelete xxxxr>є line dones>>> delete linesdelete liner▌)rиrс┌deleterrrr┌ testDeleteаs■z RecvlineLoopbackMixin.testDeletecCs|аdtdddddgбS)Ns third ineщєl dones>>> third lines third liner▌)rиrсrrrr┌ testInsertжs■z RecvlineLoopbackMixin.testInsertcCs"|аdtdtddddgбS)Nsfourth xiner>rэs>>> fourth linesfourth liner▌)rиrсrрrrrr┌testTypeoverмs■z"RecvlineLoopbackMixin.testTypeovercCs|аtdtddddgбS)Ns blah lines home dones >>> home lines home liner▌)rиrр┌homerrrr┌testHome▓s■zRecvlineLoopbackMixin.testHomecCs"|аdtdtddddgбS)Nsend r>rщs>>> end linesend liner▌)rиrс┌endrrrr┌testEnd╕s■zRecvlineLoopbackMixin.testEndN)rRrSrTrXrМr▐rтrхrшrыrюrяrёrєrrrrr┌Жsr┌c@seZdZdS)┌RecvlineLoopbackTelnetTestsNrПrrrrrЇ┐srЇc@seZdZdS)┌RecvlineLoopbackSSHTestsNrПrrrrrї├srїzBTerminal requirements missing, can't run recvline tests over stdioc@seZdZdS)┌RecvlineLoopbackStdioTestsNrПrrrrrЎ╟srЎc@s(eZdZeZddДZddДZddДZdS)┌HistoricRecvlineLoopbackMixincCs|аdtddddddgбS)Nsfirst line є doner█r▄r▌)rи┌uprrrr┌testUpArrow╧s √■z)HistoricRecvlineLoopbackMixin.testUpArrowcCs"|аdttddddddgбS)aH Pressing down arrow to visit an entry that was added to the history by pressing the up arrow instead of return does not raise a L{TypeError}. @see: U{http://twistedmatrix.com/trac/ticket/9031} @return: A L{defer.Deferred} that fires when C{b"done"} is echoed back. sfirst line partial liner°r█r▄s>>> partial linespartial liner▌йrиr∙┌downrrrr┌$test_DownArrowToPartialLineInHistory█s√■zBHistoricRecvlineLoopbackMixin.test_DownArrowToPartialLineInHistoryc Cs*|аdtdtddddddddgбS) Nsfirst line second line rуr°r█r▄s>>> second linerцr▌r√rrrr┌ testDownArrowЄs∙■z+HistoricRecvlineLoopbackMixin.testDownArrowN)rRrSrTrXrМr·r¤r■rrrrrў╠srўc@seZdZdS)┌#HistoricRecvlineLoopbackTelnetTestsNrПrrrrr sr c@seZdZdS)┌ HistoricRecvlineLoopbackSSHTestsNrПrrrrrsrzKTerminal requirements missing, can't run historic recvline tests over stdioc@seZdZdS)┌"HistoricRecvlineLoopbackStdioTestsNrПrrrrr src@seZdZdZddДZdS)┌TransportSequenceTestsz5 L{twisted.conch.recvline.TransportSequence} cCs|аttjбdS)zh Initializing a L{recvline.TransportSequence} with no args raises an assertion. N)┌assertRaises┌AssertionErrorrZTransportSequencerrrr┌test_invalidSequencesz+TransportSequenceTests.test_invalidSequenceN)rRrSrT┌__doc__rrrrrrsr)er┌osr╩ZunittestrZ twisted.conchrZtwisted.conch.insultsrZtwisted.credrr╬rrZtwisted.internet.testingrZtwisted.pythonr r rZtwisted.python.compatrZtwisted.python.reflectr Ztwisted.trial.unittestrrr╔┌dict┌environr╧┌pathsep┌join┌pathrrUrVZtwisted.conch.test.loopbackrWrrXrсrфr∙r№rрrЁrъrЄrчr]Ztwisted.conch.manhole_sshr^r_r`rarbZtwisted.conch.sshrcrdrerfrgrh┌ImportErrorr▓rjrir}r|rБrАZSSHClientTransportrЖrЛrНrОZregisterAdapterZISessionrУrРrЧrкZtwisted.conch.testr╛r┤ZTestProtocolr┐rr└r┴r├r┌rЇrїrЎrўr rrrrrrr┌sа $ "' 7>9 5 ■