File: //lib/python3/dist-packages/twisted/pair/test/__pycache__/test_ip.cpython-310.pyc
o
�b�; � @ sd d dl mZ d dlmZmZ d dlmZ d dlmZ e� ej
�G dd� d��Z G dd� dej �Z
d S )
� )� interface)�ip�raw)�
components)�unittestc @ s$ e Zd Zdd� 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 C s� | j sJ d��| j �d�\}}|�� }tt|��}t� }|D ]}|| || ks9J d|� d|| �d|| ����q ||ksHJ d|�d|����d S )Nz(Got a packet when not expecting anymore.r z Expected �=z, got )r
�pop�keysr �sorted�locals)r �data�partial�source�dest�protocol�version�ihl�tos�tot_len� fragment_id�fragment_offset�
dont_fragment�more_fragments�ttl�
expectData�expectKw� expectKwKeys�localVariables�kr r r
�datagramReceived s � zMyProtocol.datagramReceivedc C s d S r r )r �num�protor r r
�addProto- s zMyProtocol.addProtoN)�__name__�
__module__� __qualname__r 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� |jr3J d|j ��d S �N� foobarr �1.2.3.4�5.6.7.8� � � � � �� �� � � �
r r r r r r r r r r r r r! � T ޭ��FEfoobar�dummy�r r r r �5Should not expect any more packets, but still want %r�r �
IPProtocolr r* r' r
�r r) �p1r r r
�testPacketParsing3 s> ���� ��zIPTests.testPacketParsingc C s� t �� }tddddddddd d
d dd d
d�
fdd ddddddd d
d dd d
d�
fg�}|�d|� |jdddddd� |jdd dddd� |jrNJ d|j ��d S )Nr0 r r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< � quux�5.4.3.2�6.7.8.9r= r> r? � T ޭ��FE quuxr@ rA rC r r r
�testMultiplePacketsc sl ������ ( � ��zIPTests.testMultiplePacketsc C s� t �� }tddddddddd d
d dd d
d�
fg�}tddddddddd d
d dd d
d�
fg�}|�d|� |�d|� |jdddddd� |jrNJ d|j ��|jrXJ d|j ��d S r/ rA �r r) rD �p2r r r
�testMultipleSameProtos� sl �������� ���z IPTests.testMultipleSameProtosc C s4 t �� }tg �}|�d|� |jdddddd� d S )Nr: r= r r> r? )r rB r r* r' rC r r r
�testWrongProtoNotSeen s
�z IPTests.testWrongProtoNotSeenc C s4 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� |jr�J d|j ��|jr�J d|j ��d S )Nr0 r r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< rF rG rH �
s T ޭ��
FE quuxr> r? r= rI s T ޭ��
FEfoobarr@ rA rK r r r
� testDemuxing s� ������ (������ ( � � � ���zIPTests.testDemuxingc C s6 t �� }z |�dd� W td�� tjy Y dS w )z2Adding a wrong level protocol raises an exception.�* � silliness�2addProto must raise an exception for bad protocolsN)r rB r* r � CannotAdapt�AssertionError�r �er r r
� testAddingBadProtos_WrongLevel� s ��z&IPTests.testAddingBadProtos_WrongLevelc
C �X t �� }z
|�dtg �� W td�� ty+ } z|jdkr n� W Y d}~dS d}~ww )z=Adding a protocol with a negative number raises an exception.���)z'Added protocol must be positive or zeroNrS �r rB r* r � TypeError�argsrU rV r r r
� testAddingBadProtos_TooSmall� � �
���z$IPTests.testAddingBadProtos_TooSmallc
C rY )�<Adding a protocol with a number >=2**32 raises an exception.l �z"Added protocol must fit in 32 bitsNrS r[ rV r r r
�testAddingBadProtos_TooBig� r_ z"IPTests.testAddingBadProtos_TooBigc
C rY )r` l ra NrS r[ rV r r r
�testAddingBadProtos_TooBig2� r_ z#IPTests.testAddingBadProtos_TooBig2N) r+ r, r- rE rJ rM rN rP rX r^ rb rc r r r r
r. 2 s 0RL #
r. N)�zoper � twisted.pairr r �twisted.pythonr �
twisted.trialr � implementer�IRawDatagramProtocolr �TestCaser. r r r r
�<module> s
'