File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_channel.cpython-38.pyc
U
W[C/ � @ s� d Z ddlmZmZ ddlmZ z\ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ dd
lmZ d ZW n ek
r� d ZeZ
Y nX dd
lmZ ddlmZ G dd� de
�Zddd�Z G dd� dej �Z d S )z
Test ssh/channel.py.
� )�division�absolute_import)� verifyObject)�channel)�SSHTransportAddress)�SSHServerTransport)�
SSHService)�
interfaces)� IPv4Address)�StringTransportNzConch SSH not supported.)�unittest)�
intToBytesc @ s8 e Zd ZdZdd� 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.
c C s i | _ i | _i | _d S )N)�data�extData�closes��self� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_channel.py�__init__) s zMockConnection.__init__c C s dS )z,
Return our logging prefix.
r r r r r r � logPrefix/ s zMockConnection.logPrefixc C s | j �|g ��|� dS )z'
Record the sent data.
N)r �
setdefault�append)r r r r r r �sendData6 s zMockConnection.sendDatac C s | j �|g ��||f� dS )z0
Record the sent extended data.
N)r r r )r r �typer r r r �sendExtendedData= s z MockConnection.sendExtendedDatac C s d| j |< dS )z?
Record that the channel sent a close message.
TN)r )r r r r r � sendCloseD s zMockConnection.sendCloseN) �__name__�
__module__� __qualname__�__doc__r r r r r r r r r r s r c C s& 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 ZmakeConnectionr Z
setService)�servicer" r# Z transportr r r �connectSSHTransportL s �r% c @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )!� ChannelTestsz"
Tests for L{SSHChannel}.
c C s& 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�remoteMaxPackets channelN)r r( r �
SSHChannel�namer r r r �setUph s
�zChannelTests.setUpc C s | � ttj| j�� dS )zK
L{SSHChannel} instances provide L{interfaces.ITransport}.
N)�
assertTruer r Z
ITransportr r r r r �test_interfacet s zChannelTests.test_interfacec C s t j| 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( i i � r � � � � � � � N) r r* r( � assertEqualZlocalWindowSizeZlocalWindowLeftZlocalMaxPacket�remoteWindowLeftr) � assertIsNoner Zavatar)r �cZc2r r r � test_init{ s$
zChannelTests.test_initc C s. | � 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%<SSHChannel channel (lw 131072 rw 0)>r/ �Z localWindowz <SSHChannel None (lw 1 rw 0)>N)r6 �strr r* r r r r �test_str� s ��zChannelTests.test_strc C s. | � | j�� d� | � tjdd��� d� dS )z
Test that bytes(SSHChannel) works, gives the channel name and
local and remote windows at a glance..
s% <SSHChannel channel (lw 131072 rw 0)>r/ r; s <SSHChannel None (lw 1 rw 0)>N)r6 r � __bytes__r* r r r r �
test_bytes� s ��zChannelTests.test_bytesc C s | � | 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)r6 r r r r r r �test_logPrefix� s zChannelTests.test_logPrefixc s dg� � fdd�}|| j _| j �d� | j �dd� | j �d� | �| j jd� | �| j j� | �� d � | �| j j d � | �| j
j | j dg� | �| j j g � | �| j
j
| j d
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 s d� d< d S �NTr r r ��cbr r �stubStartWriting� s z:ChannelTests.test_addWindowBytes.<locals>.stubStartWriting� testr/ �2 �* r � )r/ rE � sP aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaN)r Z startWriting�write�
writeExtended�addWindowBytesr6 r7 r- �
areWriting�bufr( r �extBufr � assertFalse�loseConnection)r rD r rB r �test_addWindowBytes� s(
z ChannelTests.test_addWindowBytesc C sL dd� | j _| �| j �dd�� | �| j �dd�� | �| j �dd�� dS )zn
Test that requestReceived handles requests by dispatching them to
request_* methods.
c S s | dkS )NrH r )r r r r �<lambda>� rH z3ChannelTests.test_requestReceived.<locals>.<lambda>s test-methodrH � as
bad-methodN)r Zrequest_test_methodr- ZrequestReceivedrP r r r r �test_requestReceived� s z!ChannelTests.test_requestReceivedc C s* | � | jj� | j�� | �| jj� dS )zM
Test that the default closeReceieved closes the connection.
N)rP r �closingZ
closeReceivedr- r r r r �test_closeReceieved� s
z ChannelTests.test_closeReceievedc s. dg� � fdd�}|| j _| j �d� | j �d� | �| j j� | �� d � | j �d� | j �d� | jj| j }| � |d dg� | � | j j
d
� | j �d � | � |d dd d
g� | � | j j
d� d� d< | j �d� | �| j j� | �� d � | � |d dd d
dg� | � | j j d� | � | j j
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.
Fc s d� d< d S rA r r rB r r �stubStopWriting� s z0ChannelTests.test_write.<locals>.stubStopWriting� drT r rI s ta� da� � 12345678901�
1234567890� 1r3 � 123456� 12345� 6N) r � stopWritingrJ rP rM r- rL r( r r6 r7 rN �r rX r r rB r �
test_write� s, zChannelTests.test_writec sR dg� � fdd�}|| j _| j �dd� | j �dd� | j �dd� | �| j j� | �� d � | j �d
� | j �dd� | jj| j }| � |d d d
g� | � | j j
d� | j �dd� | � |d d d
ddg� | � | j j
d� d� d < | j �dd� | �| j j� | �� d � | � |d d d
dddg� | � | j j ddgg� | � | j j
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.
Fc s d� d< d S rA r r rB r r rX s z8ChannelTests.test_writeExtended.<locals>.stubStopWritingr/ rY rT r0 � tr rI )r/ rZ )r0 re )r0 rT r[ r1 r\ )r1 r] )r1 r^ r3 r2 r_ )r2 r` ra N) r rb rK rP rM r- rL r( r r6 r7 rO rc r rB r �test_writeExtended s< � �z ChannelTests.test_writeExtendedc C s> | j �d� | j �tttd��� | �| jj| j dg� dS )zS
Test that writeSequence is equivalent to write(''.join(sequece)).
rI r' s
0123456789N) r rL Z
writeSequence�mapr
�ranger6 r( r r r r r �test_writeSequence8 s z ChannelTests.test_writeSequencec C s� | 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.
s datar/ s datadatar2 � N)
r rJ rK rQ r8 r( r �getrL r- r r r r �test_loseConnectionA s
z ChannelTests.test_loseConnectionc C s6 t ddd�}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% r r( r6 r ZgetPeer)r Zpeerr r r � test_getPeerP s zChannelTests.test_getPeerc C s6 t ddd�}t| jj|d� | �t|�| j�� � dS )z�
L{SSHChannel.getHost} returns the same object as the underlying
transport's C{getHost} method returns.
rm z 127.0.0.1i90 )r$ r" N)r
r% r r( r6 r ZgetHost)r Zhostr r r � test_getHost[ s zChannelTests.test_getHostN)r r r r! �skipTest�skipr, r. r: r= r? r@ rR rU rW rd rf ri rl rn ro r r r r r& a s" #& r& )NN) r! Z
__future__r r Zzope.interface.verifyr Ztwisted.conch.sshr Ztwisted.conch.ssh.addressr Ztwisted.conch.ssh.transportr Ztwisted.conch.ssh.servicer Ztwisted.internetr Ztwisted.internet.addressr
Ztwisted.test.proto_helpersr rp � ImportError�objectZ
twisted.trialr Ztwisted.python.compatr
r r% ZTestCaser& r r r r �<module> s&
0