File: //usr/lib/python3/dist-packages/twisted/pair/test/__pycache__/test_ip.cpython-38.pyc
U
W[,; � @ sd d dl mZ d dlmZ d dlmZmZ d dlmZ e� ej
�G dd� d��Z G dd� dej �Z
d S )
� )�unittest)�
components)�ip�raw)� interfacec @ 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_ip.py�__init__ s zMyProtocol.__init__c K s� | j std��| j �d�\}}|�� }tt|��}|�� }tt|��}||ks^td||f ��|D ].}|| || ksbtd||| || f ��qb||ks�td||f ��||ks�td||f ��d S )Nz(Got a packet when not expecting anymore.r zExpected %r, got %rzExpected %s=%r, got %r)r �AssertionError�pop�keysr �sorted)r
�data�kwZ
expectDataZexpectKwZ expectKwKeysZkwKeys�kr 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 )�IPTestsc C sj t �� }tddddddddd d
d dd d
d�
fg�}|�d|� |jdddddd� |jrftd|j ��d S �N� foobarr �1.2.3.4�5.6.7.8� � � � � �� �� � � �
�partial�dest�source�protocol�versionZihlZtosZtot_lenZ fragment_idZfragment_offsetZ
dont_fragmentZmore_fragmentsZttl� T ޭ��FEfoobar�dummy�r( r) r* r+ �5Should not expect any more packets, but still want %r�r �
IPProtocolr �addProtor r r �r
�proto�p1r r r �testPacketParsing s8 �� ��zIPTests.testPacketParsingc C s� t �� }tddddddddd d
d dd d
d�
fdd ddddddd d
d dd d
d�
fg�}|�d|� |jdddddd� |jdd dddd� |jr�td|j ��d S )Nr r r r r r r r! r" r# r$ r% r& r' � quux�5.4.3.2�6.7.8.9r- r. r/ � T ޭ��FE quuxr0 r1 r4 r r r �testMultiplePacketsJ sb ���# � ��zIPTests.testMultiplePacketsc C s� t �� }tddddddddd d
d dd d
d�
fg�}tddddddddd d
d dd d
d�
fg�}|�d|� |�d|� |jdddddd� |jr�td|j ��|jr�td|j ��d S r r1 �r
r5 r6 Zp2r r r �testMultipleSameProtos� sb ���� ���z IPTests.testMultipleSameProtosc C s4 t �� }tg �}|�d|� |jdddddd� d S )Nr% r- r r. r/ )r r2 r r3 r r4 r r r �testWrongProtoNotSeen� s �z IPTests.testWrongProtoNotSeenc C s8 t �� }tddddddddd d
d dd d
d�
fdd ddddddd d
d dd d
d�
fg�}|�d|� tdd ddddddd d
d dd d
d�
fddddddddd d
d dd d
d�
fg�}|�d|� |jdd dddd� |jdddddd� |jdd dddd� |jdddddd� |j�r td|j ��|j�r4td|j ��d S )Nr r r r r r r r! r" r# r$ r% r& r' r8 r9 r: �
s T ޭ��
FE quuxr. r/ r- r; s T ޭ��
FEfoobarr0 r1 r= r r r � testDemuxing� s� ���# ���$ � � � �
�
�zIPTests.testDemuxingc C s< t �� }z|�dd� W n tjk
r. Y n
X td��dS )z2Adding a wrong level protocol raises an exception.�* Z silliness�2addProto must raise an exception for bad protocolsN)r r2 r3 r Z CannotAdaptr �r
�er r r � testAddingBadProtos_WrongLevel� s z&IPTests.testAddingBadProtos_WrongLevelc
C sZ t �� }z|�dtg �� 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 zeroNrC �r r2 r3 r � TypeError�argsr rD r r r � testAddingBadProtos_TooSmall� s
z$IPTests.testAddingBadProtos_TooSmallc
C sZ t �� }z|�dtg �� W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )�<Adding a protocol with a number >=2**32 raises an exception.l �z"Added protocol must fit in 32 bitsNrC rH rD r r r �testAddingBadProtos_TooBig� s
z"IPTests.testAddingBadProtos_TooBigc
C sZ t �� }z|�dtg �� W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )rL l rM NrC rH rD r r r �testAddingBadProtos_TooBig2� s
z#IPTests.testAddingBadProtos_TooBig2N) r r r r7 r< r> r? rA rF rK rN rO r r r r r s +KC
r N)Z
twisted.trialr Ztwisted.pythonr Z twisted.pairr r �zoper Z implementerZIRawDatagramProtocolr ZTestCaser r r r r �<module> s