Your IP : 216.73.216.213
���]P � @ 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 d d l Z
d d l m Z m Z m
Z
m Z m Z m Z m Z d d l m Z d d l Z
d d l Z Gd d � d e j � Z Gd d � d e j � Z Gd
d � d � Z Gd d
� d
� Z Gd d � d e � Z Gd d � d e j � Z Gd d � d � Z Gd d � d � Z! Gd d � d e j" � Z# Gd d � d � Z$ Gd d � d � Z% d d � Z& d d � Z' Gd d! � d! e
j( j) � Z* Gd" d# � d# e
j( j+ � Z, Gd$ d% � d% e
j( j) � Z- Gd& d' � d' � Z. Gd( d) � d) e j � Z/ d* d+ � Z0 Gd, d- � d- e j � Z1 Gd. d/ � d/ e j � Z2 Gd0 d1 � d1 e j � Z3 e4 d2 k r�e j5 � d S)3� N)�support)� test_urllib)�Request�OpenerDirector�HTTPBasicAuthHandler� HTTPPasswordMgrWithPriorAuth� _parse_proxy� _proxy_bypass_macosx_sysconf�AbstractDigestAuthHandler)�urlparsec @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� TrivialTestsc C s� x� d D]� } i } t d | | � | d =| d k r� t j d k r� | j d � | j d
� } } | j | j d � | j | j d � xD | j � D]6 \ } } | j | j d | d
| | | j f � q� Wq Wd S)N�request�response�parse�error� robotparserzfrom urllib.%s import *� __builtins__�nt� url2pathname� pathname2urlZ
nturl2pathz urllib.%sz.%r is exposed in 'urllib.%s' but defined in %r)r
r r r r )�exec�os�name�pop� assertEqual�
__module__�items)�self�module�context�u�p�k�v� r$ �1/opt/alt/python35/lib64/python3.5/test_urllib2.py� test___all__ s
zTrivialTests.test___all__c C s� | j t t j j d � t j j t j j � j t j
d � } t j d k r\ d | } n
d | } t j j | � } | j � | j
� d S)Nz bogus url�/r z
file:///%sz file://%s)� assertRaises�
ValueError�urllibr
�urlopenr �path�abspath�__file__�replace�sepr �read�close)r ZfnameZfile_url�fr$ r$ r% � test_trivial- s '
zTrivialTests.test_trivialc
C s� d d d d g f d d d g f d d d d d
d d d
g f d d d d g f g } x0 | D]( \ } } | j t j j | � | � q^ Wd S)Nza,b,c�a�b�czpath"o,l"og"i"cal, examplezpath"o,l"og"i"calZexampleza, b, "c", "d", "e,f", g, hz"c"z"d"z"e,f"�g�hz a="b\"c", d="e\,f", g="h\\i"za="b"c"zd="e,f"zg="h\i")r r* r
Zparse_http_list)r �tests�string�listr$ r$ r% �test_parse_http_list? s z!TrivialTests.test_parse_http_listc C s/ t j j d � } | j | j t | � � d S)N�reason)r* r �URLError�assertInr> �str)r �errr$ r$ r% �test_URLError_reasonstrJ s z$TrivialTests.test_URLError_reasonstrN)�__name__r � __qualname__r& r4 r= rC 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 S)
�RequestHdrsTestsc C s\ d } | j t | d d d i �j d d � | j t | d d d i �j d d � d S)Nzhttp://example.com�headersz Spam-eggs�blahz spam-EggS)r r rG )r �urlr$ r$ r% �test_request_headers_dictQ s
z*RequestHdrsTests.test_request_headers_dictc C s7 d } t | d d d i �} | j | j d � � | j | j � d g � | j d d � | j t | j � � d
d g � | j | j d � � | j | j d � � | j | j d d � d � | j
d � | j | j d � � | j d
d � | j | j d
� � | j
d
� | j | j d
� � d S)Nzhttp://example.comrG � Spam-eggsrH zFoo-Bar�baz�Foo-barz Not-there�defaultzUnredirected-spamZEggs)rK rH )rM rL )rK rH ) r �
assertTrue�
has_headerr Z header_items�
add_header�sorted� assertFalse� assertIsNone�
get_headerZ
remove_header�add_unredirected_header)r rI �reqr$ r$ r% � test_request_headers_methodsf s
z-RequestHdrsTests.test_request_headers_methodsc C sZ t j j � } | j } | j } | d d d d � | d d d d � | d d d d � | d d
d d � | d
d d d � | d
d d d � | d d d d � | d d d d � | d d d d � | d d d d � | j | d d � d# � | j | d d � d$ � | j | d d � d% � | j | d d � d& � | j | d d � d' � | j | d d � d( � | j | d d
� d) � | j | d
d � d* � | j | d d � d+ � | j | d d � d, � | j | d d! � d- � | j | d d" � d. � | j | d d � d/ � | j | d d � d0 � | j | d d � d1 � d S)2Nz
Some Realmzhttp://example.com/�joe�passwordzhttp://example.com/ni�nir7 zhttp://example.com/foo�foozhttp://example.com/bar�bar�ninir6 �firstrH �second�spamr5 zhttp://example.com�1zhttp://c.example.com:3128�3z
d.example.com�4�dze.example.com:3128�5�ez example.comzhttp://example.com/spamz http://example.com/spam/spamzhttp://a.example.com/z
c.example.comzc.example.com:3128)rY rZ )rY rZ )rY rZ )rY rZ )rY rZ )r\ r[ )r] r^ )r` ra )rb r5 )NN)NN)rc r7 )rc r7 )rd re )rf rg )r* r
�HTTPPasswordMgr� add_password�find_user_passwordr )r �mgr�add�find_user_passr$ r$ r% �test_password_manager� s\ z&RequestHdrsTests.test_password_managerc C s t j j � } | j } | j } | d d d d � | d d d d � | d d
d d � | d
d d d � | j | d d � d � | j | d d � d � | j | d d � d � | j | d d � d! � | j | d d � d" � | j | d d � d# � | j | d d � d$ � | j | d d � d% � | j | d d � d& � | j | d d � d' � | j | d d
� d( � | j | d d � d) � | j | d
d � d* � | j | d
d � d+ � | j | d
d � d, � | j | d
d � d- � d S).Nr3 zhttp://g.example.com:80�10�jr8 zhttp://h.example.com�11r" r9 zi.example.com:80�12�l�iz
j.example.com�13�mzg.example.com:100zg.example.com:80z
g.example.comzhttp://g.example.com:100zhttp://g.example.comz
h.example.comzh.example.com:80zhttp://h.example.com:80z
i.example.comzhttp://i.example.com:80zj.example.com:80zhttp://j.example.comzhttp://j.example.com:80)NN)ro rp )NN)NN)ro rp )ro rp )rq r" )rq r" )rq r" )NN)rr rs )rr rs )ru rv )NN)ru rv )NN)r* r
rh ri rj r )r rk rl rm r$ r$ r% �"test_password_manager_default_port� sD z3RequestHdrsTests.test_password_manager_default_portN)rD r rE rJ rX rn rw r$ r$ r$ r% rF O s $ ;rF c @ s7 e Z d Z g Z d e j d d � Z d d � Z d S)�
MockOpenerNc C s! | | | | _ | _ | _ d S)N)rW �data�timeout)r rW ry rz r$ r$ r% �open� s zMockOpener.openc G s | | | _ | _ d S)N)�proto�args)r r| r} r$ r$ r% r � s zMockOpener.error)rD r rE �
addheaders�socket�_GLOBAL_DEFAULT_TIMEOUTr{ r r$ r$ r$ r% rx � s rx c @ s: e Z d Z d d d � Z d d d � Z d d � Z d S)�MockFileNc C s d S)Nr$ )r �countr$ r$ r% r1 � s z
MockFile.readc C s d S)Nr$ )r r� r$ r$ r% �readline� s zMockFile.readlinec C s d S)Nr$ )r r$ r$ r% r2 s zMockFile.close)rD r rE r1 r� r2 r$ r$ r$ r% r� � s r� c @ s e Z d Z d d � Z d S)� MockHeadersc C s t | j � � S)N)r< �values)r r r$ r$ r% �
getheaders s zMockHeaders.getheadersN)rD r rE r� r$ r$ r$ r% r� s r� c @ s7 e Z d Z d d d � Z d d � Z d d � Z d S)� MockResponseNc C sA t j j | | � | | | | f \ | _ | _ | _ | _ d S)N)�io�StringIO�__init__�code�msgrG rI )r r� r� rG ry rI r$ r$ r% r�
s zMockResponse.__init__c C s | j S)N)rG )r r$ r$ r% �info s zMockResponse.infoc C s | j S)N)rI )r r$ r$ r% �geturl s zMockResponse.geturl)rD r rE r� r� r� r$ r$ r$ r% r� s r� c @ s( e Z d Z d d � Z d d � Z d S)�
MockCookieJarc C s
| | _ d S)N)�ach_req)r r
r$ r$ r% �add_cookie_header s z MockCookieJar.add_cookie_headerc C s | | | _ | _ d S)N)�ec_req�ec_r)r r r
r$ r$ r% �extract_cookies s z MockCookieJar.extract_cookiesN)rD r rE r� r� r$ r$ r$ r% r� s r� c @ s( e Z d Z d d � Z d d � Z d S)�
FakeMethodc C s | | _ | | _ | | _ d S)N)� meth_name�handle�action)r r� r� r� r$ r$ r% r� s zFakeMethod.__init__c G s | j | j | j | � S)N)r� r� r� )r r} r$ r$ r% �__call__# s zFakeMethod.__call__N)rD r 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 S)
�MockHTTPResponsec C s1 | | _ | | _ | | _ | | _ d | _ d S)N�� )�fpr� �statusr> r� )r r� r� r� r> r$ r$ r% r� ( s
zMockHTTPResponse.__init__c C s d S)N� r$ )r r$ r$ r% r1 / s zMockHTTPResponse.readc C s i S)Nr$ )r r$ r$ r% r� 2 s zMockHTTPResponse.infoc C s | j S)N)rI )r r$ r$ r% r� 5 s zMockHTTPResponse.geturlN)rD r rE r� r1 r� r� r$ r$ r$ r% r� ' s r� c @ sv e Z d Z d d � Z e j d d � Z d d � Z d d d d � Z d d d
d � Z d d
� Z
d d � Z d S)�
MockHTTPClassc C s: d | _ g | _ d | _ d | _ d | _ i | _ d S)Nr F)�level� req_headersry �raise_on_endheaders�sock�_tunnel_headers)r r$ r$ r% r� : s zMockHTTPClass.__init__c C s | | _ | | _ | S)N)�hostrz )r r� rz r$ r$ r% r� B s zMockHTTPClass.__call__c C s
| | _ d S)N)r� )r r� r$ r$ r% �set_debuglevelG s z MockHTTPClass.set_debuglevelNc C s5 | | _ | | _ | r$ | | _ n
| j j � d S)N)� _tunnel_hostZ _tunnel_portr� �clear)r r� �portrG r$ r$ r% �
set_tunnelJ s
zMockHTTPClass.set_tunnelc C se | | _ | | _ | d k r3 | j | j � 7_ | j j � | rO | | _ | j ra t � � d S)N)�method�selectorr� r �sortry r� �OSError)r r� rI ZbodyrG r$ r$ r% r
R s
zMockHTTPClass.requestc C s t t � i d d � S)Nr� �OK)r� r� )r r$ r$ r% � getresponse] s zMockHTTPClass.getresponsec C s d S)Nr$ )r r$ r$ r% r2 ` s zMockHTTPClass.close) rD r rE r� r r� r� r� r� r
r� r2 r$ r$ r$ r% r� 9 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 S)� MockHandleri� c C s | j | � d S)N)�_define_methods)r �methodsr$ r$ r% r� i s zMockHandler.__init__c C sk xd | D]\ } t | � d k r. | \ } } n
| d } } t | | | j � } t | j | | � q Wd S)N� )�lenr� r� �setattr� __class__)r r� �specr r� �methr$ r$ r% r� l s
zMockHandler._define_methodsc O s% | j j j | | | | f � | d k r/ d S| d k r? | S| d k rd t d d i d � } | S| d k rz t d � S| j d � r| | j d � d
d � } y t | � } Wn t k
r� Yn Xt d d i d � } | j j d | d | | d i � S| d
k r!t
j j d � � d S)Nz return selfzreturn responser� r� r� zreturn requestz http://blah/r � � �httpr �raiserH ) �parent�calls�appendr� r �
startswith�rfind�intr) r r* r? )r Zfn_namer� r} �kwds�resr� r$ r$ r% r� u s*
# zMockHandler.handlec C s d S)Nr$ )r r$ r$ r% r2 � s zMockHandler.closec C s | | _ g | j _ d S)N)r� r� )r r� r$ r$ r% �
add_parent� s zMockHandler.add_parentc C s# t | d � s d S| j | j k S)N�
handler_orderT)�hasattrr� )r �otherr$ r$ r% �__lt__� s zMockHandler.__lt__N)
rD r rE r� r� r� r� r2 r� r� r$ r$ r$ r% r� d s r� c C s� g } d } xp | D]h } Gd d � d t � } | | � } | j | 7_ | j | � | d } | j | � | j | � q W| S)Nr c @ s e Z d Z d S)z6add_ordered_mock_handlers.<locals>.MockHandlerSubclassN)rD r rE r$ r$ r$ r% �MockHandlerSubclass� s r� r� )r� r� r� r� � add_handler)�opener� meth_spec�handlersr� �methsr� r9 r$ r$ r% �add_ordered_mock_handlers� s
r� c G s+ t � } x | D] } | j | � q W| S)N)r r� )Zhandler_instancesr� r9 r$ r$ r% �build_test_opener� s
r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�MockHTTPHandlerc C s | | _ | | _ | j � d S)N)r� rG �reset)r r� rG r$ r$ r% r� � s zMockHTTPHandler.__init__c C s d | _ g | _ d S)Nr )�_count�requests)r r$ r$ r% r� � s zMockHTTPHandler.resetc C s� d d l } d d l } | j j | j | � � | j d k r� | j d | _ t j j | j } | j
| j � } | j j
d | t � | j | | � S| | _ | j
d � } t d d | d | j � � Sd S)Nr r� r� z
r� r� r� )�email�copyr� r� �deepcopyr� r� �client� responsesr� Zmessage_from_stringrG r� r r� rW r� � get_full_url)r rW r� r� r r� r$ r$ r% � http_open� s zMockHTTPHandler.http_openN)rD r rE r� r� r� r$ r$ r$ r% r� � s r� c @ s+ e Z d Z d d d � Z d d � Z d S)�MockHTTPSHandlerr c C s) t j j j | d | �t � | _ d S)N�
debuglevel)r* r
�AbstractHTTPHandlerr� r� �httpconn)r r� r$ r$ r% r� � s zMockHTTPSHandler.__init__c C s | j | j | � S)N)�do_openr� )r rW r$ r$ r% �
https_open� s zMockHTTPSHandler.https_openN)rD r rE r� r� r$ r$ r$ r% r� � s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�MockHTTPHandlerCheckAuthc C s | | _ d | _ d S)NF)r� �has_auth_header)r r� r$ r$ r% r� � s z!MockHTTPHandlerCheckAuth.__init__c C s
d | _ d S)NF)r� )r r$ r$ r% r� � s z MockHTTPHandlerCheckAuth.resetc C sM | j d � r d | _ t j j | j } t | j | t � d | j � � S)N�
AuthorizationTr� ) rP r� r� r� r� r� r� r� r� )r rW r r$ r$ r% r� � s z"MockHTTPHandlerCheckAuth.http_openN)rD r rE r� r� r� r$ r$ r$ r% r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�MockPasswordManagerc C s( | | _ | | _ | | _ | | _ d S)N)�realmrI �userrZ )r r� Zurir� rZ r$ r$ r% ri � s z MockPasswordManager.add_passwordc C s" | | _ | | _ | j | j f S)N)� target_realm�
target_urlr� rZ )r r� Zauthurir$ r$ r% rj s z&MockPasswordManager.find_user_passwordN)rD r rE ri rj r$ r$ r$ r% r� � s r� c @ sd 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 S)�OpenerDirectorTestsc C s6 Gd d � d t � } | j t t � j | � � d S)Nc @ s e Z d Z d S)z<OpenerDirectorTests.test_add_non_handler.<locals>.NonHandlerN)rD r rE r$ r$ r$ r% �
NonHandler s r� )�objectr( � TypeErrorr r� )r r� r$ r$ r% �test_add_non_handler
s z(OpenerDirectorTests.test_add_non_handlerc C s� d d l m } t � } d d g d
g g } t | | � | j t j j � � x( d D] } | j | | j | d
� qX Wd S)Nr )r? r� � return self�
proxy_open�redirect_request�do�proxy�redirectz://example.com/)r� r� )r� r� )r� r� )r� r� r� )
� urllib.errorr? r r� r� r* r
ZUnknownHandlerr( r{ )r r? �or� �schemer$ r$ r% �test_badly_named_methods s
z,OpenerDirectorTests.test_badly_named_methodsc
C s� t � } d d d g d g d g d g g } t | | � } t d � } | j | � } | j | | d � | d d f | d d f g } x[ t | | j � D]G \ } } | \ } }
} } | j | |
f | � | j | | f � q� Wd S)
Nr� �ftp_open�http_error_302� return selfzhttp://example.com/r� r )r� r� )r� r� )r r� r r{ r �zipr� )
r r� r� r� rW �rr� �expected�got�handlerr r} r� r$ r$ r% � test_handled( s z OpenerDirectorTests.test_handledc C s� t � } g } xq d g d f d g d f g D]Q \ } } Gd d � d t � } | | � } | | _ | j | � | j | � q. W| j d � | j | j d d | d � | j | j d d | d � d S)
Nr� � return selfi� r c @ s e Z d Z d S)zCOpenerDirectorTests.test_handler_order.<locals>.MockHandlerSubclassN)rD r rE r$ r$ r$ r% r� G s r� zhttp://example.com/r� )r� r )r r� r� r� r� r{ r r� )r r� r� r� r� r� r9 r$ r$ r% �test_handler_orderB s
z&OpenerDirectorTests.test_handler_orderc C s t � } d g d g g } t | | � } t d � } | j t j j | j | � | j | j | d d | f i f g � d S)Nr� r� � return selfzhttp://example.com/r )r� r� )r� r )
r r� r r( r* r r? r{ r r� )r r� r� r� rW r$ r$ r% �
test_raiseT s z OpenerDirectorTests.test_raisec C s t � } d g d d g d d d g d g g } t | | � } Gd d
� d
� } t d � } | j | � | d d | f f | d
d | | � d d i f f g } xc t | | j � D]O \ } } | \ } }
} | j | |
f | d d
� � | j | | d
� q� Wd S)Nr� � error 302�http_error_400r� r� �return responseZhttp_error_303Z
http_errorc @ s e Z d Z d d � Z d S)z4OpenerDirectorTests.test_http_error.<locals>.Unknownc S s d S)NTr$ )r r� r$ r$ r% �__eq__o s z;OpenerDirectorTests.test_http_error.<locals>.Unknown.__eq__N)rD r rE r r$ r$ r$ r% �Unknownn s r zhttp://example.com/r r� i. r� )r� r )r r� )r� r
)r r� r r{ r� r� r ) r r� r� r� r rW r� r r r Z method_namer} r$ r$ r% �test_http_errora s"
z#OpenerDirectorTests.test_http_errorc C sn t � } d d
g d d g g } t | | � } t d � } | j | � | d d f | d d f | d d f | d d f g } x� t | j � D]� \ } \ } } } }
| d k r� | j | | f | | � | j t | � d � | j | d t � q� | j | | f | | � | j t | � d � | j | d t � | d d k r� | j | d t � q� Wd S)
N� http_request�return request�
http_response�return responsezhttp://example.com/r r� r� )r r )r r )r r )r r )
r r� r r{ � enumerater� r r� ZassertIsInstancer� ) r r� r� r� rW r� rt r r r} r� r$ r$ r% �test_processors} s(
( z#OpenerDirectorTests.test_processorsN)
rD r rE r� r� r r r r
r r$ r$ r$ r% r� s
r� c C sy y | j d � Wn! t k
r4 t j d � � Yn Xt j j | � } t j d k ru | j d � ru | d d � } | S)Nzutf-8z path is not encodable to utf8r z///r� )
�encode�UnicodeEncodeError�unittestZSkipTestr* r
r r r r� )r, �urlpathr$ r$ r% �sanepathname2url� 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 � 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# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z e j e j d- k d. � d/ d0 � � Z d1 d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z! d: d; � Z" d< d= � Z# d> d? � Z$ d@ dA � Z% dB dC � Z& dD dE � Z' dF dG � Z( dH dI � Z) dJ S)K� HandlerTestsc sm Gd d � d � � G� f d d � d t j j � } d d l } d } | | � } t � | _ xd d | j d d d
d d g d
d f d d | j d d d
d d g d
d f d d | j d d d
d d g d
d f d d | j d d d
d d g d
d f d d d d d d d d g d d f d d | j d d d g d d f g D] \ } } } } } }
} } }
t | � } d | _ | j | � } | j
| j | � | j
| j | � | j
| j
t j | � � | j
| j | � | j
| j | � | j
| j j | � | j
| j j |
� | j � } | j
| j d � |
� | j
t | d � t | � � qGWd S) Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z-HandlerTests.test_ftp.<locals>.MockFTPWrapperc S s
| | _ d S)N)ry )r ry r$ r$ r% r� � s z6HandlerTests.test_ftp.<locals>.MockFTPWrapper.__init__c S s2 | | | _ | _ t j | j � t | j � f S)N)�filename�filetyper� r� ry r� )r r r r$ r$ r% �retrfile� s z6HandlerTests.test_ftp.<locals>.MockFTPWrapper.retrfilec S s d S)Nr$ )r r$ r$ r% r2 � s z3HandlerTests.test_ftp.<locals>.MockFTPWrapper.closeN)rD r rE r� r r2 r$ r$ r$ r% �MockFTPWrapper� s r c s4 e Z d Z d d � Z e j � f d d � Z d S)z-HandlerTests.test_ftp.<locals>.NullFTPHandlerc S s
| | _ d S)N)ry )r ry r$ r$ r% r� � s z6HandlerTests.test_ftp.<locals>.NullFTPHandler.__init__c sH | | | _ | _ | | | _ | _ | | _ � | j � | _ | j S)N)r� �passwdr� r� �dirsry �
ftpwrapper)r r� r r� r� r rz )r r$ r% � connect_ftp� s
z9HandlerTests.test_ftp.<locals>.NullFTPHandler.connect_ftpN)rD r rE r� r r� r! r$ )r r$ r% �NullFTPHandler� s r" r zrheum rhaponicumz ftp://localhost/foo/bar/baz.html� localhostr� �Ir\ r] zbaz.htmlz text/htmlz'ftp://parrot@localhost/foo/bar/baz.htmlZparrotz*ftp://%25parrot@localhost/foo/bar/baz.htmlz%parrotz,ftp://%2542parrot@localhost/foo/bar/baz.htmlz %42parrotzftp://localhost:80/foo/bar/�P �Dz ftp://localhost/baz.gif;type=a�Azbaz.gifz Content-typezContent-length)r* r
Z
FTPHandler�ftplibrx r� ZFTP_PORTr rz r� r r� r r� r �
gethostbynamer� r r r r r� �getr� r� )r r" r( ry r9 rI r� r� r� r Ztype_r r ZmimetyperW r� rG r$ )r r% �test_ftp� sL " 1 zHandlerTests.test_ftpc # C sw d d l } t j j � } t � } | _ t j } t t j
j | � � } d } d | d | d t j
d � | f g } y t j
t j � � } Wn t j k
r� d } Yn X| r� | j d | | f � x"| D]} t | d � }
z� z |
j | � Wd |
j � X| j t | � � } z( | j � } | j � }
| j � } Wd | j � Xt j | � } | j j | j d d
�} Wd t j | � X| j | | � | j |
d d � | j |
d
d � | j |
d | � | j | | � q� Wx� d | d d d t j
d � t j � | f d t j � | f g D]k } zT t | d � }
z |
j | � Wd |
j � X| j t j! j" | j t | � � Wd t j | � Xq;Wt j j � } t � } | _ x� d d d d d g D]� \ } } t | � } y | j | � Wn% t j! j" k
r/| j# | � Yn' X| j$ | j% | � | j | j& d � | j | j& d k | � q�Wd S)!Nr s
hello, world
zfile://localhost%sz file://%sz file://%s%sr# r� �wbZusegmtTz Content-typez
text/plainzContent-lengthru z
Last-modifiedzfile://localhost:80%sz file:///file_does_not_exist.txtz)file://not-a-local-host.com//dir/file.txtzfile://%s:80%s/%sz,file://somerandomhost.ontheinternet.com%s/%s� file://ftp.example.com//foo.txtF� file://ftp.example.com///foo.txt� file://ftp.example.com/foo.txt�"file://somehost//foo/something.txt�#file://localhost//foo/something.txt�ftp)r- F)r. F)r/ F)r0 F)r1 F)'Z email.utilsr* r
� FileHandlerrx r� r �TESTFNr r r, r- r r) � gethostname�gaierrorr� r{ �writer2 Z file_openr r1 r� r� �statZutilsZ
formatdate�st_mtime�remover �getcwdr( r r? rS �assertIsrW �type)r r� r9 r� r4 r Ztowrite�urlsZ localaddrrI r3 r� ry rG ZrespurlZstatsZmodifiedr2 rW r$ r$ r% � test_file� s�
zHandlerTests.test_filec C s� t j j � } t � } | _ d } xFd' d( g D]8\ } } t | | d d i � } d | _ | j d d � t � } | j | | � } | j
| j | j | j
| j | j d k d
f | j � } | j | j | j | j
� | � | j | j d � | j | j d � | j | j | � | j | j d
� | j | j d) d* d+ g � | j | j | � q2 Wd | _ | j t j j | j | | � t d d � } | j t | j | � d, g | _ x}d- D]u} t d | � } t d d
i d � } | j | � }
| d k r<| j! d | j" � | j! d | j" � n. | j | j" d d � | j | j" d d � | j | j" d d � | j | j" d d � | j d d � | j d d � | j d d � | j d d � | j | � }
| j | j" d d � | j | j" d d � | j | j" d d � | j | j" d d � q�Wd d � } x] i d d i f D]I } t d | � | � } | s�| j t# | j | � qf| j | � }
qfWt$ j% � }
|
j& d � xy i d d i f D]e } t d |
| � } | s| j t# | j | � q�| j | � }
| j t' |
j( d � � d � q�W|
j) � t* j* d! d" d# d$ d% g � } xZ i d d& i f D]F } t d | | � } | j | � }
| j t' |
j( d � � d& � q�Wd S).Nzhttp://example.com/�GET�POST� blah�Foor] �Spam�eggsr� r� z example.comr r' �
Connectionr2 TZbadpost� r� zContent-lengthz Content-type�0z!application/x-www-form-urlencoded�Hostr\ rL c s s d Vd Vd Vd S)Ns ones twos threer$ r$ r$ r$ r% �
iterable_body� s z-HandlerTests.test_http.<locals>.iterable_bodyzContent-Length� s Something
Something
Something
� r$ r� r� � � � )r@ N)rA rB )rF r2 )rC r] )rD rE )rD rE )rG N)+r* r
r� rx r� r rz rV r� r� r1 r� r� r� r� r� r* � __contains__r r� r� r� r� r� ry r� r( r r? r� � do_request_r~ r� � assertNotIn�unredirected_hdrsr) r� �BytesIOr7 r� rU r2 �array)r r9 r� rI r� ry rW r� r� �hdrs�newreqrJ rG Zfile_objZiterable_arrayr$ r$ r% � test_httpK s�
#
zHandlerTests.test_httpc C sI t � } t d d � } | j | � | j d � | j | j d � d S)Nr� r� zhttps://www.example.com)r r� r� r{ r Z _debuglevel)r r� r9 r$ r$ r% � test_http_handler_debuglevel� s
z)HandlerTests.test_http_handler_debuglevelc C s� t j j � } t � | _ d } d d d d g } xy | D]q } t | | � } | j | � } | j | j d d � | j d d � | j | � } | j | j d d � q: Wd S) NrG z#http://example.com/foo/bar/baz.htmlz$http://example.com//foo/bar/baz.htmlz$http://example.com/foo//bar/baz.htmlz$http://example.com/foo/bar//baz.htmlrI z example.comzsomeproxy:3128)
r* r
r� rx r� r rQ r rS � set_proxy)r r9 ry Zds_urlsZds_urlZds_reqZ np_ds_reqZp_ds_reqr$ r$ r% �test_http_doubleslash� s
z"HandlerTests.test_http_doubleslashc C s� d d d g } t d � } xw | D]o } | | _ t | � } | j | j � | � | j | j pe d | j � | j t | j � � j | j � q" Wd S)Nz http://example.com?foo=bar#bazz)http://example.com?foo=bar&spam=eggs#bashzhttp://example.comr� )r �full_urlr r r� �fragmentZquery)r r> r� rI Zparsedr$ r$ r% �test_full_url_setter� s
z!HandlerTests.test_full_url_setterc C sI t d � } | ` | j | j � | j | j � | j | j d � d S)Nzhttp://www.example.comr� )r r\ rT r] r r� )r r� r$ r$ r% �test_full_url_deleter� s
z"HandlerTests.test_full_url_deleterc C s� t j j � } t � | _ d } t | � } | j | � } | j | j d � | j | j d � d } t | � } | j | � } | j | j d � | j | j d � d S)Nz http://www.python.org?getspamzwww.python.orgz /?getspamzhttp://www.python.orgr� )
r* r
r� rx r� r rQ r r� r� )r r9 Z weird_urlrW rW Zurl_without_pathr$ r$ r% �test_fixpath_in_weirdurls� s z&HandlerTests.test_fixpath_in_weirdurlsc C s� t j j � } t � } | _ d } t | � } t d d i d | � } | j | | � } | j | | � | j t
| d � � t d d i d | � } | j | | � } | j | | � | j t
| d � � t d d i d | � } | j | | � } | j | | � | j t
| d � � t d
d i d | � } | j | j | | � � | j | j
d � | j | j | | d
d i f � d S)
Nzhttp://example.com/r� r� r� r| �� ZAccepted�� zPartial contenti� z Bad gatewayr� )r* r
ZHTTPErrorProcessorrx r� r r� r r< rS r� rT r r| r} )r r9 r� rI rW r� �newrr$ r$ r% � test_errors
s( zHandlerTests.test_errorsc C s� t � } t j j | � } t � | _ t d � } t d d i d � } | j | � } | j | j
| � | j | j
| � | j | j d � | j
| j � | j | | � } | j | j | � | j | j | � | j | | � d S)Nzhttp://example.com/r� r� r� z example.com)r� r* r
�HTTPCookieProcessorrx r� r r� r r< r� r �origin_req_hostrS Z unverifiabler r� r� )r �cjr9 rW r� rW rc r$ r$ r% � test_cookies# s zHandlerTests.test_cookiesc C s* d } d } t j j � } t � } | _ x�d D]�} x�d D]�} t | d | � } t | | � } t j | _ | j
d d
� | d k r� | j
d t t | � � � | j
d d
� y) | | t � | d t d | i � � Wn5 t j j k
r| j | d � | j | � Yn X| j | j j � | � y | j | j j � d � Wn% t k
ry| j | j j � Yn Xd d � | j j D� } | j d | � | j d | � | j | j j d d
� | j d | j j � | j d | j j � q? Wq2 Wt | � } t j | _ | d d � }
t | d d �} d } t j | _ y% x |
| | d � | d } qRWWn1 t j j k
r�| j | t j j j � Yn Xt | d d �} d } t j | _ y) x" |
| | d | � | d } q�WWn1 t j j k
r%| j | t j j j � Yn Xd S) Nzhttp://example.com/a.htmlzhttp://example.com/b.html�- �. �/ �3 �
blah
blah
z
http_error_%sZNonsensezviking=withholdzContent-LengthrD ra �Blah�locationr@ c S s g | ] } | j � � q Sr$ )�lower)�.0�xr$ r$ r% �
<listcomp>S s z.HandlerTests.test_redirect.<locals>.<listcomp>zcontent-lengthz content-typec S s, | j | t � d d t d | i � � d S)Ni. rn ro )r� r� r� )r9 rW rI r$ r$ r% r� ` s z,HandlerTests.test_redirect.<locals>.redirectrf z example.comr zhttp://example.com/r� zhttp://example.com/%d)ri rj rk rl )Nrm ) r* r
�HTTPRedirectHandlerrx r� �getattrr r r� rz rQ rA r� rV r� r� r � HTTPErrorr �assertIsNotNonerW r� �
get_method�AttributeErrorrS ry rG rR rS Z max_repeatsZmax_redirections) r �from_urlZto_urlr9 r� r� ry r� rW rG r� r� r$ r$ r% �
test_redirect4 sj
zHandlerTests.test_redirectc C s d } d d d g } d d d g } d } t j j � } t � } | _ t | � } t j | _ xS | D]K } | d | } | j t j
j | j | t
� d
d t d | i � � qh Wx] | D]U } | d | }
| j | t
� d
d
t d |
i � � | j | j j � |
� q� Wd S)Nzhttp://example.com/a.htmlr� �httpsr2 �file�imapZldapzexample.com/b.htmlz://i. zSecurity Loopholero z That's fine)r* r
rt rx r� r r r� rz r( r rv r� r� r� r rW r� ) r rz Z
valid_schemesZinvalid_schemesZschemeless_urlr9 r� rW r� Z invalid_url� valid_urlr$ r$ r% �test_invalid_redirect~ s$
z"HandlerTests.test_invalid_redirectc C s� d } d } t j j � } t � } | _ t | � } t j | _ t j j
| | � } | j | t � d d t
d | i � � | j | j j � | � d S)Nzhttp://example.com/a.htmlz/b.htmli. z That's finero )r* r
rt rx r� r r r� rz r Zurljoinr� r� r� r rW r� )r rz Z relative_urlr9 r� rW r r$ r$ r% �test_relative_redirect� s z#HandlerTests.test_relative_redirectc C s� d d l m } d d l m } | � } | | d d � t d d � } t j j � } t j j � } t j j | � } t
| | | | � } | j d � | j | j
j d � � d S) Nr )� CookieJar)�interact_netscapezhttp://www.example.com/z spam=eggsi. z%Location: http://www.cracker.com/
�Cookie)Zhttp.cookiejarr� Ztest.test_http_cookiejarr� r� r* r
�HTTPDefaultErrorHandlerrt re r� r{ rS rW rP ) r r� r� rg �hh�hdeh�hrh�cpr� r$ r$ r% �test_cookie_redirect� s
z!HandlerTests.test_cookie_redirectc C sx d } t d d | � } t j j � } t j j � } t | | | � } | j d � } | j | j � | j � � d S)Nz(http://www.example.com/index.html#OK
i. z
Location: zhttp://www.example.com)
r� r* r
r� rt r� r{ r r� �strip)r Zredirected_urlr� r� r� r� r� r$ r$ r% �test_redirect_fragment� s z#HandlerTests.test_redirect_fragmentc s� t j j � d } t j | � t j _ � j t t j d � � t d � � � � � f d d � } | t j j _ t j j
d � } � j | j � d � d S)
Ns( HTTP/1.1 302 Found
Location: ?query
�HTTPConnection�/path� /path?queryc s? � j | t � � � � j | | | | | � d | j _ d S)Ns HTTP/1.1 200 OK
Hello!)r �nextr
r� Zfakedata)�connr� rI �pos�kw)�
real_classr r> r$ r% r
� s z3HandlerTests.test_redirect_no_path.<locals>.requestzhttp://python.org/pathz http://python.org/path?query)r� r� )
r� r� r� r �fakehttpZ
addCleanupr� �iterr
r* r+ r r� )r Z response1r
r� r$ )r� r r> r% �test_redirect_no_path� s z"HandlerTests.test_redirect_no_pathc
C s� Gd d � d t j j � } | � } t j j | � } d } x� | D]� \ } } | j t | � � �d t j d | d
� | _ | j d � } d | d
} | j
} | j | j | � t | � � Wd QRXqD Wd S)Nc @ s e Z d Z d d � Z d S)z4HandlerTests.test_redirect_encoding.<locals>.Handlerc S s: | j | j | � } | j j | _ t j d � | _ | S)Ns) HTTP/1.1 200 OK
Content-Length: 3
123)r� �
connection�buf�last_bufr r� )r rW �resultr$ r$ r% r� � s
z>HandlerTests.test_redirect_encoding.<locals>.Handler.http_openN)rD r rE r� r$ r$ r$ r% �Handler� s r� � /på-dansk/� /p%C3%A5-dansk/� /spaced%20path/�
/spaced path/� /?på-dansk� /?p%C3%A5-dansks! HTTP/1.1 302 Redirect
Location: s
zhttp://example.com/s GET � �r� r� �r� r� �r� r� �r� r� )r� r� r� r� )
r* r
� HTTPHandler� build_openerZsubTest�reprr r� r� r{ r� rO r� )
r r� r r� r: ro r� r r r
r$ r$ r% �test_redirect_encoding� s z#HandlerTests.test_redirect_encodingc C s� t � } t j j t d d � � } | j | � d
g g } t | | � } t d � } | j | j d � | j
| � | j | j d � | j | d d f g d d � | j D� � d S) Nr� zproxy.example.com:3128r� �return responsezhttp://acme.example.com/zacme.example.comr c S s g | ] } | d d � � q S)r r� r$ )rq �tupr$ r$ r% rs � s z+HandlerTests.test_proxy.<locals>.<listcomp>)r� r� ) r r* r
� ProxyHandler�dictr� r� r r r� r{ r� )r r� �phr� r� rW r$ r$ r% �
test_proxy� s
zHandlerTests.test_proxyc C s� d t j d <t � } t j j t d d � � } | j | � t d � } | j | j
d � | j | � | j | j
d � t d � } | j | j
d � | j | � | j | j
d � t j d =d S) Nz
python.org�no_proxyr� zproxy.example.comzhttp://www.perl.org/z www.perl.orgzhttp://www.python.orgzwww.python.org) r �environr r* r
r� r� r� r r r� r{ )r r� r� rW r$ r$ r% �test_proxy_no_proxy� s
z HandlerTests.test_proxy_no_proxyc C s� d t j d <t � } t j j t d d � � } | j | � t d � } | j | j
d � | j | � | j | j
d � t j d =d S)N�*r� r� zproxy.example.comzhttp://www.python.orgzwww.python.org) r r� r r* r
r� r� r� r r r� r{ )r r� r� rW r$ r$ r% �test_proxy_no_proxy_all s
z$HandlerTests.test_proxy_no_proxy_allc C s� t � } t j j t d d � � } | j | � d
g g } t | | � } t d � } | j | j d � | j
| � | j | j d � | j | d d f g d d � | j D� � d S) Nr| zproxy.example.com:3128r� �return responsezhttps://www.example.com/zwww.example.comr c S s g | ] } | d d � � q S)r r� r$ )rq r� r$ r$ r% rs % s z1HandlerTests.test_proxy_https.<locals>.<listcomp>)r� r� ) r r* r
r� r� r� r� r r r� r{ r� )r r� r� r� r� rW r$ r$ r% �test_proxy_https s
z HandlerTests.test_proxy_httpsc C s t � } t j j t d d � � } | j | � t � } | j | � t d � } | j d d � | j d d � | j | j
d � | j | j � | j
| � | j d
| j j � | j d | j j � | j | j � | j | j
d � | j | j d � d � d S) Nr| zproxy.example.com:3128zhttps://www.example.com/�Proxy-Authorization�FooBar�
User-Agent�Grailzwww.example.comzProxy-authorization)r� r� )r� r� )r r* r
r� r� r� r� r rQ r r� rT r� r{ rR r� r� r@ rw rU )r r� r� Z
https_handlerrW r$ r$ r% �$test_proxy_https_proxy_authorization' s$
z1HandlerTests.test_proxy_https_proxy_authorization�darwinzonly relevant for OSXc C s� d d d d d d d d g i } x+ d D]# } | j t | | � d | � q( Wx+ d D]# } | j t | | � d | � qV Wd d d g i } | j t d | � � d S)NZexclude_simpleF�
exceptions�foo.barz *.bar.com� 127.0.0.1z10.10z10.0/16� www.bar.com� 10.10.0.1�10.0.0.1z expected bypass of %s to be True� abc.foo.bar�bar.com� 127.0.0.2� 10.11.0.1� notinbypassz!expected bypass of %s to be FalseT�test)r� r� r� r� r� )r� r� r� r� r� )rO r rS )r Zbypassr� r$ r$ r% �test_osx_proxy_bypass> s
z"HandlerTests.test_osx_proxy_bypass�"c C s� t � } t � } t j j | � } d } t d d | | | f � } | j | � | j | � | j | | d | | | d d � d S)NzACME Widget Storei� z(WWW-Authenticate: Basic realm=%s%s%s
r� z!http://acme.example.com/protected)r r� r* r
r r� r� �_test_basic_auth)r �
quote_charr� �password_manager� auth_handlerr� � http_handlerr$ r$ r% �test_basic_authT s
z HandlerTests.test_basic_authc C s | j d d � d S)Nr� �')r� )r r$ r$ r% �(test_basic_auth_with_single_quoted_realmd s z5HandlerTests.test_basic_auth_with_single_quoted_realmc C s� t � } t � } t j j | � } d } t d d | � } | j | � | j | � | j t � �' | j | | d | | | d d � Wd QRXd S)NzACME Widget Storei� z$WWW-Authenticate: Basic realm=%s
r� z!http://acme.example.com/protected)
r r� r* r
r r� r� Z assertWarns� UserWarningr� )r r� r� r� r� r� r$ r$ r% �#test_basic_auth_with_unquoted_realmg s
z0HandlerTests.test_basic_auth_with_unquoted_realmc C s� t � } t j j t d d � � } | j | � t � } t j j | � } d } t d d | � } | j | � | j | � | j | | d | | | d d � d S)Nr� zproxy.example.com:3128z
ACME Networksi� z(Proxy-Authenticate: Basic realm="%s"
zProxy-authorizationz&http://acme.example.com:3128/protected)
r r* r
r� r� r� r� ZProxyBasicAuthHandlerr� r� )r r� r� r� r� r� r� r$ r$ r% �test_proxy_basic_authw s
z"HandlerTests.test_proxy_basic_authc
C s� Gd d � d t � } Gd d � d t j j � } Gd d � d t j j � } | � } t � } | | � } | | � } d } t d d | � } | j | � | j | � | j | � | j | | d
| | | d d � | j | j
d d
g d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)zQHandlerTests.test_basic_and_digest_auth_handlers.<locals>.RecordingOpenerDirectorc S s t j | � g | _ d S)N)r r� �recorded)r r$ r$ r% r� � s
zZHandlerTests.test_basic_and_digest_auth_handlers.<locals>.RecordingOpenerDirector.__init__c S s | j j | � d S)N)r� r� )r r� r$ r$ r% �record� s zXHandlerTests.test_basic_and_digest_auth_handlers.<locals>.RecordingOpenerDirector.recordN)rD r rE r� r� r$ r$ r$ r% �RecordingOpenerDirector� s r� c @ s e Z d Z d d � Z d S)zOHandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestDigestAuthHandlerc _ s- | j j d � t j j j | | | � d S)N�digest)r� r� r* r
�HTTPDigestAuthHandler�http_error_401)r r} r� r$ r$ r% r� � s z^HandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestDigestAuthHandler.http_error_401N)rD r rE r� r$ r$ r$ r% �TestDigestAuthHandler� s r� c @ s e Z d Z d d � Z d S)zNHandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestBasicAuthHandlerc _ s- | j j d � t j j j | | | � d S)N�basic)r� r� r* r
r r� )r r} r� r$ r$ r% r� � s z]HandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestBasicAuthHandler.http_error_401N)rD r rE r� r$ r$ r$ r% �TestBasicAuthHandler� s r� z
ACME Networksi� z&WWW-Authenticate: Basic realm="%s"
r� z!http://acme.example.com/protectedr� r� r� ) r r* r
r� r r� r� r� r� r r� )
r r� r� r� r� r� Zdigest_handlerZ
basic_handlerr� r� r$ r$ r% �#test_basic_and_digest_auth_handlers� s$
z0HandlerTests.test_basic_and_digest_auth_handlersc C s^ t � } t j j d � } t d d � } | j | � | j | � | j t | j d � d S)Ni� z WWW-Authenticate: Kerberos
zhttp://www.example.com) r r* r
r� r� r� r( r) r{ )r r� Zdigest_auth_handlerr� r$ r$ r% �$test_unsupported_auth_digest_handler� s
z1HandlerTests.test_unsupported_auth_digest_handlerc C s^ t � } t j j d � } t d d � } | j | � | j | � | j t | j d � d S)Ni� zWWW-Authenticate: NTLM
zhttp://www.example.com) r r* r
r r� r� r( r) r{ )r r� Zbasic_auth_handlerr� r$ r$ r% �#test_unsupported_auth_basic_handler� s
z0HandlerTests.test_unsupported_auth_basic_handlerc C s� d d l } d \ }
} | j | | |
| � | j | | j � | j | | j � | j |
| j � | j | | j � | j | � | j | j | � | j | j | � | j t
| j � d � | j | j d j
| � � t d |
| f d � } d | j | � j � j � }
| j | j d j | � |
� | j | j d j | |
� d | _ | _ | j � | j | � | j t
| j � d � | j | j d j
| � � d S)
Nr �wile�coyoter� z%s:%s�asciizBasic r� )r� r� )�base64ri r r� rI r� rZ r{ r� r� r� r� rS rP �bytesZ encodebytesr� �decoderU rS r� )r r� r� Z auth_headerr� r� r� � request_urlZ
protected_urlr� r� rZ ZuserpassZauth_hdr_valuer$ r$ r% r� � s0
z HandlerTests._test_basic_authc C s� d \ } } d } t d � } t � } t | � } | j d | | | d d �| j | � } | j | � t � } | j | � | j | � | j | � | j | j � d S)Nr� r� z!http://acme.example.com/protectedr� �is_authenticatedT)r� r� )
r� r r ri r� rO r r� r{ r� ) r r� rZ r� r� � pwd_manager�auth_prior_handler�is_authr� r$ r$ r% � test_basic_prior_auth_auto_send� s
z,HandlerTests.test_basic_prior_auth_auto_sendc
C s d \ } } d } d } t � } t | � } | j | | | | � | j | � } | j | � t � } | j | � t d d d � } | j | � | j | � | j | � } | j | � t
d � } | j | j � t � } | j | � | j | � | j | � | j | j � d S) Nr� r� z!http://acme.example.com/protectedZACMEi� z&WWW-Authenticate: Basic realm="%s"
r� )r� r� ) r r ri r� rS r r� r� r{ rO r� r� )
r r� rZ r� r� r� r� r� r� r� r$ r$ r% �.test_basic_prior_auth_send_after_first_success
s.
z;HandlerTests.test_basic_prior_auth_send_after_first_successc C s� x� d D]� \ } } d j | � } t j | j � | � } t j j � } t d � } d | _ | j | | � � } | j
� Wd QRX| j | j j
d j | � � q Wd S)
N�Connection: close� data�Transfer-Encoding: chunked� 4
data
0
�Content-Length: 4zHTTP/1.1 200 OK
{}
z
http://dummy/z Connection not closed with {!r}�r� r� �r� r� �r� r� )r� r� r� )�formatr r� r r* r
r� r rz r� r1 rO �fakesock�closed)r Ztransferry �headerr� r rW Zrespr$ r$ r% �test_http_closed5 s z HandlerTests.test_http_closedc
C sz t j d � } t j j � } t d � } d | _ | j t j j
� � | j | | � Wd QRX| j | j
j d � d S)NrG z
http://dummy/zConnection not closed)r r� r* r
r� r rz r( r� r� Z
BadStatusLiner� rO r� r� )r r� r rW r$ r$ r% �test_invalid_closedF s z HandlerTests.test_invalid_closedN)*rD r rE r+ r? rX rY r[ r^ r_ r` rd rh r{ r� r� r� r� r� r� r� r� r� r� r� r �
skipUnless�sys�platformr� r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% r � sF @ \ l J
# ' 2
% ( r c @ ss e Z d Z d d � Z d d � Z e j e j d � d � d d � � Z d d
� Z
d d � Z d
d � Z d S)� MiscTestsc s- | j t � f d d � | j D� � � d S)Nc 3 s | ] } | j � k Vq d S)N)r� )rq r9 )�
handler_classr$ r% � <genexpr>T s z/MiscTests.opener_has_handler.<locals>.<genexpr>)rO �anyr� )r r� r r$ )r r% �opener_has_handlerS s z MiscTests.opener_has_handlerc C s� Gd d � d t j j � } Gd d � d t j j � } Gd d � d t j j � } t j j } | | | � } | j | | � | j | | � | | | � � } | j | | � | j | | � | | � } | j | | � | � } | j | t j j � | t j j � } | j | t j j � | t j j � � } | j | t j j � Gd d � d t j j � } | | | � } | j | | � | j | | � d S) Nc @ s e Z d Z d S)z2MiscTests.test_build_opener.<locals>.MyHTTPHandlerN)rD r rE r$ r$ r$ r% �
MyHTTPHandlerX s r c @ s e Z d Z d d � Z d S)z/MiscTests.test_build_opener.<locals>.FooHandlerc S s d S)Nr$ )r r$ r$ r% �foo_open\ s z8MiscTests.test_build_opener.<locals>.FooHandler.foo_openN)rD r rE r r$ r$ r$ r% �
FooHandler[ s r c @ s e Z d Z d d � Z d S)z/MiscTests.test_build_opener.<locals>.BarHandlerc S s d S)Nr$ )r r$ r$ r% �bar_open` s z8MiscTests.test_build_opener.<locals>.BarHandler.bar_openN)rD r rE r
r$ r$ r$ r% �
BarHandler_ s r c @ s e Z d Z d S)z7MiscTests.test_build_opener.<locals>.MyOtherHTTPHandlerN)rD r rE r$ r$ r$ r% �MyOtherHTTPHandler| s r )r* r
r� � BaseHandlerr� r )r r r r r� r� r r$ r$ r% �test_build_openerW s, zMiscTests.test_build_openerZnetworkz test requires network accessc C s� t j d � �� t j j � } t j j d � } | j d | j � | j | d j d � � | j d | j � | j d | j
d � � | j | d j d � � | j d | j � | j d | j
d � � Wd QRXd S) Nzhttp://www.example.com/rb zus-ascii� 1zContent-lengthZ
1234567890s
1234567890ro ) r Ztransient_internetr* r
r� r r ry r{ r rU )r r� r
r$ r$ r% �test_issue16464� s zMiscTests.test_issue16464c C s� d } d } } } d } t j j | | | | | � } | j t | d � � | j | j d � | j t | d � � | j | j d � d | j | j f } | j t
| � | � d | j | j f } | j t | � | � d S)Nzsomething bad happenedzContent-Length: 42r> rG zHTTP Error %s: %sz<HTTPError %s: %r>) r* r rv rO r� r r> rG r� r� rA r� )r r� rI r� r� rV rB Zexpected_errmsgr$ r$ r% �test_HTTPError_interface� s z"MiscTests.test_HTTPError_interfacec C sq d d d d d d d d d! d# d% d' g } x* | D]" \ } } | j t | � | � q1 W| j t t d � f d S)(N�proxy.example.com�proxy.example.com:3128� joe:password@proxy.example.comrY rZ �#joe:password@proxy.example.com:3128�http://proxy.example.com/r� � http://proxy.example.com:3128/�&http://joe:password@proxy.example.com/�*http://joe:password@proxy.example.com:3128�1ftp://joe:password@proxy.example.com/rubbish:3128r2 �%http://joe:password@proxy.example.comzfile:/ftp.example.com�NNNr )r r �NNNr )r r �NNNr )r r �NNNr )r r �NrY rZ r )r r �NrY rZ r )r r! �r� NNr )r r" �r� NNr )r r# �r� rY rZ r )r r$ �r� rY rZ r )r r% �r2 rY rZ r )r r& �r� rY rZ r )r r' )r r r( r) )r Zparse_proxy_test_casesZtcr r$ r$ r% �test_parse_proxy� s4 zMiscTests.test_parse_proxyc
C sL t � } | j t � � } | j d � Wd QRX| j t | j � d � d S)N�invalidz5Unsupported digest authentication algorithm 'invalid')r
r( r) Zget_algorithm_implsr rA � exception)r r �excr$ r$ r% �test_unsupported_algorithm� s z$MiscTests.test_unsupported_algorithmN)
rD r rE r r r r� r Zis_resource_enabledr r r( r, r$ r$ r$ r% r Q s , %r c @ s� e Z d Z Gd d � d e � 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 d � Z d d � Z d d � Z d d � Z d d � Z d S) � RequestTestsc @ s e Z d Z d Z d S)zRequestTests.PutRequest�PUTN)rD r rE r� r$ r$ r$ r% �
PutRequest� s r/ c C sp t d � | _ t d d d d d i �| _ t d d d �| _ | j d � | _ | j d d d �| _ d S) Nz http://www.python.org/~jeremy/ry rG zX-Testr� r� �HEADrA )r r* �post�headr/ �put�
force_post)r r$ r$ r% �setUp� s zRequestTests.setUpc C s� | j d | j j � � | j d | j j � � | j d | j j � � | j d | j j � � | j d | j j � � d S)NrA r@ r0 r. )r r1 rx r* r2 r3 r4 )r r$ r$ r% � test_method� s
zRequestTests.test_methodc C sh | j | j j � | j d | j j � � d | j _ | j | j j � | j d | j j � � d S)Nr@ ra rA )rS r* ry r rx rO )r r$ r$ r% � test_data� s
zRequestTests.test_datac C si | j d | j j � | j j d d � | j d | j j d � d | j _ | j d | j j � d S)NzContent-length�* ra )rR r* rS rV r ry )r r$ r$ r% �.test_setting_data_should_remove_content_length� s
z;RequestTests.test_setting_data_should_remove_content_lengthc C sr | j d | j j � d | j _ | j j d d � | j d | j j d � | j ` | j d | j j � d S)NzContent-lengthr\ rM )rR r* rS ry rV r )r r$ r$ r% �/test_deleting_data_should_remove_content_length� s z<RequestTests.test_deleting_data_should_remove_content_lengthc C s | j d | j j � � d S)Nz http://www.python.org/~jeremy/)r r* r� )r r$ r$ r% �test_get_full_url s z RequestTests.test_get_full_urlc C s9 | j d | j j � t d � } | j d | j � d S)Nz /~jeremy/zhttp://www.python.org/r' )r r* r� r )r rW r$ r$ r% �
test_selector s zRequestTests.test_selectorc C s | j d | j j � d S)Nr� )r r* r= )r r$ r$ r% �
test_get_type s zRequestTests.test_get_typec C s | j d | j j � d S)Nzwww.python.org)r r* r� )r r$ r$ r% �
test_get_host s zRequestTests.test_get_hostc C s# t d � } | j d | j � d S)Nzhttp://www.%70ython.org/zwww.python.org)r r r� )r rW r$ r$ r% �test_get_host_unquote s z"RequestTests.test_get_host_unquotec C so | j | j j � � | j j d d � | j | j j � � | j d | j j � | j d | j j � d S)Nz www.perl.orgr� zwww.python.org)rS r* Z has_proxyrZ rO r rf r� )r r$ r$ r% r� s
zRequestTests.test_proxyc C s# t d � } | j d | j � d S)Nz<URL:http://www.python.org>zwww.python.org)r r r� )r rW r$ r$ r% �test_wrapped_url s z RequestTests.test_wrapped_urlc C sj t d � } | j d | j � t d � } | j d | j � d } t | � } | j | j � | � d S)Nz-http://www.python.org/?qs=query#fragment=truez
/?qs=queryz http://www.python.org/#fun=truer' z.http://docs.python.org/library/urllib2.html#OK)r r r� r� )r rW rI r$ r$ r% �test_url_fragment! s z RequestTests.test_url_fragmentc C sI d d d g } x3 | D]+ } t | � } | j | j � | j � q Wd S)Nzhttp://docs.python.orgz.http://docs.python.org/library/urllib2.html#OKz-http://www.python.org/?qs=query#fragment=true)r r r� r\ )r r> rI rW r$ r$ r% � test_url_fullurl_get_full_url, s
z*RequestTests.test_url_fullurl_get_full_urlN)rD r rE r r/ r5 r6 r7 r9 r: r; r< r= r> r? r� r@ rA rB r$ r$ r$ r% r- � s
r- �__main__)6r r� r r r r� r rU r Zurllib.requestr* r r r r r r r
Z urllib.parser r� Z http.clientr� ZTestCaser rF rx r� r� r� r� r� r� r� �IOBaser� r� r� r� r� r
r
r� r� r� r� r� r� r r r r- rD �mainr$ r$ r$ r% �<module> sN 4 3�
+6 "
� � � � ��a