Your IP : 216.73.216.213
�
f fs, � @ s� d Z d d l Z d d l Z d d l m Z e j d � Z d d l Z d d l Z d d l Z e j
� d d � � Z Gd d � d e j � Z
Gd d
� d
e j � Z Gd d � d e � Z Gd
d � d e � Z d d � Z e d k r� e � n d S)z&Unit tests for socket timeout feature.� N)�support�networkc C s> t j | � �) t j | | t j t j � d d SWd QXd S)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)�hostZport� r
�6/opt/alt/python34/lib64/python3.4/test/test_timeout.py�resolve_address s r c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d S)�CreationTestCasez9Test case for socket.gettimeout() and socket.settimeout()c C s t j t j t 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 | j 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 j d � | j | j j � d � | j j d � | j | j j � d � | j j d � | j | j j � d � d S)Ng�z�Ga @� )r �
settimeoutr r )r r
r
r �testFloatReturnValue) s z%CreationTestCase.testFloatReturnValuec C sn | j j d � | j t | j j � � t d � � | j j d � | j t | j j � � t d � � d S)N� g �?g333333@)r r r �typer )r r
r
r �testReturnType4 s %z CreationTestCase.testReturnTypec C s� | j j d � | j j d � | j j d � | j j d � | j t | j j d � | j t | j j d � | j t | j j f � | j t | j j g � | j t | j j i � | j t | j j d � d S)Nr g � y )r r � assertRaises� TypeError)r r
r
r �
testTypeCheck< s z CreationTestCase.testTypeCheckc C sO | j t | j j d � | j t | j j d � | j t | j j d � d S)Nr g �?���r g �)r �
ValueErrorr r )r r
r
r �testRangeCheckI s z CreationTestCase.testRangeCheckc C s� | j j d � | j j d � | j | j j � d � | j j d � | j | j j � d � | j j d � | j j d � | j | j j � d � | j j d � | j | j j � d � d S)N�
r r g )r r � setblockingr r )r r
r
r �testTimeoutThenBlockingO s z(CreationTestCase.testTimeoutThenBlockingc C sv | j j d � | j j d � | j | j j � d � | j j d � | j j d � | 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__�__doc__r r r r r r r" r% r& r
r
r
r r
s
r
c @ s= e Z d Z d Z e j Z d d � Z e Z d d � Z d S)�TimeoutTestCaseg @c C s
t � � d S)N)�NotImplementedError)r r
r
r r s s zTimeoutTestCase.setUpc G s� | j j | � t | j | � } xw t | � D]\ } t j � } y | | � Wq/ t j k
r� } z t j � | } PWYd d } ~ Xq/ Xq/ W| j d � | j | | | j � | j
| | d � d S)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�timer �timeoutZfailZ
assertLess�fuzzZ
assertGreater) r �countr0 �method�args�iZt1�eZdeltar
r
r �_sock_operationx s
z TimeoutTestCase._sock_operationN)
r' r( r) r1 r ZHOST� localhostr r r7 r
r
r
r r+ h s
r+ c @ sv e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d S)�TCPTimeoutTestCasez3TCP test case for socket.socket() timeout functionsc C s1 t j t j t j � | _ t d d � | _ 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 sJ t d d � } t d d � } d } t j t j t j � } d } | j | � zl y | j | � WnT t j k
rz Yn@ t k
r� } z | j t j k r� d } n WYd d } ~ Xn XWd | j
� ~ X| r| j d j | d | d
| | d | d
� � n | | _
t j | j
d � � | j d
d d | j
� Wd QXd 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< r0 �OSError�errnoZ ECONNREFUSEDr ZskipTest�formatr; r r r7 )r Z blackholeZ whitehole�skipr r0 �errr
r
r �testConnectTimeout� s4
z%TCPTimeoutTestCase.testConnectTimeoutc C sJ t j | j d � �. | j j | j � | j d d d d � Wd QXd S)Nr r g �?Zrecvi )r r r; r r< r7 )r r
r
r �testRecvTimeout� s z"TCPTimeoutTestCase.testRecvTimeoutc C s= t j | j | j � | j j d � | j d d d � d S)N� r g �?Zaccept)r � bind_portr r8 �listenr7 )r r
r
r �testAcceptTimeout� s z$TCPTimeoutTestCase.testAcceptTimeoutc
C su t j t j t j � �W } t j | | j � | j d � | j j | j � � | j
d d d d d � Wd QXd S)NrD �d g �?�send� Xi@
) r r r r rE r8 rF r r< � getsocknamer7 )r �servr
r
r �testSend� s
zTCPTimeoutTestCase.testSendc
C s~ t j t j t j � �` } t j | | j � | j d � | j j | j � � | j
d d d d d | j � � Wd QXd S)NrD rH g �?ZsendtorJ i@
) r r r r rE r8 rF r r< rK r7 )r rL r
r
r �
testSendto s
z TCPTimeoutTestCase.testSendtoc
C su t j t j t j � �W } t j | | j � | j d � | j j | j � � | j
d d d d d � Wd QXd S)NrD rH g �?ZsendallrJ i@
) r r r r rE r8 rF r r< rK r7 )r rL r
r
r � testSendall s
z TCPTimeoutTestCase.testSendallN) r' r( r) r* r r rB rC rG rM rN rO r
r
r
r r9 � s T
r9 c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �UDPTimeoutTestCasez3UDP test case for socket.socket() timeout functionsc C s t j t j t 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 s0 t j | j | j � | j d d d d � d S)Nr g �?Zrecvfromi )r rE r r8 r7 )r r
r
r �testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r' r( r) r* r r rQ r
r
r
r rP s rP c C s$ t j d � t j t t t � d S)Nr )r ZrequiresZ run_unittestr
r9 rP r
r
r
r � test_main& s
rR �__main__)r* � functoolsZunittestZtestr Zis_resource_enabledZ
skip_expectedr/ r> r � lru_cacher ZTestCaser
r+ r9 rP rR r' r
r
r
r �<module> s M'�