← Back
Editing: 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$ ;
Save File
Cancel