File: //usr/lib/python3/dist-packages/twisted/protocols/test/__pycache__/test_basic.cpython-38.pyc
U
W[�� � @ sz d Z ddlmZmZ ddlZddlZddlmZ ddlm Z ddl
m Z m Z ddl
mZ ddlmZ dd lmZ dd
lmZmZ dd lmZ dd lmZ d
ZG dd� dej�Z G dd� dej�Z G dd� dej �Z G dd� dej �Z!G dd� dej�Z"G dd� dej �Z#G dd� dej �Z$G d d � d �Z%G d d � d e%ej&�Z'G d d!� d!�Z(G d"d#� d#ej e(�Z)G d$d%� d%e(�Z*G d&d'� d'e+�Z,G d(d)� d)e%ej-�Z.G d*d+� d+ej e*e,�Z/G d,d-� d-e%ej0�Z1G d.d/� d/ej e*e,�Z2G d0d1� d1e1e+�Z3G d2d3� d3e2�Z4G d4d5� d5e%ej5�Z6G d6d7� d7ej e*e,�Z7G d8d9� d9e+�Z8G d:d;� d;ej�Z9G d<d=� d=ej �Z:G d>d?� d?ej;�Z<G d@dA� dAej;�Z=dS )Bz,
Test cases for L{twisted.protocols.basic}.
� )�division�absolute_importN)�BytesIO)� verifyObject)�_PY3� iterbytes)�unittest)�basic)�reflect)�protocol�task)� IProducer)�
proto_helpersz&All classes are new style on Python 3.c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�FlippingLineTesterzT
A line receiver that flips between line and raw data modes after one byte.
�
c C s
g | _ d S �N)�lines��self� r �C/usr/lib/python3/dist-packages/twisted/protocols/test/test_basic.py�__init__# s zFlippingLineTester.__init__c C s | j �|� | �� dS )z&
Set the mode to raw.
N)r �append�
setRawMode�r �liner r r � lineReceived' s z FlippingLineTester.lineReceivedc C s | � |dd� � dS )z,
Set the mode back to line.
� N)� setLineMode�r �datar r r �rawDataReceived/ s z"FlippingLineTester.rawDataReceivedN)�__name__�
__module__� __qualname__�__doc__� delimiterr r r! r r r r r s
r c @ sB e Zd ZdZdZdZddd�Zdd� Zd d
� Zd d � Z d
d� Z
dS )�
LineTestera�
A line receiver that parses data received and make actions on some tokens.
@type delimiter: C{bytes}
@ivar delimiter: character used between received lines.
@type MAX_LENGTH: C{int}
@ivar MAX_LENGTH: size of a line when C{lineLengthExceeded} will be called.
@type clock: L{twisted.internet.task.Clock}
@ivar clock: clock simulating reactor callLater. Pass it to constructor if
you want to use the pause/rawpause functionalities.
r �@ Nc C s
|| _ dS )z@
If given, use a clock to make callLater calls.
N)�clock)r r) r r r r G s zLineTester.__init__c C s
g | _ dS �z;
Create/clean data received on connection.
N��receivedr r r r �connectionMadeN s zLineTester.connectionMadec C s� | j �|� |dkr | �� n�|dkr@| �� | j�d| j� n�|dkrv| �� | �� | j �d� | j�d| j� nd|dkr�| �� nR|dd� dkr�t|dd� �| _ n.|�
d �r�| j � | d
� n|�
d �r�| j �
� dS ) z|
Receive line and make some action for some tokens: pause, rawpause,
stop, len, produce, unproduce.
� � pauser � rawpause� stopN� s len � produceF� unproduce)r, r r �pauseProducingr) Z callLater�resumeProducing�
stopProducing�int�length�
startswith� transportZregisterProducerZunregisterProducerr r r r r U s&
zLineTester.lineReceivedc C sZ |d| j � || j d� }}| j t|� | _ | jd | | jd< | j dkrV| �|� dS )zj
Read raw data, until the quantity specified by a previous 'len' line is
reached.
N���r )r9 �lenr, r )r r �restr r r r! o s
zLineTester.rawDataReceivedc C s. t |�| jd kr*| �|| jd d� � dS )z<
Adjust line mode when long lines received.
r N)r= �
MAX_LENGTHr r r r r �lineLengthExceeded{ s z LineTester.lineLengthExceeded)N) r" r# r$ r% r& r? r r- r r! r@ r r r r r' 7 s
r' c @ s( e Zd ZdZdZdZdd� Zdd� ZdS ) �LineOnlyTesterz)
A buffering line only receiver.
r r( c C s
g | _ dS r* r+ r r r r r- � s z LineOnlyTester.connectionMadec C s | j �|� dS )z%
Save received data.
N�r, r r r r r r � s zLineOnlyTester.lineReceivedN)r" r# r$ r% r& r? r- r r r r r rA � s
rA c
@ s� e Zd ZdZdZddddddd d
d d d
ddg
Zdd� ZdZdddgZedg Z dd� Z
dZ ddddd gZ ddddddgZ
d d � Zd Zddd gZd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4S )5�LineReceiverTestsz\
Test L{twisted.protocols.basic.LineReceiver}, using the C{LineTester}
wrapper.
s� len 10
0123456789len 5
1234
len 20
foo 123
0123456789
012345678len 0
foo 5
1234567890123456789012345678901234567890123456789012345678901234567890
len 1
as len 10�
0123456789s len 5s 1234
s len 20s foo 123s 0123456789
012345678s len 0s foo 5r. s 67890s len 1� ac C s� t dd�D ]r}t�� }t� }|�t�|�� t t| j�| d �D ](}| j|| |d | � }|� |� qB| �
| j |j � q
dS )zl
Test buffering for different packet size, checking received matches
expected data.
r �
N)
�ranger �StringIOWithoutClosingr' �makeConnectionr � FileWrapperr= �buffer� dataReceived� assertEqual�outputr, �r � packet_size�t�a�i�sr r r � test_buffer� s z LineReceiverTests.test_buffers! twiddle1
twiddle2
pause
twiddle3
s twiddle1s twiddle2r/ s twiddle3c C s� t dd�D ]�}t�� }t�� }t|�}|�t�|�� t t | j
�| d �D ](}| j
|| |d | � }|� |� qL| � | j
|j� |�d� | � | j|j� q
dS )zo
Test pause inside data receiving. It uses fake clock to see if
pausing/resuming work.
r rF r N)rG r rH r �Clockr' rI r rJ r= �pauseBufrL rM � pauseOutput1r, �advance� pauseOutput2�r rP rQ r) rR rS rT r r r � test_pausing� s
z LineReceiverTests.test_pausings/ twiddle1
twiddle2
len 5
rawpause
12345twiddle3
r0 s 12345c C s� t dd�D ]�}t�� }t�� }t|�}|�t�|�� t t | j
�| d �D ](}| j
|| |d | � }|� |� qL| � | j
|j� |�d� | � | j|j� q
dS )z7
Test pause inside raw date receiving.
r rF r N)rG r rH r rV r' rI r rJ r= � rawpauseBufrL rM �rawpauseOutput1r, rY �rawpauseOutput2r[ r r r �test_rawPausing� s
z!LineReceiverTests.test_rawPausings" twiddle1
twiddle2
stop
more
stuff
r1 c C s� t dd�D ]r}t�� }t� }|�t�|�� t t| j�| d �D ](}| j|| |d | � }|� |� qB| �
| j |j � q
dS )z-
Test stop inside producing.
r rF N)
rG r rH r' rI r rJ r= �stop_bufrL rM � stop_outputr, rO r r r �test_stopProducing� s z$LineReceiverTests.test_stopProducingc C sB t � }t�� }|�t�|�� |�d� | �|jddddg� dS )z6
Test produce/unproduce in receiving.
s& produce
hello world
unproduce
goodbye
r3 s hello worldr4 � goodbyeN) r' r rH rI r rJ rL rM r, )r rR rQ r r r �test_lineReceiverAsProducer s
�z-LineReceiverTests.test_lineReceiverAsProducerc C sh G dd� dt j�}|� }|�d� | �|jd� | �|jd� |�d� | �|jd� | �|jd� d S )
z�
L{LineReceiver.clearLineBuffer} removes all buffered data and returns
it as a C{bytes} and can be called from beneath C{dataReceived}.
c @ s e Zd Zdd� ZdS )z@LineReceiverTests.test_clearLineBuffer.<locals>.ClearingReceiverc S s || _ | �� | _d S r )r ZclearLineBufferr> r r r r r s zMLineReceiverTests.test_clearLineBuffer.<locals>.ClearingReceiver.lineReceivedN)r" r# r$ r r r r r �ClearingReceiver s rf s
foo
bar
baz� foos bar
bazs quux
s quuxr. N)r � LineReceiverrL rM r r> )r rf r r r r �test_clearLineBuffer s
z&LineReceiverTests.test_clearLineBufferc C sP t � }t�� }|�t�|�� t�� }|�d| � | � d| d�
|j �� dS )zC
Test switching modes many times on the same data.
s x
x� xr. N) r r rH rI r rJ �sys�getrecursionlimitrL rM �joinr )r �protor; �limitr r r �test_stackRecursion' s z%LineReceiverTests.test_stackRecursionc C sB t �� }t�� }|�|� |�d|jd d � | �|j� dS �zx
C{LineReceiver} disconnects the transport if it receives a line longer
than its C{MAX_LENGTH}.
rj r �
rN� r rh r �StringTransportrI rL r? �
assertTrue�
disconnecting�r rn r; r r r �test_maximumLineLength3 s
z(LineReceiverTests.test_maximumLineLengthc C s� t � }d|_t�� }|�|� d|jd }|�|� |�|jdd� � |�|jdd� | � | �|j� | � t
|j �d� | � ||j d � dS )a~
C{LineReceiver} doesn't disconnect the transport when it
receives a finished line as long as its C{MAX_LENGTH}, when
the second-to-last packet ended with a pattern that could have
been -- and turns out to have been -- the start of a
delimiter, and that packet causes the total input to exceed
C{MAX_LENGTH} + len(delimiter).
r2 rj r Nr< r ) r' r? r rt rI rL r& � assertFalserv rM r= r, )r rn rQ r r r r �&test_maximumLineLengthPartialDelimiter? s
z8LineReceiverTests.test_maximumLineLengthPartialDelimiterc C sX t �� }d|_t�� }|�|� |�d|j |jdt|j�d � � | � |j
� dS )aO
C{LineReceiver} doesn't disconnect the transport it if
receives a non-finished line whose length, counting the
delimiter, is longer than its C{MAX_LENGTH} but shorter than
its C{MAX_LENGTH} + len(delimiter). (When the first part that
exceeds the max is the beginning of the delimiter.)
s
rj Nr ) r rh r& r rt rI rL r? r= ry rv rw r r r �(test_notQuiteMaximumLineLengthUnfinishedV s
�z:LineReceiverTests.test_notQuiteMaximumLineLengthUnfinishedc C sF t �� }dd� |_t�� }|�|� |�� |�d�}| �|t � dS )zf
C{LineReceiver.dataReceived} forwards errors returned by
C{rawDataReceived}.
c S s t d�S )NZoops)� RuntimeError�r r r r �<lambda>o r. z5LineReceiverTests.test_rawDataError.<locals>.<lambda>s dataN)
r rh r! r rt rI r rL ZassertIsInstancer| )r rn r; Zwhyr r r �test_rawDataErrori s
z#LineReceiverTests.test_rawDataErrorc C s t �� }| �t|jd� dS )z�
When L{LineReceiver.rawDataReceived} is not overridden in a
subclass, calling it raises C{NotImplementedError}.
�fooN)r rh � assertRaises�NotImplementedErrorr! �r rn r r r �"test_rawDataReceivedNotImplementedw s z4LineReceiverTests.test_rawDataReceivedNotImplementedc C s t �� }| �t|jd� dS )z�
When L{LineReceiver.lineReceived} is not overridden in a subclass,
calling it raises C{NotImplementedError}.
r� N)r rh r� r� r r� r r r � test_lineReceivedNotImplemented� s z1LineReceiverTests.test_lineReceivedNotImplementedN)r" r# r$ r% rK rN rU rW rX rZ r\ r] r^ r_ r` ra rb rc re ri rp rx rz r{ r r� r� r r r r rC � sJ �
�
rC c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ExcessivelyLargeLineCatcherz�
Helper for L{LineReceiverLineLengthExceededTests}.
@ivar longLines: A L{list} of L{bytes} giving the values
C{lineLengthExceeded} has been called with.
c C s
g | _ d S r )� longLinesr r r r r- � s z*ExcessivelyLargeLineCatcher.connectionMadec C s dS )z/
Disregard any received lines.
Nr r r r r r � s z(ExcessivelyLargeLineCatcher.lineReceivedc C s | j �|� dS )zF
Record any data that exceeds the line length limits.
N)r� r r r r r r@ � s z.ExcessivelyLargeLineCatcher.lineLengthExceededN)r" r# r$ r% r- r r@ r r r r r� � s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�#LineReceiverLineLengthExceededTestszO
Tests for L{twisted.protocols.basic.LineReceiver.lineLengthExceeded}.
c C s, t � | _d| j_t�� | _| j�| j� d S )N� )r� rn r? r rt r; rI r r r r �setUp� s
z)LineReceiverLineLengthExceededTests.setUpc C s6 d| j jd d }| j �|� | �|g| j j� dS )z�
If more bytes than C{LineReceiver.MAX_LENGTH} arrive containing no line
delimiter, all of the bytes are passed as a single string to
L{LineReceiver.lineLengthExceeded}.
rj � N)rn r? rL rM r� �r Z excessiver r r �test_longUnendedLine� s z8LineReceiverLineLengthExceededTests.test_longUnendedLinec C sB d| j jd d }| j �d| j j | � | �|g| j j� dS )a*
If L{LineReceiver.dataReceived} is called with bytes representing a
short line followed by bytes that exceed the length limit without a
line delimiter, L{LineReceiver.lineLengthExceeded} is called with all
of the bytes following the short line's delimiter.
rj r� N)rn r? rL r& rM r� r� r r r �test_longLineAfterShortLine� s z?LineReceiverLineLengthExceededTests.test_longLineAfterShortLinec C sF | j j�d| j jd d gd �}| j �|� | �|g| j j� dS )a
If L{LineReceiver.dataReceived} is called with more than
C{LineReceiver.MAX_LENGTH} bytes containing a line delimiter somewhere
not in the first C{MAX_LENGTH} bytes, the entire byte string is passed
to L{LineReceiver.lineLengthExceeded}.
rj r� N)rn r& rm r? rL rM r� r� r r r �test_longLineWithDelimiter� s
� z>LineReceiverLineLengthExceededTests.test_longLineWithDelimiterc C sB d| j jd d | j j d }| j �|� | �|g| j j� dS )a
If L{LineReceiver.dataReceived} is called with more than
C{LineReceiver.MAX_LENGTH} bytes containing multiple line delimiters
somewhere not in the first C{MAX_LENGTH} bytes, the entire byte string
is passed to L{LineReceiver.lineLengthExceeded}.
rj r� N)rn r? r& rL rM r� r� r r r �test_multipleLongLines� s
� z:LineReceiverLineLengthExceededTests.test_multipleLongLinesc C sB t �� }t�� }|�|� |�d|jd d � | �|j� dS rq rs rw r r r rx � s
z:LineReceiverLineLengthExceededTests.test_maximumLineLengthc C sD t �� }t�� }|�|� |�d|jt|j� � | � |j
� dS )z�
C{LineReceiver} disconnects the transport it if receives a non-finished
line longer than its C{MAX_LENGTH}.
rj N) r rh r rt rI rL r? r= r& ru rv rw r r r � test_maximumLineLengthRemaining� s
zCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingN) r" r# r$ r% r� r� r� r� r� rx r� r r r r r� � s
r� c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�LineOnlyReceiverTestsz@
Tests for L{twisted.protocols.basic.LineOnlyReceiver}.
s7 foo
bleakness
desolation
plastic forks
c C sT t �� }t� }|�|� t| j�D ]}|�|� q"| �|j| j� d�dd� � dS )zW
Test buffering over line protocol: data received should match buffer.
r Nr< )
r rt rA rI r rK rL rM r, �split)r rQ rR �cr r r rU s
z!LineOnlyReceiverTests.test_bufferc C sJ t � }t�� }|�|� |�d|jt|j� d d � | �|j � dS )z�
C{LineOnlyReceiver} disconnects the transport if it receives a
line longer than its C{MAX_LENGTH} + len(delimiter).
rj r rr N)
rA r rt rI rL r? r= r& ru rv rw r r r �!test_greaterThanMaximumLineLength s
���z7LineOnlyReceiverTests.test_greaterThanMaximumLineLengthc C s t �� }| �t|jd� dS )z�
When L{LineOnlyReceiver.lineReceived} is not overridden in a subclass,
calling it raises C{NotImplementedError}.
r� N)r �LineOnlyReceiverr� r� r r� r r r r� s z5LineOnlyReceiverTests.test_lineReceivedNotImplementedN)r" r# r$ r% rK rU r� r� r r r r r� � s
r� c @ s, e Zd Zdd� Zdd� ZdZdZdd� Zd S )
� TestMixinc C s
g | _ d S r r+ r r r r r- ( s zTestMixin.connectionMadec C s | j �|� d S r rB �r rT r r r �stringReceived, s zTestMixin.stringReceived�2 r c C s
d| _ d S )Nr )�closed)r �reasonr r r �connectionLost3 s zTestMixin.connectionLostN)r" r# r$ r- r� r? r� r� r r r r r� & s
r� c @ s e Zd Zdd� ZdS )�
TestNetstringc C s | j �|� | j�|� d S r )r, r r; �writer� r r r r� : s z TestNetstring.stringReceivedN)r" r# r$ r� r r r r r� 8 s r� c @ s$ e Zd Zg ZdZdd� Zdd� ZdS )�LPTestCaseMixinNc C s t �� }| �� }|�|� |S )z�
Return a new instance of C{self.protocol} connected to a new instance
of L{proto_helpers.StringTransport}.
)r rt r rI )r rQ rR r r r � getProtocolF s
zLPTestCaseMixin.getProtocolc C s> | j D ]2}| �� }t|�D ]}|�|� q| �|jj� qdS )zO
Assert that illegal strings cause the transport to be closed.
N)�illegalStringsr� r rL ru r; rv �r rT �rr� r r r � test_illegalQ s
z LPTestCaseMixin.test_illegal)r" r# r$ r� r r� r� r r r r r� @ s r� c @ s� e Zd ZdZdddddddgZd d
d d gZeZd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z d9S ):�NetstringReceiverTestszA
Tests for L{twisted.protocols.basic.NetstringReceiver}.
� hello� worlds hows ares you123s :todays aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaas 9999999999999999999999s abcs 4:abcdes7 51:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab,c C s$ t �� | _t� | _| j�| j� d S r )r rt r; r� �netstringReceiverrI r r r r r� j s
z NetstringReceiverTests.setUpc C s� t dd�D ]�}t�� }t� }d|_|�|� | jD ]}|�|� q2|�� }t t |�| d �D ]*}||| |d | � }|r^|�
|� q^| � |j | j� q
dS )zI
Strings can be received in chunks of different lengths.
r rF i� N)
rG r rt r� r? rI �strings�
sendString�valuer= rL rM r, )r rP rQ rR rT �outrS r r r rU p s
z"NetstringReceiverTests.test_bufferc C s" | j �d� | �| j jdg� dS )zE
Empty netstrings (with length '0') can be received.
s 0:,r. N�r� rL rM r, r r r r �test_receiveEmptyNetstring� s z1NetstringReceiverTests.test_receiveEmptyNetstringc C s" | j �d� | �| j jdg� dS )z;
One-character netstrings can be received.
� 1:a,rE Nr� r r r r �test_receiveOneCharacter� s z/NetstringReceiverTests.test_receiveOneCharacterc C s" | j �d� | �| j jdg� dS )z;
Two-character netstrings can be received.
s 2:ab,� abNr� r r r r �test_receiveTwoCharacters� s z0NetstringReceiverTests.test_receiveTwoCharactersc C s" | j �d� | �| j jdg� dS )z�
Netstrings with embedded netstrings. This test makes sure that
the parser does not become confused about the ',' and ':'
characters appearing inside the data portion of the netstring.
s 4:1:a,,r� Nr� r r r r �test_receiveNestedNetstring� s z2NetstringReceiverTests.test_receiveNestedNetstringc C s | j �d� | �| jj� dS )zL
Netstrings containing more data than expected are refused.
s 2:aaa,N�r� rL ru r; rv r r r r �test_moreDataThanSpecified� s z1NetstringReceiverTests.test_moreDataThanSpecifiedc C s | j �d� | �| jj� dS )z�
Netstrings that should be empty according to their length
specification are refused if they contain data.
s 0:a,Nr� r r r r �$test_moreDataThanSpecifiedBorderCase� s z;NetstringReceiverTests.test_moreDataThanSpecifiedBorderCasec C s | j �d� | �| jj� dS )z`
Netstrings without leading digits that specify the length
are refused.
s :aaa,Nr� r r r r �test_missingNumber� s z)NetstringReceiverTests.test_missingNumberc C s | j �d� | �| jj� dS )zg
Netstrings without a colon between length specification and
data are refused.
s 3aaa,Nr� r r r r �test_missingColon� s z(NetstringReceiverTests.test_missingColonc C s | j �d� | �| jj� dS )zY
Netstrings that have no leading digits nor a colon are
refused.
s aaa,Nr� r r r r �test_missingNumberAndColon� s z1NetstringReceiverTests.test_missingNumberAndColonc C s | j �d� | �| jj� dS )zA
Netstrings consisting only of data are refused.
s aaaNr� r r r r �
test_onlyData� s z$NetstringReceiverTests.test_onlyDatac C s> | j �d� | j �d� | �| j jdg� | �| j �� � dS )z=
Netstrings can be received in two portions.
s 4:aas aa,s aaaaN)r� rL rM r, ru �_payloadCompleter r r r � test_receiveNetstringPortions_1� s z6NetstringReceiverTests.test_receiveNetstringPortions_1c C s, dD ]}| j �|� q| �| j jdg� dS )z�
Netstrings can be received in more than two portions, even if
the length specification is split across two portions.
)� 1s 0:01234s 56789� ,rD Nr� �r �partr r r � test_receiveNetstringPortions_2� s z6NetstringReceiverTests.test_receiveNetstringPortions_2c C s, dD ]}| j �|� q| �| j jdg� dS )zE
Netstrings can be received one character at a time.
)� 2� :rE � br� r� Nr� r� r r r � test_receiveNetstringPortions_3� s z6NetstringReceiverTests.test_receiveNetstringPortions_3c C sR | j �d� | �| j �� � | �| j jdg� | j �d� | �| j jddg� dS )z�
A stream of two netstrings can be received in two portions,
where the first portion contains the complete first netstring
and the length specification of the second netstring.
s 1:a,1rE s :b,r� N)r� rL ru r� rM r, r r r r �test_receiveTwoNetstrings� s
z0NetstringReceiverTests.test_receiveTwoNetstringsc C s> | j jd }| j �d�t|�dd| f�� | �| jj� dS )zs
Netstrings with a length specification exceeding the specified
C{MAX_LENGTH} are refused.
r r. r� rE N)r� r? rL rm �bytesru r; rv )r ZtooLongr r r �test_maxReceiveLimit s
�z+NetstringReceiverTests.test_maxReceiveLimitc C s& d| j _| j �� | �| j jd� dS )zw
C{_consumeLength} returns the expected length of the
netstring, including the trailing comma.
� 12:�
N)r� �_remainingData�_consumeLengthrM �_expectedPayloadSizer r r r �test_consumeLength s
z)NetstringReceiverTests.test_consumeLengthc C s. d| j _d| j _| j �� | �| j jd� dS )z�
C{_consumeLength} works as expected if the length specification
contains the value of C{MAX_LENGTH} (border case).
r� � r� N)r� r� r? r� rM r� r r r r � test_consumeLengthBorderCase1 s
z4NetstringReceiverTests.test_consumeLengthBorderCase1c C s& d| j _d| j _| �tj| j j� dS )z�
C{_consumeLength} raises a L{basic.NetstringParseError} if
the length specification exceeds the value of C{MAX_LENGTH}
by 1 (border case).
r� � N�r� r� r? r� r ZNetstringParseErrorr� r r r r � test_consumeLengthBorderCase2# s
�z4NetstringReceiverTests.test_consumeLengthBorderCase2c C s& d| j _d| j _| �tj| j j� dS )z�
C{_consumeLength} raises a L{basic.NetstringParseError} if
the length specification exceeds the value of C{MAX_LENGTH}
by more than 1.
s 1000:r� Nr� r r r r � test_consumeLengthBorderCase3/ s
�z4NetstringReceiverTests.test_consumeLengthBorderCase3c C s t �� }| �t|jd� dS )z�
When L{NetstringReceiver.stringReceived} is not overridden in a
subclass, calling it raises C{NotImplementedError}.
r� N)r �NetstringReceiverr� r� r� r� r r r �!test_stringReceivedNotImplemented; s z8NetstringReceiverTests.test_stringReceivedNotImplementedN) r" r# r$ r% r� r� r� r r� rU 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� ] s@ � �
r� c @ sP e Zd ZdZdZdZdZdZdd� Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dS )�IntNTestCaseMixinz4
TestCase mixin for int-prefixed protocols.
Nc C sR | � � }| jD ].}tt�|jt|��| �D ]}|�|� q,q| �|j | j� dS )z>
Test receiving data find the same data send.
N)
r� r� r �struct�pack� structFormatr= rL rM r, )r r� rT r� r r r � test_receiveO s
z IntNTestCaseMixin.test_receivec C s> | j D ]2}| �� }t|�D ]}|�|� q| �|jg � qdS )zK
Send partial data, nothing should be definitely received.
N)�partialStringsr� r rL rM r, r� r r r � test_partialZ s
z IntNTestCaseMixin.test_partialc C s6 | � � }|�d� | �|j�� t�|jd�d � dS )z2
Test sending data over protocol.
� bbbbbbbbbbbbbbbb� N)r� r� rM r; r� r� r� r� �r r� r r r � test_sende s
�zIntNTestCaseMixin.test_sendc C s@ g }| � � }|j|_d|_|�t�|jd�� | �|dg� dS )z�
When a length prefix is received which is greater than the protocol's
C{MAX_LENGTH} attribute, the C{lengthLimitExceeded} method is called
with the received length prefix.
rF r� N) r� r �lengthLimitExceededr? rL r� r� r� rM )r r9 r� r r r �test_lengthLimitExceededo s z*IntNTestCaseMixin.test_lengthLimitExceededc C s8 | � � }d|_|�t�|jd�d � | �|jg � dS )z�
If a length prefix for a string longer than C{MAX_LENGTH} is delivered
to C{dataReceived} at the same time as the entire string, the string is
not passed to C{stringReceived}.
rF r� s xxxxxxxxxxxN)r� r? rL r� r� r� rM r, r� r r r �test_longStringNotDelivered} s �z-IntNTestCaseMixin.test_longStringNotDeliveredc C s t �� }| �t|jd� dS )z�
When L{IntNStringReceiver.stringReceived} is not overridden in a
subclass, calling it raises C{NotImplementedError}.
r� N)r ZIntNStringReceiverr� r� r� r� r r r r� � s z3IntNTestCaseMixin.test_stringReceivedNotImplemented)r" r# r$ r% r r� r� r� r� r� r� r� r� r� r r r r r� E s
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�RecvdAttributeMixinz�
Mixin defining tests for string receiving protocols with a C{recvd}
attribute which should be settable by application code, to be combined with
L{IntNTestCaseMixin} on a L{TestCase} subclass
c C s t �|jt|��| S )zg
Return C{data} prefixed with message length in C{protocol.structFormat}
form.
)r� r� r� r= )r r r r r r � makeMessage� s z RecvdAttributeMixin.makeMessagec sd g �| � � � � �fdd�}|� _t�� jd�d }t�� jd�d }� �|| � | ��|g� dS )z�
In stringReceived, recvd contains the remaining data that was passed to
dataReceived that was not part of the current message.
c s �� � j� d S r �r �recvd�ZreceivedString�r� �resultr r r� � s zKRecvdAttributeMixin.test_recvdContainsRemainingData.<locals>.stringReceived� � aaaaas bbbbN)r� r� r� r� r� rL rM )r r� ZcompleteMessageZincompleteMessager r� r � test_recvdContainsRemainingData� s z3RecvdAttributeMixin.test_recvdContainsRemainingDatac st | � � �g �d}| ��|�� � ��fdd�}|�_d}d}| ��|�}| ��|�}��|| � | ��||g� dS )z�
In stringReceived, if recvd is changed, messages should be parsed from
it rather than the input to dataReceived.
s cccccc s �s
� �_ ��| � d S r )r� r r� �ZmessageCr� r� r r r� � s z=RecvdAttributeMixin.test_recvdChanged.<locals>.stringReceivedr� s bbbbbN)r� r� r� rL rM )r ZpayloadCr� ZpayloadAZpayloadBZmessageAZmessageBr r� r �test_recvdChanged� s z%RecvdAttributeMixin.test_recvdChangedc s� | � � �g }d� | jD ] }|�| ��|�� |�� � qg �� ��fdd�}|�_��d�|�� ��d� | ��| j� | ��jd� dS )z�
Data already parsed by L{IntNStringReceiver.dataReceived} is not
reparsed if C{stringReceived} consumes some of the
L{IntNStringReceiver.recvd} buffer.
s c s" �� | � �jt� �d � �_d S r )r r� r= r� �ZSWITCHrn r� r r r� � s
z:RecvdAttributeMixin.test_switching.<locals>.stringReceivedr. � N) r� r� r r� r� rL rm rM r� )r ZmixrT r� r r� r �test_switching� s
z"RecvdAttributeMixin.test_switchingc sp | � � � d}t|�d � _| �� |�}g �� �fdd�}|� _� �|� | ��d t|�� | ��d |� dS )z�
The L{IntNStringReceiver.recvd} buffer contains all data not yet
processed by L{IntNStringReceiver.dataReceived} if the
C{lengthLimitExceeded} event occurs.
s too longr c s �� | � �� � j� d S r r� )r9 �rn r� r r r� � s
zPRecvdAttributeMixin.test_recvdInLengthLimitExceeded.<locals>.lengthLimitExceededr N)r� r= r? r� r� rL rM )r ZDATA�messager� r r� r � test_recvdInLengthLimitExceeded� s
z3RecvdAttributeMixin.test_recvdInLengthLimitExceededN) r" r# r$ r% r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZdS )� TestInt32z�
A L{basic.Int32StringReceiver} storing received strings in an array.
@ivar received: array holding received strings.
N�r" r# r$ r% r r r r r� � s r� c @ s4 e Zd ZdZeZddgZdgZdddgZdd � Z d
S ) �
Int32Testsz/
Test case for int32-prefixed protocol
rE r� s
aaaaaas � hello therer. c C sB | � � }|�d� | �|j�� d� |�d� | �|jdg� dS )z?
Test specific behavior of the 32-bits length.
rg s foos ubar� ubarN�r� r� rM r; r� rL r, r� r r r � test_data s
zInt32Tests.test_dataN)
r" r# r$ r% r� r r� r� r� r� r r r r r� s
r� c @ s e Zd ZdZdS )� TestInt16z�
A L{basic.Int16StringReceiver} storing received strings in an array.
@ivar received: array holding received strings.
Nr� r r r r r� s r� c @ s< e Zd ZdZeZddgZdgZdddgZdd � Z d
d � Z
d S )
�
Int16Testsz/
Test case for int16-prefixed protocol
rE r� s aaaaaa� r� r. c C sB | � � }|�d� | �|j�� d� |�d� | �|jdg� dS )z?
Test specific behavior of the 16-bits length.
rg s foos ubarr� Nr� r� r r r r� / s
zInt16Tests.test_datac C s2 | � � }dd|jd d }| �t|j|� dS �zA
Send too much data: that should cause an error.
r� r� � r N�r� Z prefixLengthr� �AssertionErrorr� �r r� ZtooSendr r r �test_tooLongSend: s zInt16Tests.test_tooLongSendN) r" r# r$ r% r� r r� r� r� r� r r r r r r % s
r c @ s e Zd ZdZdS )�NewStyleTestInt16z0
A new-style class version of TestInt16
Nr� r r r r r D s r c @ s e Zd ZdZereZeZdS )�NewStyleInt16Testszn
This test case verifies that IntNStringReceiver still works when inherited
by a new-style class.
N) r" r# r$ r% r �_PY3NEWSTYLESKIP�skipr r r r r r r K s r c @ s e Zd ZdZdS )�TestInt8z�
A L{basic.Int8StringReceiver} storing received strings in an array.
@ivar received: array holding received strings.
Nr� r r r r r W s r c @ s< e Zd ZdZeZddgZdgZdddgZdd � Z d
d � Z
d S )
� Int8Testsz.
Test case for int8-prefixed protocol
rE r� s aaaaaa� s dzadzr. c C sB | � � }|�d� | �|j�� d� |�d� | �|jdg� dS )z>
Test specific behavior of the 8-bits length.
rg s foos ubarr� Nr� r� r r r r� k s
zInt8Tests.test_datac C s2 | � � }dd|jd d }| �t|j|� dS r r r r r r r v s zInt8Tests.test_tooLongSendN) r" r# r$ r% r r r� r� r� r� r r r r r r
` s
r
c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d S ) �OnlyProducerTransportzm
Transport which isn't really a transport, just looks like one to
someone not looking very hard.
Fc C s
g | _ d S r r} r r r r r � s z OnlyProducerTransport.__init__c C s
d| _ d S )NT��pausedr r r r r5 � s z$OnlyProducerTransport.pauseProducingc C s
d| _ d S )NFr r r r r r6 � s z%OnlyProducerTransport.resumeProducingc C s | j �|� d S r )r r )r r� r r r r� � s zOnlyProducerTransport.writeN)
r" r# r$ r% r rv r r5 r6 r� r r r r r � s r c @ s e Zd ZdZdd� ZdS )�ConsumingProtocolzC
Protocol that really, really doesn't want any more bytes.
c C s | j �|� | �� d S r )r; r� r5 r r r r r � s z ConsumingProtocol.lineReceivedN)r" r# r$ r% r r r r r r � s r c @ s e Zd ZdZdd� ZdS )�
ProducerTestszM
Tests for L{basic._PausableMixin} and L{basic.LineReceiver.paused}.
c C s� t � }t� }|�|� |�d� | �|jg � | �|j� | �|j� |�d� | �|jdg� | �|j� | �|j� |� � | �|jdg� | �|j� | �|j� |�d� | �|jddg� | �|j� | �|j� |� � | �|jdddg� | �|j� | �|j� |�d� | �|jdddg� | �|j� | �|j� |� � | �|jddddg� | �|j� | �|j� |� � | �|jddddg� | �|j� | �|j� d S )
a?
When L{basic.LineReceiver} is paused, it doesn't deliver lines to
L{basic.LineReceiver.lineReceived} and delivers them immediately upon
being resumed.
L{ConsumingProtocol} is a L{LineReceiver} that pauses itself after
every line, and writes that line to its transport.
s hello, s world
s hello, worlds hello
world
r� r� s goodbye
rd N)
r r rI rL rM r ry r ru r6 )r �prQ r r r �test_pauseResume� sR
�
� z ProducerTests.test_pauseResumeN)r" r# r$ r% r r r r r r � s r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�FileSenderTestsz(
Tests for L{basic.FileSender}.
c C s t �� }| �tt|�� dS )zP
L{basic.FileSender} implements the L{IPullProducer} interface.
N)r �
FileSenderru r r
)r �senderr r r �test_interface� s z FileSenderTests.test_interfacec C sB t d�}t�� }t�� }|�||� | �|j|� | �|j � dS )z�
When L{basic.FileSender.beginFileTransfer} is called, it registers
itself with provided consumer, as a non-streaming producer.
� Test contentN)
r r rt r r �beginFileTransferrM �producerry Z streaming)r �source�consumerr r r r �test_producerRegistered� s z'FileSenderTests.test_producerRegisteredc C sf t d�}t�� }t�� }|�||�}|�� |�� | �|j� | � d| �
|�� | � d|� � � dS )z�
L{basic.FileSender} sends the content of the given file using a
C{IConsumer} interface via C{beginFileTransfer}. It returns a
L{Deferred} which fires with the last byte sent.
r � tN) r r rt r r r r6 Z assertIsNoner rM �successResultOfr� �r r r r �dr r r �
test_transfer s z FileSenderTests.test_transferc C s� t d�}t�� }t�� }d|_|�||�}|�� | �d|� � � |�� | �d|� � � |�� | �d|� � � |�� | �d| �
|�� | �d|� � � dS )zj
L{basic.FileSender} reads at most C{CHUNK_SIZE} every time it resumes
producing.
r r2 s Tests Test conr N) r r rt r r Z
CHUNK_SIZEr r6 rM r� r! r" r r r �test_transferMultipleChunks s z+FileSenderTests.test_transferMultipleChunksc C sd dd� }t d�}t�� }t�� }|�|||�}|�� |�� | �d| �|�� | �d|� � � dS )z�
L{basic.FileSender.beginFileTransfer} takes a C{transform} argument
which allows to manipulate the data on the fly.
c S s | � � S r )�swapcase)�chunkr r r � transform< s z=FileSenderTests.test_transferWithTransform.<locals>.transformr � Ts tEST CONTENTN)
r r rt r r r r6 rM r! r� )r r( r r r r# r r r �test_transferWithTransform6 s z*FileSenderTests.test_transferWithTransformc C sV t d�}t�� }t�� }|�||�}|�� | �|�}|�t � | �
dt |j �� dS )z�
The C{Deferred} returned by L{basic.FileSender.beginFileTransfer} fails
with an C{Exception} if C{stopProducing} when the transfer is not
complete.
r z#Consumer asked us to stop producingN)
r r rt r r r r7 ZfailureResultOfZtrap� ExceptionrM �strr� )r r r r r# Zfailurer r r �test_abortedTransferK s
�z$FileSenderTests.test_abortedTransferN)
r" r# r$ r% r r r$ r% r* r- r r r r r � s
r c @ s e Zd ZdZerdZdd� ZdS )�MiceDeprecationTestsz2
L{twisted.protocols.mice} is deprecated.
z7twisted.protocols.mice is not being ported to Python 3.c C s: t �d� | �� }| �dt|�� | �d|d d � dS )zM
L{twisted.protocols.mice} is deprecated since Twisted 16.0.
ztwisted.protocols.micer zatwisted.protocols.mice was deprecated in Twisted 16.0.0: There is no replacement for this module.r r� N)r
ZnamedAnyZ
flushWarningsrM r= )r Z
warningsShownr r r �test_MiceDeprecationg s
�z)MiceDeprecationTests.test_MiceDeprecationN)r" r# r$ r% r r r/ r r r r r. _ s r. )>r% Z
__future__r r rk r� �ior Zzope.interface.verifyr Ztwisted.python.compatr r Z
twisted.trialr Ztwisted.protocolsr Ztwisted.pythonr
Ztwisted.internetr r Ztwisted.internet.interfacesr
Z twisted.testr r
rh r r' r� rA ZSynchronousTestCaserC r� r� r� r� r� r� r� r� r� �objectr� ZInt32StringReceiverr� r� ZInt16StringReceiverr� r r r ZInt8StringReceiverr r
r r r ZTestCaser r. r r r r �<module> s^ M qU. iOi � � � Ln