o пbXcу@sbdZddlZddlZddlmZddlmZddlmZddl m Z ddlmZm Z ddlmZmZmZdd lmZdd 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@y╞d"ZAYnGwd#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)┌ components┌filepath┌reflect)┌ iterbytes)┌ requireModule)┌StringTransport)┌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·B/usr/lib/python3/dist-packages/twisted/conch/test/test_recvline.py┌"sz#ArrowsTests.setUp..)r ┌underlyingTransportr┌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йєxyzє)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. cєИjа|dбSrйrr'й┌chrrrr7єz3ArrowsTests.test_horizontalArrows..r%r$)єxyr#)r!syz)r&r%N)rr(rr)r┌RIGHT_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. cr+rr,r-rrrr]r/z*ArrowsTests.test_newline..єxyz abc 123 ййr%єabcє123rєcєbєaє ))r%r9r:scbarN)rr(r┌currentHistoryBufferr3rrr┌test_newlineXs zArrowsTests.test_newlinecs6З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. cr+rr,r-rrrrur/z1ArrowsTests.test_verticalArrows..r6r7йr&r&))r%r9)r:)r:r&))r%)r9r:)r9r&)rr8r$щN) rr(rr?r)r┌UP_ARROW┌range┌ DOWN_ARROW)rr4r.┌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. cr+rr,r-rrrrЦr/z'ArrowsTests.test_home..єhello, worldйrHr&)r&rHN)rr(rr)r┌HOMEr3rrr┌ 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. cr+rr,r-rrrrдr/z&ArrowsTests.test_end..rHrIN)rr(rr)rrJ┌ENDr3rrr┌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. cr+rr,r-rrrr│r/z,ArrowsTests.test_backspace..r%r$йr0r&)r&r"N)rr(rr)r┌ BACKSPACEr2r3rrr┌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. cr+rr,r-rrrr╚r/z)ArrowsTests.test_delete..r%r$rN)r!r&rAN)rr(rr)r┌DELETEr2r3rrr┌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. cr+rr,r-rrrrхr/z)ArrowsTests.test_insert..r%єA)єxyAr#єB)єxyBsAzN)rrr2r(rr)r3rrr┌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. cr+rr,r-rrrr∙r/z+ArrowsTests.test_typeover..r%rS)rTr&rU)rVr&N)rr┌INSERTr2r(rr)r3rrr┌ test_typeoverЄs zArrowsTests.test_typeovercsrЗfddД}Иj}|j|j|j|j|j|j|j|j|j |j |j|j|j |jfD]}||ГИаИjабdбq'dS)zб When L{HistoricRecvLine} receives a keystroke for an unprintable function key with no assigned behavior, the line buffer is unmodified. cr+rr,r-rrrr r/z8ArrowsTests.test_unprintableCharacters..rAN)r┌F1┌F2┌F3┌F4┌F5┌F6┌F7┌F8┌F9┌F10┌F11┌F12┌PGUP┌PGDNr(rr))rr4rr.rrr┌test_unprintableCharacterss(Єяz&ArrowsTests.test_unprintableCharactersN)┌__name__┌ __module__┌__qualname__r r*r5r@rGrKrMrPrRrWrYrhrrrrrs'#r)┌telnet)┌helper)┌ LoopbackRelayc@єeZdZddДZdS)┌ EchoServercCs |jа|d|j|jбdS)Nr>)┌terminal┌write┌ps┌pn)r┌linerrr┌lineReceived*s zEchoServer.lineReceivedN)rirjrkrvrrrrrp)єrps[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)┌SessionChannelssessioncOє<tjj|g|вRi|дО||_||_||_||_||_dSr)r┌ SSHChannel┌__init__r┌protocolArgs┌protocolKwArgs┌width┌heightйrrrЙrКrЛrМ┌a┌kwrrrrИSє zSessionChannel.__init__cCshtаd|j|jddfdб}|jа|d|б|jа|ddб|j|ji|jдО|_ ||j _ |j а|бdS)Nsvt102rr&spty-reqsshell)rВ┌packRequest_pty_reqrМrЛ┌conn┌sendRequestrrЙrК┌_protocolInstancerr)r┌data┌termrrr┌channelOpen_s zSessionChannel.channelOpencCs|jаtаббdSr)rФ┌connectionLostr┌ConnectionDonerrrr┌closedlszSessionChannel.closedcCє|jа|бdSr)rФ┌dataReceivedйrrХrrrrЬoєzSessionChannel.dataReceivedN)rirjrk┌namerИrЧrЪrЬrrrrrЕPs rЕc@є$eZdZddДZddДZddДZdS)┌TestConnectioncOrЖr)rА┌ SSHConnectionrИrrЙrКrЛrМrНrrrrИsrРzTestConnection.__init__cCs,t|j|j|j|j|jГ|_|а|jбdSr)rЕrrЙrКrЛrМ┌_TestConnection__channel┌openChannelrrrr┌serviceStarteds√zTestConnection.serviceStartedcCє|jа|бSr)rгrrrЭrrrrrЙєzTestConnection.writeN)rirjrkrИrеrrrrrrrбrs rбc@єeZdZddДZddДZdS)┌TestAuthcOs&tjj||g|вRi|дО||_dSr)rД┌SSHUserAuthClientrИ┌password)r┌usernamerлrОrПrrrrИНs zTestAuth.__init__cCstа|jбSr)r┌succeedrлrrrr┌getPasswordСrзzTestAuth.getPasswordN)rirjrkrИrоrrrrrйМsrйc@s,eZdZddДZddДZddДZddДZd S) ┌ TestTransportc Os.||_||_||_||_||_||_||_dSr)rrЙrКrмrлrЛrМ) rrrЙrКrмrлrЛrМrОrПrrrrИХs zTestTransport.__init__cCs tаdбS)NT)rrн)r┌hostKey┌fingerprintrrr┌ verifyHostKeyйs zTestTransport.verifyHostKeycCs8t|j|j|j|j|jГ|_|аt|j |j |jГбdSr)rбrrЙrКrЛrМ┌_TestTransport__connection┌requestServicerйrмrлrrrr┌connectionSecureмs√ zTestTransport.connectionSecurecCrжr)r│rrrЭrrrrr╕rзzTestTransport.writeN)rirjrkrИr▓r╡rrrrrrrпФs rпc@ro)┌TestSessionTransportcCs|jjjjабSr)┌avatarrТrГr┌serverProtocolrrrrr╝rЮz$TestSessionTransport.protocolFactoryN)rirjrkrrrrrr╢╗rwr╢c@seZdZeZdS)┌TestSessionN)rirjrkr╢┌transportFactoryrrrrr╣┐sr╣c@єeZdZdS)┌TestUserNйrirjrkrrrrr╝┬єr╝c@rа)┌NotifyingExpectableBuffercCstаб|_tаб|_dSr)r┌Deferred┌onConnection┌onDisconnectionrrrrrИ╔s z"NotifyingExpectableBuffer.__init__cCstjа|б|jа|бdSr)rm┌ExpectableBuffer┌connectionMader┴┌callbackrrrrr─═sz(NotifyingExpectableBuffer.connectionMadecCrЫr)r┬┌errback)r┌reasonrrrrШ╤rЮz(NotifyingExpectableBuffer.connectionLostN)rirjrkrИr─rШrrrrr┐╚sr┐c@s$eZdZdZdZddДZddДZdS)┌ _BaseMixinщPщc Csк|jабаб}|dg|jt|Гd}|аt|Гt|Гбtt|ГГD]-}|а||||dа|td|dГ|dЕбddа|td|dГ|dЕббq%dS)Nr&щrs != ) ┌recvlineClient┌ __bytes__┌ splitlines┌HEIGHT┌lenr(rD┌join┌max)r┌lines┌ receivedLines┌ expectedLinesrFrrr┌ _assertBuffer┘s ■¤ z_BaseMixin._assertBuffercs.Иjаdб}Иа|бЗЗfddД}|а|бS)NsdonecsИаИбdSr)r╓)┌ignй┌outputrrr┌finishedыsz)_BaseMixin._trivialTest..finished)r╠┌expect┌ _testwrite┌addCallback)r┌ inputLiner┘┌doner┌rr╪r┌_trivialTestцs z_BaseMixin._trivialTestN)rirjrk┌WIDTHr╧r╓rрrrrrr╚╒s r╚c@rи)┌ _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)stestuserstestpasscєИSrrrй┌ insultsServerrrr№єz!_SSHMixin.setUp..i)┌keySizesssh-rsacrуrrrй┌recvlineServerrrrrцcrуrrrйr╠rrrrцcrуrrrй┌ insultsClientrrrrцr)&┌sshrr{r╝┌userFactory┌chainedProtocolFactoryry┌'InMemoryUsernamePasswordDatabaseDontUse┌addUserr┌Portal┌registerCheckerrzrБ┌_getPersistentRSAKeyr ┌FilePath┌mktemp┌ publicKeys┌privateKeysr╕┌startFactoryrr┌ buildProtocolrnr┐┌ClientProtocolrпrсr╧rr╠┌ sshClient┌ sshServer┌clientTransport┌serverTransportr┴)r┌ur┌rlm┌checker┌ptl┌ sshFactory┌sshKeyr¤r■r№r rйrьrхr╠rщrr ЄsL z_SSHMixin.setUpcCrЫr)r№rrrЭrrrr▄$rЮz_SSHMixin._testwriteNйrirjrkr r▄rrrrrтёs2rт)┌test_telnetc@r╗)┌TestInsultsClientProtocolNr╜rrrrr +r╛r c@r╗)┌TestInsultsServerProtocolNr╜rrrrr /r╛r c@rи)┌_TelnetMixincsд|абЙtЗfddДГЙtаЗfddДб}t|Г}tГЙtЗfddДГЙtаЗfddДб}t|Г}|а|б|а|б|аб|абИ|_ ||_ ||_||_Иj S)Ncrуrrrrшrrr6rцz$_TelnetMixin.setUp..crуrrrrфrrr7rцcrуrrrrъrrr;rцcrуrrrrыrrr<rц)r╕r rl┌TelnetTransportrnr┐r r┌clearBufferr╠┌telnetClientr■r r┴)r┌telnetServerr■rr rrrr 4s" z_TelnetMixin.setUpcCrЫr)rrrrЭrrrr▄LrЮz_TelnetMixin._testwriteNrrrrrr3src@rа)┌_StdioMixincs┤tГЙtаЗfddДб}tа|б}tj}tj}|аdбs!|аdбr'|ddЕ}||t а |jбg}ddlm }|j|||tdd Н}И|_|_||_||_tаtd|jИаd бgГбS)Ncrуrrrй┌testTerminalrrr[rцz#_StdioMixin.setUp..z.pycz.pyoщ r)┌reactorT)┌env┌usePTYs>>> )r┐rr√┌stdio┌TerminalProcessProtocol┌sys┌ executable┌__file__┌endswithr ┌qualr╕┌twisted.internetr┌spawnProcess┌ properEnvr╠r┌ processClientr■r┌ gatherResults┌filterr┴r█)rrьr!┌exe┌module┌argsrr■rrrr Qs$ z_StdioMixin.setUpc sDzИjаdбWntjtfyYnwЗfddД}Иjjа|бS)N┌KILLcs.|аtjбИа|jjбИа|jjdбdS)Nщ )┌trapr┌ProcessTerminated┌assertIsNone┌value┌exitCoder(┌status)┌failurerrrr)Гsz"_StdioMixin.tearDown..trap)r■┌ signalProcessr┌ProcessExitedAlready┌OSErrorrr┬┌ addErrback)rr)rrr┌tearDown|s z_StdioMixin.tearDowncCrЫr)r■rrrЭrrrr▄КrЮz_StdioMixin._testwriteN)rirjrkr r4r▄rrrrrPs+rc@sXeZdZeZddДZddДZddДZddДZd d ДZ ddДZ d dДZddДZddДZ dS)┌RecvlineLoopbackMixincCs|аdgdвбS)Nsfirst line done)є>>> first lineє first lineє>>> done)rрrrrr┌ testSimpleСs z RecvlineLoopbackMixin.testSimplecCs |аtdtddgdвбS)Nr7rBє xxxx done)s>>> first xxxxs first xxxxr8)rр┌insert┌leftrrrr┌ testLeftArrowЦє■z#RecvlineLoopbackMixin.testLeftArrowcCs(|аtdtdtddgdвбS)Ns right linerBщsxx done)s>>> right lixxs right lixxr8)rрr;r<┌rightrrrr┌testRightArrowЬs■z$RecvlineLoopbackMixin.testRightArrowcCє|аdtddgdвбS)Nєsecond linerBr:)s>>> second xxxxssecond xxxxr8)rр┌ backspacerrrr┌ testBackspaceвє■z#RecvlineLoopbackMixin.testBackspacecCs$|аdtdtddgdвбS)Nsdelete xxxxrBє line done)s>>> delete linesdelete liner8)rрr<┌deleterrrr┌ testDeleteиs■z RecvlineLoopbackMixin.testDeletecCrB)Ns third ineщєl done)s>>> third lines third liner8)rрr<rrrr┌ testInsertоrFz RecvlineLoopbackMixin.testInsertcCє |аdtdtdgdвбS)Nsfourth xinerBrK)s>>> fourth linesfourth liner8)rрr<r;rrrr┌testTypeover┤r>z"RecvlineLoopbackMixin.testTypeovercCs|аtdtdgdвбS)Ns blah lines home done)s >>> home lines home liner8)rрr;┌homerrrr┌testHome║rFzRecvlineLoopbackMixin.testHomecCrM)Nsend rBrG)s>>> end linesend liner8)rрr<┌endrrrr┌testEnd└r>zRecvlineLoopbackMixin.testEndN)rirjrkrpr╕r9r=rArErIrLrNrPrRrrrrr5Оsr5c@r╗)┌RecvlineLoopbackTelnetTestsNr╜rrrrrS╟r╛rSc@r╗)┌RecvlineLoopbackSSHTestsNr╜rrrrrT╦r╛rTzBTerminal requirements missing, can't run recvline tests over stdioc@r╗)┌RecvlineLoopbackStdioTestsNr╜rrrrrU╧srUc@s(eZdZeZddДZddДZddДZdS)┌HistoricRecvlineLoopbackMixincCs|аdtdgdвбS)Nsfirst line є done)r6r7r6r7r8)rр┌uprrrr┌testUpArrow╫s ■z)HistoricRecvlineLoopbackMixin.testUpArrowcCs|аdttdgdвб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 linerW)r6r7s>>> partial linespartial liner8йrрrX┌downrrrr┌$test_DownArrowToPartialLineInHistoryуs■zBHistoricRecvlineLoopbackMixin.test_DownArrowToPartialLineInHistorycCrM)Nsfirst line second line r?rW)r6r7є>>> second linerCr]rCr8rZrrrr┌ testDownArrow·r>z+HistoricRecvlineLoopbackMixin.testDownArrowN)rirjrkrpr╕rYr\r^rrrrrV╘s rVc@r╗)┌#HistoricRecvlineLoopbackTelnetTestsNr╜rrrrr_ єr_c@r╗)┌ HistoricRecvlineLoopbackSSHTestsNr╜rrrrrar`razKTerminal requirements missing, can't run historic recvline tests over stdioc@r╗)┌"HistoricRecvlineLoopbackStdioTestsNr╜rrrrrbsrbc@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┌AssertionErrorr┌TransportSequencerrrr┌test_invalidSequence$sz+TransportSequenceTests.test_invalidSequenceN)rirjrk┌__doc__rgrrrrrcsrc)erh┌osr┌unittestr┌ twisted.conchr┌twisted.conch.insultsr┌twisted.credrrrr┌twisted.pythonrr r ┌twisted.python.compatr┌twisted.python.reflectr┌twisted.test.proto_helpersr ┌twisted.trial.unittestrrr┌dict┌environr ┌pathsepr╤┌pathrrlrm┌twisted.conch.test.loopbackrnrrpr<r@rXr[r;rOrHrQrDry┌twisted.conch.manhole_sshrzr{r|r}r~┌twisted.conch.sshrrАrБrВrГrД┌ImportErrorrэrЗrЕrвrбrкrй┌SSHClientTransportrпr╢r╣r╝┌registerAdapter┌ISessionr├r┐r╚rт┌twisted.conch.testrr√┌TestProtocolr rr rrr5rSrTrUrVr_rarbrcrrrr┌sШ $ "' 7>9 5 ■