File: //usr/local/lib/python3.7/test/__pycache__/test_timeout.cpython-37.pyc
B
��g�, � @ s� d Z ddlZddlZddlmZ e�d� ZddlZddlZddl Z e�
� dd� �Z G dd� dej �Z
G d d
� d
ej �ZG d d � d e�ZG d
d� de�Zdd� Zedkr�e� dS )z&Unit tests for socket timeout feature.� N)�support�networkc C s2 t �| �� t�| |tjtj�d d S Q R X dS )z�Resolve an (host, port) to an address.
We must perform name resolution before timeout tests, otherwise it will be
performed by connect().
r � N)r �transient_internet�socketZ getaddrinfo�AF_INET� SOCK_STREAM)ZhostZport� r �-/usr/local/lib/python3.7/test/test_timeout.py�resolve_address s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�CreationTestCasez9Test case for socket.gettimeout() and socket.settimeout()c C s t � t jt j�| _d S )N)r r r �sock)�selfr r r
�setUp s zCreationTestCase.setUpc C s | j �� d S )N)r
�close)r r r r
�tearDown! s zCreationTestCase.tearDownc C s | � | j�� d d� d S )Nz timeout not disabled by default)� assertEqualr
�
gettimeout)r r r r
�testObjectCreation$ s z#CreationTestCase.testObjectCreationc C s^ | j �d� | �| j �� d� | j �d� | �| j �� d� | j �d � | �| j �� d � d S )Ng�z�Ga @� )r
�
settimeoutr r )r r r r
�testFloatReturnValue) s z%CreationTestCase.testFloatReturnValuec C sP | j �d� | �t| j �� �td�� | j �d� | �t| j �� �td�� d S )N� g �?g333333@)r
r r �typer )r r r r
�testReturnType4 s z CreationTestCase.testReturnTypec C s� | j �d� | j �d� | j �d� | j �d � | �t| j jd� | �t| j jd� | �t| j jd� | �t| j jg � | �t| j ji � | �t| j jd� d S )Nr g � r y )r
r � assertRaises� TypeError)r r r r
�
testTypeCheck< s z CreationTestCase.testTypeCheckc C s: | � t| jjd� | � t| jjd� | � t| jjd� d S )N���g �)r �
ValueErrorr
r )r r r r
�testRangeCheckI s z CreationTestCase.testRangeCheckc C s� | j �d� | j �d� | �| j �� d � | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� | j �d� | �| j �� d � d S )N�
r r g )r
r � setblockingr r )r r r r
�testTimeoutThenBlockingO s z(CreationTestCase.testTimeoutThenBlockingc C sX | j �d� | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� d S )Nr r )r
r# r r r )r r r r
�testBlockingThenTimeout] s z(CreationTestCase.testBlockingThenTimeoutN)
�__name__�
__module__� __qualname__�__doc__r r r r r r r! r$ r% r r r r
r s
r c @ s* e Zd ZdZejZdd� ZeZdd� Z dS )�TimeoutTestCaseg @c C s
t � �d S )N)�NotImplementedError)r r r r
r s s zTimeoutTestCase.setUpc G s� | j �|� t| j |�}xbt|�D ]L}t�� }y ||� W q" tjk
rl } zt�� | }P W dd}~X Y q"X q"W | �d� | � ||| j
� | � ||d � dS )z�
Test the specified socket method.
The method is run at most `count` times and must raise a socket.timeout
within `timeout` + self.fuzz seconds.
Nz socket.timeout was not raisedg �?) r
r �getattr�range�timeZ monotonicr �timeoutZfailZ
assertLess�fuzzZ
assertGreater) r �countr/ �method�args�iZt1�eZdeltar r r
�_sock_operationx s
z TimeoutTestCase._sock_operationN)
r& r' r( r0 r ZHOST� localhostr r r6 r r r r
r* h s
r* c @ s\ e Zd ZdZdd� Zdd� Ze�dd�dd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )�TCPTimeoutTestCasez3TCP test case for socket.socket() timeout functionsc C s" t � t jt j�| _tdd�| _d S )Nzwww.python.org.�P )r r r r
r � addr_remote)r r r r
r � s zTCPTimeoutTestCase.setUpc C s | j �� d S )N)r
r )r r r r
r � s zTCPTimeoutTestCase.tearDownTz*need to replace these hosts; see bpo-35518c
C s t dd�}t dd�}d}t�tjtj�}d}|�|� zZy|�|� W nF tjk
r\ Y n2 tk
r� } z|jtj kr|d}W d d }~X Y nX W d |�
� ~X |r�| � d� |d |d
||d |d
�� || _
t�| j
d �� | �d
d d | j
� W d Q R X d S )
Nzblackhole.snakebite.netiZ� zwhitehole.snakebite.neti[� Tr Fz�We didn't receive a connection reset (RST) packet from {}:{} within {} seconds, so we're unable to test connect timeout against the corresponding {}:{} (which is configured to silently drop packets).r r g����MbP?�connect)r r r r r r; r/ �OSError�errnoZ ECONNREFUSEDr ZskipTest�formatr: r r r6 )r Z blackholeZ whitehole�skipr
r/ �errr r r
�testConnectTimeout� s4
z%TCPTimeoutTestCase.testConnectTimeoutc C s> t �| jd ��$ | j�| j� | �dddd� W d Q R X d S )Nr r g �?Zrecvi )r r r: r
r; r6 )r r r r
�testRecvTimeout� s z"TCPTimeoutTestCase.testRecvTimeoutc C s, t �| j| j� | j�� | �ddd� d S )Nr g �?Zaccept)r � bind_portr
r7 �listenr6 )r r r r
�testAcceptTimeout� s
z$TCPTimeoutTestCase.testAcceptTimeoutc C sZ t � t jt j��@}t�|| j� |�� | j�|� � � | �
ddddd � W d Q R X d S )N�d g �?�send� Xi@
) r r r r rC r7 rD r
r; � getsocknamer6 )r �servr r r
�testSend� s
zTCPTimeoutTestCase.testSendc C s` t � t jt j��F}t�|| j� |�� | j�|� � � | �
ddddd |� � � W d Q R X d S )NrF g �?ZsendtorH i@
) r r r r rC r7 rD r
r; rI r6 )r rJ r r r
�
testSendto s z TCPTimeoutTestCase.testSendtoc C sZ t � t jt j��@}t�|| j� |�� | j�|� � � | �
ddddd � W d Q R X d S )NrF g �?ZsendallrH i@
) r r r r rC r7 rD r
r; rI r6 )r rJ r r r
� testSendall
s
z TCPTimeoutTestCase.testSendallN)r& r' r( r) r r �unittestZskipIfrA rB rE rK rL rM r r r r
r8 � s U
r8 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UDPTimeoutTestCasez3UDP test case for socket.socket() timeout functionsc C s t � t jt j�| _d S )N)r r Z
SOCK_DGRAMr
)r r r r
r s zUDPTimeoutTestCase.setUpc C s | j �� d S )N)r
r )r r r r
r s zUDPTimeoutTestCase.tearDownc C s$ t �| j| j� | �dddd� d S )Nr g �?Zrecvfromi )r rC r
r7 r6 )r r r r
�testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r& r' r( r) r r rP r r r r
rO s rO c C s t �d� t �ttt� d S )Nr )r ZrequiresZ run_unittestr r8 rO r r r r
� test_main' s
rQ �__main__)r) � functoolsrN Ztestr Zis_resource_enabledZ
skip_expectedr. r= r � lru_cacher ZTestCaser r* r8 rO rQ r& r r r r
�<module> s M'