File: //lib/python3/dist-packages/twisted/pair/test/__pycache__/test_rawudp.cpython-310.pyc
o
�b�( � @ sL d dl mZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�Z dS ) � )�protocol)�rawudp)�unittestc @ 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 s J d��| j �d�\}}}||ks J d�||���||ks,J d�||���||ks:J d||||f ��d S )Nz(Got a packet when not expecting anymore.r z Expected data {!r}, got {!r}z Expected host {!r}, got {!r}z&Expected port %d=0x%04x, got %d=0x%04x)r �pop�format)r �data�peer�host�port�
expectData�
expectHost�
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� |jr,J d|j ��d S � N�� foobar� testHosti�C �� � C�� ޭfoobarr � dummyr �
�partial�dest�sourcer �version�ihl�tos�tot_len� fragment_id�fragment_offset�
dont_fragment�more_fragments�ttl�5Should not expect any more packets, but still want %r�r �RawUDPProtocolr �addProtor r �r �proto�p1r r r
�testPacketParsing# s2 �� ��z RawUDPTests.testPacketParsingc C s� t �� }tddg�}|�d|� |jddddddddddddddd� |jd ddd
ddddddddddd� |jr@J d |j ��d S ) Nr �� quux� otherHosti�3 r r r r r r! � 3�� ޭquuxr8 r. r/ r2 r r r
�testMultiplePacketsE sT �� ���z RawUDPTests.testMultiplePacketsc C s� t �� }tdg�}tdg�}|�d|� |�d|� |jddddddddddddddd� |jr7J d|j ��|jrAJ d|j ��d S r r/ �r r3 r4 �p2r r r
�testMultipleSameProtos{ sB ���� ���z"RawUDPTests.testMultipleSameProtosc C sF t �� }tg �}|�d|� |jddddddddddddddd� d S )N� r r r r r! )r r0 r r1 r r2 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 � |jrrJ d|j ��|jr|J d|j ��d S )Nr r6 r )r7 r8 i� )r r i� iP� s ��P ޭquuxr r r8 r! r r r9 s ��P ޭfoobarr. r/ r; r r r
� testDemuxing� s� �� �� ������zRawUDPTests.testDemuxingc
C sT t �� }z |�dd� W td�� ty) } z|jdkr n� W Y d}~dS d}~ww )z2Adding a wrong level protocol raises an exception.�* � silliness)z6Added protocol must be an instance of DatagramProtocolN�2addProto must raise an exception for bad protocols)r r0 r1 � TypeError�args�AssertionError�r �er r r
� testAddingBadProtos_WrongLevel, s �
���z*RawUDPTests.testAddingBadProtos_WrongLevelc
C �X t �� }z
|�dt�� � 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 zeroNrC �r r0 r1 r �DatagramProtocolrD rE rF rG r r r
� testAddingBadProtos_TooSmall9 � �
���z(RawUDPTests.testAddingBadProtos_TooSmallc
C rJ )�<Adding a protocol with a number >=2**16 raises an exception.i �z"Added protocol must fit in 16 bitsNrC rL rG r r r
�testAddingBadProtos_TooBigF rO z&RawUDPTests.testAddingBadProtos_TooBigc
C rJ )rP i rQ NrC rL rG r r r
�testAddingBadProtos_TooBig2S rO z'RawUDPTests.testAddingBadProtos_TooBig2N) r r r r5 r: r= r? r@ rI rN rR rS r r r r
r " s "6-j
r N)
�twisted.internetr � twisted.pairr �
twisted.trialr rM r �TestCaser r r r r
�<module> s