File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_telnet.cpython-38.pyc
U
W[�g � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ ee j�G d
d � d ��ZG d d
� d
e
j�ZG dd� de
j�ZG dd� de j�ZG dd� de
j�ZdS )z$
Tests for L{twisted.conch.telnet}.
� )�absolute_import�division)� implementer)� verifyObject)�defer)�telnet)�unittest)�
proto_helpers)� iterbytesc @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )� TestProtocol� c C s. d| _ g | _g | _g | _g | _g | _g | _d S )N� )�data�subcmd�calls� enabledLocal�
enabledRemote�
disabledLocal�disabledRemote��selfr r �@/usr/lib/python3/dist-packages/twisted/conch/test/test_telnet.py�__init__ s zTestProtocol.__init__c sL i }|_ � j|d< |j�� }|_dD ] }|f� fdd� |tt|�< q(d S )N� ) �NOP�DM�BRK�IP�AO�AYT�EC�EL�GAc s � j �|�S �N)r �append)�arg�cmdr r r �<lambda>- r
z-TestProtocol.makeConnection.<locals>.<lambda>)ZnegotiationMap�neg_TEST_COMMAND�
commandMap�copy�getattrr )r � transport�dr& r r r �makeConnection' s
zTestProtocol.makeConnectionc C s | j |7 _ d S r# )r �r r r r r � dataReceived0 s zTestProtocol.dataReceivedc C s d S r# r )r �reasonr r r �connectionLost4 s zTestProtocol.connectionLostc C s
|| _ d S r# )r )r Zpayloadr r r r( 8 s z TestProtocol.neg_TEST_COMMANDc C s || j kr| j�|� dS dS �NTF)�localEnableabler r$ �r Zoptionr r r � enableLocal< s
zTestProtocol.enableLocalc C s | j �|� d S r# )r r$ r5 r r r � disableLocalC s zTestProtocol.disableLocalc C s || j kr| j�|� dS dS r3 )�remoteEnableabler r$ r5 r r r � enableRemoteG s
zTestProtocol.enableRemotec C s | j �|� d S r# )r r$ r5 r r r �
disableRemoteN s zTestProtocol.disableRemoteN)�__name__�
__module__� __qualname__r4 r8 r r. r0 r2 r( r6 r7 r9 r: r r r r r s r c @ s e Zd Zdd� ZdS )�InterfacesTestsc C s t �� }tt j|� dS )zO
L{telnet.TelnetProtocol} implements L{telnet.ITelnetProtocol}
N)r ZTelnetProtocolr �ITelnetProtocol)r �pr r r �test_interfaceT s z InterfacesTests.test_interfaceN)r; r< r= rA r r r r r> S s r> c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zg g g g fd$d%�Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJS )K�TelnetTransportTestsz.
Tests for L{telnet.TelnetTransport}.
c C s( t �t�| _t�� | _| j�| j� d S r# )r ZTelnetTransportr r@ r �StringTransport�tr. r r r r �setUpa s
zTelnetTransportTests.setUpc C sF | j j}ddddddg}|D ]}| j �|� q | �|jd�|�� d S )Ns here are some bytes la la las some more arrive heres lots of bytes to play withs la la las ta de das dumr
)r@ �protocolr0 � assertEqualr �join�r �h�L�br r r �testRegularBytesg s �z%TelnetTransportTests.testRegularBytesc C s� | j j}ddddg}|D ]}| j �|� q| �|j|d d d� d |d d d� d |d
d d� d |d d d� d � d S ) Ns here is the first line
s here is the second line
s here is the third line
s here is the last line
r ����
� �
� � )r@ rF r0 rG r rI r r r �testNewlineHandlingy s* �������z(TelnetTransportTests.testNewlineHandlingc C sH | j j}dddg}|D ]}| j �|� q| �|jd�|��dd�� d S )Ns* here are some bytes�� with an embedded IACs& and here is a test of a border escape�s � did you get that IAC?r
s ��� �)r@ rF r0 rG r rH �replacerI r r r �
testIACEscape� s �z"TelnetTransportTests.testIACEscapec C sn | j j}tjtt|� }dd| d g}|D ]}| j �|� q,| �|j|g� | �|jd� |��
|d�� d S )Ns Here's some bytes, tra la las But ono!s
an interruptr
) r@ rF r �IACr+ r0 rG r r rH rV )r ZcmdNamerJ r& rK rL r r r �_simpleCommandTest� s
�z'TelnetTransportTests._simpleCommandTestc C s | � d� d S )Nr �rY r r r r �
testInterrupt� s z"TelnetTransportTests.testInterruptc C s | � d� d S )Nr rZ r r r r �testNoOperation� s z$TelnetTransportTests.testNoOperationc C s | � d� d S )Nr rZ r r r r � testDataMark� s z!TelnetTransportTests.testDataMarkc C s | � d� d S )Nr rZ r r r r � testBreak� s z TelnetTransportTests.testBreakc C s | � d� d S )Nr rZ r r r r �testAbortOutput� s z$TelnetTransportTests.testAbortOutputc C s | � d� d S )Nr rZ r r r r �testAreYouThere� s z$TelnetTransportTests.testAreYouTherec C s | � d� d S )Nr rZ r r r r �testEraseCharacter� s z'TelnetTransportTests.testEraseCharacterc C s | � d� d S )Nr! rZ r r r r �
testEraseLine� s z"TelnetTransportTests.testEraseLinec C s | � d� d S )Nr" rZ r r r r � testGoAhead� s z TelnetTransportTests.testGoAheadc C s| | j j}tjtj d tj tj }d| dg}|D ]}| j �|� q4| �|jd� |��
|d�� | �|j t t
d��� d S )Ns hello worlds These are some bytes but soons there will be some morer
s hello world)r@ rF r rX �SB�SEr0 rG r rH rV r �listr
�r rJ r& rK rL r r r �testSubnegotiation� s � z'TelnetTransportTests.testSubnegotiationc C s� | j j}tjtj d tj tj tj }d| d dg}|D ]}| j �|� q>| �|jd� |��
|d�� | �|j tjg� d S )Nr s Some bytes are heres and herer
) r@ rF r rX rd re r0 rG r rH rV r rg r r r � testSubnegotiationWithEmbeddedSE� s"
����
� z5TelnetTransportTests.testSubnegotiationWithEmbeddedSEc C s� t jt j d t j d t j t j }tt|��D ]�}t� }| j_|� | j� |d |� ||d � }}d| |d g}|D ]}| j�
|� q~| � |j d�
|��|d�� | � |jt jgttd�� � q2d S )Nr s hellos
first parts last partr
)r rX rd re �range�lenr r@ rF r. r0 rG r rH rV r rf r
)r r& �irJ �arL rK r r r r �testBoundarySubnegotiation� s,
����� � z/TelnetTransportTests.testBoundarySubnegotiationc C s< | � |j|� | � |j|� | � |j|� | � |j|� d S r# )rG r r r r )r �o�eL�eR�dL�dRr r r �_enabledHelper s z#TelnetTransportTests._enabledHelperc C sr t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j �
� t jt j d � | � | jj� d S �Nr s surrounding bytess to spice things upr
)
r rX �WILLr@ r0 rG rF r rV rD �value�DONTrt �r r& r r r r �testRefuseWill s z#TelnetTransportTests.testRefuseWillc C sr t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j �
� t jt j d � | � | jj� d S ru )
r rX �DOr@ r0 rG rF r rV rD rw �WONTrt ry r r r � testRefuseDo s z!TelnetTransportTests.testRefuseDoc C sh t jt j d }d| d }| jj}d|_| j�|� | �| j� � t jt j
d � | j |dgd� d S )N� � paddings trailer)r~ )rp ) r rX r{ r@ rF r4 r0 rG rD rw rv rt �r r& r rJ r r r � testAcceptDo* s z!TelnetTransportTests.testAcceptDoc C sh t jt j d }d| d }| jj}d|_| j�|� | �| j� � t jt j
d � | j |dgd� d S )N� �s headerr )r� �rq ) r rX rv r@ rF r8 r0 rG rD rw r{ rt r� r r r �testAcceptWill8 s z#TelnetTransportTests.testAcceptWillc C s� t jt j d }| j�d�}d|j_d| }| j�|� | �| jj j
|� |d�� | �| j �
� t jt j d � | �|jjd� | j| jj dgd� d S )N� )�yess
fiddle deer
�no�rs )r rX r| r@ �getOptionState�him�stater0 rG rF r rV rD rw rx rt �r r& �sr r r r �testAcceptWontE s z#TelnetTransportTests.testAcceptWontc C s� t jt j d }| j�d�}d|j_d| }| j�|� | �| jj j
|� |d�� | �| j �
� t jt j d � | �|jjd� | j| jj dgd� d S )Nr� r� s fiddle dum r
r� )rr )r rX rx r@ r� �usr� r0 rG rF r rV rD rw r| rt r� r r r �testAcceptDontZ s z#TelnetTransportTests.testAcceptDontc C sf t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j �
� d� | � | jj� d S �N� G�
dum de dum� tra la lar
) r rX r| r@ r0 rG rF r rV rD rw rt ry r r r �testIgnoreWonto s z#TelnetTransportTests.testIgnoreWontc C sf t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j �
� d� | � | jj� d S r� ) r rX rx r@ r0 rG rF r rV rD rw rt ry r r r �testIgnoreDont| s z#TelnetTransportTests.testIgnoreDontc C sz t jt j d }| j�d�}d|j_d| d }| j�|� | �| jj j
|� |d�� | �| j �
� d� | �| jj � d S �N� Vr� r� r� r
)r rX rv r@ r� r� r� r0 rG rF r rV rD rw rt r� r r r �testIgnoreWill� s z#TelnetTransportTests.testIgnoreWillc C sz t jt j d }| j�d�}d|j_d| d }| j�|� | �| jj j
|� |d�� | �| j �
� d� | �| jj � d S r� )r rX r{ r@ r� r� r� r0 rG rF r rV rD rw rt r� r r r � testIgnoreDo� s z!TelnetTransportTests.testIgnoreDoc sv �j �d�}�j j� d� _���j�� tjtj d � �j �
tjtj d � |� �jd� |� � �fdd�� |S )N� B�r� Tc s �j � dgd�S )Nr� r� )rt ��_�rJ r r r r' � r
z@TelnetTransportTests.testAcceptedEnableRequest.<locals>.<lambda>)
r@ �dorF r8 rG rD rw r rX r{ r0 rv � addCallback�r r- r r� r �testAcceptedEnableRequest� s z.TelnetTransportTests.testAcceptedEnableRequestc s� d�j j_�j �d�}���j�� tjtj d � �j �
d�� ��� j j d� ��� j
j d� ��� j j� ��� j
j� �j �tjtj d � ��|tj�}|��fdd�� |�� �fdd�� |S )z�
If the peer refuses to enable an option we request it to enable, the
L{Deferred} returned by L{TelnetProtocol.do} fires with an
L{OptionRefused} L{Failure}.
r� r� r� c s � � � jj�S r# �rt r@ rF �Zignoredr r r r' � r
z@TelnetTransportTests.test_refusedEnableRequest.<locals>.<lambda>c s �� � jj�S r# )� assertFalser� � negotiatingr� �r� r r r r' � r
)r@ rF r8 r� rG rD rw r rX r{ r� r� r� r� �
assertTruer� r� r0 r| �
assertFailure�
OptionRefusedr� r� r r� r �test_refusedEnableRequest� s
�z.TelnetTransportTests.test_refusedEnableRequestc s� d�j j_�j �d�}���j�� tjtj d � �j �
d�� ��� j j d� ��� j
j d� ��� j j� ��� j
j� �j �tjtj d � ��|tj�}|��fdd�� |�� �fdd�� |S )z�
If the peer refuses to allow us to enable an option, the L{Deferred}
returned by L{TelnetProtocol.will} fires with an L{OptionRefused}
L{Failure}.
r� r� r� c s � � � jj�S r# r� r� r r r r' � r
z>TelnetTransportTests.test_refusedEnableOffer.<locals>.<lambda>c s �� � jj�S r# )r� r� r� r� r� r r r' � r
)r@ rF r4 ZwillrG rD rw r rX rv r� r� r� r� r� r� r� r0 rx r� r� r� r� r r� r �test_refusedEnableOffer� s
�z,TelnetTransportTests.test_refusedEnableOfferc sz � j �d�}d|j_� j �d�}� �� j�� tj tj
d � � j � tj tj d � |�
� jd� |�
� fdd�� |S )Nr� r� Tc s � j � jjdgd�S )Nr� r� r� r� r r r r' s
�zATelnetTransportTests.testAcceptedDisableRequest.<locals>.<lambda>)r@ r� r� r� �dontrG rD rw r rX rx r0 r| r� �r r� r- r r r �testAcceptedDisableRequest s z/TelnetTransportTests.testAcceptedDisableRequestc sd � j �d�}d|j_� j �d� � fdd�}� fdd�}� fdd�}|d �}|�|� |�|� |S ) N� $r� c s � j �d�}� �|tj�S �Nr� )r@ r� r� r �AlreadyNegotiating��xr- r r r �_do s zITelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._doc s � j �d�}� �|tj�S r� )r@ r� r� r r� r� r r r �_dont" s zKTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._dontc s~ � j �tjtj d � � j� j jdgd� d� j j_� j �d�}� j �tjtj d � |�
� j d� |�
� fdd�� |S )Nr� r� )r� Tc s � j � jjdgdgd�S )Nr� )rq rs r� r� r r r r' 0 s
�z^TelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final.<locals>.<lambda>) r@ r0 r rX r| rt rF r8 r� rv r� rG r� r r r �_final' s
zLTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final)r@ r� r� r� r� r� )r r� r� r� r� r- r r r �'testNegotiationBlocksFurtherNegotiation s
z<TelnetTransportTests.testNegotiationBlocksFurtherNegotiationc C s | j �d�}| �|tj�S )N� �)r@ r� r� r ZAlreadyDisabledr� r r r �#testSuperfluousDisableRequestRaises; s z8TelnetTransportTests.testSuperfluousDisableRequestRaisesc C s. | j �d�}d|j_| j �d�}| �|tj�S )Nr� r� )r@ r� r� r� r� r� r ZAlreadyEnabledr� r r r �"testSuperfluousEnableRequestRaisesA s z7TelnetTransportTests.testSuperfluousEnableRequestRaisesc C sx | j �d�}| j �d�}| j �d�}G dd� dt�}| j �|d�� | �||�}| �||�}| �||�}t�|||g�S )Nr � #� 4c @ s e Zd ZdS )zLTelnetTransportTests.testLostConnectionFailsDeferreds.<locals>.TestExceptionN)r; r< r= r r r r �
TestExceptionN s r� zTotal failure!)r@ r� � Exceptionr2 r� r Z
gatherResults)r Zd1Zd2Zd3r� r r r � testLostConnectionFailsDeferredsI s z5TelnetTransportTests.testLostConnectionFailsDeferredsN)(r; r< r= �__doc__rE rM rT rW rY r[ r\ r] r^ r_ r` ra rb rc rh ri rn rt rz r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rB ] sJ
&rB c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
TestTelnetzP
A trivial extension of the telnet protocol class useful to unit tests.
c C s t j�| � g | _d S r# )r �Telnetr �eventsr r r r r ] s zTestTelnet.__init__c C s | j �d|f� dS )z:
Record the given data in C{self.events}.
�bytesN�r� r$ r/ r r r �applicationDataReceivedb s z"TestTelnet.applicationDataReceivedc C s | j �d||f� dS )z=
Record the given command in C{self.events}.
�commandNr� �r r� r r r r �unhandledCommandi s zTestTelnet.unhandledCommandc C s | j �d||f� dS )zL
Record the given subnegotiation command in C{self.events}.
� negotiateNr� r� r r r �unhandledSubnegotiationp s z"TestTelnet.unhandledSubnegotiationN)r; r< r= r� r r� r� r� r r r r r� Y s
r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#� TelnetTestsz�
Tests for L{telnet.Telnet}.
L{telnet.Telnet} implements the TELNET protocol (RFC 854), including option
and suboption negotiation, and option state tracking.
c C s t � | _dS )zM
Create an unconnected L{telnet.Telnet} to be used by tests.
N)r� rF r r r r rE s zTelnetTests.setUpc C s | � | j�d�� dS )z�
L{telnet.Telnet.enableLocal} should reject all options, since
L{telnet.Telnet} does not know how to implement any options.
� N)r� rF r6 r r r r �test_enableLocal� s z TelnetTests.test_enableLocalc C s | � | j�d�� dS )z�
L{telnet.Telnet.enableRemote} should reject all options, since
L{telnet.Telnet} does not know how to implement any options.
r� N)r� rF r9 r r r r �test_enableRemote� s z TelnetTests.test_enableRemotec C s | � t| jjd� dS )a
It is an error for L{telnet.Telnet.disableLocal} to be called, since
L{telnet.Telnet.enableLocal} will never allow any options to be enabled
locally. If a subclass overrides enableLocal, it must also override
disableLocal.
r� N)� assertRaises�NotImplementedErrorrF r7 r r r r �test_disableLocal� s z TelnetTests.test_disableLocalc C s | � t| jjd� dS )a
It is an error for L{telnet.Telnet.disableRemote} to be called, since
L{telnet.Telnet.enableRemote} will never allow any options to be
enabled remotely. If a subclass overrides enableRemote, it must also
override disableRemote.
r� N)r� r� rF r: r r r r �test_disableRemote� s z TelnetTests.test_disableRemotec C s6 t �� }| j�|� | j�dd� | �|�� d� dS )z�
L{telnet.Telnet.requestNegotiation} formats the feature byte and the
payload bytes into the subnegotiation format and sends them.
See RFC 855.
� s s ����N�r rC rF r. ZrequestNegotiationrG rw �r r, r r r �test_requestNegotiation� s �z#TelnetTests.test_requestNegotiationc C s6 t �� }| j�|� | j�dd� | �|�� d� dS )z�
If the payload for a subnegotiation includes I{IAC}, it is escaped by
L{telnet.Telnet.requestNegotiation} with another I{IAC}.
See RFC 855.
r� rU s ������Nr� r� r r r �!test_requestNegotiationEscapesIAC� s �z-TelnetTests.test_requestNegotiationEscapesIACc G s, g }| j _| j �|� | �|t|�� dS )z
Pass the given bytes to the protocol's C{dataReceived} method and
assert that the given events occur.
N)rF r� r0 rG rf )r r ZexpectedZreceivedr r r �_deliver� s zTelnetTests._deliverc C s | � dd� dS )zc
One application-data byte in the default state gets delivered right
away.
� a)r� r� N�r� r r r r �test_oneApplicationDataByte� s z'TelnetTests.test_oneApplicationDataBytec C s | � dd� dS )za
Two application-data bytes in the default state get delivered
together.
� bc)r� r� Nr� r r r r � test_twoApplicationDataBytes� s z(TelnetTests.test_twoApplicationDataBytesc C s | � dtj d� dS )z~
Three application-data bytes followed by a control byte get
delivered, but the control byte doesn't.
� def)r� r� N�r� r rX r r r r � test_threeApplicationDataBytes� s z*TelnetTests.test_threeApplicationDataBytesc C s, | � tj� | � tjd dtjd f� dS )zy
IAC in the escaped state gets delivered and so does another
application-data byte following it.
� gr� Nr� r r r r �test_escapedControl� s z TelnetTests.test_escapedControlc C s� | � d� | � dd� | � dd� | � d� | � dd� | � dd� | � d� | � dd � | � d
d � | � d� | � tjtj d d dtj d f� d
S )a�
A carriage return only puts the protocol into the newline state. A
linefeed in the newline state causes just the newline to be
delivered. A nul in the newline state causes a carriage return to
be delivered. An IAC in the newline state causes a carriage return
to be delivered and puts the protocol into the escaped state.
Anything else causes a carriage return and that thing to be
delivered.
rQ rO )r� rO s
r� )r� rQ s
r� )r� �
ar� � xr� Nr� r r r r �test_carriageReturn� s
�z TelnetTests.test_carriageReturnc C s& | � dtj tj ddtjdf� dS )zt
Application bytes received before a command are delivered before the
command is processed.
r� )r� r� r� N)r� r rX r r r r r �'test_applicationDataBeforeSimpleCommand s
�z3TelnetTests.test_applicationDataBeforeSimpleCommandc C s2 i | j _| �dtj tj d ddtjdf� dS )z~
Application bytes received before a WILL/WONT/DO/DONT are delivered
before the command is processed.
� yr� )r� r� r� N)rF r) r� r rX rv r r r r �!test_applicationDataBeforeCommand s
�z-TelnetTests.test_applicationDataBeforeCommandc C s6 | � dtj tj d tj tj ddddgf� dS )z�
Application bytes received before a subnegotiation command are
delivered before the negotiation is processed.
� zs Qx)r� r� r� � Qr� N)r� r rX rd re r r r r �(test_applicationDataBeforeSubnegotiation$ s
�z4TelnetTests.test_applicationDataBeforeSubnegotiationN)r; r< r= r� rE r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� x s"
r� N)r� Z
__future__r r Zzope.interfacer Zzope.interface.verifyr Ztwisted.internetr Z
twisted.conchr Z
twisted.trialr Z twisted.testr Ztwisted.python.compatr
r? r ZTestCaser> rB r� r� r� r r r r �<module> s$ ;