o Ņó·i2ć@sźdZddlmZddlmZddlmZz.ddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZd ZWn eyOd ZeZ YnwddlmZGdd„de ƒZ  dddd„ZeeeƒGdd„deƒƒZdS)z Test ssh/channel.py. é)Ś annotations)ŚskipIf)Ś verifyObject)Śchannel)ŚSSHTransportAddress)Ś SSHService)ŚSSHServerTransport)Ś interfaces)Ś IPv4Address)ŚStringTransportŚzConch SSH not supported.)ŚTestCasec@sBeZdZdZddd„Zddd„Zdd d„Zddd„Zddd„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. ŚreturnŚNonecCsi|_i|_i|_dS)N)ŚdataŚextDataŚcloses©Śself©rśeC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/conch/test/test_channel.pyŚ__init__*s zMockConnection.__init__ŚstrcCsdS)z, Return our logging prefix. rNrrrrrŚ logPrefix/szMockConnection.logPrefixrśchannel.SSHChannelrŚbytescCs|j |g” |”dS)z' Record the sent data. N)rŚ setdefaultŚappend)rrrrrrŚsendData5ózMockConnection.sendDataŚtypeŚintcCs|j |g” ||f”dS)z0 Record the sent extended data. N)rrr)rrr!rrrrŚsendExtendedData;szMockConnection.sendExtendedDatacCsd|j|<dS)z? Record that the channel sent a close message. TN)r)rrrrrŚ sendCloseCszMockConnection.sendCloseN©rr)rr)rrrrrr)rrr!r"rrrr)rrrr) Ś__name__Ś __module__Ś __qualname__Ś__doc__rrrr#r$rrrrrs   rNŚservicerŚ hostAddressśinterfaces.IAddress | NoneŚ peerAddressrrcCs&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¦eZdZdZd#dd„Zd#dd„Zd#dd „Zd#d d „Zd#d d „Zd#dd„Z d#dd„Z d#dd„Z d#dd„Z d#dd„Z d#dd„Zd#dd„Zd#dd„Zd#dd„Zd#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)rr2rŚ SSHChannelŚnamerrrrŚsetUpis zChannelTests.setUpcCs| ttj|jƒ”dS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)Ś assertTruerr Z ITransportrrrrrŚtest_interfacesr zChannelTests.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. )r2ii€réééééééN) rr4r2Ś assertEqualZlocalWindowSizeZlocalWindowLeftZlocalMaxPacketŚremoteWindowLeftr3Ś 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%r9©Z localWindowzN)r@rrr4rrrrŚ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%r9rFsN)r@rŚ __bytes__r4rrrrŚ 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‰d‡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. Frrcó dˆd<dS©NTrrr©ŚcbrrŚstubStartWritingæó z:ChannelTests.test_addWindowBytes..stubStartWritingótestr9é2é*ró)r9rQésPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaNr%)rZ startWritingŚwriteŚ writeExtendedŚaddWindowBytesr@rAr7Ś areWritingŚbufr2rŚ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)rZrequest_test_methodr7ZrequestReceivedr\rrrrŚtest_requestReceived×s z!ChannelTests.test_requestReceivedcCs*| |jj”|j ”| |jj”dS)zM Test that the default closeReceieved closes the connection. N)r\rŚclosingZ closeReceivedr7rrrrŚtest_closeReceievedįs z ChannelTests.test_closeReceievedcs&dg‰d‡fdd„ }||j_|j d”|j d”| |jj”| ˆd”|j d ”|j d ”|jj|j}|  |d d g”|  |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. FrrcrKrLrrrMrrŚstubStopWritingńrPz0ChannelTests.test_write..stubStopWritingódr`rrUótaódaéó 12345678901)rgrfó 1234567890ó1r=ó123456)rgrfrjrkó12345ó6Nr%) rŚ stopWritingrVr\rYr7rXr2rr@rArZ©rrdrrrMrŚ test_writeés,      zChannelTests.test_writecsDdg‰d‡fdd„ }||j_|j dd”|j dd”|j d d ”| |jj”| ˆd ”|j d ”|j d d”|jj|j}|  |gd ¢”|  |jj d”|j dd”|  |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. FrrcrKrLrrrMrrrdrPz8ChannelTests.test_writeExtended..stubStopWritingr9rer`r:ótrrU)©r9rg©r:rr©r:r`rhr;ri)rsrtru©r;rj©r;rkr=r<rl)rsrtrurvrw)r<rmrnNr%) rrorWr\rYr7rXr2rr@rAr[rprrMrŚtest_writeExtended s8 ’ž 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Ś Cs€z2ChannelTests.test_writeSequence..r1s 0123456789N)rrXZ writeSequenceŚranger@r2rrrrrŚ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. sdatar9sdatadatar<éN) rrVrWr]rBr2rŚgetrXr7rrrrŚtest_loseConnectionFs    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/rr2r@rZgetPeer)rZpeerrrrŚ test_getPeerTó 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/rr2r@rZgetHost)rŚhostrrrŚ test_getHost^r„zChannelTests.test_getHostNr%)r&r'r(r)r6r8rErGrIrJr^rarcrqrxr}r€rƒr†rrrrr0cs"       $ 1   r0)NN)r*rr+r,r-r,rr)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/r0rrrrŚs0           ž / ż