File: //opt/cppython/lib/python3.8/test/__pycache__/test_timeout.cpython-38.pyc
U
.D�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 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�socket� getaddrinfo�AF_INET� SOCK_STREAM)�host�port� r �0/opt/cppython/lib/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 �countr8 �method�args�i�t1�eZdeltar r r
�_sock_operationx s
z TimeoutTestCase._sock_operationN)
r. r/ r0 r9 r ZHOST� localhostr r r@ r r r r
r2 h s
r2 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 rE r8 �OSError�errnoZ ECONNREFUSEDZskipTest�formatrD r r r@ )r Z blackholeZ whitehole�skipr r8 �errr r r
�testConnectTimeout� s8
��z%TCPTimeoutTestCase.testConnectTimeoutc C s> t �| jd ��$ | j�| j� | �dddd� W 5 Q R X d S )Nr r � �?�recv� )r r rD r rE r@ r r r r
�testRecvTimeout� s z"TCPTimeoutTestCase.testRecvTimeoutc C s, t �| j| j� | j�� | �ddd� d S )Nr rL �accept)r � bind_portr rA �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 rL �send� X�@
� r r r r rQ rA rR r rE � 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 )NrT rL �sendtorV rW rX rZ 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 )NrT rL �sendallrV rW rX rZ r r r
� testSendall
s
z TCPTimeoutTestCase.testSendallN)r. r/ r0 r1 r r �unittestZskipIfrK rO rS r[ r] r_ r r r r
rB � s
T
rB 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 �
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 rL �recvfromrN )r rQ r rA r@ r r r r
�testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r. r/ r0 r1 r r rd r r r r
ra s ra c C s t �d� t �ttt� d S )Nr )r ZrequiresZ run_unittestr rB ra r r r r
� test_main' s
�re �__main__)r1 � functoolsr` �testr Zis_resource_enabledZ
skip_expectedr6 rG r � lru_cacher ZTestCaser r2 rB ra re r. r r r r
�<module> s"
M'