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