Current File : //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_udp_internals.cpython-312.pyc
�
Ϫ�f� � �� � d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl
m Z e dk( rdd l m
Z ndd
l mZ G d � d � Z G d
� de� Z G d� de j" � Zy)zK
Tests for the internal implementation details of L{twisted.internet.udp}.
� )� annotationsN)�udp)�DatagramProtocol)� platformType)�unittest�win32)�WSAEWOULDBLOCK)� EWOULDBLOCKc �( � e Zd ZdZdd�Zdd�Zdd�Zy) �StringUDPSocketa
A fake UDP socket object, which returns a fixed sequence of strings and/or
socket errors. Useful for testing.
@ivar retvals: A C{list} containing either strings or C{socket.error}s.
@ivar connectedAddr: The address the socket is connected to.
c � � || _ d | _ y �N)�retvals�
connectedAddr)�selfr s �J/usr/lib/python3/dist-packages/twisted/internet/test/test_udp_internals.py�__init__zStringUDPSocket.__init__ s � � �� �,0��� c � � || _ y r )r )r �addrs r �connectzStringUDPSocket.connect$ s
� � !��r c �x � | j j d� }t |t j � r|�|dfS )zH
Return (or raise) the next value from C{self.retvals}.
r N)r �pop�
isinstance�socket�error)r �size�rets r �recvfromzStringUDPSocket.recvfrom' s5 � � �l�l� � �q�!�� �c�6�<�<� (��I��D�y�r N)r zlist[bytes | socket.error]�return�None)r �objectr r! )r �intr ztuple[bytes, None])�__name__�
__module__� __qualname__�__doc__r r r � r r r r s � ��1�"�r r c � � e Zd ZdZdd�Zdd�Zy)� KeepReadsz%
Accumulate reads in a list.
c � � g | _ y r )�reads)r s r r zKeepReads.__init__6 s � �"$��
r c �: � | j j |� y r )r, �append)r �datar s r �datagramReceivedzKeepReads.datagramReceived9 s � � �
�
���$� r N�r r! )r/ �bytesr r" r r! )r$ r% r& r' r r0 r( r r r* r* 1 s � ��%� r r* c �0 � e Zd ZdZdd�Zdd�Zdd�Zdd�Zy)� ErrorsTestsz/
Error handling tests for C{udp.Port}.
c � � t j j d� | j t j j d� t � }t j d|� }t ddt j d� dt j d� g� |_ |j � | j |j ddg� |j � | j |j g d�� y)z�
Socket reads with some good data followed by a socket error which can
be ignored causes reading to stop, and no log messages to be logged.
i��N� result� 123� 456)r6 r7 r8 )
r �_sockErrReadIgnorer. �
addCleanup�remover* �Portr r r �doRead� assertEqualr, �r �protocol�ports r �test_socketReadNormalz!ErrorsTests.test_socketReadNormalB s� � � � � �%�%�e�,� ����.�.�5�5�u�=� �;���x�x��h�'�� &�
��� � �U� 3�V�V�\�\�%�=P� Q�
�� �
� � �
� ���� � �)�V�)<�=� � � �
� ���� � �)D�Er c � � t j j d� | j t j j d� t � }d� |_ t j d|� }t dt j d� dt j t � g� |_ |j � | j |j dg� |j � | j |j ddg� y)z�
If the socket is unconnected, socket reads with an immediate
connection refusal are ignored, and reading stops. The protocol's
C{connectionRefused} method is not called.
��c � � ddz S )N� r r( r( r r �<lambda>z5ErrorsTests.test_readImmediateError.<locals>.<lambda>e s
� �Q��U� r N� a� b)r �_sockErrReadRefuser. r: r; r* �connectionRefusedr< r r r r
r= r> r, r? s r �test_readImmediateErrorz#ErrorsTests.test_readImmediateErrorX s� � � � � �%�%�e�,� ����.�.�5�5�u�=� �;��%2��"��x�x��h�'�� &�
�6�<�<��&��f�l�l�;�.G� H�
�� �
� � �
� ���� � �$��0� � � �
� ���� � �$�� �6r c � �� t j j d� | j t j j d� t � }g ��fd�|_ t j d|� }t dt j d� dt j t � g� |_ |j dd� |j � | j |j dg� | j �dg� |j � | j |j ddg� | j �dg� y) z�
If the socket connected, socket reads with an immediate
connection refusal are ignored, and reading stops. The protocol's
C{connectionRefused} method is called.
rD c �& �� � j d� S )NT)r. )�refuseds �r rG z>ErrorsTests.test_connectedReadImmediateError.<locals>.<lambda>� s �� �W�^�^�D�-A� r NrH rI z 127.0.0.1i' T)r rJ r. r: r; r* rK r< r r r r
r r= r> r, )r r@ rA rO s @r � test_connectedReadImmediateErrorz,ErrorsTests.test_connectedReadImmediateErrort s� �� � � � �%�%�e�,� ����.�.�5�5�u�=� �;����%A��"��x�x��h�'��%�
�6�<�<��&��f�l�l�;�.G� H�
�� �
� � �[�$�'�
� � �
� ���� � �$��0� ����4�&�)�
� � �
� ���� � �$�� �6� ����4�&�)r c � � t � }t j d|� }t dt j
d� g� |_ | j
t j
|j � | j |j dg� y)zG
Socket reads with an unknown socket error are raised.
Ns goodi���)
r* r r< r r r � assertRaisesr= r> r, r? s r �test_readUnknownErrorz!ErrorsTests.test_readUnknownError� se � � �;���x�x��h�'�� &�w�� � �U�0C�&D�E�� � ���&�,�,�� � �4� ���� � �'��3r Nr1 )r$ r% r& r' rB rL rP rS r( r r r4 r4 = s � ��F�,7�8 *�>
4r r4 )r' �
__future__r r �twisted.internetr �twisted.internet.protocolr �twisted.python.runtimer �
twisted.trialr �errnor r
r r* �SynchronousTestCaser4 r( r r �<module>r[ sW ��� #�
� � 6� /� "��7��3�!�� �6 � � �`4�(�.�.� `4r
Mini Shell