File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_stateful.cpython-310.pyc
o
�b� � @ st d Z ddlmZmZmZ ddlmZ ddlmZ ddl m
Z
G dd� de�Z G dd � d e �Z G d
d � d e
ej
�Zd S )
z+
Test cases for twisted.protocols.stateful
� )�calcsize�pack�unpack)�StatefulProtocol)�
test_basic)�TestCasec @ 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 �
z%MyInt32StringReceiver.getInitialStatec C s | j �� d S �N)� transport�loseConnection)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 �)z
Override this.
)�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 �
sendString0 s z MyInt32StringReceiver.sendStringN)�__name__�
__module__� __qualname__�__doc__r r r � 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 �connectionMade8 r zTestInt32.connectionMadec C s | j �|� d S r )r) �append)r
�sr r r r ; s zTestInt32.stringReceived�2 r c C s
d| _ d S )N� )�closed)r
�reasonr r r �connectionLostA r zTestInt32.connectionLostN)r# r$ r% r* r r r/ r1 r r r r r( 7 s r( c @ s. e Zd ZeZddgZdgZg d�Zdd� ZdS )�
Int32Tests� as bbbbbbbbbbbbbbbbs
aaaaaa)s s hello there� c C sT | � � }d}| jd D ]
}|tdt|��| 7 }q |�|� | �|j| jd � d S )Nr4 r
r )� getProtocol�stringsr r � dataReceived� assertEqualr) )r
�r�bigr, r r r �test_bigReceiveK s
zInt32Tests.test_bigReceiveN) r# r$ r% r( �protocolr6 �illegalStrings�partialStringsr; r r r r r2 E s r2 N)r& �structr r r �twisted.protocols.statefulr �twisted.protocols.testr �twisted.trial.unittestr r r( �IntNTestCaseMixinr2 r r r r �<module> s '