Your IP : 216.73.216.213
���]o1 � @ 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 �4/opt/alt/python35/lib64/python3.5/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 S)� AuthTestsN)�__name__�
__module__� __qualname__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�url�response�sockr 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& r7 �Requestr8 � 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& r7 r; r8 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& r7 � build_openerr; r Z header_itemsr0 r Z
has_headerZ
add_headerr< Z
get_header)r r Zopenerr7 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& r7 r8 r6 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�logging� getLogger�debugr& r7 rB r0 r r' r( ZsubTest�
isinstance�tupler r �TIMEOUT�OSError�typeZassertIsInstanceZtime_outZsocket_peer_resetZioerror_peer_resetrD �len�socket�timeout�print�sys�stderrr �sleep)r r+ �handlersr. rG rH rJ r8 r r> Z expected_errr9 �err�msg�bufr r r r) � s<
- z OtherNetworkTests._test_urlsc C sC g } t j j � } | j | j � | j d � | j | � | S)N� )r& r7 ZCacheFTPHandler�
addCleanupZ clear_cacheZ
setTimeout�append)r rW Zcfhr r r r* � s
z!OtherNetworkTests._extra_handlersN)
r r r r% �skip_ftp_test_on_travisr, r: r@ rA rC rE 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.comrR ) � assertIsNonerQ �getdefaulttimeoutr r r r\ 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` rQ ra r r �setdefaulttimeoutr r\ r r< r rb rc rd )r r re 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 rR )
r` rQ ra r r rg r r\ r r rb rc rd )r r re 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.comrR �x )
r r r r\ r r< r rb rc rd )r r re 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)NrR )
r` rQ ra r r �FTP_HOSTr r\ r r rb rc rd )r re 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` rQ ra r r rl rg r r\ r r< r rb rc rd )r re 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 rR )r` rQ ra r r rl rg r r\ r r rb rc rd )r re 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)NrR r ) r r rl r r\ r r< r rb rc rd )r re r r r �test_ftp_timeoutL s z TimeoutTest.test_ftp_timeoutN)
r r r rf rh ri rk rl r^ rm rn ro rp r r r r r_ s r_ �__main__) ZunittestZtestr Ztest.test_urllib2r r2 rQ Z urllib.errorr& Zurllib.requestrT ZrequiresrM r r ZskipIf�environr^ r7 r8 r' r( r ZTestCaser r r$ r_ r �mainr r r r �<module> s*
'
�R