Your IP : 216.73.217.129
a
XC?h, � @ s� d Z ddlZddlZddlmZ ddlmZ e�d� Zddl Z ddl
Z
ddl Z e� � dd� �Z
G dd � d ej�ZG d
d � d ej�ZG d d
� d
e�ZG dd� de�Zdd� Zedkr�e�� dS )z&Unit tests for socket timeout feature.� N)�support)�
socket_helper�networkc C sJ t �| ��, t�| |tjtj�d d W d � S 1 s<0 Y 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/python39/lib64/python3.9/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 �testReturnType5 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 �testRangeCheckJ 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�
TFr )r r � setblockingr r r r
r
r �testTimeoutThenBlockingP s z(CreationTestCase.testTimeoutThenBlockingc C sX | j �d� | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� d S )NFr T)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 t s zTimeoutTestCase.setUpc G s� | j �|� t| j |�}t|�D ]Z}t�� }z ||� W q tjyx } z&t�� | }W Y d}~ q�W Y d}~q d}~0 0 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 �countr6 �method�args�i�t1�eZdeltar
r
r �_sock_operationy s &
z TimeoutTestCase._sock_operationN)
r, r- r. r7 r ZHOST� localhostr r r> r
r
r
r r0 i s
r0 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�}tj}|�|� zdz|�|� W nF tjy\ Y n4 t y� } z |j
t
j krzd}W Y d }~n
d }~0 0 W |� � ~n |� � ~0 |r�| �
d�|d |d ||d |d �� || _t�| jd ��" | �d d
d | j� W d � n1 �s0 Y d S ) Nzblackhole.snakebite.netiZ� zwhitehole.snakebite.neti[� TFz�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 ZLOOPBACK_TIMEOUTr rC r6 �OSError�errnoZ ECONNREFUSEDr ZskipTest�formatrB r r r> )r Z blackholeZ whitehole�skipr r6 �errr
r
r �testConnectTimeout� s<
���z%TCPTimeoutTestCase.testConnectTimeoutc C sR t �| jd ��. | j�| j� | �dddd� W d � n1 sD0 Y d S )Nr r � �?Zrecv� )r r rB r rC r> r r
r
r �testRecvTimeout� s z"TCPTimeoutTestCase.testRecvTimeoutc C s, t �| j| j� | j�� | �ddd� d S )Nr rJ Zaccept)r � bind_portr r? �listenr> r r
r
r �testAcceptTimeout� s
z$TCPTimeoutTestCase.testAcceptTimeoutc C sn t � t jt j��J}t�|| j� |�� | j�|� � � | �
ddddd � W d � n1 s`0 Y d S )N�d rJ �send� X�@
� r r r r rM r? rN r rC Z getsocknamer> �r Zservr
r
r �testSend� s
zTCPTimeoutTestCase.testSendc C st t � t jt j��P}t�|| j� |�� | j�|� � � | �
ddddd |� � � W d � n1 sf0 Y d S )NrP rJ ZsendtorR rS rT rU r
r
r �
testSendto s �z TCPTimeoutTestCase.testSendtoc C sn t � t jt j��J}t�|| j� |�� | j�|� � � | �
ddddd � W d � n1 s`0 Y d S )NrP rJ ZsendallrR rS rT rU r
r
r � testSendall s
z TCPTimeoutTestCase.testSendallN)r, r- r. r/ r r �unittestZskipIfrI rL rO rV rW rX r
r
r
r r@ � s
Q
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 rJ ZrecvfromrK )r rM r r? r> r r
r
r �testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r, r- r. r/ r r r[ r
r
r
r rZ s rZ c C s t �d� d S )Nr )r Zrequiresr
r
r
r � setUpModule% s r\ �__main__)r/ � functoolsrY �testr Z test.supportr Zis_resource_enabledZ
skip_expectedr4 rE r � lru_cacher ZTestCaser
r0 r@ rZ r\ r, �mainr
r
r
r �<module> s$
M'