Current File : //usr/lib/python3/dist-packages/twisted/pair/test/__pycache__/test_rawudp.cpython-312.pyc
�
Ϫ�f�) � � � d dl mZ d dlmZ d dlmZ d dlmZ G d� dej � Z G d� dej � Z y )
� )� annotations)�protocol)�rawudp)�unittestc � � e Zd Zdd�Zdd�Zy)�
MyProtocolc �$ � t |� | _ y )N)�list� expecting)�selfr s �?/usr/lib/python3/dist-packages/twisted/pair/test/test_rawudp.py�__init__zMyProtocol.__init__
s
� � �i� ��� c � � |\ }}| j sJ d� �| j j d� \ }}}||k( sJ dj ||� � �||k( sJ dj ||� � �||k( sJ d||||fz � �y )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�
expectPorts r
�datagramReceivedzMyProtocol.datagramReceived s� � �� ��t��~�~�I� I�I�~�-1�^�^�-?�-?��-B�*�
�J�
��T�!�
�#A�#H�#H� ��$
�
�!� �T�!�
�#A�#H�#H� ��$
�
�!� �T�!�
�#K� � � � � O
� $
�
�!r N)r z list[tuple[bytes, bytes, int]]�return�None)r �bytesr ztuple[bytes, int]r r )�__name__�
__module__� __qualname__r r � r r
r r s � �)�
r r c �T � e Zd Zd d�Zd d�Zd d�Zd d�Zd d�Zd d�Zd d�Z d d�Z
d d �Z y
) � RawUDPTestsc �� � t j � }t dg� }|j d|� |j dddddddddddddd�� |j
rJ d|j
z � �y � 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�p1s r
�testPacketParsingz RawUDPTests.testPacketParsing$ s� � ��%�%�'��
�0�
�
��
���v�r�"�
� � �
�
�� � � ��� � � $�"�#��% �
�* �<�<�
� C�b�l�l� R�
� �<r c �& � t j � }t ddg� }|j d|� |j dddddddddddddd�� |j d ddd
dddddddddd�� |j
rJ d |j
z � �y ) Nr& �� quux� otherHosti�3 r) r* r r+ r( r, � 3�� ޭquuxrD r9 r: r= s r
�testMultiplePacketsz RawUDPTests.testMultiplePacketsF s� � ��%�%�'��
�0�/�
�
�� ���v�r�"�
� � �
�
�� � � ��� � � $�"�#��% �
�( � � �
�
�� � � ��� � � $�"�#��% �
�* �<�<�
� C�b�l�l� R�
� �<r c �b � t j � }t dg� }t dg� }|j d|� |j d|� |j dddddddddddddd�� |j
rJ d|j
z � �|j
rJ d|j
z � �y r% r: �r r> r? �p2s r
�testMultipleSameProtosz"RawUDPTests.testMultipleSameProtos| s� � ��%�%�'��
�0�
�
�� �0�
�
�� ���v�r�"�
���v�r�"�
� � �
�
�� � � ��� � � $�"�#��% �
�* �<�<�
� C�b�l�l� R�
� � �<�<�
� C�b�l�l� R�
� �<r c � � t j � }t g � }|j d|� |j dddddddddddddd�� y )N� r* r r+ r( r, )r r; r r< r r= s r
�testWrongProtoNotSeenz!RawUDPTests.testWrongProtoNotSeen� sh � ��%�%�'��
��^��
���q�"� �
� � �
�
�� � � ��� � � $�"�#��% �
r c � � t j � }t ddg� }|j d|� t ddg� }|j d|� |j ddd d
d d d d d d d d d d � � |j d dd d
d d d d d d d d d d � � |j ddd d
d d d d d d d d d d � � |j ddd d
d d d d d d d d d d � � |j
rJ d|j
z � �|j
rJ d|j
z � �y )Nr& rB r) )rC rD i� )r' r( i� iP� s ��P ޭquuxr r+ rD r, r* r( rE s ��P ޭfoobarr9 r: rH s r
� testDemuxingzRawUDPTests.testDemuxing� s� � ��%�%�'��
�0�/�
�
�� ���v�r�"�
�/�0�
�
�� ���v�r�"�
� � �
�
�� � � ��� � � $�"�#��% �
�( � � �
�
�� � � ��� � � $�"�#��% �
�( � � �
�
�� � � ��� � � $�"�#��% �
�( � � �
�
�� � � ��� � � $�"�#��% �
�* �<�<�
� C�b�l�l� R�
� � �<�<�
� C�b�l�l� R�
� �<r c � � t j � } |j dd� t d� �# t $ r}|j
dk( rn� Y d}~yd}~ww xY w)z2Adding a wrong level protocol raises an exception.�* � silliness�2addProto must raise an exception for bad protocols)z6Added protocol must be an instance of DatagramProtocolN)r r; r< �AssertionError� TypeError�args�r �es r
� testAddingBadProtos_WrongLevelz*RawUDPTests.testAddingBadProtos_WrongLevel- s\ � � � !� !� #�� W�
�J�J�r�;� '� !�!U�V� V��
� ��v�v�T�T��� �� �s �3 � A�A�Ac �� � t j � } |j dt j � � t d� �# t $ r}|j dk( rn� Y d}~yd}~ww xY w)z=Adding a protocol with a negative number raises an exception.���rS )z'Added protocol must be positive or zeroN�r r; r< r �DatagramProtocolrT rU rV rW s r
� testAddingBadProtos_TooSmallz(RawUDPTests.testAddingBadProtos_TooSmall: se � � � !� !� #�� W�
�J�J�r�8�4�4�6� 7� !�!U�V� V��
� ��v�v�E�E��� �� �� �$A � A)�A$�$A)c �� � t j � } |j dt j � � t d� �# t $ r}|j dk( rn� Y d}~yd}~ww xY w)�<Adding a protocol with a number >=2**16 raises an exception.i rS �z"Added protocol must fit in 16 bitsNr\ rW s r
�testAddingBadProtos_TooBigz&RawUDPTests.testAddingBadProtos_TooBigG se � � � !� !� #�� W�
�J�J�u�h� 7� 7� 9� :� !�!U�V� V��
� ��v�v�@�@��� �� �r_ c �� � t j � } |j dt j � � t d� �# t $ r}|j dk( rn� Y d}~yd}~ww xY w)ra i rS rb Nr\ rW s r
�testAddingBadProtos_TooBig2z'RawUDPTests.testAddingBadProtos_TooBig2T se � � � !� !� #�� W�
�J�J�y�(�";�";�"=� >� !�!U�V� V��
� ��v�v�@�@��� �� �r_ N)r r ) r r r r@ rF rJ rM rO rY r^ rc re r! r r
r# r# # s8 � �
�D4
�l+
�Z
�4h
�T W� W� W� Wr r# N) �
__future__r �twisted.internetr � twisted.pairr �
twisted.trialr r] r �TestCaser# r! r r
�<module>rk s= �� #� %� � #�
��*�*�
�.|W�(�#�#� |Wr
Mini Shell