| Current File : //proc/self/root/lib/python3/dist-packages/twisted/test/__pycache__/test_stateful.cpython-312.pyc |
�
Ϫ�f� � � � d Z ddlmZmZmZ ddlmZ ddlmZ ddl m
Z
G d� de� Z G d� d e � Z G d
� d e
ej � Zy )
z+
Test cases for twisted.protocols.stateful
� )�calcsize�pack�unpack)�StatefulProtocol)�
test_basic)�TestCasec �L � e Zd ZdZdZdZ ee� Zd� Zd� Z d� Z
d� Z d� Z d � Z
y
) �MyInt32StringReceiverz)
A stateful Int32StringReceiver.
i�� z!Ic � � | j dfS �N� )�
_getHeader��selfs �</usr/lib/python3/dist-packages/twisted/test/test_stateful.py�getInitialStatez%MyInt32StringReceiver.getInitialState s � �����!�!� c �8 � | j j � y �N)� transport�loseConnection)r �lengths r �lengthLimitExceededz)MyInt32StringReceiver.lengthLimitExceeded s � � ���%�%�'r c �| � t d|� \ }|| j kD r| j |� y | j |fS )N�!i)r �
MAX_LENGTHr �
_getString)r �msgr s r r z MyInt32StringReceiver._getHeader s< � ��4��%� �� �D�O�O� #� � $� $�V� ,� �����&�&r c �@ � | j |� | j dfS r )�stringReceivedr �r r s r r z MyInt32StringReceiver._getString& s � � ���C� �����!�!r c � � t �)z
Override this.
)�NotImplementedErrorr! s r r z$MyInt32StringReceiver.stringReceived* s
� � "�!r c �z � | j j t | j t |� � |z � y)zS
Send an int32-prefixed string to the other end of the connection.
N)r �writer � structFormat�len)r �datas r �
sendStringz MyInt32StringReceiver.sendString0 s, � �
��� � �T�$�"3�"3�S��Y� ?�$� F�Gr N)�__name__�
__module__� __qualname__�__doc__r r& r � prefixLengthr r r r r r) � r r r
r
s< � �� �J��L��L�)�L�"�(�'�"�"� Hr r
c �& � e Zd Zd� Zd� ZdZdZd� Zy)� TestInt32c � � g | _ y r )�receivedr s r �connectionMadezTestInt32.connectionMade8 s � ���
r c �: � | j j |� y r )r3 �append)r �ss r r zTestInt32.stringReceived; s � � �
�
� � �Q� r �2 r c � � d| _ y )N� )�closed)r �reasons r �connectionLostzTestInt32.connectionLostA s � ��� r N)r* r+ r, r4 r r r; r= r/ r r r1 r1 7 s � �� � �J�
�F�r r1 c �, � e Zd ZeZddgZdgZg d�Zd� Zy)�
Int32Tests� as bbbbbbbbbbbbbbbbs
aaaaaa)s s hello therer c � � | j � }d}| j dz D ] }|t dt |� � |z z
}� |j |� | j |j | j dz � y )Nr r
r )� getProtocol�stringsr r' � dataReceived� assertEqualr3 )r �r�bigr7 s r �test_bigReceivezInt32Tests.test_bigReceiveK sp � � � � � ����� � ��!� *�A� �4��c�!�f�%��)� )�C� *� ���s�� ������T�\�\�A�%5�6r N) r* r+ r, r1 �protocolrC �illegalStrings�partialStringsrH r/ r r r? r? E s# � ��H��Y� �G�/�0�N�;�N�7r r? N)r- �structr r r �twisted.protocols.statefulr �twisted.protocols.testr �twisted.trial.unittestr r
r1 �IntNTestCaseMixinr? r/ r r �<module>rQ sL ��
� *� )� 7� -� +�$H� ,� $H�N �%� � 7��:�7�7� 7r