U  W[C/ć@sādZddlmZmZddlmZz\ddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZd ZWnek r˜d ZeZ YnXdd lmZddlmZGdd„de ƒZddd„ZGdd„dejƒZd S)z Test ssh/channel.py. é)ŚdivisionŚabsolute_import)Ś verifyObject)Śchannel)ŚSSHTransportAddress)ŚSSHServerTransport)Ś SSHService)Ś interfaces)Ś IPv4Address)ŚStringTransportNzConch SSH not supported.)Śunittest)Ś intToBytesc@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__)szMockConnection.__init__cCsdS)z, Return our logging prefix. rrrrrrŚ logPrefix/szMockConnection.logPrefixcCs|j |g” |”dS)z' Record the sent data. N)rŚ setdefaultŚappend)rrrrrrŚsendData6szMockConnection.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Ś sendCloseDszMockConnection.sendCloseN) Ś__name__Ś __module__Ś __qualname__Ś__doc__rrrrrrrrrrs  rcCs&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)rZmakeConnectionr Z setService)Śservicer"r#Z transportrrrŚconnectSSHTransportLs ’r%c@sŒeZdZdZeZdd„Zdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zdd„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)rr(rŚ SSHChannelŚnamerrrrŚsetUphs ’zChannelTests.setUpcCs| ttj|jƒ”dS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)Ś assertTruerr Z ITransportrrrrrŚtest_interfacetszChannelTests.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(Ś assertEqualZlocalWindowSizeZlocalWindowLeftZlocalMaxPacketŚremoteWindowLeftr)Ś assertIsNonerZavatar)rŚcZc2rrrŚ test_init{s$   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%r/©Z localWindowzN)r6Ś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%r/r;sN)r6rŚ __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)r6rrrrrrŚtest_logPrefix“szChannelTests.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. Fcs dˆd<dS©NTrrr©ŚcbrrŚstubStartWritingĆsz:ChannelTests.test_addWindowBytes..stubStartWritingótestr/é2é*ró)r/rEésPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaN)rZ startWritingŚwriteŚ writeExtendedŚaddWindowBytesr6r7r-Ś areWritingŚbufr(rŚextBufrŚ assertFalseŚloseConnection)rrDrrBrŚ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)NrHr)rrrrŚąrHz3ChannelTests.test_requestReceived..s test-methodrHóas bad-methodN)rZrequest_test_methodr-ZrequestReceivedrPrrrrŚtest_requestReceivedŪs z!ChannelTests.test_requestReceivedcCs*| |jj”|j ”| |jj”dS)zM Test that the default closeReceieved closes the connection. N)rPrŚclosingZ closeReceivedr-rrrrŚ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 ”|  |d dd d g”|  |jj d”dˆd<|j d”| |jj”| ˆd”|  |d dd d dg”|  |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. Fcs dˆd<dSrArrrBrrŚstubStopWritingösz0ChannelTests.test_write..stubStopWritingódrTrrIstaódaéó 12345678901ó 1234567890ó1r3ó123456ó12345ó6N) rŚ stopWritingrJrPrMr-rLr(rr6r7rN©rrXrrrBrŚ test_writeļs,       zChannelTests.test_writec sRdg‰‡fdd„}||j_|j dd”|j dd”|j dd”| |jj”| ˆd ”|j d ”|j dd”|jj|j}|  |d d d g”|  |jj d”|j dd”|  |d d d ddg”|  |jj d”dˆd <|j dd”| |jj”| ˆd ”|  |d d d dddg”|  |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. Fcs dˆd<dSrArrrBrrrXsz8ChannelTests.test_writeExtended..stubStopWritingr/rYrTr0ótrrI)r/rZ)r0re)r0rTr[r1r\)r1r])r1r^r3r2r_)r2r`raN) rrbrKrPrMr-rLr(rr6r7rOrcrrBrŚtest_writeExtendeds<   ’ ’zChannelTests.test_writeExtendedcCs>|j d”|j tttdƒƒ”| |jj|jdg”dS)zS Test that writeSequence is equivalent to write(''.join(sequece)). rIr's 0123456789N) rrLZ writeSequenceŚmapr Śranger6r(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. sdatar/sdatadatar2éN) rrJrKrQr8r(rŚgetrLr-rrrrŚtest_loseConnectionAs    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(r6rZgetPeer)rZpeerrrrŚ test_getPeerPs 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. rmz 127.0.0.1i90)r$r"N)r r%rr(r6rZgetHost)rZhostrrrŚ test_getHost[s zChannelTests.test_getHostN)rrr r!ŚskipTestŚskipr,r.r:r=r?r@rRrUrWrdrfrirlrnrorrrrr&as"     #&  r&)NN)r!Z __future__rrZzope.interface.verifyrZtwisted.conch.sshrZtwisted.conch.ssh.addressrZtwisted.conch.ssh.transportrZtwisted.conch.ssh.servicerZtwisted.internetr Ztwisted.internet.addressr Ztwisted.test.proto_helpersr rpŚ ImportErrorŚobjectZ twisted.trialr Ztwisted.python.compatr rr%ZTestCaser&rrrrŚs&           0