U W[y~@sdZddlmZmZddlZddlmZedZedZerfddl m Z m Z m Z m Z ddlmZmZnGd d d Zdd lmZmZdd lmZmZdd lmZddlmZddlmZmZmZddlmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&Gddde'Z(Gdddej)Z*Gddde'Z+ddl m,Z,erHe,-e+e*e j.Gdddej/Z0GdddZ1Gdd d Z2Gd!d"d"Z3Gd#d$d$Z4edk redk rdd%lm5Z5dd&l m6Z6m7Z7m8Z8m9Z9dd'l m:Z:m;Z;Gd(d)d)ZGd,d-d-e8j?Z@Gd.d/d/ZAGd0d1d1eAe:jBZCGd2d3d3eAe:jDZEGd4d5d5e;jFZGGd6d7d7e7jHZIGd8d9d9e6jJZKd:d;ZLGdd?d?e$jMZOGd@dAdAe$jMZPGdBdCdCe$jMZQdS)Dz! Tests for L{twisted.conch.ssh}. )divisionabsolute_importN) requireModule cryptographypyasn1)common forwardingsession_kex)avatarerrorc@seZdZGdddZdS)r c@s eZdZdS)zavatar.ConchUserN)__name__ __module__ __qualname__rr=/usr/lib/python3/dist-packages/twisted/conch/test/test_ssh.py ConchUsersrN)r rrrrrrrr sr )publicRSA_opensshprivateRSA_openssh)publicDSA_opensshprivateDSA_openssh)portal)UnauthorizedLogin)deferprotocolreactor)ProcessTerminated)failurelog)unittest) LoopbackRelayc@s$eZdZdZdZddZddZdS)ConchTestRealma" A realm which expects a particular avatarId to log in once and creates a L{ConchTestAvatar} for that request. @ivar expectedAvatarID: The only avatarID that this realm will produce an avatar for. @ivar avatar: A reference to the avatar after it is requested. NcCs ||_dSN)expectedAvatarID)selfr#rrr__init__1szConchTestRealm.__init__cGsN||jkr8|jdk rtdt|_|d|j|jjfStd|j|fdS)z Return a new L{ConchTestAvatar} if the avatarID matches the expected one and this is the first avatar request. NzOnly one login allowedrzOnly %r may log in, not %r)r#r rConchTestAvatarlogout)r$ZavatarIDZmindZ interfacesrrr requestAvatar5s   zConchTestRealm.requestAvatar)r rr__doc__r r%r(rrrrr!%s r!c@sLeZdZdZesdZdZddZddZdd Z d d Z d d Z ddZ dS)r&z An avatar against which various SSH features can be tested. @ivar loggedOut: A flag indicating whether the avatar logout method has been called. cannot run without cryptographyFcCsBtj|i|_i|_|jtjt j d|j dt idS)N)sessions direct-tcpipcrazy) r rr% listenersglobalRequestsZ channelLookupupdater Z SSHSessionrZopenConnectForwardingClientZsubsystemLookupCrazySubsystemr$rrrr%Ps zConchTestAvatar.__init__cCs||jd<dS)Nfoor.r$datarrr global_fooZs zConchTestAvatar.global_foocCs||jd<dS)Nfoo_2)r3datar4r5rrr global_foo_2_s zConchTestAvatar.global_foo_2cCsft|\}}z&tj|t|j||ftj|d}WntddYdSX||j ||f<dSdS)N)Z interfacez2something went wrong with remote->local forwardingrr3) runpackGlobal_tcpip_forwardrZ listenTCPZSSHListenForwardingFactoryconnZ SSHListenServerForwardingChannelrerrr-r$r6ZhostZportlistenerrrrglobal_tcpip_forwardds   z$ConchTestAvatar.global_tcpip_forwardcCs@t|\}}|j||fd}|s(dS|j||f=|dS)Nrr3)rr;r-get stopListeningr>rrrglobal_cancel_tcpip_forwardts z+ConchTestAvatar.global_cancel_tcpip_forwardcCs0d|_|jD]}td||qdS)NTzstopListening %s) loggedOutr-valuesrmsgrB)r$r?rrrr'~szConchTestAvatar.logoutN) r rrr)rskiprDr%r7r:r@rCr'rrrrr&Ds  r&c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)ConchSessionForTestAvatarz2 An ISession adapter for ConchTestAvatar. cCs4||_||j_d|_d|_d|_d|_t|_dS)zq Initialize the session and create a reference to it on the avatar for later inspection. NFr) r _testSessioncmdprotoptyReqeofrDeferredonClose)r$r rrrr%sz"ConchSessionForTestAvatar.__init__cCs td||_||_d|_dS)Nzpty reqT)rrF _terminalType _windowSizerL)r$ZtermZ windowSizeZattrsrrrgetPtys z ConchSessionForTestAvatar.getPtycCs"td||_t|d|_dS)Nz opening shellshell)rrFrK EchoTransportrJ)r$rKrrr openShells z#ConchSessionForTestAvatar.openShellcCs||_||_|d}|dkr8t|}td|jn|dkrdt|}||dd|nb|dkrt |}||dd|n6|dkrt |}||dd|n t dd |j jj_dS) Nrfalsesechossechoseechobad execr3)rJrKsplitFalseTransportrZ callLaterloseConnectionrTwriteSuperEchoTransportErrEchoTransportr ConchErrorr r< transportexpectedLoseConnection)r$rKrJftrrr execCommands(     z%ConchSessionForTestAvatar.execCommandcCs d|_dSNr3)rMr1rrr eofReceivedsz%ConchSessionForTestAvatar.eofReceivedcCs,td|j|jjj|_|jddS)Nzclosed cmd "%s") rrFrJrKr ZremoteWindowLeftremoteWindowLeftAtCloserOcallbackr1rrrcloseds z ConchSessionForTestAvatar.closedN) r rrr)r%rRrUrergrjrrrrrHsrH) componentsc@seZdZddZddZdS)r0cOsdSr"r)r$argskwrrrr%szCrazySubsystem.__init__cCsdS)z good ... good Nrr1rrrconnectionMadeszCrazySubsystem.connectionMadeN)r rrr%rnrrrrr0sr0c@s eZdZdZddZddZdS)r[a False transport should act like a /bin/false execution, i.e. just exit with nonzero status, writing nothing to the terminal. @ivar proto: The protocol associated with this transport. @ivar closed: A flag tracking whether C{loseConnection} has been called yet. cCs||_||d|_dS)zY @type p L{twisted.conch.ssh.session.SSHSessionProcessProtocol} instance rNrKmakeConnectionrjr$prrrr%s zFalseTransport.__init__cCsL|jr dSd|_|j|j|j|jttddddS)zI Disconnect the protocol associated with this transport. Nr3 rjrKZinConnectionLostZoutConnectionLostZerrConnectionLostZ processEndedrZFailurerr1rrrr\s   zFalseTransport.loseConnectionN)r rrr)r%r\rrrrr[s r[c@s$eZdZddZddZddZdS)rTcCs||_||d|_dSNrrorqrrrr%s zEchoTransport.__init__cCs:tt||j||jdd|kr6|dS)N )rrFreprrK outReceivedr\r5rrrr]s   zEchoTransport.writecCsL|jr dSd|_|j|j|j|jttddddSNr3rrtr1rrrr\ s   zEchoTransport.loseConnectionNr rrr%r]r\rrrrrTsrTc@s$eZdZddZddZddZdS)r_cCs||_||d|_dSrurorqrrrr%s zErrEchoTransport.__init__cCs|j||jddSNrv)rK errReceivedr5rrrr]s zErrEchoTransport.writecCsL|jr dSd|_|j|j|j|jttddddSrzrtr1rrrr\s   zErrEchoTransport.loseConnectionNr{rrrrr_sr_c@s$eZdZddZddZddZdS)r^cCs||_||d|_dSrurorqrrrr%&s zSuperEchoTransport.__init__cCs4|j||jd|j||jddSr|)rKryr}r5rrrr]+s   zSuperEchoTransport.writecCsL|jr dSd|_|j|j|j|jttddddSrzrtr1rrrr\1s   z!SuperEchoTransport.loseConnectionNr{rrrrr^$sr^)checkers)channel connectionfactorykeys)rauserauthc@seZdZejfZddZdS)ConchTestPasswordCheckercCs.|jdkr |jdkr t|jSttdS)NtestusertestpasszBad credentials)ZusernameZpasswordrsucceedfail Exception)r$Z credentialsrrrrequestAvatarIdBs z(ConchTestPasswordChecker.requestAvatarIdN)r rrr~ZIUsernamePasswordZcredentialInterfacesrrrrrr?src@seZdZddZdS)ConchTestSSHCheckercCs"|dkst|j|dkrdSdS)NrFT)lenZsuccessfulCredentials)r$ZavatarIdrrrareDoneJszConchTestSSHChecker.areDoneN)r rrrrrrrrHsrc@sFeZdZdZejejdZddZ ddZ ddZ d d Z d d Z d S)ConchTestServerFactoryr)s ssh-userauthssh-connectioncCs4t}|j|_||_t|dr*|j|_||_|S)Nrb)ConchTestServerZ privateKeysrZsupportedPublicKeysrhasattrrbrK)r$ZaddrrKrrr buildProtocolWs  z$ConchTestServerFactory.buildProtocolcCstjttjtdSN)sssh-rsasssh-dss)rKey fromStringrrr1rrr getPublicKeysbs  z$ConchTestServerFactory.getPublicKeyscCstjttjtdSr)rrrrrr1rrrgetPrivateKeyshs  z%ConchTestServerFactory.getPrivateKeyscCsdtdgiS)a% Diffie-Hellman primes that can be used for the diffie-hellman-group-exchange-sha1 key exchange. @return: The primes and generators. @rtype: L{dict} mapping the key size to a C{list} of C{(generator, prime)} tupple. isdiffie-hellman-group14-sha1)r ZgetDHGeneratorAndPrimer1rrr getPrimesns z ConchTestServerFactory.getPrimescCstj|||Sr")r SSHFactory getService)r$Ztransnamerrrrsz!ConchTestServerFactory.getServiceN)r rrZnoisyrZSSHUserAuthServerr SSHConnectionservicesrrrrrrrrrrOs rc@s(eZdZdZddZddZddZdS) ConchTestBasercCs0|jr dSt|ds&td||fd|_dS)Nrbz"unexpectedly lost connection %s %sr3)donerrFailTestr$reasonrrrconnectionLosts  zConchTestBase.connectionLostcCs4d|_|tjkr(ttd|||f|dS)Nr3z*got disconnect for %s: reason %s, desc: %s)rbraZDISCONNECT_BY_APPLICATIONrr=rr\)r$Z reasonCodeZdescrrr receiveErrors zConchTestBase.receiveErrorcCs"td|fd|_|dS)Nzgot unimplemented: seqid %sr3)rrrbr\)r$ZseqIDrrrreceiveUnimplementedsz"ConchTestBase.receiveUnimplementedN)r rrrrrrrrrrrsrc@seZdZddZdS)rcCst||tj||dSr")rrraSSHServerTransportrrrrrs zConchTestServer.connectionLostN)r rrrrrrrrsrc@s0eZdZdZddZddZddZdd Zd S) ConchTestClientz @ivar _channelFactory: A callable which accepts an SSH connection and returns a channel which will be attached to a new channel on that connection. cCs ||_dSr")_channelFactoryr$ZchannelFactoryrrrr%szConchTestClient.__init__cCst||tj||dSr")rrraSSHClientTransportrrrrrs zConchTestClient.connectionLostcCs<|tjtk}|dk}|r.|r.tdSttdS)Ns/85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:dar3zKey or fingerprint mismatch) rrrrZblobrrrr)r$keyfpZkeyMatchZfingerprintMatchrrr verifyHostKeys  zConchTestClient.verifyHostKeycCs|tdt|jdS)Nr)ZrequestServiceConchTestClientAuthConchTestClientConnectionrr1rrrconnectionSecuresz ConchTestClient.connectionSecureN)r rrr)r%rrrrrrrrs rc@s8eZdZdZdZdZddZddZddZdd Z d S) rrcCs(|js|jrtdtj||dS)Nz2got USERAUTH_SUCCESS before password and publickey)canSucceedPasswordcanSucceedPublicKeyrrrSSHUserAuthClientssh_USERAUTH_SUCCESS)r$Zpacketrrrrs  z(ConchTestClientAuth.ssh_USERAUTH_SUCCESScCsd|_tdS)Nr3r)rrrr1rrr getPasswordszConchTestClientAuth.getPasswordcCsd|_ttjtSrf)rrrrrrrr1rrr getPrivateKeysz!ConchTestClientAuth.getPrivateKeycCs tjtSr")rrrrr1rrr getPublicKeysz ConchTestClientAuth.getPublicKeyN) r rrZ hasTriedNonerrrrrrrrrrrsrc@s,eZdZdZdZdZdZddZddZd S) rz @ivar _completed: A L{Deferred} which will be fired when the number of results collected reaches C{totalResults}. rrcCstj|||_dSr")rrr%rrrrrr%s z"ConchTestClientConnection.__init__cCs||j|ddS)Nr<)Z openChannelrr1rrrserviceStartedsz(ConchTestClientConnection.serviceStartedN) r rrr)rZresultsZ totalResultsr%rrrrrrs rc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)SSHTestChannelcOs:||_||_g|_g|_t|_tjj |f||dSr") r_openedreceived receivedExtrrNrOr SSHChannelr%)r$rZopenedrlkwargsrrrr%s  zSSHTestChannel.__init__cCs|j|dSr")rZerrbackrrrr openFailedszSSHTestChannel.openFailedcCs|j|dSr")rri)r$ignorerrr channelOpenszSSHTestChannel.channelOpencCs|j|dSr")rappendr5rrr dataReceivedszSSHTestChannel.dataReceivedcCs,|tjkr|j|ntd|fdS)NzUnrecognized extended data: %r)rZEXTENDED_DATA_STDERRrrrrF)r$ZdataTyper6rrr extReceiveds zSSHTestChannel.extReceivedcCstd|\|_dS)Nz>L)structZunpackstatus)r$rrrrrequest_exit_status sz"SSHTestChannel.request_exit_statuscCs d|_dS)NT) eofCalledr1rrrrgszSSHTestChannel.eofReceivedcCs|jddSr")rOrir1rrrrjszSSHTestChannel.closedN) r rrr%rrrrrrgrjrrrrrs rcCs"tdtjtgi}t|S)z Produces a SSHPublicKeyChecker with an in-memory key mapping with a single use: 'testuser' @return: L{twisted.conch.checkers.SSHPublicKeyChecker} r)r~ZInMemorySSHKeyDBrrrrZSSHPublicKeyChecker)ZconchTestPublicKeyDBrrrconchTestPublicKeyCheckersrc@sjeZdZdZesdZesdZdddZddZd d Z d d Z d dZ ddZ ddZ ddZddZdS)SSHProtocolTestsz^ Tests for communication between L{SSHServerTransport} and L{SSHClientTransport}. can't run without cryptographyCannot run without PyASN1r+c sttd|_t|j}t}|t|t ||t }||_| | d|_ t|j |_tfdd|_t|j|_|j |j|j|jS)z Create a connected SSH client and server protocol pair and return a L{Deferred} which fires with an L{SSHTestChannel} instance connected to a channel on that SSH connection. rNcstfd|iS)Nr<)rrrrresultrrCz:SSHProtocolTests._ourServerOurClientTest..)rrNr!realmrZPortalrZregisterCheckerrrr startFactoryrZserverr ZclientTransportrZclientZserverTransportrp)r$rrrrZsshpcZfacrrr_ourServerOurClientTest0s&        z(SSHProtocolTests._ourServerOurClientTestcsd}fdd}||fdd}||fdd}||fdd}|||S) z Run the Conch server against the Conch client. Set up several different channels which exercise different behaviors and wait for them to complete. Verify that the channels with errors log them. c s&|_|j|dtddtS)N subsystems not-crazyr3r assertFailurer< sendRequestrNSrrr1rr cbSubsystemTszFSSHProtocolTests.test_subsystemsAndGlobalRequests..cbSubsystemcsj}|j|dtddS)Nrr,r3rr<rrrignoredrr1rrcbNotCrazyFailed\szKSSHProtocolTests.test_subsystemsAndGlobalRequests..cbNotCrazyFailedcs\j}|jddd}|jddd}|jd|jdddt}t|||gS)Nfoobarr3sfoo-2bar2r9) rr<ZsendGlobalRequest addCallback assertEqualrrr gatherResults)rrZd1Zd2Zd3r1rrcbGlobalRequestsbszKSSHProtocolTests.test_subsystemsAndGlobalRequests..cbGlobalRequestscs>jjjdddj}d|jj_|j| dS)Nrr)r2r8T) rrr r.rr<rarbZserviceStoppedr\rr1rr disconnectps  zESSHProtocolTests.test_subsystemsAndGlobalRequests..disconnectrr)r$rrrrrrr1r test_subsystemsAndGlobalRequestsLs       z1SSHProtocolTests.test_subsystemsAndGlobalRequestscst}tdddfdd}||fdd}||fdd }||fd d }|||S) z L{SSHChannel.sendRequest} can open a shell with a I{pty-req} request, specifying a terminal type and window size. conch-test-termPrrrcs|_|j|ddS)Nspty-reqr3)rr<rrr6r$rr cbChannelsz.SSHProtocolTests.test_shell..cbChannelcsdjjjjdj}|jjj|jd|jd |j j }|j |dddS)NrrrrSrr3) rr r<Zchannelsr ZassertIsrrPrQ assertTruerLrr)rr rr1rrcbPtys z*SSHProtocolTests.test_shell..cbPtycs6jdjjjtjjjjj jgS)Nstesting the shell!) rr]r<ZsendEOFrrrOrr rIrr1rrcbShells   z,SSHProtocolTests.test_shell..cbShellcsZjjdkr tdjjfdjjdjjj j j j dS)Nrzshell exit status was not 0: %irstesting the shell! ) rrrrFrjoinrrrrr rIrMrr1rrcbExiteds    z-SSHProtocolTests.test_shell..cbExited)rr ZpackRequest_pty_reqr)r$rrrrrrrr test_shell}s     zSSHProtocolTests.test_shellcs8}fdd}||fdd}|||S)z If L{SSHChannel.sendRequest} issues an exec which the server responds to with an error, the L{Deferred} it returns fires its errback. c s&|_|j|dtddtS)Nexecs jumboliahr3rrr1rrrsz3SSHProtocolTests.test_failedExec..cbChannelcs$tj}|djjddS)Nr)rYN)flushLoggedErrorsr r`rvaluerlrerrorsr1rrcbFaileds z2SSHProtocolTests.test_failedExec..cbFailedr)r$rrrrr1rtest_failedExecs     z SSHProtocolTests.test_failedExeccsN}fdd}||fdd}||fdd}|||S)z When the process started by a L{SSHChannel.sendRequest} exec request exits, the exit status is reported to the channel. cs|_|j|dtddS)NrrVr3rrr1rrrsz5SSHProtocolTests.test_falseChannel..cbChannelcsjjSr"rrOrr1rrcbExecsz2SSHProtocolTests.test_falseChannel..cbExeccs$jjgjjddSru)rrrZassertNotEqualrrr1rrcbClosedsz4SSHProtocolTests.test_falseChannel..cbClosedrr$rrrrrr1rtest_falseChannels      z"SSHProtocolTests.test_falseChannelcsTjddd}fdd}||fdd}||fdd }|||S) z Bytes sent over the extended channel for stderr data are delivered to the channel's C{extReceived} method. rWZ localWindowZlocalMaxPacketcs|_|j|dtddS)Nrs eecho hellor3rrr1rrrsz5SSHProtocolTests.test_errorChannel..cbChannelcstjjjjjjgSr"rrrrOrr rIrr1rrrs z2SSHProtocolTests.test_errorChannel..cbExeccspjjgdjjdjjdjjjjdjjj j j j dS)Nrhello rr) rrrrrrrrlocalWindowLeftrr rIrhrr1rrrs z4SSHProtocolTests.test_errorChannel..cbClosedrrrr1rtest_errorChannels     z"SSHProtocolTests.test_errorChannelcs,dt}fdd}|||S)z When an attempt is made to open an unknown channel type, the L{Deferred} returned by L{SSHChannel.sendRequest} fires its errback. scrazy-unknown-channelcs4tj}|djjdt|ddS)Nr)zunknown channelr3)rr r`rrrlrrr1rrrs z6SSHProtocolTests.test_unknownChannel..cbFailed)rrrr)r$drrr1rtest_unknownChannels  z$SSHProtocolTests.test_unknownChannelcsTjddd}fdd}||fdd}||fdd }|||S) zc An L{SSHChannel} can be configured with a maximum packet size to receive. r3rcs|_|j|dtddS)Nrs secho hellor3rrr1rrrsz2SSHProtocolTests.test_maxPacket..cbChannelcsjjSr"rrr1rrrsz/SSHProtocolTests.test_maxPacket..cbExeccs^jjddjjddjjdjjdjjdS)Nrrrr) rrrrrrr rrrr1rrr s z1SSHProtocolTests.test_maxPacket..cbClosedrrrr1rtest_maxPacket s      zSSHProtocolTests.test_maxPacketcsTjddd}fdd}||fdd}||fdd }|||S) ze Normal standard out bytes are sent to the channel's C{dataReceived} method. rrWrcs|_|j|dtddS)Nrs echo hellor3rrr1rrr1sz-SSHProtocolTests.test_echo..cbChannelcstjjjjjjgSr"rrr1rrcbEcho7s z*SSHProtocolTests.test_echo..cbEchocs`jjddjjdjjdjjjjjj j j dS)Nrrrr) rrrrrr rrrr rIrhrr1rrr=s z,SSHProtocolTests.test_echo..cbClosedr)r$rrrrrr1r test_echo*s      zSSHProtocolTests.test_echoN)r+)r rrr)rrGrrrrrrr r rrrrrrr$s 1/ rc@sFeZdZesdZesdZdddZddZdd Zd d Z d d Z dS)SSHFactoryTestsrrNcs6t}dd}fdd|_||_|_||S)NcSsdtdiS)Nzssh-rsa)rrrrrrrTrz0SSHFactoryTests.makeSSHFactory..csSr"rrprimesrrrUr)rrrrrr)r$rZ sshFactoryZgpkrrrmakeSSHFactoryRs  zSSHFactoryTests.makeSSHFactorycCs$|}|d}||tjdS)zc By default, buildProtocol() constructs an instance of SSHServerTransport. N)rrZassertIsInstancerar)r$rrrrrtest_buildProtocol[s z"SSHFactoryTests.test_buildProtocolcs:gfdd}|}||_|d|dgdS)zc buildProtocol() calls 'self.protocol()' to construct a protocol instance. cs|tSr")rrar)rlZcallsrr makeProtocolks zHSSHFactoryTests.test_buildProtocolRespectsProtocol..makeProtocolNr)rrrr)r$rrrrr"test_buildProtocolRespectsProtocoles   z2SSHFactoryTests.test_buildProtocolRespectsProtocolcCs6|jdd}|d}|d|j|d|jdS)zg Group key exchanges are not supported when we don't have the primes database. Nr"diffie-hellman-group-exchange-sha1$diffie-hellman-group-exchange-sha256)rrZ assertNotInsupportedKeyExchanges)r$f1Zp1rrrtest_buildProtocolNoPrimests  z*SSHFactoryTests.test_buildProtocolNoPrimescCs:|jddid}|d}|d|j|d|jdS)zU Group key exchanges are supported when we have the primes database. r3)rr rNrr)rrZassertInr)r$f2Zp2rrrtest_buildProtocolWithPrimess z,SSHFactoryTests.test_buildProtocolWithPrimes)N) r rrrrGrrrrrr rrrrrJs  rc@sVeZdZdZesdZesdZer*eej Z ddZ ddZ dd Z d d Z d d ZdS)MPTestszs Tests for L{common.getMP}. @cvar getMP: a method providing a MP parser. @type getMP: C{callable} rrcCs||dddS)z L{common.getMP} should parse the a multiple precision integer from a string: a 4-byte length followed by length bytes of the integer. s)r3rNrgetMPr1rrr test_getMPszMPTests.test_getMPcCs||dddS)zv L{common.getMP} should be able to parse a big enough integer (that doesn't fit on one byte). s)irNr"r1rrrtest_getMPBigIntegerszMPTests.test_getMPBigIntegercCs||ddddS)zg L{common.getMP} has the ability to parse multiple integer in the same string. sr)r3rrNr"r1rrrtest_multipleGetMPs zMPTests.test_multipleGetMPcCs||dddS)zu When more data than needed is sent to L{common.getMP}, it should return the remaining data. s foo)r3rNr"r1rrrtest_getMPRemainingDataszMPTests.test_getMPRemainingDatacCs|tj|jddS)z When the string passed to L{common.getMP} doesn't even make 5 bytes, it should raise a L{struct.error}. sN)Z assertRaisesrr r#r1rrrtest_notEnoughDataszMPTests.test_notEnoughDataN)r rrr)rrGr staticmethodrr#r$r%r&r'r(rrrrr!s     r!c@s eZdZdZesdZddZdS)GMPYInstallDeprecationTestszI Tests for the deprecation of former GMPY accidental public API. r*cCs>t||jg}|t|d||ddddS)zD L{twisted.conch.ssh.common.install} is deprecated. r3rmessagezAtwisted.conch.ssh.common.install was deprecated in Twisted 16.5.0N)rinstallZ flushWarningstest_deprecatedrr)r$warningsrrrr-s z+GMPYInstallDeprecationTests.test_deprecatedN)r rrr)rrGr-rrrrr*sr*)Rr)Z __future__rrrZtwisted.python.reflectrrrZtwisted.conch.sshrrr r Z twisted.conchr r Ztwisted.conch.test.keydatarrrrZ twisted.credrZtwisted.cred.errorrZtwisted.internetrrrZtwisted.internet.errorrZtwisted.pythonrrZ twisted.trialrZtwisted.conch.test.loopbackr objectr!rr&rHrkZregisterAdapterZISessionZProtocolr0r[rTr_r^r~rrrrrarrZSSHProtocolCheckerrrrrrrrrrrrrrrrZTestCaserrr!r*rrrrsf      BC     5* (HA