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