File: //usr/lib/python3/dist-packages/twisted/pair/test/__pycache__/test_rawudp.cpython-38.pyc
U
W[3 � @ sL d dl mZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�Z dS ) � )�unittest)�protocol)�rawudpc @ s e Zd Zdd� Zdd� ZdS )�
MyProtocolc C s t |�| _d S )N)�list� expecting)�selfr � r �?/usr/lib/python3/dist-packages/twisted/pair/test/test_rawudp.py�__init__ s zMyProtocol.__init__c C sx |\}}| j std��| j �d�\}}}||ks@td||f ��||ksXtd||f ��||ksttd||||f ��d S )Nz(Got a packet when not expecting anymore.r zExpected data %r, got %rzExpected host %r, got %rz&Expected port %d=0x%04x, got %d=0x%04x)r �AssertionError�pop)r �dataZpeerZhostZportZ
expectDataZ
expectHostZ
expectPortr r r
�datagramReceived s zMyProtocol.datagramReceivedN)�__name__�
__module__� __qualname__r r r r r r
r
s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )� RawUDPTestsc C s\ t �� }tdg�}|�d|� |jddddddddddddddd� |jrXtd|j ��d S � N�� foobar� testHosti�C �� � C�� ޭfoobarr � dummyr �
�partial�dest�sourcer �versionZihlZtosZtot_lenZ fragment_idZfragment_offsetZ
dont_fragmentZmore_fragmentsZttl�5Should not expect any more packets, but still want %r�r �RawUDPProtocolr �addProtor r r �r �proto�p1r r r
�testPacketParsing s. � ��z RawUDPTests.testPacketParsingc C s� t �� }tddg�}|�d|� |jddddddddddddddd� |jd ddd
ddddddddddd� |jr�td |j ��d S ) Nr �� quux� otherHosti�3 r r r r r r � 3�� ޭquuxr* r r! r$ r r r
�testMultiplePackets8 sN � ���z RawUDPTests.testMultiplePacketsc C s� t �� }tdg�}tdg�}|�d|� |�d|� |jddddddddddddddd� |jrntd|j ��|jr�td|j ��d S r r! �r r% r& Zp2r r r
�testMultipleSameProtosl s< �� ���z"RawUDPTests.testMultipleSameProtosc C sF t �� }tg �}|�d|� |jddddddddddddddd� d S )N� r r r r r )r r" r r# r r$ r r r
�testWrongProtoNotSeen� s$ �z!RawUDPTests.testWrongProtoNotSeenc C s� t �� }tddg�}|�d|� tddg�}|�d|� |jddd d
d d d d d d d d d d d � |jd dd d
d d d d d d d d d d d � |jddd d
d d d d d d d d d d d � |jddd d
d d d d d d d d d d d � |jr�td|j ��|jr�td|j ��d S )Nr r( r )r) r* i� )r r i� iP� s ��P ޭquuxr r r* r r r r+ s ��P ޭfoobarr r! r- r r r
� testDemuxing� s� � � ������zRawUDPTests.testDemuxingc
C sV t �� }z|�dd� W n0 tk
rH } z|jdkr6n� W 5 d}~X Y n
X td��dS )z2Adding a wrong level protocol raises an exception.�* Z silliness)z6Added protocol must be an instance of DatagramProtocolN�2addProto must raise an exception for bad protocols)r r" r# � TypeError�argsr �r �er r r
� testAddingBadProtos_WrongLevel s
z*RawUDPTests.testAddingBadProtos_WrongLevelc
C sZ t �� }z|�dt�� � W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )z=Adding a protocol with a negative number raises an exception.���)z'Added protocol must be positive or zeroNr3 �r r" r# r �DatagramProtocolr4 r5 r r6 r r r
� testAddingBadProtos_TooSmall! s
z(RawUDPTests.testAddingBadProtos_TooSmallc
C sZ t �� }z|�dt�� � W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )�<Adding a protocol with a number >=2**16 raises an exception.i �z"Added protocol must fit in 16 bitsNr3 r: r6 r r r
�testAddingBadProtos_TooBig/ s
z&RawUDPTests.testAddingBadProtos_TooBigc
C sZ t �� }z|�dt�� � W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )r= i r> Nr3 r: r6 r r r
�testAddingBadProtos_TooBig2< s
z'RawUDPTests.testAddingBadProtos_TooBig2N) r r r r' r, r. r0 r1 r8 r< r? r@ r r r r
r s 4*d
r N)
Z
twisted.trialr Ztwisted.internetr Z twisted.pairr r; r ZTestCaser r r r r
�<module> s