Your IP : 216.73.216.213
3
��2ao, � @ s� d dl Z d dlZd dlmZ ejd� Zd dlZd dlZd dlZe j � dd� �Z
G dd� dej �Z G dd � d ej �Z
G d
d � d e
�ZG d d
� d
e
�Zdd� Zedkr�e� dS )� N)�support�networkc C s2 t j| �� tj| |tjtj�d d S Q R X d S )Nr � )r �transient_internet�socketZ getaddrinfo�AF_INET� SOCK_STREAM)�hostZport� r
�1/opt/alt/python36/lib64/python3.6/test_timeout.py�resolve_address s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�CreationTestCasec C s t j t jt j�| _d S )N)r r r �sock)�selfr
r
r �setUp s zCreationTestCase.setUpc C s | j j� d S )N)r �close)r r
r
r �tearDown! s zCreationTestCase.tearDownc C s | j | jj� d d� d S )Nz timeout not disabled by default)� assertEqualr �
gettimeout)r r
r
r �testObjectCreation$ s z#CreationTestCase.testObjectCreationc C s^ | j jd� | j| j j� d� | j jd� | j| j j� d� | j jd � | j| j j� d � d S )Ng�z�Ga @� )r �
settimeoutr r )r r
r
r �testFloatReturnValue) s z%CreationTestCase.testFloatReturnValuec C sP | j jd� | jt| j j� �td�� | j jd� | jt| j j� �td�� d S )N� g �?g333333@)r r r �typer )r r
r
r �testReturnType4 s z CreationTestCase.testReturnTypec C s� | j jd� | j jd� | j jd� | j jd � | jt| j jd� | jt| j jd� | jt| j jf � | jt| j jg � | jt| j ji � | jt| j jd� d S )Nr g � y )r r � assertRaises� TypeError)r r
r
r �
testTypeCheck< s z CreationTestCase.testTypeCheckc C s: | j t| jjd� | j t| jjd� | j t| jjd� d S )Nr g �?���r g �)r �
ValueErrorr r )r r
r
r �testRangeCheckI s z CreationTestCase.testRangeCheckc C s� | j jd� | j jd� | j| j j� d � | j jd� | j| j j� d� | j jd� | j jd� | j| j j� d� | j jd� | j| j j� d � d S )N�
r r g )r r � setblockingr r )r r
r
r �testTimeoutThenBlockingO s z(CreationTestCase.testTimeoutThenBlockingc C sX | j jd� | j jd� | j| j j� d� | j jd� | j jd� | j| j j� d� d S )Nr r )r r$ r r r )r r
r
r �testBlockingThenTimeout] s z(CreationTestCase.testBlockingThenTimeoutN) �__name__�
__module__� __qualname__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 j|� t| j |�}xbt|�D ]L}tj� }y ||� W q" tjk
rl } ztj� | }P W Y d d }~X q"X q"W | jd� | j||| j � | j
||d � d S )Nz socket.timeout was not raisedg �?) r r �getattr�range�timer �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 @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TCPTimeoutTestCasec C s" t j 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 j� d S )N)r r )r r
r
r r � s zTCPTimeoutTestCase.tearDownc C s t dd�}t dd�}d}tjtjtj�}d}|j|� zZy|j|� W nF tjk
r\ Y n2 tk
r� } z|jtj kr|d}W Y d d }~X nX W d |j
� ~X |r�| j dj |d |d
||d |d
�� || _
tj| j
d �� | jd
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 j| jd ��$ | jj| j� | jdddd� 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� | jj� | jddd� d S )Nr g �?Zaccept)r � bind_portr r7 �listenr6 )r r
r
r �testAcceptTimeout� s
z$TCPTimeoutTestCase.testAcceptTimeoutc
C sZ t j t jt j��@}tj|| j� |j� | jj|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 j t jt j��F}tj|| j� |j� | jj|j � � | j
ddddd |j � � 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 j t jt j��@}tj|| j� |j� | jj|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 rA rB rE rK rL rM r
r
r
r r8 � s T
r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�UDPTimeoutTestCasec C s t j t jt j�| _d S )N)r r Z
SOCK_DGRAMr )r r
r
r r s zUDPTimeoutTestCase.setUpc C s | j j� d S )N)r r )r r
r
r r s zUDPTimeoutTestCase.tearDownc C s$ t j| j| j� | jdddd� d S )Nr g �?Zrecvfromi )r rC r r7 r6 )r r
r
r �testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r' r( r) r r rO r
r
r
r rN s rN c C s t jd� t jttt� d S )Nr )r ZrequiresZ run_unittestr
r8 rN r
r
r
r � test_main& s
rP �__main__)� functoolsZunittestZtestr Zis_resource_enabledZ
skip_expectedr. r= r � lru_cacher ZTestCaser
r* r8 rN rP r' r
r
r
r �<module> s M'