U ª€lj2ã@sòdZddlmZddlmZddlmZz\ddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZd ZWnek r d ZeZ YnXddlmZGdd„de ƒZddddddœdd„ZeeeƒGdd„deƒƒZdS)z Test ssh/channel.py. é)Ú annotations)ÚskipIf)Ú verifyObject)Úchannel)ÚSSHTransportAddress)Ú SSHService)ÚSSHServerTransport)Ú interfaces)Ú IPv4Address)ÚStringTransportÚzConch SSH not supported.)ÚTestCasec@sbeZdZdZddœdd„Zddœdd„Zd d dd œd d „Zd dd ddœdd„Zd ddœdd„ZdS)Ú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. ÚNone©ÚreturncCsi|_i|_i|_dS)N)ÚdataÚextDataÚcloses©Úself©rúC/tmp/pip-unpacked-wheel-aghyhbke/twisted/conch/test/test_channel.pyÚ__init__*szMockConnection.__init__ÚstrcCsdS)z, Return our logging prefix. rrrrrrÚ logPrefix/szMockConnection.logPrefixzchannel.SSHChannelÚbytes)rrrcCs|j |g¡ |¡dS)z' Record the sent data. N)rÚ setdefaultÚappend)rrrrrrÚsendData5szMockConnection.sendDataÚint)rÚtyperrcCs|j |g¡ ||f¡dS)z0 Record the sent extended data. N)rrr)rrr!rrrrÚsendExtendedData;szMockConnection.sendExtendedData)rrcCsd|j|<dS)z? Record that the channel sent a close message. TN)r)rrrrrÚ sendCloseCszMockConnection.sendCloseN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr"r#rrrrrs  rNrzinterfaces.IAddress | Noner)ÚserviceÚ hostAddressÚ peerAddressrcCs&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} )r)r*N)rZmakeConnectionr Z setService)r(r)r*Ú transportrrrÚconnectSSHTransportJs  ÿr,c@sâeZdZdZddœdd„Zddœdd„Zddœdd „Zddœd d „Zddœd d „Zddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Zddœdd„Zddœdd„Zddœdd„Zddœd d!„Zd"S)#Ú ChannelTestsz" Tests for L{SSHChannel}. rrcCs&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)rr/rÚ SSHChannelÚnamerrrrÚsetUpiszChannelTests.setUpcCs| ttj|jƒ¡dS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)Ú assertTruerr Z ITransportrrrrrÚtest_interfacesszChannelTests.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) rr1r/Ú assertEqualZlocalWindowSizeZlocalWindowLeftZlocalMaxPacketÚremoteWindowLeftr0Ú assertIsNonerZavatar)rÚcÚc2rrrÚ test_initys$   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%r6©Z localWindowzN)r=rrr1rrrrÚ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%r6rCsN)r=rÚ __bytes__r1rrrrÚ 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‰ddœ‡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 ¡dS)zq Test that addWindowBytes adds bytes to the window and resumes writing if it was paused. Frrcs dˆd<dS©NTrrr©ÚcbrrÚstubStartWriting¿sz:ChannelTests.test_addWindowBytes..stubStartWritingótestr6é2é*ró)r6rLésPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaN)rZ startWritingÚwriteÚ writeExtendedÚaddWindowBytesr=r>r4Ú areWritingÚbufr/rÚextBufrÚ assertFalseÚloseConnection)rrKrrIrÚ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)NrOr)rrrrÚÜrOz3ChannelTests.test_requestReceived..s test-methodrOóas bad-methodN)rZrequest_test_methodr4ZrequestReceivedrWrrrrÚtest_requestReceived×s z!ChannelTests.test_requestReceivedcCs*| |jj¡|j ¡| |jj¡dS)zM Test that the default closeReceieved closes the connection. N)rWrÚclosingZ closeReceivedr4rrrrÚtest_closeReceievedás z ChannelTests.test_closeReceievedcs4dg‰ddœ‡fdd„ }||j_|j d¡|j d¡| |jj¡| ˆd¡|j d ¡|j d ¡|jj|j}|  |d d g¡|  |jj d ¡|j d ¡|  |d d ddg¡|  |jj d¡dˆd<|j d¡| |jj¡| ˆd¡|  |d d dddg¡|  |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. Frrcs dˆd<dSrHrrrIrrÚstubStopWritingñsz0ChannelTests.test_write..stubStopWritingódr[rrPstaódaéó 12345678901ó 1234567890ó1r:ó123456ó12345ó6N) rÚ stopWritingrQrWrTr4rSr/rr=r>rU©rr_rrrIrÚ test_writeés,      zChannelTests.test_writec sXdg‰ddœ‡fdd„ }||j_|j dd¡|j dd¡|j d d ¡| |jj¡| ˆd ¡|j d ¡|j d d¡|jj|j}|  |d ddg¡|  |jj d¡|j dd¡|  |d ddddg¡|  |jj d¡dˆd <|j dd¡| |jj¡| ˆd ¡|  |d dddddg¡|  |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. Frrcs dˆd<dSrHrrrIrrr_sz8ChannelTests.test_writeExtended..stubStopWritingr6r`r[r7ótrrP)r6ra)r7rl)r7r[rbr8rc)r8rd)r8rer:r9rf)r9rgrhN) rrirRrWrTr4rSr/rr=r>rVrjrrIrÚtest_writeExtended sF  ÿúþ zChannelTests.test_writeExtendedcCsB|j d¡|j dd„tdƒDƒ¡| |jj|jdg¡dS)zS Test that writeSequence is equivalent to write(''.join(sequece)). rPcss|]}d|fVqdS)s%dNr)Ú.0ÚirrrÚ Csz2ChannelTests.test_writeSequence..r.s 0123456789N)rrSZ writeSequenceÚranger=r/rrrrrÚtest_writeSequence>s 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. sdatar6sdatadatar9éN) rrQrRrXr?r/rÚgetrSr4rrrrÚtest_loseConnectionFs    z ChannelTests.test_loseConnectioncCs6tdddƒ}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=rZgetPeer)rZpeerrrrÚ test_getPeerTs zChannelTests.test_getPeercCs6tdddƒ}t|jj|d| t|ƒ|j ¡¡dS)z€ L{SSHChannel.getHost} returns the same object as the underlying transport's C{getHost} method returns. rvz 127.0.0.1i90)r(r)N)r r,rr/r=rZgetHost)rÚhostrrrÚ test_getHost^s zChannelTests.test_getHostN)r$r%r&r'r3r5rBrDrFrGrYr\r^rkrmrrrurwryrrrrr-cs     $1 r-)NN)r'Ú __future__rZunittestrZzope.interface.verifyrZtwisted.conch.sshrZtwisted.conch.ssh.addressrZtwisted.conch.ssh.servicerZtwisted.conch.ssh.transportrZtwisted.internetr Ztwisted.internet.addressr Ztwisted.internet.testingr ZskipTestÚ ImportErrorÚobjectZtwisted.trial.unittestr rr,r-rrrrÚs,            /ý