File: //lib/python3/dist-packages/twisted/pair/test/__pycache__/test_ethernet.cpython-38.pyc
U
W[� � @ sb d dl mZ d dlmZ d dlmZmZ d dlmZ eej �G dd� d��Z
G dd� dej �Z d S )
� )�unittest)�
components)�ethernet�raw)� implementerc @ s e Zd Zdd� Zdd� ZdS )�
MyProtocolc C s t |�| _d S )N)�list� expecting)�selfr � r �A/usr/lib/python3/dist-packages/twisted/pair/test/test_ethernet.py�__init__ s zMyProtocol.__init__c K s> | j std��| j �d�}|||fks:td|||ff ��d S )Nz(Got a packet when not expecting anymore.r zExpected %r, got %r)r �AssertionError�pop)r
�data�kwZexpectr 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 )�
EthernetTestsc C sR t �� }tdddddd�fg�}|�d|� |jddd� |jrNtd |j ��d S �
N� foobarr � 123456� 987654� ��partial�dest�sourceZprotocol� 123456987654 foobar�r �5Should not expect any more packets, but still want %r�r �EthernetProtocolr �addProtor r r �r
�proto�p1r r r �testPacketParsing s ��
��z EthernetTests.testPacketParsingc C sp t �� }tdddddd�fddd d
dd�fg�}|�d|� |jd dd � |jd
dd � |jrltd|j ��d S )Nr r r r r r � quux� � 012345� abcdefr r! � 012345abcdef quuxr" r# r& r r r �testMultiplePackets- s2 ��� ���z!EthernetTests.testMultiplePacketsc C s� t �� }tdddddd�fg�}tdddddd�fg�}|�d|� |�d|� |jddd� |jrrtd |j ��|jr�td |j ��d S r r# �r
r' r( Zp2r r r �testMultipleSameProtosK s8 �� �� ���z$EthernetTests.testMultipleSameProtosc C s< t �� }tg �}|�d|� |jddd� |jddd� d S )Ni r r r! r. r+ )r r$ r r% r r& r r r �testWrongProtoNotSeenn s ��z#EthernetTests.testWrongProtoNotSeenc C s� t �� }tdddddd�fddd d
dd�fg�}|�d|� tddd d
d d�fddddd d�fg�}|�d |� |jd dd
� |jddd
� |jddd
� |jddd
� |jr�td|j ��|jr�td|j ��d S )Nr r r r r r r* r+ r, r- i r r! s 012345abcdefquuxs 123456987654foobarr. r" r# r0 r r r � testDemuxingx sb ��� ��� ������zEthernetTests.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 r$ r% r Z CannotAdaptr �r
�er r r � testAddingBadProtos_WrongLevel� s z,EthernetTests.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 zeroNr5 �r r$ r% r � TypeError�argsr r6 r r r � testAddingBadProtos_TooSmall� s
z*EthernetTests.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**16 raises an exception.i �z"Added protocol must fit in 16 bitsNr5 r: r6 r r r �testAddingBadProtos_TooBig� s
z(EthernetTests.testAddingBadProtos_TooBigc
C sZ t �� }z|�dtg �� W n0 tk
rL } z|jdkr:n� W 5 d}~X Y n
X td��dS )r> i r? Nr5 r: r6 r r r �testAddingBadProtos_TooBig2� s
z)EthernetTests.testAddingBadProtos_TooBig2N) r r r r) r/ r1 r2 r3 r8 r= r@ rA r r r r r s #
6
r N)
Z
twisted.trialr Ztwisted.pythonr Z twisted.pairr r Zzope.interfacer ZIRawPacketProtocolr ZTestCaser r r r r �<module> s