Your IP : 216.73.216.52
U
i�f�, � @ 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 s> t �| ��* t�| |tjtj�d d W 5 Q R � 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 �6/opt/alt/python38/lib64/python3.8/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 r
�r �closer r r r
�tearDown! s zCreationTestCase.tearDownc C s | � | j�� d d� d S )Nz timeout not disabled by default)� assertEqualr �
gettimeoutr 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� � �?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 � � r y )r r � assertRaises� TypeErrorr 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 r )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 r
)�NotImplementedErrorr r r r
r s s zTimeoutTestCase.setUpc G s� | j �|� t| j |�}t|�D ]T}t�� }z ||� W q tjk
rr } zt�� | }W Y � q�W 5 d}~X Y q X q | �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 raisedr ) r r �getattr�range�time� monotonicr �timeoutZfailZ
assertLess�fuzzZ
assertGreater) r �countr5 �method�args�i�t1�eZdeltar r r
�_sock_operationx s
z TimeoutTestCase._sock_operationN)
r+ r, r- r6 r ZHOST� localhostr r r= 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_remoter r r r
r � s zTCPTimeoutTestCase.setUpc C s | j �� d S 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}|�|� zZz|�|� W nF tjk
r\ Y n2 tk
r� } z|j t j
kr|d}W 5 d }~X Y nX W 5 |�� ~X |r�| � d� |d |d
||d |d
�� || _
t�| j
d �� | �d
d d | j
� W 5 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 rB r5 �OSError�errnoZ ECONNREFUSEDZskipTest�formatrA r r r= )r Z blackholeZ whitehole�skipr r5 �errr r r
�testConnectTimeout� s8
��z%TCPTimeoutTestCase.testConnectTimeoutc C s> t �| jd ��$ | j�| j� | �dddd� W 5 Q R X d S )Nr r � �?Zrecv� )r r rA r rB r= r r r r
�testRecvTimeout� s z"TCPTimeoutTestCase.testRecvTimeoutc C s, t �| j| j� | j�� | �ddd� d S )Nr rI Zaccept)r � bind_portr r> �listenr= r r r r
�testAcceptTimeout� s
z$TCPTimeoutTestCase.testAcceptTimeoutc C sZ t � t jt j��@}t�|| j� |�� | j�|� � � | �
ddddd � W 5 Q R X d S )N�d rI �send� X�@
� r r r r rL r> rM r rB Z getsocknamer= �r Zservr r r
�testSend� s
zTCPTimeoutTestCase.testSendc C s` t � t jt j��F}t�|| j� |�� | j�|� � � | �
ddddd |� � � W 5 Q R X d S )NrO rI ZsendtorQ rR rS rT r r r
�
testSendto s �z TCPTimeoutTestCase.testSendtoc C sZ t � t jt j��@}t�|| j� |�� | j�|� � � | �
ddddd � W 5 Q R X d S )NrO rI ZsendallrQ rR rS rT r r r
� testSendall
s
z TCPTimeoutTestCase.testSendallN)r+ r, r- r. r r �unittestZskipIfrH rK rN rU rV rW r r r r
r? � s
T
r? c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UDPTimeoutTestCasez3UDP test case for socket.socket() timeout functionsc C s t � t jt j�| _d S r
)r r Z
SOCK_DGRAMr r r r r
r s zUDPTimeoutTestCase.setUpc C s | j �� d S r
r r r r r
r s zUDPTimeoutTestCase.tearDownc C s$ t �| j| j� | �dddd� d S )Nr rI ZrecvfromrJ )r rL r r> r= r r r r
�testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r+ r, r- r. r r rZ r r r r
rY s rY c C s t �d� t �ttt� d S )Nr )r ZrequiresZ run_unittestr r? rY r r r r
� test_main' s
�r[ �__main__)r. � functoolsrX Ztestr Zis_resource_enabledZ
skip_expectedr3 rD r � lru_cacher ZTestCaser r/ r? rY r[ r+ r r r r
�<module> s"
M'