File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_stateful.cpython-38.pyc
U
W[� � @ st d Z ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
G dd� de�Z G dd � d e �Z G d
d � d eej
�Zd S )
z+
Test cases for twisted.protocols.stateful
� )�TestCase)�
test_basic)�StatefulProtocol)�pack�unpack�calcsizec @ sP e Zd ZdZdZdZee�Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dS )�MyInt32StringReceiverz)
A stateful Int32StringReceiver.
i�� z!Ic C s
| j dfS �N� )�
_getHeader��self� r �</usr/lib/python3/dist-packages/twisted/test/test_stateful.py�getInitialState s z%MyInt32StringReceiver.getInitialStatec C s | j �� d S �N)� transportZloseConnection)r
�lengthr r r �lengthLimitExceeded s z)MyInt32StringReceiver.lengthLimitExceededc C s. t d|�\}|| jkr$| �|� d S | j|fS )N�!i)r �
MAX_LENGTHr �
_getString)r
�msgr r r r r s
z MyInt32StringReceiver._getHeaderc C s | � |� | jdfS r )�stringReceivedr �r
r r r r r ( s
z MyInt32StringReceiver._getStringc C s t �dS )z
Override this.
N)�NotImplementedErrorr r r r r - s z$MyInt32StringReceiver.stringReceivedc C s | j �t| jt|��| � dS )zS
Send an int32-prefixed string to the other end of the connection.
N)r �writer � structFormat�len)r
�datar r r �
sendString4 s z MyInt32StringReceiver.sendStringN)�__name__�
__module__� __qualname__�__doc__r r r Z prefixLengthr r r r r r r r r r r s r c @ s, e Zd Zdd� Zdd� ZdZdZdd� Zd S )
� TestInt32c C s
g | _ d S r )�receivedr r r r �connectionMade= s zTestInt32.connectionMadec C s | j �|� d S r )r& �append)r
�sr r r r A s zTestInt32.stringReceived�2 r c C s
d| _ d S )N� )�closed)r
�reasonr r r �connectionLostH s zTestInt32.connectionLostN)r! r" r# r' r r r, r. r r r r r% < s
r% c @ s0 e Zd ZeZddgZdgZdddgZdd� Zd S )
�
Int32Tests� as bbbbbbbbbbbbbbbbs
aaaaaas s hello there� c C sT | � � }d}| jd D ]}|tdt|��| 7 }q|�|� | �|j| jd � d S )N� r
r )Z getProtocol�stringsr r Z dataReceivedZ assertEqualr& )r
�rZbigr) r r r �test_bigReceiveS s
zInt32Tests.test_bigReceiveN) r! r" r# r% Zprotocolr3 ZillegalStringsZpartialStringsr5 r r r r r/ M s
r/ N)r$ Ztwisted.trial.unittestr Ztwisted.protocols.testr Ztwisted.protocols.statefulr Zstructr r r r r% ZIntNTestCaseMixinr/ r r r r �<module> s ,