File: //usr/lib/python3/dist-packages/twisted/protocols/test/__pycache__/test_basic.cpython-310.pyc
o
�b� � @ sJ d Z ddlZddlZddlmZ ddlmZmZmZ ddl m
Z
ddl m Z m
Z
ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ 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 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'�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� d1e&ej3�Z4G d2d3� d3ej!e+e,�Z5G d4d5� d5�Z6G d6d7� d7ej �Z7G d8d9� d9ej!�Z8G d:d;� d;ej9�Z:dS )<z,
Test cases for L{twisted.protocols.basic}.
� N)�BytesIO)�List�Optional�Type)� verifyObject)�protocol�task)� IProducer)�connectionDone)�basic)� iterbytes)�Failure)�
proto_helpers)�unittestc @ �, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�FlippingLineTesterzT
A line receiver that flips between line and raw data modes after one byte.
�
c C �
g | _ d S �N)�lines��self� r �C/usr/lib/python3/dist-packages/twisted/protocols/test/test_basic.py�__init__! �
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 B �
zLineTester.__init__c C �
g | _ dS �z;
Create/clean data received on connection.
N��receivedr r r r �connectionMadeH r. zLineTester.connectionMadec C s� | j �|� |dkr| �� dS |dkr"| �� | j�d| j� dS |dkr>| �� | �� | j �d� | j�d| j� dS |dkrH| �� dS |dd� dkr[t|dd� �| _ dS |�
d �ri| j � | d
� dS |�
d �ru| j �
� dS dS ) z|
Receive line and make some action for some tokens: pause, rawpause,
stop, len, produce, unproduce.
� � pauser � rawpause� stopN� s len � produceF� unproduce)r2 r r �pauseProducingr- � callLater�resumeProducing�
stopProducing�int�length�
startswith� transport�registerProducer�unregisterProducerr r r r r N s(
�zLineTester.lineReceivedc C s^ |d| j � || j d� }}| j t|� | _ | jd | | jd< | j dkr-| �|� dS dS )zj
Read raw data, until the quantity specified by a previous 'len' line is
reached.
N���r )r@ �lenr2 r" )r r$ �restr r r r% g s
�zLineTester.rawDataReceivedc C s2 t |�| jd kr| �|| jd d� � dS dS )z<
Adjust line mode when long lines received.
r! N)rF �
MAX_LENGTHr" r r r r �lineLengthExceededr s �z LineTester.lineLengthExceededr ) r& r' r( r) r* rH r r3 r r% rI r r r r r+ 2 s
r+ c @ s( e Zd ZdZdZdZdd� Zdd� ZdS ) �LineOnlyTesterz)
A buffering line only receiver.
r r, c C r/ r0 r1 r r r r r3 � r. z LineOnlyTester.connectionMadec C � | j �|� dS )z%
Save received data.
N�r2 r r r r r r � � zLineOnlyTester.lineReceivedN)r& r' r( r) r* rH r3 r r r r r rJ z s rJ c @ s� e Zd ZdZdZg d�Zdd� ZdZg d�Zedg Z d d
� Z
d Z g d �Z g d
�Z
dd� ZdZg d�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&S )'�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
a)
s len 10�
0123456789� len 5s 1234
s len 20s foo 123s 0123456789
012345678s len 0s foo 5r4 s 67890s len 1� ac C � t dd�D ]9}t�� }t� }|�t�|�� t t| j�| d �D ]}| j|| |d | � }|� |� q!| �
| j |j � qdS )zl
Test buffering for different packet size, checking received matches
expected data.
r! �
N)
�ranger �StringIOWithoutClosingr+ �makeConnectionr � FileWrapperrF �buffer� dataReceived� assertEqual�outputr2 �r � packet_size�t�a�i�sr r r � test_buffer� s �z LineReceiverTests.test_buffers! twiddle1
twiddle2
pause
twiddle3
)� twiddle1� twiddle2r5 � twiddle3c C � t dd�D ]K}t�� }t�� }t|�}|�t�|�� t t | j
�| d �D ]}| j
|| |d | � }|� |� q&| � | j
|j� |�d� | � | j|j� qdS )zo
Test pause inside data receiving. It uses fake clock to see if
pausing/resuming work.
r! rS r N)rT r rU r �Clockr+ rV r rW rF �pauseBufrY rZ � pauseOutput1r2 �advance� pauseOutput2�r r] r^ r- r_ r` ra r r r � test_pausing� s
�z LineReceiverTests.test_pausings/ twiddle1
twiddle2
len 5
rawpause
12345twiddle3
)rc rd rP r6 r4 )rc rd rP r6 s 12345re c C rf )z7
Test pause inside raw date receiving.
r! rS r N)rT r rU r rg r+ rV r rW rF � rawpauseBufrY rZ �rawpauseOutput1r2 rj �rawpauseOutput2rl r r r �test_rawPausing� s
�z!LineReceiverTests.test_rawPausings" twiddle1
twiddle2
stop
more
stuff
)rc rd r7 c C rR )z-
Test stop inside producing.
r! rS N)
rT r rU r+ rV r rW rF �stop_bufrY rZ � stop_outputr2 r\ r r r �test_stopProducing� s �z$LineReceiverTests.test_stopProducingc C s> t � }t�� }|�t�|�� |�d� | �|jg d�� dS )z6
Test produce/unproduce in receiving.
s& produce
hello world
unproduce
goodbye
)r9 s hello worldr: � goodbyeN) r+ r rU rV r rW rY rZ r2 )r r_ r^ 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 @ � e Zd Zdd� ZdS )z@LineReceiverTests.test_clearLineBuffer.<locals>.ClearingReceiverc S s || _ | �� | _d S r )r �clearLineBufferrG r r r r r s zMLineReceiverTests.test_clearLineBuffer.<locals>.ClearingReceiver.lineReceivedN)r& r' r( r r r r r �ClearingReceiver � ry s
foo
bar
baz� foos bar
bazs quux
s quuxr4 N)r � LineReceiverrY rZ r rG )r ry 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� xr4 N) r r rU rV r rW �sys�getrecursionlimitrY rZ �joinr )r �protorB �limitr r r �test_stackRecursion* s z%LineReceiverTests.test_stackRecursionc C �B t �� }t�� }|�|� |�d|jd d � | �|j� dS �zx
C{LineReceiver} disconnects the transport if it receives a line longer
than its C{MAX_LENGTH}.
r~ r! �
rN� r r| r �StringTransportrV rY rH �
assertTrue�
disconnecting�r r� rB r r r �test_maximumLineLength5 �
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).
r8 r~ r! NrE r ) r+ rH r r� rV rY r* � assertFalser� rZ rF r2 )r r� r^ 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
r~ Nr! ) r r| r* r r� rV rY rH rF r� r� r� 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 )N�oops)� RuntimeError�r$ r r r �<lambda>o s z5LineReceiverTests.test_rawDataError.<locals>.<lambda>s dataN)
r r| r% r r� rV r rY �assertIsInstancer� )r r� rB �whyr r r �test_rawDataErrori s
z#LineReceiverTests.test_rawDataErrorc C � t �� }| �t|jd� dS )z�
When L{LineReceiver.rawDataReceived} is not overridden in a
subclass, calling it raises C{NotImplementedError}.
�fooN)r r| � assertRaises�NotImplementedErrorr% �r r� r r r �"test_rawDataReceivedNotImplementedv � z4LineReceiverTests.test_rawDataReceivedNotImplementedc C r� )z�
When L{LineReceiver.lineReceived} is not overridden in a subclass,
calling it raises C{NotImplementedError}.
r� N)r r| r� r� r r� r r r � test_lineReceivedNotImplemented~ r� z1LineReceiverTests.test_lineReceivedNotImplementedN)r& r' r( r) rX r[ rb rh ri rk rm rn ro rp rq rr rs rt rv r} r� r� r� r� r� r� r� r r r r rN � s2
rN 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 r r )� longLinesr r r r r3 � r z*ExcessivelyLargeLineCatcher.connectionMadec C s dS )z/
Disregard any received lines.
Nr r r r r r � s z(ExcessivelyLargeLineCatcher.lineReceivedc C rK )zF
Record any data that exceeds the line length limits.
N)r� r r# r r r rI � rM z.ExcessivelyLargeLineCatcher.lineLengthExceededN)r& r' r( r) r3 r rI 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� r� rH r r� rB rV 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}.
r~ � N)r� rH rY rZ r� �r � 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.
r~ r� N)r� rH rY r* rZ 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}.
r~ r� N)r� r* r� rH rY rZ 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}.
r~ r� N)r� rH r* rY rZ r� r� r r r �test_multipleLongLines� s z:LineReceiverLineLengthExceededTests.test_multipleLongLinesc C r� r� r� r� r r r r� � r� 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}.
r~ N) r r| r r� rV rY rH rF r* r� r� r� r r r � test_maximumLineLengthRemaining� s
zCLineReceiverLineLengthExceededTests.test_maximumLineLengthRemainingN) r& r' r( r) r� r� r� r� r� r� r� r r r r r� � s
r� c @ r )
�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 NrE )
r r� rJ rV r rX rY rZ r2 �split)r r^ r_ �cr r r rb � 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).
r~ r! r� N)
rJ r r� rV rY rH rF r* r� r� r� r r r �!test_greaterThanMaximumLineLength s
�z7LineOnlyReceiverTests.test_greaterThanMaximumLineLengthc C r� )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� r� z5LineOnlyReceiverTests.test_lineReceivedNotImplementedN)r& r' r( r) rX rb r� r� r r r r r� � s
r� c @ s: e Zd Zdd� Zdd� ZdZdZefdedd fd
d �Z d S ) � TestMixinc C r r r1 r r r r r3 r zTestMixin.connectionMadec C � | j �|� d S r rL �r ra r r r �stringReceived � zTestMixin.stringReceived�2 r �reason�returnNc C �
d| _ d S )Nr! )�closed)r r� r r r �connectionLost# r zTestMixin.connectionLost)
r& r' r( r3 r� rH r� r
r
r� r r r r r� s r� c @ rw )�
TestNetstringc C s | j �|� | j�|� d S r )r2 r rB �writer� r r r r� ( s z TestNetstring.stringReceivedN)r& r' r( r� r r r r r� ' rz r� c @ s> e Zd ZU g Zeee ed< dZded< dd� Z dd� Z
dS ) �LPTestCaseMixin�illegalStringsN�!Optional[Type[protocol.Protocol]]r c C s t �� }| �� }|�|� |S )z�
Return a new instance of C{self.protocol} connected to a new instance
of L{proto_helpers.StringTransport}.
)r r� r rV )r r^ r_ r r r � getProtocol2 s
zLPTestCaseMixin.getProtocolc C s> | j D ]}| �� }t|�D ]}|�|� q
| �|jj� qdS )zO
Assert that illegal strings cause the transport to be closed.
N)r� r� r rY r� rB r� �r ra �rr� r r r � test_illegal< �
�z LPTestCaseMixin.test_illegal) r& r' r( r� r r �bytes�__annotations__r r� r� r r r r r� - s
r� c @ s� e Zd ZdZg d�Zg d�ZeZdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0S )1�NetstringReceiverTestszA
Tests for L{twisted.protocols.basic.NetstringReceiver}.
)� hello� worlds hows ares you123s :todays aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)s 9999999999999999999999s abcs 4:abcdes7 51:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab,c C s$ t �� | _t� | _| j�| j� d S r )r r� rB r� �netstringReceiverrV r r r r r� W s
z NetstringReceiverTests.setUpc C s� t dd�D ]H}t�� }t� }d|_|�|� | jD ]}|�|� q|�� }t t |�| d �D ]}||| |d | � }|rD|�
|� q/| � |j | j� qdS )zI
Strings can be received in chunks of different lengths.
r! rS i� N)
rT r r� r� rH rV �strings�
sendString�valuerF rY rZ r2 )r r] r^ r_ ra �outr` r r r rb \ s
��z"NetstringReceiverTests.test_bufferc C �" | j �d� | �| j jdg� dS )zE
Empty netstrings (with length '0') can be received.
s 0:,r4 N�r� rY rZ r2 r r r r �test_receiveEmptyNetstringn � z1NetstringReceiverTests.test_receiveEmptyNetstringc C r� )z;
One-character netstrings can be received.
� 1:a,rQ Nr� r r r r �test_receiveOneCharacteru r� z/NetstringReceiverTests.test_receiveOneCharacterc C r� )z;
Two-character netstrings can be received.
s 2:ab,� abNr� r r r r �test_receiveTwoCharacters| r� z0NetstringReceiverTests.test_receiveTwoCharactersc C r� )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 � | j �d� | �| jj� dS )zL
Netstrings containing more data than expected are refused.
s 2:aaa,N�r� rY r� rB r� r r r r �test_moreDataThanSpecified� � z1NetstringReceiverTests.test_moreDataThanSpecifiedc C r� )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� � z;NetstringReceiverTests.test_moreDataThanSpecifiedBorderCasec C r� )z`
Netstrings without leading digits that specify the length
are refused.
s :aaa,Nr� r r r r �test_missingNumber� r� z)NetstringReceiverTests.test_missingNumberc C r� )zg
Netstrings without a colon between length specification and
data are refused.
s 3aaa,Nr� r r r r �test_missingColon� r� z(NetstringReceiverTests.test_missingColonc C r� )zY
Netstrings that have no leading digits nor a colon are
refused.
s aaa,Nr� r r r r �test_missingNumberAndColon� r� z1NetstringReceiverTests.test_missingNumberAndColonc C r� )zA
Netstrings consisting only of data are refused.
s aaaNr� r r r r �
test_onlyData� r� 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� rY rZ r2 r� �_payloadCompleter r r r � test_receiveNetstringPortions_1� s z6NetstringReceiverTests.test_receiveNetstringPortions_1c C �, 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� ,rO Nr� �r �partr r r � test_receiveNetstringPortions_2� s z6NetstringReceiverTests.test_receiveNetstringPortions_2c C r� )zE
Netstrings can be received one character at a time.
)� 2� :rQ � 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,1rQ s :b,r� N)r� rY r� r� rZ r2 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! r4 r� rQ N)r� rH rY r� r� r� rB r� )r �tooLongr 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�_consumeLengthrZ �_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� rH r� rZ r� r r r r � test_consumeLengthBorderCase1� s
z4NetstringReceiverTests.test_consumeLengthBorderCase1c C �& 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� rH r� r �NetstringParseErrorr� r r r r � test_consumeLengthBorderCase2� �
�z4NetstringReceiverTests.test_consumeLengthBorderCase2c C r� )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
r z4NetstringReceiverTests.test_consumeLengthBorderCase3c C r� )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 r� z8NetstringReceiverTests.test_stringReceivedNotImplementedN) r& r' r( r) r� r� r� r r� rb 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� G s6
r� c @ s� e Zd ZU dZdZded< dZeee ed< dZ
eee ed< dZ eee ed< dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� ZdS )�IntNTestCaseMixinz4
TestCase mixin for int-prefixed protocols.
Nr� r r� r� �partialStringsc C sR | � � }| jD ]}tt�|jt|��| �D ]}|�|� qq| �|j | j� dS )z>
Test receiving data find the same data send.
N)
r� r� r �struct�pack� structFormatrF rY rZ r2 )r r� ra r� r r r � test_receive) s
�z IntNTestCaseMixin.test_receivec C s> | j D ]}| �� }t|�D ]}|�|� q
| �|jg � qdS )zK
Send partial data, nothing should be definitely received.
N)r r� r rY rZ r2 r� r r r � test_partial3 r� z IntNTestCaseMixin.test_partialc C s6 | � � }|�d� | �|j�� t�|jd�d � dS )z2
Test sending data over protocol.
� bbbbbbbbbbbbbbbb� N)r� r� rZ rB r� r r r �r r� r r r � test_send= 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.
rS r� N) r� r �lengthLimitExceededrH rY r r r rZ )r r@ r� r r r �test_lengthLimitExceededG 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}.
rS r� s xxxxxxxxxxxN)r� rH rY r r r rZ r2 r r r r �test_longStringNotDeliveredT s z-IntNTestCaseMixin.test_longStringNotDeliveredc C r� )z�
When L{IntNStringReceiver.stringReceived} is not overridden in a
subclass, calling it raises C{NotImplementedError}.
r� N)r �IntNStringReceiverr� r� r� r� r r r r _ r� z3IntNTestCaseMixin.test_stringReceivedNotImplemented)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 @ 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 rF )r r r$ r r r � makeMessageo 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��receivedString�r� �resultr r r� ~ r� zKRecvdAttributeMixin.test_recvdContainsRemainingData.<locals>.stringReceived� � aaaaas bbbbN)r� r� r r r rY rZ )r r� �completeMessage�incompleteMessager r r � test_recvdContainsRemainingDatav 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 ��messageCr� r r r r� � s z=RecvdAttributeMixin.test_recvdChanged.<locals>.stringReceivedr s bbbbbN)r� r r� rY rZ )r �payloadCr� �payloadA�payloadB�messageA�messageBr r! r �test_recvdChanged� s z%RecvdAttributeMixin.test_recvdChangedc s� | � � �g }d� | jD ]}|�| ��|�� |�� � q g �� ��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 rF r ��SWITCHr� r r r r� � s
z:RecvdAttributeMixin.test_switching.<locals>.stringReceivedr4 � N) r� r� r r r� rY r� rZ r )r �mixra 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 )r@ �r� r r r r � s
zPRecvdAttributeMixin.test_recvdInLengthLimitExceeded.<locals>.lengthLimitExceededr N)r� rF rH r r rY rZ )r �DATA�messager r r. r � test_recvdInLengthLimitExceeded� s
z3RecvdAttributeMixin.test_recvdInLengthLimitExceededN) r& r' r( r) r r r( r- r1 r r r r r h s r c @ � 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 r3 � � r3 c @ s2 e Zd ZdZeZddgZdgZg d�Zdd� Z dS ) �
Int32Testsz/
Test case for int32-prefixed protocol
rQ r s
aaaaaa)s � hello therer4 c C �B | � � }|�d� | �|j�� d� |�d� | �|jdg� dS )z?
Test specific behavior of the 32-bits length.
r{ s foos ubar� ubarN�r� r� rZ rB r� rY r2 r r r r � test_data� �
zInt32Tests.test_dataN)
r& r' r( r) r3 r r� r� r r; r r r r r6 � s r6 c @ r2 )� TestInt16z�
A L{basic.Int16StringReceiver} storing received strings in an array.
@ivar received: array holding received strings.
Nr4 r r r r r= � r5 r= c @ �: e Zd ZdZeZddgZdgZg d�Zdd� Z dd � Z
d
S ) �
Int16Testsz/
Test case for int16-prefixed protocol
rQ r s aaaaaa)� r7 r4 c C r8 )z?
Test specific behavior of the 16-bits length.
r{ s foos ubarr9 Nr: r r r r r; r< zInt16Tests.test_datac C �2 | � � }dd|jd d }| �t|j|� dS �zA
Send too much data: that should cause an error.
r� r� � r! N�r� � prefixLengthr� �AssertionErrorr� �r r� �tooSendr r r �test_tooLongSend � zInt16Tests.test_tooLongSendN) r& r' r( r) r= r r� r� r r; rI r r r r r? � �
r? c @ r2 )�TestInt8z�
A L{basic.Int8StringReceiver} storing received strings in an array.
@ivar received: array holding received strings.
Nr4 r r r r rL r5 rL c @ r> ) � Int8Testsz.
Test case for int8-prefixed protocol
rQ r s aaaaaa)� s dzadzr4 c C r8 )z>
Test specific behavior of the 8-bits length.
r{ s foos ubarr9 Nr: r r r r r; & r< zInt8Tests.test_datac C rA rB rD rG r r r rI 0 rJ zInt8Tests.test_tooLongSendN) r& r' r( r) rL r r� r� r r; rI r r r r rM rK rM 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 r r r� r r r r r B r z OnlyProducerTransport.__init__c C r� )NT��pausedr r r r r; E r z$OnlyProducerTransport.pauseProducingc C r� )NFrP r r r r r= H r z%OnlyProducerTransport.resumeProducingc C r� r )r$ r )r r� r r r r� K r� zOnlyProducerTransport.writeN)
r& r' r( r) rQ r� r r; r= r� r r r r rO 9 s rO c @ � e Zd ZdZdd� ZdS )�ConsumingProtocolzC
Protocol that really, really doesn't want any more bytes.
c C s | j �|� | �� d S r )rB r� r; r r r r r T s z ConsumingProtocol.lineReceivedN)r& r' r( r) r r r r r rS O � rS c @ rR )�
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� |� � | �|jg d�� | �|j� | �|j� |�d� | �|jg d�� | �|j� | �|j� |� � | �|jg d�� | �|j� | �|j� |� � | �|jg d�� | �|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
� hello, worlds hello
world
r� )rV r� r� s goodbye
)rV r� r� ru N)
rS rO rV rY rZ r$ r� rQ r� r= )r �pr^ r r r �test_pauseResume^ sF
z ProducerTests.test_pauseResumeN)r& r' r( r) rX r r r r rU Y rT rU 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 �
FileSenderr� 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 r� r rZ �beginFileTransferrZ �producerr� � 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 r� r rZ r^ r= � assertIsNoner_ rZ �successResultOfr� �r ra rb 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] r8 s Tests Test conrd N) r r r� r rZ �
CHUNK_SIZEr^ r= rZ r� rf rg 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 r� r rZ r^ r= rZ rf r� )r rn ra rb r[ rh r r r �test_transferWithTransform� 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 r� r rZ r^ r> �failureResultOf�trap� ExceptionrZ �strr� )r ra rb r[ rh �failurer r r �test_abortedTransfer� s
z$FileSenderTests.test_abortedTransferN)
r& r' r( r) r\ rc ri rk rp rv r r r r rY � s rY );r) r r �ior �typingr r r �zope.interface.verifyr �twisted.internetr r �twisted.internet.interfacesr �twisted.internet.protocolr
�twisted.protocolsr �twisted.python.compatr �twisted.python.failurer
� twisted.testr �
twisted.trialr r| r r+ r� rJ �SynchronousTestCaserN r� r� r� r� r r� r� r� r r �Int32StringReceiverr3 r6 �Int16StringReceiverr= r? �Int8StringReceiverrL rM rO rS rU �TestCaserY r r r r �<module> sN H yO, YIj
I