a
5êdga ã @ sz d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZ z d dlZW n ey‚ dZY n0 ej e¡Zej ed¡Zej ed¡ZG dd„ dejjƒZG dd „ d ejƒZ G d
d „ d ƒZ G d d
„ d
ejj ƒZ G dd„ dejj ƒZ G dd„ de j!ƒZ"e #d¡G dd„ de j!ƒƒZ$dd„ Z%G dd„ de j!ƒZ&da'dd„ Z(dd „ Z)e*d krve +¡ dS ) é N)Úsupport)Úhashlib_helperz keycert.pemz keycert2.pemc @ s e Zd ZdZdd„ Zdd„ ZdS )ÚLoopbackHttpServerz^HTTP server w/ a few modifications that make it useful for
loopback testing purposes.
c C s" t jj | ||¡ | j d¡ d S )Ngš™™™™™¹?)ÚhttpÚserverÚ
HTTPServerÚ__init__ÚsocketÚ
settimeout)ÚselfZserver_addressZRequestHandlerClass© r ú6/usr/local/lib/python3.9/test/test_urllib2_localnet.pyr ! s
þzLoopbackHttpServer.__init__c C s | j ¡ \}}| d¡ ||fS )z HTTPServer method, overridden.g $@)r Úacceptr
)r ÚrequestZclient_addressr r r
Ú get_request* s
z LoopbackHttpServer.get_requestN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r r r
r s r c @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚLoopbackHttpServerThreadz2Stoppable thread that runs a loopback http server.c C s< t j | ¡ d| _t ¡ | _d|_td|ƒ| _| jj | _
d S )NFzHTTP/1.0)z 127.0.0.1r ) Ú threadingÚThreadr Ú _stop_serverÚEventÚreadyZprotocol_versionr ÚhttpdZ server_portÚport©r Úrequest_handlerr r r
r 9 s
ÿz!LoopbackHttpServerThread.__init__c C s d| _ | ¡ | j ¡ dS )z.Stops the webserver if it's currently running.TN)r Újoinr Z server_close©r r r r
ÚstopB s z LoopbackHttpServerThread.stopc C s | j ¡ | js | j ¡ q
d S ©N)r Úsetr r Úhandle_requestr r r r
ÚrunJ s
z LoopbackHttpServerThread.runN)r r r r r r! r% r r r r
r 6 s r c @ sX e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dS )ÚDigestAuthHandlerz-Handler for performing digest authentication.c C s" d| _ g | _i | _d| _d| _d S )Nr z
Test RealmÚauth)Ú _request_numÚ_noncesÚ_usersÚ _realm_nameÚ_qopr r r r
r T s
zDigestAuthHandler.__init__c C s
|| _ d S r" )r, )r Zqopr r r
Úset_qop[ s zDigestAuthHandler.set_qopc C s t |tƒsJ ‚|| _d S r" )Ú
isinstanceÚdictr* )r Zusersr r r
Ú set_users^ s zDigestAuthHandler.set_usersc C s
|| _ d S r" )r+ )r Zrealmr r r
Ú set_realmb s zDigestAuthHandler.set_realmc C s8 | j d7 _ t t| j ƒ d¡¡ ¡ }| j |¡ |S )Né Úascii)r( ÚhashlibÚmd5ÚstrÚencodeÚ hexdigestr) Úappend)r Únoncer r r
Ú_generate_noncee s z!DigestAuthHandler._generate_noncec C s‚ | d¡}||d d … }| d¡}i }|D ]P}| d¡\}}| ¡ }|d dkrl|d dkrl|dd… }n| ¡ }|||< q,|S )Nú r2 ú,ú=r ú"éÿÿÿÿ)ÚfindÚsplitÚstrip)r Zauth_strZfirst_space_indexÚpartsÚ auth_dictÚpartÚnameÚvaluer r r
Ú_create_auth_dictk s
z#DigestAuthHandler._create_auth_dictc C s– i }| |¡ ||d< ||d< ||d< d| }t | d¡¡ ¡ }d| }t | d¡¡ ¡ } ||d< | |d< d | }
t |
d¡¡ ¡ } | |d
kS ) NÚpasswordÚmethodÚuriz#%(username)s:%(realm)s:%(password)sr3 z%(method)s:%(uri)sÚHA1ÚHA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)sÚresponse)Úupdater4 r5 r7 r8 ) r rE rJ rK rL Z
final_dictZHA1_strrM ZHA2_strrN Z response_strrO r r r
Ú_validate_auth| s
ÿz DigestAuthHandler._validate_authc C sN | dd¡ | dd¡ | dd| j| j| ¡ f ¡ | ¡ |j d¡ dS ) Ni— z Proxy Authentication Requiredú Content-Typeú text/htmlzProxy-Authenticatez(Digest realm="%s", qop="%s",nonce="%s", s Proxy Authentication Required.F)Ú
send_responseÚ send_headerr+ r, r; Ú end_headersÚwfileÚwriter r r r
Ú_return_auth_challengeŽ s ýÿ z(DigestAuthHandler._return_auth_challengec C sÊ t | jƒdkrdS d|jvr&| |¡S | |jd ¡}|d | jv rT| j|d }n
| |¡S | d¡| jvrx| |¡S | j |d ¡ d}|j|j fD ]}|
|||j |¡r˜d}q˜|sÂ| |¡S dS dS )a Performs digest authentication on the given HTTP request
handler. Returns True if authentication was successful, False
otherwise.
If no users have been set, then digest auth is effectively
disabled and this method will always return True.
r TzProxy-AuthorizationÚusernamer: FN) Úlenr* ÚheadersrY rI Úgetr) ÚremoveÚpathÚ
short_pathrQ Zcommand)r r rE rJ Zauth_validatedr_ r r r
r$ s0
ÿ
ý
z DigestAuthHandler.handle_requestN)
r r r r r r- r0 r1 r; rI rQ rY r$ r r r r
r& Q s r& c @ sf e Zd ZdZdZdZdZdeef Ze e
d¡¡ d¡Z dd„ Z
d d
„ Zd d „ Zd
d„ Zdd„ ZdS )ÚBasicAuthHandlerz,Handler for performing basic authentication.ÚtestUserÚtestPassÚTestz%s:%sr3 c O s t jjj| g|¢R i |¤Ž d S r" )r r ÚBaseHTTPRequestHandlerr )r ÚargsÚkwargsr r r
r Ó s zBasicAuthHandler.__init__c G s d S r" r ©r Úformatrf r r r
Ú log_messageÖ s z BasicAuthHandler.log_messagec C s" | d¡ | dd¡ | ¡ d S )NéÈ ú Content-typerS )rT rU rV r r r r
Údo_HEADÚ s
zBasicAuthHandler.do_HEADc C s4 | d¡ | dd| j ¡ | dd¡ | ¡ d S )Ni‘ zWWW-AuthenticatezBasic realm="%s"rl rS )rT rU ÚREALMrV r r r r
Ú do_AUTHHEADß s
z BasicAuthHandler.do_AUTHHEADc C sh | j dd¡s$| ¡ | j d¡ n@| j dd¡d| j kr\| d¡ | ¡ | j d¡ n| ¡ d S )NZ
AuthorizationÚ s No Auth header receivedzBasic rk s It works)r\ r] ro rW rX Ú ENCODED_AUTHrT rV r r r r
Údo_GETå s ÿÿ
zBasicAuthHandler.do_GETN)r r r r ÚUSERÚPASSWDrn Z USER_PASSWDÚbase64Z b64encoder7 Údecoderq r rj rm ro rr r r r r
ra Ê s ra c @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚFakeProxyHandlerzÏThis is a 'fake proxy' that makes it look like the entire
internet has gone down due to a sudden zombie invasion. It main
utility is in providing us with authentication support for
testing.
c O s& || _ tjjj| g|¢R i |¤Ž d S r" )Údigest_auth_handlerr r re r )r rx rf rg r r r
r ý s zFakeProxyHandler.__init__c G s d S r" r rh r r r
rj s z FakeProxyHandler.log_messagec C sv t j | jd¡\}}}}}}|| _| j | ¡rr| dd¡ | dd¡ | ¡ | j
t d| j dƒ¡ | j
d¡ d S ) Nr rk ZOKrR rS zYou've reached %s!
r3 sF Our apologies, but our server is down due to a sudden zombie invasion.)
ÚurllibÚparseÚurlparser_ r` rx r$ rT rU rV rW rX Úbytes)r ZscmÚnetlocr_ ÚparamsÚqueryÚfragmentr r r
rr s ÿ ÿzFakeProxyHandler.do_GETN)r r r r r rj rr r r r r
rw ö s rw c sP e Zd ZdZdZdZdZ‡ fdd„Zdd„ Z‡ fd d
„Z d d „ Z
d
d„ Z ‡ Z S )ÚBasicAuthTestsrb rc Z Incorrectrd c sT t t| ƒ ¡ dd„ }t|ƒ| _| | j¡ d| jj | _| j ¡ | jj
¡ d S )Nc _ s t | i |¤ŽS r" )ra ©rf rg r r r
Ú#http_server_with_basic_auth_handler s zABasicAuthTests.setUp..http_server_with_basic_auth_handlerzhttp://127.0.0.1:%s) Úsuperr ÚsetUpr r Ú
addCleanupÚ stop_serverr Ú
server_urlÚstartr Úwait)r rƒ ©Ú __class__r r
r… s
zBasicAuthTests.setUpc C s | j ¡ d | _ d S r" ©r r! r r r r
r‡ ( s
zBasicAuthTests.stop_serverc s t t| ƒ ¡ d S r" )r„ r ÚtearDownr r‹ r r
rŽ , s zBasicAuthTests.tearDownc C sz t j ¡ }| | j| j| j| j¡ t j t j |¡¡ z|
t j | j¡¡ W n& t j j
yt | d| j ¡ Y n0 d S )Nz!Basic auth failed for the url: %s)ry r ÚHTTPBasicAuthHandlerÚ add_passwordrn rˆ rs rt Úinstall_openerÚ build_openerÚ
assertTrueÚurlopenÚerrorÚ HTTPErrorÚfail©r Zahr r r
Útest_basic_auth_success/ s
z&BasicAuthTests.test_basic_auth_successc C sR t j ¡ }| | j| j| j| j¡ t j t j |¡¡ |
t j j t jj
| j¡ d S r" )ry r r r rn rˆ rs ÚINCORRECT_PASSWDr‘ r’ Ú assertRaisesr• r– r” r˜ r r r
Útest_basic_auth_httperror8 s
z(BasicAuthTests.test_basic_auth_httperror)
r r r rs rt rš rn r… r‡ rŽ r™ rœ Ú
__classcell__r r r‹ r
r s r r5 c sT e Zd ZdZdZdZdZ‡ fdd„Zdd„ Zd d
„ Z d d „ Z
d
d„ Z dd„ Z ‡ Z
S )ÚProxyAuthTestszhttp://localhostZtesterZtest123Z TestRealmc sÜ t tˆ ƒ ¡ dd„ }ˆ |tj ¡ ¡ dtjd< dtjd< tƒ ˆ _ˆ j ˆ j
ˆ j i¡ ˆ j ˆ j
¡ ‡ fdd„}t|ƒˆ _ˆ ˆ j¡ ˆ j ¡ ˆ jj ¡ dˆ jj }tj d |i¡}tj ¡ ˆ _tj |ˆ j¡ˆ _d S )
Nc S s t j ¡ t j | ¡ d S r" ©ÚosÚenvironÚclearrP ©Z old_environr r r
Úrestore_environJ s
z-ProxyAuthTests.setUp..restore_environrp ÚNO_PROXYÚno_proxyc s t ˆ jg| ¢R i |¤ŽS r" )rw rx r‚ r r r
Úcreate_fake_proxy_handlerU s z7ProxyAuthTests.setUp..create_fake_proxy_handlerzhttp://127.0.0.1:%dr ) r„ rž r… r† r r¡ Úcopyr& rx r0 rs rt r1 rn r r r‡ r‰ r rŠ r ry r Z ProxyHandlerZProxyDigestAuthHandlerÚproxy_digest_handlerr’ Úopener)r r¤ r§ Z proxy_urlÚhandlerr‹ r r
r… G s&
ÿzProxyAuthTests.setUpc C s | j ¡ d | _ d S r" r r r r r
r‡ b s
zProxyAuthTests.stop_serverc C sF | j | j| j| j| jd ¡ | j d¡ | t j
j | j j
| j¡ d S )NZbadr' )r© r rn ÚURLrs rt rx r- r› ry r• r– rª Úopenr r r r
Ú-test_proxy_with_bad_password_raises_httperrorf s ÿ
þz.FakeHTTPRequestHandlerz TestHTTP/éP c S s* | ¡ }|r&| j |¡}||d … }qd S r" )Ú send_headrW rX )r ÚbodyÚdoner r r
rr – s z8GetRequestHandler..FakeHTTPRequestHandler.do_GETc S s2 | j d }| j t|ƒ¡}| ¡ | j |¡ d S )NzContent-Length)r\ Zrfiler± Úintrr Úrequestsr9 )r Zcontent_lengthZ post_datar r r
Údo_POSTœ s
z9GetRequestHandler..FakeHTTPRequestHandler.do_POSTc s| | j ˆ _| j | j¡ ˆ d¡\}}}| |¡ |D ] \}}| ||d| ji ¡ q4|rp| dd¡ | ¡ |S | ¡ d S )Nr r rl z
text/plain)
r\ Úheaders_receivedr¼ r9 r_ ÚpoprT rU r rV )r Z
response_coder\ r¹ ÚheaderrH ©ÚFakeHTTPRequestHandlerÚ responsesr r
r¸ ¢ s
z;GetRequestHandler..FakeHTTPRequestHandler.send_headc W s d S r" r )r rf r r r
rj ± s z=GetRequestHandler..FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr¼ r¾ r rr r½ r¸ rj r rÁ r r
r s r )r r re )rà r rÁ r
ÚGetRequestHandler s &rÄ c s¾ e Zd ZdZ‡ fdd„Zd+dd„Zdd„ Zd,d d
„Zd-d d „Zd
d„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd „ Zd d „ Zd d „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Z‡ ZS ).Ú TestUrlopenaN Tests urllib.request.urlopen using the network.
These tests are not exhaustive. Assuming that testing using files does a
good job overall of some of the basic interface features. There are no
tests exercising the optional 'data' and 'proxies' arguments. No tests
for transparent redirection have been written.
c sN t t| ƒ ¡ | tjj¡ dd„ }| |tj ¡ ¡ dtjd< dtjd< d S )Nc S s t j ¡ t j | ¡ d S r" rŸ r£ r r r
r¤ È s
z*TestUrlopen.setUp..restore_environÚ*r¥ r¦ )
r„ rÅ r… r† ry r Ú
urlcleanupr r¡ r¨ )r r¤ r‹ r r
r… Á s
zTestUrlopen.setUpNc K sx g }t jj||fi |¤Ž}zH| | d¡¡ | | ¡ ¡ | | d¡¡ | | ¡ ¡ W | ¡ n
| ¡ 0 d |¡S )Nrk i ó )
ry r r” ÚextendÚ readlinesr9 Úreadliner± Úcloser )r ÚurlÚdatarg ÚlÚfr r r
r” Ï s zTestUrlopen.urlopenc C s | j ¡ d | _ d S r" r r r r r
r‡ Ü s
zTestUrlopen.stop_serverc C sZ |d u rdg dfg}t |ƒ}t|ƒ| _| | j¡ | j ¡ | jj ¡ | jj}||_|S )Nrk s
we don't care) rÄ r r r† r‡ r‰ r rŠ r )r rà r« r r r r
Ú start_serverà s
zTestUrlopen.start_serverc K s^ t tjdƒs| d¡ ddlm} |d u r6dg dfg}t|ƒ}|| fd|i|¤Ž}|j|_|S )NZ HTTPSHandlerzssl support requiredr )Úmake_https_serverrk ó
we care a bitZ
handler_class)Úhasattrry r ÚskipTestZtest.ssl_serversrÒ rÄ r )r rà rg rÒ r« r r r r
Ústart_https_serverí s
z TestUrlopen.start_https_serverc C sV d}ddgdfdg |fg}| |¡}| d|j ¡}| ||¡ | |jddg¡ d S ) Ns We got here...i. )ZLocationz(http://localhost:%(port)s/somewhere_elserp rk úhttp://localhost:%s/ú/z/somewhere_else©rÑ r” r Ú assertEqualr¼ )r Úexpected_responserà r« rÎ r r r
Útest_redirectionø s ÿý
z TestUrlopen.test_redirectionc C s@ d}d}ddg|fg}| |¡}| d|j ¡}| ||¡ d S )Ns hello worlds a
hello worl
1
d
0
rk )zTransfer-EncodingZchunkedr× )rÑ r” r rÚ )r rÛ Z
chunked_startrO r« rÎ r r r
Ú test_chunked s ÿ
zTestUrlopen.test_chunkedc
C sŽ d}| dg |fg¡}z| d|j ¡ W n8 tjjyb } z | ¡ }| ¡ W Y d }~nd }~0 0 | d¡ | ||¡ | |j
dg¡ d S )Ns Bad bad bad...i” zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble) rÑ r” r ry r• rµ r± rÌ r— rÚ r¼ )r rÛ r« rÐ rÎ r r r
Útest_404 s
zTestUrlopen.test_404c C sF d}| dg |fg¡}| d|j ¡}| ||¡ | |jdg¡ d S )Nó
pycon 2008...rk úhttp://localhost:%s/bizarreú/bizarrerÙ ©r rÛ r« rÎ r r r
Útest_200" s
zTestUrlopen.test_200c C sJ d}| dg |fg¡}| d|j d¡}| ||¡ | |jddg¡ d S )Nrß rk rà s get=with_feelingrá rÙ râ r r r
Útest_200_with_parameters) s ÿ z$TestUrlopen.test_200_with_parametersc C s8 | ¡ }tjtd}| jd|j |d}| |d¡ d S )N©Zcafileú https://localhost:%s/bizarre©ÚcontextrÓ )rÖ ÚsslÚcreate_default_contextÚCERT_localhostr” r rÚ )r r« rè rÎ r r r
Ú
test_https1 s zTestUrlopen.test_httpsc C sî | j td}t dtf¡À | jd|j td}| |d¡ | t j
j ¡$}| jd|j t d W d ƒ n1 st0 Y | j t d}| t j
j ¡$}| jd|j t d W d ƒ n1 sÂ0 Y W d ƒ n1 sà0 Y d S )N©Úcertfilerp ræ rå rÓ )
rÖ rë r Úcheck_warningsÚDeprecationWarningr” r rÚ r› ry r• rµ ÚCERT_fakehostname)r r« rÎ Úcmr r r
Útest_https_with_cafile7 s ÿ ÿ$ ÿz"TestUrlopen.test_https_with_cafilec C s€ | j td}t dtf¡R | tjj¡$}| j d|j
dd W d ƒ n1 sT0 Y W d ƒ n1 sr0 Y d S )Nrí rp ræ T)Z cadefault) rÖ rë r rï rð r› ry r• rµ r” r )r r« rò r r r
Útest_https_with_cadefaultH s ÿz%TestUrlopen.test_https_with_cadefaultc s† t d u r| d¡ t js"| d¡ d ‰ ‡ fdd„}t t j¡}| |¡ | j|td}t jtd}| j d|j
|d | ˆ d ¡ d S )
Nzssl module requiredz SNI support required in OpenSSLc s |‰ d S r" r )Zssl_sockZ server_nameZinitial_context©Zsni_namer r
Úcb_sniV s z*TestUrlopen.test_https_sni..cb_sni)rè rî rå zhttps://localhost:%srç Ú localhost) ré rÕ ZHAS_SNIZ
SSLContextZPROTOCOL_TLS_SERVERZset_servername_callbackrÖ rë rê r” r rÚ )r rö rè r« r rõ r
Útest_https_sniP s
zTestUrlopen.test_https_snic C sd | ¡ }tjjd|j ddid}tj |¡ W d ƒ n1 sD0 Y | |jd d¡ d S )Nr× ZRangez bytes=20-39)r\ )rÑ ry r ZRequestr r” rÚ r¾ )r r« Zreqr r r
Útest_sending_headers` s ÿ z TestUrlopen.test_sending_headersc C sn | ¡ }tj d|j ¡@}dD ]}| t||ƒd| ¡ q | | ¡ d¡ W d ƒ n1 s`0 Y d S )Núhttp://localhost:%s)r± rÌ ÚinfoÚgeturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)rÑ ry r r” r r“ rÔ r± )r r« Úopen_urlÚattrr r r
Ú
test_basich s ÿzTestUrlopen.test_basicc C sl | ¡ }tj d|j ¡}| | ¡ }W d ƒ n1 s<0 Y | |tjj d¡ |
| ¡ d¡ d S )Nrú zEobject returned by 'info' is not an instance of email.message.MessageZplain) rÑ ry r r” r rû ZassertIsInstanceÚemailÚmessageZMessagerÚ Zget_content_subtype)r r« rý Zinfo_objr r r
Ú test_infop s ÿ& ÿzTestUrlopen.test_infoc C s\ | ¡ }tj d|j ¡}| | ¡ }W d ƒ n1 s<0 Y | |d|j ¡ d S )Nrú )rÑ ry r r” r rü rÚ )r r« rý rÍ r r r
Ú test_geturl{ s
&zTestUrlopen.test_geturlc C sB d}| dg |fg¡}tj d|j ¡}|D ]}| ||¡ q,d S )Nrß rk rú )rÑ ry r r” r rÚ )r rÛ r« rÎ Úliner r r
Útest_iterationƒ s
zTestUrlopen.test_iterationc
C s” ddddd d g}d |¡}| dg |fg¡}tj d |j ¡}t|ƒD ]0\}}| ||| d
|t|| ƒt|ƒf ¡ qJ| |d t|ƒ¡ d S ) Ns We
s got
s here
s verylong i ó
rÈ rk rú zQFetched line number %s doesn't match expected:
Expected length was %s, got %sr2 ) r rÑ ry r r” r Ú enumeraterÚ r[ )r ÚlinesrÛ r« rÎ Úindexr r r r
Útest_line_iterationŠ s
þÿz TestUrlopen.test_line_iteration)N)N)N)r r r r r… r” r‡ rÑ rÖ rÜ rÝ rÞ rã rä rì ró rô rø rù rÿ r r r r
r r r r‹ r
rÅ ¸ s*
rÅ c C s t ¡ ad S r" )r Zthreading_setupÚ threads_keyr r r r
Ú setUpModule™ s r c C s t rtjt Ž d S r" )r r Zthreading_cleanupr r r r
ÚtearDownModuleŸ s r
Ú__main__),ru r r Ú urllib.parsery Zurllib.requestZ http.serverr r Zunittestr4 Útestr Z test.supportr ré Ú ImportErrorr_ ÚdirnameÚ__file__Úherer rë rñ r r r r r r& re ra rw ZTestCaser Zrequires_hashdigestrž rÄ rÅ r r r
r Úmainr r r r
Ú sB
y,!(M+ `