Your IP : 216.73.216.85
��Yfo1 � @ sG 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 j d
e j k 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 rCe j � d S)� N)�support)�sanepathname2urlZnetwork�< c O sc xV t d � D]H } y | | | � SWq
| k
rT } z | } w
WYd d } ~ Xq
Xq
W| � d S)N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �9/opt/alt/python35/lib64/python3.5/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 ZTRAVISz%bpo-35411: skip FTP test on Travis CIc @ 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 QRXd S)Nzhttp://www.example.com/)r �transient_internet�_urlopen_with_retry�fp� assertFalse�closed�close�
assertTrue)�self�urlZresponseZsockr r r �
test_closeT s
zCloseSocketTest.test_closeN)r r r r" r r r r r R s r c @ s� e Z d Z d d � Z e 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://www.pythontest.net/READMEz*ftp://www.pythontest.net/non-existent-file)�urllib�error�URLError�
_test_urls�_extra_handlers)r �urlsr r r �test_ftpi 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_filer s
zOtherNetworkTests.test_filec
C s[ d } t j | � �? t j j | � } t j j | � } | j | j � d � Wd QRXd 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 s[ d } t j | � �? t j j | � } t j j | � } | j | j � d � Wd QRXd 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 QRXd 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 QRXWn2 t k
rj } z | j d � WYd d } ~ Xn X| j | � t j j | � } | j � } | j | � Wd QRXd 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 } | rZ t | t j j
� } x�| D]{} | j d | � �`t | t
� r� | \ } } }
n
d } }
t j | � �y | | | t � } Wnc t k
r0} zC |
rd |
| | t | � | f }
| j | |
|
� n � WYd d } ~ Xn� XyW t j �G t j �6 t j �% | j � } | d t | � � Wd QRXWd QRXWd QRXWn, t j k
r�t d | d t j �Yn X| j � Wd QRX| j d � Wd QRXqa 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$ �skip_ftp_test_on_travisr+ r9 r? r@ rB rD r( r) r r r r r# _ s -
'r# c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d Z e d
d � � Z e d d
� � Z
e d d � � Z e d d � � Z d S)� TimeoutTestc C sr | j t j � � d } t j | d d ��= t | � } | j | j � | j | j j j
j � � Wd QRXd 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 QRXd S)Nzhttp://www.example.comr )r\ rO r] r r �setdefaulttimeoutr rX r r; r r^ r_ r` )r r! ra 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 QRXd S)Nzhttp://www.example.comr rP )
r\ rO r] r r rc r rX r r r^ r_ r` )r r! ra r r r �test_http_no_timeout s
z TimeoutTest.test_http_no_timeoutc C sb d } t j | � �F t | d d �} | j | j � | j | j j j j � d � Wd QRXd S)Nzhttp://www.example.comrP �x )
r r r rX r r; r r^ r_ r` )r r! ra r r r �test_http_timeout# s
z TimeoutTest.test_http_timeoutzftp://www.pythontest.net/c C su | j t j � � t j | j d d ��C t | j � } | j | j � | j | j j j
j j � � Wd QRXd S)NrP )
r\ rO r] r r �FTP_HOSTr rX r r r^ r_ r` )r ra 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 QRXd S)Nr )r\ rO r] r r rh rc r rX r r; r r^ r_ r` )r ra r r r �test_ftp_default_timeout4 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 QRXd S)Nr rP )r\ rO r] r r rh rc r rX r r r^ r_ r` )r ra r r r �test_ftp_no_timeout@ s
z TimeoutTest.test_ftp_no_timeoutc C se t j | j � �L t | j d d �} | j | j � | j | j j j j j
� d � Wd QRXd S)NrP r ) r r rh r rX r r; r r^ r_ r` )r ra r r r �test_ftp_timeoutL s z TimeoutTest.test_ftp_timeoutN)
r r r rb rd re rg rh rZ ri rj rk rl r r r r r[ s r[ �__main__) ZunittestZtestr Ztest.test_urllib2r r1 rO Z urllib.errorr% Zurllib.requestrR ZrequiresrK r r ZskipIf�environrZ r6 r7 r& r' r ZTestCaser r r# r[ r �mainr r r r �<module> s*
'
�R