o ¼>h2ć@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ś/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/conch/test/test_channel.pyŚ__init__*s zMockConnection.__init__ŚstrcCsdS)z, Return our logging prefix. rrrrrrŚ 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)rŚmakeConnectionr Ś setService)r*r+r-Ś transportrrrŚconnectSSHTransportJs  ’r1c@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)rr4rŚ SSHChannelŚnamerrrrŚsetUpis zChannelTests.setUpcCs| ttj|jƒ”dS)zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)Ś assertTruerr Ś 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. )r4ii€réééééééN) rr6r4Ś assertEqualŚlocalWindowSizeŚlocalWindowLeftŚlocalMaxPacketŚremoteWindowLeftr5Ś assertIsNonerŚavatar)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%r<©Ś localWindowzN)rCrrr6rrrrŚ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<rMsN)rCrŚ __bytes__r6rrrrŚ 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)rCrrrrrrŚ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ótestr<é2é*ró)r<rYésPaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaNr%)rŚ startWritingŚwriteŚ writeExtendedŚaddWindowBytesrCrGr9Ś areWritingŚbufr4rŚextBufrŚ assertFalseŚloseConnection)rrWrrUrŚ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)Nr\r)rrrrŚÜsz3ChannelTests.test_requestReceived..s test-methodr\óas bad-methodN)rŚrequest_test_methodr9ŚrequestReceivedrerrrrŚtest_requestReceived×s z!ChannelTests.test_requestReceivedcCs*| |jj”|j ”| |jj”dS)zM Test that the default closeReceieved closes the connection. N)rerŚclosingŚ closeReceivedr9rrrrŚ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. FrrcrSrTrrrUrrŚstubStopWritingńrXz0ChannelTests.test_write..stubStopWritingódrirr]ótaódaéó 12345678901)rsrró 1234567890ó1r@ó123456)rsrrrvrwó12345ó6Nr%) rŚ stopWritingr_rerbr9rar4rrCrGrc©rrprrrUrŚ 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. FrrcrSrTrrrUrrrprXz8ChannelTests.test_writeExtended..stubStopWritingr<rqrir=ótrr])©r<rs©r=r~©r=rirtr>ru)rr€r©r>rv©r>rwr@r?rx)rr€rr‚rƒ)r?ryrzNr%) rr{r`rerbr9rar4rrCrGrdr|rrUrŚ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)). r]css|]}d|fVqdS)s%dNr)Ś.0ŚirrrŚ Cs€z2ChannelTests.test_writeSequence..r3s 0123456789N)rraŚ writeSequenceŚrangerCr4rrrrrŚ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. sdatar<sdatadatar?éN) rr_r`rfrHr4rŚgetrar9rrrrŚ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 r1rr4rCrŚgetPeer)rŚpeerrrrŚ 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 r1rr4rCrŚgetHost)rŚhostrrrŚ test_getHost^r“zChannelTests.test_getHostNr%)r&r'r(r)r8r;rLrOrQrRrgrlror}r„rŠrr’r–rrrrr2cs"       $ 1   r2)NN)r*rr+r,r-r,rr)r)Ś __future__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.internet.testingr ŚskipTestŚ ImportErrorŚobjectŚtwisted.trial.unittestr rr1r2rrrrŚs0           ž / ż