o ¯b—/ã@sØdZddlmZddlmZz.ddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZd ZWn eyId ZeZ Ynwdd lmZGdd„de ƒZddd„ZeeeƒGdd„deƒƒZdS)z Test ssh/channel.py. é)ÚskipIf)Ú verifyObject)Úchannel)ÚSSHTransportAddress)Ú SSHService)ÚSSHServerTransport)Ú interfaces)Ú IPv4Address)ÚStringTransportÚzConch SSH not supported.)ÚTestCasec@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚMockConnectionaÝ A mock for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping channel id #s to lists of data sent by that channel. @ivar extData: a L{dict} mapping channel id #s to lists of 2-tuples (extended data type, data) sent by that channel. @ivar closes: a L{dict} mapping channel id #s to True if that channel sent a close message. cCsi|_i|_i|_dS)N)ÚdataÚextDataÚcloses©Úself©rúA/usr/lib/python3/dist-packages/twisted/conch/test/test_channel.pyÚ__init__*s zMockConnection.__init__cCsdS)z, Return our logging prefix. r rrrrrÚ logPrefix/szMockConnection.logPrefixcCs|j |g¡ |¡dS)z' Record the sent data. N)rÚ setdefaultÚappend)rrrrrrÚsendData5ózMockConnection.sendDatacCs|j |g¡ ||f¡dS)z0 Record the sent extended data. N)rrr)rrÚtyperrrrÚsendExtendedData;szMockConnection.sendExtendedDatacCsd|j|<dS)z? Record that the channel sent a close message. TN)r)rrrrrÚ sendCloseAszMockConnection.sendCloseN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrrrr s  r NcCs&tƒ}| t||d¡| |¡dS)a¯ Connect a SSHTransport which is already connected to a remote peer to the channel under test. @param service: Service used over the connected transport. @type service: L{SSHService} @param hostAddress: Local address of the connected transport. @type hostAddress: L{interfaces.IAddress} @param peerAddress: Remote address of the connected transport. @type peerAddress: L{interfaces.IAddress} )Ú hostAddressÚ peerAddressN)rÚmakeConnectionr Ú setService)Úservicer"r#Ú transportrrrÚconnectSSHTransportHs  ÿr(c@sˆeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd S)!Ú ChannelTestsz" Tests for L{SSHChannel}. cCs&tƒ|_tj|jdd|_d|j_dS)z¶ Initialize the channel. remoteMaxPacket is 10 so that data is able to be sent (the default of 0 means no data is sent because no packets are made). é )ÚconnÚremoteMaxPacketschannelN)r r+rÚ SSHChannelÚnamerrrrÚsetUpcs zChannelTests.setUpcCs| ttj|jƒ¡dS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)Ú assertTruerrÚ ITransportrrrrrÚtest_interfacemrzChannelTests.test_interfacec Cstj|jd}| |jd¡| |jd¡| |jd¡| |jd¡| |jd¡| |j|j¡|  |j ¡|  |j ¡t ddddd d d ¡}| |jd¡| |jd¡| |jd¡| |jd¡| |jd¡| |jd ¡| |j d ¡| |j d ¡d S) a‹ Test that SSHChannel initializes correctly. localWindowSize defaults to 131072 (2**17) and localMaxPacket to 32768 (2**15) as reasonable defaults (what OpenSSH uses for those variables). The values in the second set of assertions are meaningless; they serve only to verify that the instance variables are assigned in the correct order. )r+ii€réééééééN) rr-r+Ú assertEqualÚlocalWindowSizeÚlocalWindowLeftÚlocalMaxPacketÚremoteWindowLeftr,Ú assertIsNonerÚavatar)rÚcÚc2rrrÚ test_initss$   zChannelTests.test_initcCs.| t|jƒd¡| ttjddƒd¡dS)z{ Test that str(SSHChannel) works gives the channel name and local and remote windows at a glance.. z%r3©Ú localWindowzN)r:Ústrrr-rrrrÚtest_str‘sÿzChannelTests.test_strcCs.| |j ¡d¡| tjdd ¡d¡dS)z Test that bytes(SSHChannel) works, gives the channel name and local and remote windows at a glance.. s%r3rDsN)r:rÚ __bytes__r-rrrrÚ test_bytes›s ÿþzChannelTests.test_bytescCs| |j ¡d¡dS)z‹ Test that SSHChannel.logPrefix gives the name of the channel, the local channel ID and the underlying connection. z.SSHChannel channel (unknown) on MockConnectionN)r:rrrrrrÚtest_logPrefix©s ÿzChannelTests.test_logPrefixcsdg‰‡fdd„}||j_|j d¡|j dd¡|j d¡| |jjd¡| |jj¡| ˆd¡| |jj d ¡| |j j |jdg¡| |jj g¡| |j j |jd g¡dˆd<|j d ¡| ˆd¡|j d ¡|j ¡|j d ¡| ˆd¡d S)zq Test that addWindowBytes adds bytes to the window and resumes writing if it was paused. Fcó dˆd<dS©NTrrr©ÚcbrrÚstubStartWriting¹ó z:ChannelTests.test_addWindowBytes..stubStartWritingótestr3é2é*ró)r3rQésPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaN)rÚ startWritingÚwriteÚ writeExtendedÚaddWindowBytesr:r>r0Ú areWritingÚbufr+rÚextBufrÚ assertFalseÚloseConnection)rrOrrMrÚtest_addWindowBytes²s(       z ChannelTests.test_addWindowBytescCsLdd„|j_| |j dd¡¡| |j dd¡¡| |j dd¡¡dS)zn Test that requestReceived handles requests by dispatching them to request_* methods. cSs|dkS)NrTr)rrrrÚÖsz3ChannelTests.test_requestReceived..s test-methodrTóas bad-methodN)rÚrequest_test_methodr0ÚrequestReceivedr]rrrrÚtest_requestReceivedÑs z!ChannelTests.test_requestReceivedcCs*| |jj¡|j ¡| |jj¡dS)zM Test that the default closeReceieved closes the connection. N)r]rÚclosingÚ closeReceivedr0rrrrÚtest_closeReceievedÛs z ChannelTests.test_closeReceievedcs$dg‰‡fdd„}||j_|j d¡|j d¡| |jj¡| ˆd¡|j d¡|j d¡|jj|j}|  |d dg¡|  |jj d ¡|j d ¡|  |gd ¢¡|  |jj d ¡dˆd<|j d¡| |jj¡| ˆd¡|  |gd¢¡|  |jj d¡|  |jj d¡dS)z´ Test that write handles data correctly. Send data up to the size of the remote window, splitting the data into packets of length remoteMaxPacket. FcrKrLrrrMrrÚstubStopWritingërPz0ChannelTests.test_write..stubStopWritingódrarrUótaódaéó 12345678901)rkrjó 1234567890ó1r7ó123456)rkrjrnroó12345ó6N) rÚ stopWritingrWr]rZr0rYr+rr:r>r[©rrhrrrMrÚ test_writeãs,       zChannelTests.test_writecsBdg‰‡fdd„}||j_|j dd¡|j dd¡|j dd¡| |jj¡| ˆd ¡|j d ¡|j dd¡|jj|j}|  |gd ¢¡|  |jj d ¡|j d d¡|  |gd¢¡|  |jj d¡dˆd <|j dd¡| |jj¡| ˆd ¡|  |gd¢¡|  |jj ddgg¡|  |jj d ¡dS)zÇ Test that writeExtended handles data correctly. Send extended data up to the size of the window, splitting the extended data into packets of length remoteMaxPacket. FcrKrLrrrMrrrhrPz8ChannelTests.test_writeExtended..stubStopWritingr3rirar4ótrrU)©r3rk©r4rv©r4rarlr5rm)rwrxry©r5rn©r5ror7r6rp)rwrxryrzr{)r6rqrrN) rrsrXr]rZr0rYr+rr:r>r\rtrrMrÚtest_writeExtendeds8  ÿþ zChannelTests.test_writeExtendedcCsB|j d¡|j dd„tdƒDƒ¡| |jj|jdg¡dS)zS Test that writeSequence is equivalent to write(''.join(sequece)). rUcss|]}d|fVqdS)s%dNr)Ú.0ÚirrrÚ =s€z2ChannelTests.test_writeSequence..r*s 0123456789N)rrYÚ writeSequenceÚranger:r+rrrrrÚtest_writeSequence8s zChannelTests.test_writeSequencecCs‚|j d¡|j dd¡|j ¡| |jj |j¡¡|j d¡| |jj |j¡¡|j d¡|  |jj |j¡¡dS)zk Tesyt that loseConnection() doesn't close the channel until all the data is sent. sdatar3sdatadatar6éN) rrWrXr^r?r+rÚgetrYr0rrrrÚtest_loseConnection@s    z ChannelTests.test_loseConnectioncCó6tdddƒ}t|jj|d| t|ƒ|j ¡¡dS)z€ L{SSHChannel.getPeer} returns the same object as the underlying transport's C{getPeer} method returns. ÚTCPz 192.168.0.1i1Ô)r&r#N)r r(rr+r:rÚgetPeer)rÚpeerrrrÚ test_getPeerNó zChannelTests.test_getPeercCr†)z€ L{SSHChannel.getHost} returns the same object as the underlying transport's C{getHost} method returns. r‡z 127.0.0.1i90)r&r"N)r r(rr+r:rÚgetHost)rÚhostrrrÚ test_getHostXr‹zChannelTests.test_getHostN)rrr r!r/r2rCrGrIrJr_rdrgrur|r‚r…rŠrŽrrrrr)]s"    $1 r))NN)r!ÚunittestrÚzope.interface.verifyrÚtwisted.conch.sshrÚtwisted.conch.ssh.addressrÚtwisted.conch.ssh.servicerÚtwisted.conch.ssh.transportrÚtwisted.internetrÚtwisted.internet.addressr Útwisted.test.proto_helpersr ÚskipTestÚ ImportErrorÚobjectÚtwisted.trial.unittestr r r(r)rrrrÚs*          þ  +