Your IP : 216.73.216.213
�
f f�/ � @ s/ d d l Z d d l m Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l
Z
e j d � d Z d d � Z
d d � Z e e j j e j j � Z Gd
d � d e j � Z Gd d
� d
e j � Z Gd d � d e j � Z Gd d � d e j � Z e d k r+e j � n d S)� N)�support)�sanepathname2urlZnetwork�< c O sm x` t d � D]R } y | | | � SWq
| k
rT } z | } w
WYd d } ~ Xq
� Yq
Xq
W| � d S)N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �9/opt/alt/python34/lib64/python3.4/test/test_urllib2net.py�
_retry_thrice s r c s � � f d d � } | S)Nc s t � � | | � S)N)r )r r
)r r r r �wrapped s z(_wrap_with_retry_thrice.<locals>.wrappedr )r r r r )r r r �_wrap_with_retry_thrice s r c @ s e Z d Z d Z d S)� AuthTestsz&Tests urllib2 authentication features.N)�__name__�
__module__� __qualname__�__doc__r r r r r & s r c @ s e Z d Z d d � Z d S)�CloseSocketTestc C s_ d } t j | � �D t | � } | j } | j | j � | j � | j | j � Wd QXd S)Nzhttp://www.example.com/)r �transient_internet�_urlopen_with_retry�fp� assertFalse�closed�close�
assertTrue)�self�urlZresponseZsockr r r �
test_closeO s
zCloseSocketTest.test_closeN)r r r r" r r r r r M s r c @ s e 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 d � Z
d d � Z d S)�OtherNetworkTestsc C s d S)Nr )r r r r �setUp[ s zOtherNetworkTests.setUpc C s5 d d d t j j f g } | j | | j � � d S)Nz"ftp://ftp.debian.org/debian/READMEz-ftp://ftp.debian.org/debian/non-existent-file)�urllib�error�URLError�
_test_urls�_extra_handlers)r �urlsr r r �test_ftpd s zOtherNetworkTests.test_ftpc C s� t j } t | d � } zh | j d � | j � d t t j j | � � d d t j
j f g } | j | | j
� d d �Wd t j | � X| j t t j j d � d S)N�wz hi there
zfile:z file:///nonsensename/etc/passwd�retryTz./relative_path/to/file)r �TESTFN�open�writer r �os�path�abspathr% r& r' r( r) �removeZ assertRaises�
ValueError�request�urlopen)r r. �fr* r r r � test_filel s
zOtherNetworkTests.test_filec
C sZ d } t j | � �? t j j | � } t j j | � } | j | j � d � Wd QXd S)Nz)http://www.pythontest.net/index.html#frag)r r r% r6 �Requestr7 � assertEqual�geturl)r Z urlwith_frag�req�resr r r �test_urlwithfrag� s z"OtherNetworkTests.test_urlwithfragc
C sZ d } t j | � �? t j j | � } t j j | � } | j | j � d � Wd QXd S)Nz*http://www.pythontest.net/redir/with_frag/z)http://www.pythontest.net/elsewhere/#frag)r r r% r6 r: r7 r; r< )r Zredirect_url_with_fragr= r> r r r �test_redirect_url_withfrag� s z,OtherNetworkTests.test_redirect_url_withfragc
C s� d } t j | � �� t j j � } t j j | � } | j | j � � | j | � | j | j � � | j | j
d � � | j d d � | j | � | j | j
d � d � Wd QXd S)Nzhttp://www.example.comz
User-agentz
User-Agentz
Test-Agent)r r r% r6 � build_openerr: r Z header_itemsr/ r Z
has_headerZ
add_headerr; Z
get_header)r r! Zopenerr6 r r r �test_custom_headers� s
z%OtherNetworkTests.test_custom_headersc C s� d } t j | � �� y t j j | � � } Wd QXWn2 t k
ri } z | j d � WYd d } ~ Xn X| j | � t j j | � } | j � } | j | � Wd QXd S)Nzhttp://www.imdb.comzOurlopen failed for site not sending Connection:close) r r r% r6 r7 r5 Zfailr �read)r ZURLr> r r= r r r � test_sites_no_connection_close� s
z0OtherNetworkTests.test_sites_no_connection_closeTc 6 C s� d d l } d d l } | j d � j } t j j | � j } | r] t | t j j
� } n x~| D]v} | j d | � �\t | t
� r� | \ } } }
n
d } }
t j | � �y | | | t � } Wnc t k
r3} zC |
r d |
| | t | � | f }
| j | |
|
� n � WYd d } ~ Xn� XyT t j �E t j �5 t j �% | j � } | d t | � � Wd QXWd QXWd QXWn, t j k
r�t d | d t j �Yn X| j � Wd QX| j d � Wd QXqd Wd S) Nr Z test_urllib2r! z5Didn't get expected error(s) %s for %s %s, got %s: %sz
read %d bytesz
<timeout: %s>�fileg�������?) �time�loggingZ getLogger�debugr% r6 rA r/ r r& r' ZsubTest�
isinstance�tupler r �TIMEOUT�OSError�typeZassertIsInstanceZtime_outZsocket_peer_resetZioerror_peer_resetrC �len�socket�timeout�print�sys�stderrr Zsleep)r r* �handlersr- rF rG rH r7 r! r= Z expected_errr8 �err�msgZbufr r r r( � s<
* z OtherNetworkTests._test_urlsc C sC g } t j j � } | j | j � | j d � | j | � | S)N� )r% r6 ZCacheFTPHandler�
addCleanupZ clear_cacheZ
setTimeout�append)r rT Zcfhr r r r) � s
z!OtherNetworkTests._extra_handlersN) r r r r$ r+ r9 r? r@ rB rD r( r) r r r r r# Z s -
'r# c @ sv e Z d Z d d � Z d d � Z d d � Z d d � Z d Z d
d � Z d d
� Z d d � Z
d d � Z d S)� TimeoutTestc C sq | j t j � � d } t j | d d ��= t | � } | j | j � | j | j j j
j � � Wd QXd S)Nzhttp://www.example.comrP ) � assertIsNonerO �getdefaulttimeoutr r r rX r r �raw�_sock�
gettimeout)r r! �ur r r �test_http_basic� s zTimeoutTest.test_http_basicc C s� | j t j � � d } t j | � �b t j d � z t | � } | j | j � Wd t j d � X| j | j
j j j
� d � Wd QXd S)Nzhttp://www.example.comr )r[ rO r\ r r �setdefaulttimeoutr rX r r; r r] r^ r_ )r r! r` r r r �test_http_default_timeout s
z%TimeoutTest.test_http_default_timeoutc C s� | j t j � � d } t j | � �e t j d � z& t | d d �} | j | j � Wd t j d � X| j | j j
j j � � Wd QXd S)Nzhttp://www.example.comr rP )
r[ rO r\ r r rb r rX r r r] r^ r_ )r r! r` r r r �test_http_no_timeout s
z TimeoutTest.test_http_no_timeoutc C sa d } t j | � �F t | d d �} | j | j � | j | j j j j � d � Wd QXd S)Nzhttp://www.example.comrP �x )
r r r rX r r; r r] r^ r_ )r r! r` r r r �test_http_timeout s
z TimeoutTest.test_http_timeoutz ftp://ftp.debian.org/debian/c C st | j t j � � t j | j d d ��C t | j � } | j | j � | j | j j j
j j � � Wd QXd S)NrP )
r[ rO r\ r r �FTP_HOSTr rX r r r] r^ r_ )r r` r r r �test_ftp_basic& s
zTimeoutTest.test_ftp_basicc C s� | j t j � � t j | j � �h t j d � z# t | j � } | j | j � Wd t j d � X| j
| j j j j
j � d � Wd QXd S)Nr )r[ rO r\ r r rg rb r rX r r; r r] r^ r_ )r r` r r r �test_ftp_default_timeout- s
z$TimeoutTest.test_ftp_default_timeoutc C s� | j t j � � t j | j � �k t j d � z) t | j d d �} | j | j � Wd t j d � X| j | j
j
j j j
� � Wd QXd S)Nr rP )r[ rO r\ r r rg rb r rX r r r] r^ r_ )r r` r r r �test_ftp_no_timeout8 s
z TimeoutTest.test_ftp_no_timeoutc C sd t j | j � �L t | j d d �} | j | j � | j | j j j j j
� d � Wd QXd S)NrP r ) r r rg r rX r r; r r] r^ r_ )r r` r r r �test_ftp_timeoutC s z TimeoutTest.test_ftp_timeoutN) r r r ra rc rd rf rg rh ri rj rk r r r r rZ � s rZ �__main__)ZunittestZtestr Ztest.test_urllib2r r1 rO Z urllib.errorr% Zurllib.requestrR ZrequiresrK r r r6 r7 r& r' r ZTestCaser r r# rZ r �mainr r r r �<module> s&
'
�N