← Back
Editing: 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 ,
Save File
Cancel