File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_telnet.cpython-310.pyc
o
�b�h � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ eej�G d d
� d
��ZG d d � d ej�ZG d
d� dej�ZG dd� dej�ZG dd� dej�ZdS )z$
Tests for L{twisted.conch.telnet}.
� )� implementer)� verifyObject)�telnet)�defer)� iterbytes)�
proto_helpers)�unittestc @ st e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS )� 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|�< qd S )N� )
�EOR�NOP�DM�BRK�IP�AO�AYT�EC�EL�GAc s � j �|�S �N)r �append)�arg�cmdr r
r �<lambda>) s z-TestProtocol.makeConnection.<locals>.<lambda>)�negotiationMap�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 � dataReceived+ s zTestProtocol.dataReceivedc C � d S r" r
)r �reasonr
r
r �connectionLost. s zTestProtocol.connectionLostc C s
|| _ d S r" )r
)r �payloadr
r
r r( 1 s
z TestProtocol.neg_TEST_COMMANDc C � || j v r
| j�|� dS dS �NTF)�localEnableabler r# �r �optionr
r
r � enableLocal4 �
zTestProtocol.enableLocalc C � | j �|� d S r" )r r# r8 r
r
r � disableLocal: � zTestProtocol.disableLocalc C r5 r6 )�remoteEnableabler r# r8 r
r
r � enableRemote= r; zTestProtocol.enableRemotec C r< r" )r r# r8 r
r
r �
disableRemoteC r> zTestProtocol.disableRemotec C r1 r" r
r r
r
r �connectionMadeF � zTestProtocol.connectionMadec C r1 r" r
)r �command�argumentr
r
r �unhandledCommandJ rC z TestProtocol.unhandledCommandc C r1 r" r
�r rD r r
r
r �unhandledSubnegotiationN rC z$TestProtocol.unhandledSubnegotiationN)�__name__�
__module__� __qualname__r7 r? r r. r0 r3 r( r: r= r@ rA rB rF rH 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 �TelnetProtocolr �ITelnetProtocol)r �pr
r
r �test_interfaceT s z InterfacesTests.test_interfaceN)rI rJ rK rP r
r
r
r rL S s rL c @ sB 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#� Zd$d%� Zg g g g fd&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'dJdK� Z(dLS )M�TelnetTransportTestsz.
Tests for L{telnet.TelnetTransport}.
c C s( t �t�| _t�� | _| j�| j� d S r" )r �TelnetTransportr rO r �StringTransport�tr. r r
r
r �setUpa s
zTelnetTransportTests.setUpc C s> | j j}g d�}|D ]}| j �|� q
| �|jd�|�� d S )N)s here are some bytes la la las some more arrive heres lots of bytes to play withs la la las ta de das dumr )rO �protocolr0 � assertEqualr �join�r �h�L�br
r
r �testRegularBytesf s
z%TelnetTransportTests.testRegularBytesc C s� | j j}g d�}|D ]}| j �|� q
| �|j|d d d� d |d d d� d |d d d� d |d d d� d � d S ) N)s here is the first line
s here is the second line
s here is the third line
s here is the last line
r ����
� �
� � )rO rV r0 rW r rY r
r
r �testNewlineHandlingy s, ��������z(TelnetTransportTests.testNewlineHandlingc C sF | j j}g d�}|D ]}| j �|� q
| �|jd�|��dd�� d S )N)s* 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 ��� �)rO rV r0 rW r rX �replacerY 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 ) rO rV r �IACr+ r0 rW r r rX rf )r �cmdNamerZ r% r[ r\ r
r
r �_simpleCommandTest� s z'TelnetTransportTests._simpleCommandTestc C � | � d� d S )Nr �rj r r
r
r �
testInterrupt� � z"TelnetTransportTests.testInterruptc C rk )Nr rl r r
r
r �testEndOfRecord� rn z$TelnetTransportTests.testEndOfRecordc C rk )Nr rl r r
r
r �testNoOperation� rn z$TelnetTransportTests.testNoOperationc C rk )Nr rl r r
r
r � testDataMark� rn z!TelnetTransportTests.testDataMarkc C rk )Nr rl r r
r
r � testBreak� rn z TelnetTransportTests.testBreakc C rk )Nr rl r r
r
r �testAbortOutput� rn z$TelnetTransportTests.testAbortOutputc C rk )Nr rl r r
r
r �testAreYouThere� rn z$TelnetTransportTests.testAreYouTherec C rk )Nr rl r r
r
r �testEraseCharacter� rn z'TelnetTransportTests.testEraseCharacterc C rk )Nr rl r r
r
r �
testEraseLine� rn z"TelnetTransportTests.testEraseLinec C rk )Nr! rl r r
r
r � testGoAhead� rn z TelnetTransportTests.testGoAheadc C s| | j j}tjtj d tj tj }d| dg}|D ]}| j �|� q| �|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)rO rV r rh �SB�SEr0 rW r rX rf r
�listr �r rZ r% r[ r\ 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 ) rO rV r rh rx ry r0 rW r rX rf r
r{ r
r
r � testSubnegotiationWithEmbeddedSE� s " z5TelnetTransportTests.testSubnegotiationWithEmbeddedSEc C s� t jt j d t j d t j t j }tt|��D ]L}t� }| j_|� | j� |d |� ||d � }}d| |d g}|D ]}| j�
|� q?| � |j d�
|��|d�� | � |jt jgttd�� � qd S )Nr s hellos
first parts last partr )r rh rx ry �range�lenr rO rV r. r0 rW r rX rf r
rz r )r r% �irZ �ar\ r[ r r
r
r �testBoundarySubnegotiation� s0 �������
�z/TelnetTransportTests.testBoundarySubnegotiationc C s< | � |j|� | � |j|� | � |j|� | � |j|� d S r" )rW r r r r )r �o�eL�eR�dL�dRr
r
r �_enabledHelper
s z#TelnetTransportTests._enabledHelperc C �r 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 rh �WILLrO r0 rW rV r rf rT �value�DONTr� �r r% r r
r
r �testRefuseWill � z#TelnetTransportTests.testRefuseWillc C r� r� )
r rh �DOrO r0 rW rV r rf rT r� �WONTr� r� r
r
r � testRefuseDo r� z!TelnetTransportTests.testRefuseDoc C �h t jt j d }d| d }| jj}d|_| j�|� | �| j� � t jt j
d � | j |dgd� d S )N� � paddings trailer)r� )r� ) r rh r� rO rV r7 r0 rW rT r� r� r� �r r% r rZ r
r
r � testAcceptDo& s z!TelnetTransportTests.testAcceptDoc C r� )N� �s headerr� )r� �r� ) r rh r� rO rV r? r0 rW rT r� r� r� r� r
r
r �testAcceptWill3 s z#TelnetTransportTests.testAcceptWillc C � 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�r� )r rh r� rO �getOptionState�him�stater0 rW rV r rf rT r� r� r� �r r% �sr r
r
r �testAcceptWont? � z#TelnetTransportTests.testAcceptWontc C r� )Nr� r� s fiddle dum r r� )r� )r rh r� rO r� �usr� r0 rW rV r rf rT r� r� r� r� r
r
r �testAcceptDontS r� z#TelnetTransportTests.testAcceptDontc C �f t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j �
� d� | � | jj� d S �N� G�
dum de dum� tra la lar ) r rh r� rO r0 rW rV r rf rT r� r� r� r
r
r �testIgnoreWontg s z#TelnetTransportTests.testIgnoreWontc C r� r� ) r rh r� rO r0 rW rV r rf rT r� r� r� r
r
r �testIgnoreDonts s z#TelnetTransportTests.testIgnoreDontc C �z 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 rh r� rO r� r� r� r0 rW rV r rf rT r� r� r� r
r
r �testIgnoreWill� � z#TelnetTransportTests.testIgnoreWillc C r� r� )r rh r� rO r� r� r� r0 rW rV r rf rT r� r� r� r
r
r � testIgnoreDo� r� 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� )r� ��_�rZ r r
r r& � s z@TelnetTransportTests.testAcceptedEnableRequest.<locals>.<lambda>)
rO �dorV r? rW rT r� r rh r� r0 r� � addCallback�r r- r
r� r �testAcceptedEnableRequest� s z.TelnetTransportTests.testAcceptedEnableRequestc �� 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 � � � � jj�S r" �r� rO rV ��ignoredr r
r r& � � z@TelnetTransportTests.test_refusedEnableRequest.<locals>.<lambda>c � �� � jj�S r" )� assertFalser� � negotiatingr� �r� r r
r r& � r� )rO rV r? r� rW rT r� r rh r� r� r� r� r� �
assertTruer� r� r0 r� �
assertFailure�
OptionRefusedr� r� r
r� r �test_refusedEnableRequest� �
z.TelnetTransportTests.test_refusedEnableRequestc r� )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 r� r" r� r� r r
r r& � r� z>TelnetTransportTests.test_refusedEnableOffer.<locals>.<lambda>c r� r" )r� r� r� r� r� r
r r& � r� )rO rV r7 �willrW rT r� r rh r� r� r� r� r� r� r� r� r0 r� r� r� r� r� r
r� r �test_refusedEnableOffer� r� 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>)rO r� r� r� �dontrW rT r� r rh r� 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 � � j �d�}� �|tj�S �Nr� )rO r� r� r �AlreadyNegotiating��xr- r r
r �_do
� zITelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._doc r� r� )rO r� r� r r� r� r r
r �_dont r� 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� )r� r� r� r� r r
r r& s �z^TelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final.<locals>.<lambda>) rO r0 r rh r� r� rV r? r� r� r� rW r� r r
r �_final s
�zLTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final)rO 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� �)rO r� r� r �AlreadyDisabledr� r
r
r �#testSuperfluousDisableRequestRaises' s z8TelnetTransportTests.testSuperfluousDisableRequestRaisesc C s. | j �d�}d|j_| j �d�}| �|tj�S )Nr� r� )rO r� r� r� r� r� r �AlreadyEnabledr� r
r
r �"testSuperfluousEnableRequestRaises, 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)rI rJ rK r
r
r
r �
TestException8 s r� zTotal failure!)rO r� � Exceptionr3 r� r �
gatherResults)r �d1�d2�d3r� r
r
r � testLostConnectionFailsDeferreds3 s z5TelnetTransportTests.testLostConnectionFailsDeferredsN))rI rJ rK �__doc__rU r] rd rg rj rm ro rp rq rr rs rt ru rv rw r| r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r rQ \ sN
% rQ 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 H 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 �applicationDataReceivedL s z"TestTelnet.applicationDataReceivedc C � | j �d||f� dS )z=
Record the given command in C{self.events}.
rD Nr� rG r
r
r rF R � zTestTelnet.unhandledCommandc C r� )zL
Record the given subnegotiation command in C{self.events}.
� negotiateNr� rG r
r
r rH X r� z"TestTelnet.unhandledSubnegotiationN)rI rJ rK r� r r� rF rH r
r
r
r r� C 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� rV r r
r
r rU g s zTelnetTests.setUpc C � | � | 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� rV r: r r
r
r �test_enableLocalm � z TelnetTests.test_enableLocalc C r� )z�
L{telnet.Telnet.enableRemote} should reject all options, since
L{telnet.Telnet} does not know how to implement any options.
r� N)r� rV r@ r r
r
r �test_enableRemotet r� z TelnetTests.test_enableRemotec C � | � 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�NotImplementedErrorrV r= r r
r
r �test_disableLocal{ � z TelnetTests.test_disableLocalc C r� )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� rV rA r r
r
r �test_disableRemote� r z TelnetTests.test_disableRemotec C �6 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 rS rV r. �requestNegotiationrW r� �r r, r
r
r �test_requestNegotiation� s �z#TelnetTests.test_requestNegotiationc C r )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 re 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)rV r� r0 rW rz )r r �expected�receivedr
r
r �_deliver� s zTelnetTests._deliverc C � | � 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� � z'TelnetTests.test_oneApplicationDataBytec C r
)za
Two application-data bytes in the default state get delivered
together.
� bc)r� r Nr r r
r
r � test_twoApplicationDataBytes� r 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 rh r r
r
r � test_threeApplicationDataBytes� r� 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.
ra r_ )r� r_ s
r� )r� ra 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 rD N)r r rh 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 rD N)rV r) r r rh r� 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 rh rx ry r r
r
r �(test_applicationDataBeforeSubnegotiation s
�z4TelnetTests.test_applicationDataBeforeSubnegotiationN)rI rJ rK r� rU r� r� r r r r r r r r r r r r r! r
r
r
r r� _ s$ r� N)r� �zope.interfacer �zope.interface.verifyr �
twisted.conchr �twisted.internetr �twisted.python.compatr � twisted.testr �
twisted.trialr rN r �TestCaserL rQ r� r� r� r
r
r
r �<module> s" > j