o
    IXîhüe  ã                   @   s€  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Zd dl	Z	d dl
Z
d dlmZ d dlmZ d dlmZ zd dlZW n eyG   dZY nw ej e¡Zej ed¡Zej ed¡ZG dd„ dejjƒZG d	d
„ d
ejƒZG dd„ dƒZG dd„ dejjƒZG dd„ dejjƒZ G dd„ de	j!ƒZ"ej#dddG dd„ de	j!ƒƒZ$dd„ Z%G dd„ de	j!ƒZ&dd„ Z'e(dkr¾e	 )¡  dS dS )é    N)Úhashlib_helper)Úthreading_helper)Úwarnings_helperzkeycert.pemzkeycert2.pemc                   @   s    e Zd ZdZdd„ Zdd„ ZdS )ÚLoopbackHttpServerz^HTTP server w/ a few modifications that make it useful for
    loopback testing purposes.
    c                 C   s"   t jj | ||¡ | j d¡ d S )Ngš™™™™™¹?)ÚhttpÚserverÚ
HTTPServerÚ__init__ÚsocketÚ
settimeout)ÚselfZserver_addressZRequestHandlerClass© r   úD/opt/python-3.10.19/usr/lib/python3.10/test/test_urllib2_localnet.pyr	   "   s
   
þzLoopbackHttpServer.__init__c                 C   s    | j  ¡ \}}| d¡ ||fS )zHTTPServer method, overridden.g      $@)r
   Úacceptr   )r   ÚrequestZclient_addressr   r   r   Úget_request+   s   
zLoopbackHttpServer.get_requestN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r	   r   r   r   r   r   r      s    	r   c                   @   ó(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	ÚLoopbackHttpServerThreadz2Stoppable thread that runs a loopback http server.c                 C   s<   t j | ¡ d| _t  ¡ | _d|_td|ƒ| _| jj	| _
d S )NFzHTTP/1.0)z	127.0.0.1r   )Ú	threadingÚThreadr	   Ú_stop_serverÚEventÚreadyZprotocol_versionr   ÚhttpdZserver_portÚport©r   Úrequest_handlerr   r   r   r	   :   s   
ÿz!LoopbackHttpServerThread.__init__c                 C   s   d| _ |  ¡  | j ¡  dS )z.Stops the webserver if it's currently running.TN)r   Újoinr   Zserver_close©r   r   r   r   ÚstopC   s   zLoopbackHttpServerThread.stopc                 C   s(   | j  ¡  | js| j ¡  | jrd S d S ©N)r   Úsetr   r   Úhandle_requestr"   r   r   r   ÚrunK   s   

ÿzLoopbackHttpServerThread.runN)r   r   r   r   r	   r#   r'   r   r   r   r   r   7   s
    	r   c                   @   sX   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d„ Z
dd„ Zdd„ ZdS )Ú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	   U   s
   
zDigestAuthHandler.__init__c                 C   ó
   || _ d S r$   )r.   )r   Zqopr   r   r   Úset_qop\   ó   
zDigestAuthHandler.set_qopc                 C   r/   r$   )r,   )r   Zusersr   r   r   Ú	set_users_   s   
zDigestAuthHandler.set_usersc                 C   r/   r$   )r-   )r   Zrealmr   r   r   Ú	set_realmc   r1   zDigestAuthHandler.set_realmc                 C   s8   |  j d7  _ t t| j ƒ d¡¡ ¡ }| j |¡ |S )Né   Úascii)r*   ÚhashlibÚmd5ÚstrÚencodeÚ	hexdigestr+   Úappend)r   Únoncer   r   r   Ú_generate_noncef   s   z!DigestAuthHandler._generate_noncec                 C   s‚   |  d¡}||d d … }| d¡}i }|D ](}| d¡\}}| ¡ }|d dkr6|d dkr6|dd… }n| ¡ }|||< q|S )Nú r4   ú,ú=r   ú"éÿÿÿÿ)ÚfindÚsplitÚstrip)r   Zauth_strZfirst_space_indexÚpartsÚ	auth_dictÚpartÚnameÚvaluer   r   r   Ú_create_auth_dictl   s   


z#DigestAuthHandler._create_auth_dictc                 C   s–   i }|  |¡ ||d< ||d< ||d< d| }t | d¡¡ ¡ }d| }t | d¡¡ ¡ }	||d< |	|d< d	| }
t |
 d¡¡ ¡ }||d
 kS )NÚpasswordÚmethodÚuriz#%(username)s:%(realm)s:%(password)sr5   z%(method)s:%(uri)sÚHA1ÚHA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)sÚresponse)Úupdater6   r7   r9   r:   )r   rG   rL   rM   rN   Z
final_dictZHA1_strrO   ZHA2_strrP   Zresponse_strrQ   r   r   r   Ú_validate_auth}   s    
ÿz DigestAuthHandler._validate_authc                 C   sN   |  dd¡ | dd¡ | dd| j| j|  ¡ f ¡ | ¡  |j d¡ dS )	Ni—  zProxy Authentication RequiredúContent-Typeú	text/htmlzProxy-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ƒdkr	dS d|jvr|  |¡S |  |jd ¡}|d | jv r*| j|d  }n|  |¡S | d¡| jvr<|  |¡S | j |d ¡ d}|j|j	fD ]}|  
|||j|¡rYd}qL|sa|  |¡S dS )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   TzProxy-AuthorizationÚusernamer<   F)Úlenr,   Úheadersr[   rK   Úgetr+   ÚremoveÚpathÚ
short_pathrS   Zcommand)r   r    rG   rL   Zauth_validatedra   r   r   r   r&   ž   s2   	

ÿ

ý€
z DigestAuthHandler.handle_requestN)r   r   r   r   r	   r0   r2   r3   r=   rK   rS   r[   r&   r   r   r   r   r(   R   s    r(   c                   @   sf   e Zd ZdZdZdZdZdeef Ze 	e 
d¡¡ d¡Zdd„ Zd	d
„ Zdd„ Zdd„ Zdd„ ZdS )ÚBasicAuthHandlerz,Handler for performing basic authentication.ÚtestUserÚtestPassÚTestz%s:%sr5   c                 O   s    t jjj| g|¢R i |¤Ž d S r$   )r   r   ÚBaseHTTPRequestHandlerr	   )r   ÚargsÚkwargsr   r   r   r	   Ô   s    zBasicAuthHandler.__init__c                 G   ó   d S r$   r   ©r   Úformatrh   r   r   r   Úlog_message×   s   zBasicAuthHandler.log_messagec                 C   s"   |   d¡ |  dd¡ |  ¡  d S )NéÈ   úContent-typerU   )rV   rW   rX   r"   r   r   r   Údo_HEADÛ   s   
zBasicAuthHandler.do_HEADc                 C   s4   |   d¡ |  dd| j ¡ |  dd¡ |  ¡  d S )Ni‘  zWWW-AuthenticatezBasic realm="%s"ro   rU   )rV   rW   ÚREALMrX   r"   r   r   r   Údo_AUTHHEADà   s   
zBasicAuthHandler.do_AUTHHEADc                 C   sl   | j  dd¡s|  ¡  | j d¡ d S | j  dd¡d| j kr0|  d¡ |  ¡  | j d¡ d S |  ¡  d S )NZAuthorizationÚ s   No Auth header receivedzBasic rn   s   It works)r^   r_   rr   rY   rZ   ÚENCODED_AUTHrV   rX   r"   r   r   r   Údo_GETæ   s   ÿÿ
zBasicAuthHandler.do_GETN)r   r   r   r   ÚUSERÚPASSWDrq   ZUSER_PASSWDÚbase64Z	b64encoder9   Údecodert   r	   rm   rp   rr   ru   r   r   r   r   rc   Ë   s    rc   c                   @   r   )	Ú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&   || _ tjjj| g|¢R i |¤Ž d S r$   )Údigest_auth_handlerr   r   rg   r	   )r   r{   rh   ri   r   r   r   r	   þ   s    zFakeProxyHandler.__init__c                 G   rj   r$   r   rk   r   r   r   rm     s   zFakeProxyHandler.log_messagec                 C   sz   t j | jd¡\}}}}}}|| _| j | ¡r;|  dd¡ |  dd¡ |  	¡  | j
 td| j dƒ¡ | j
 d¡ d S d S )	Nr   rn   ZOKrT   rU   zYou've reached %s!<BR>r5   sF   Our apologies, but our server is down due to a sudden zombie invasion.)ÚurllibÚparseÚurlparsera   rb   r{   r&   rV   rW   rX   rY   rZ   Úbytes)r   ZscmÚnetlocra   ÚparamsÚqueryÚfragmentr   r   r   ru   	  s   ÿÿúzFakeProxyHandler.do_GETN)r   r   r   r   r	   rm   ru   r   r   r   r   rz   ÷   s
    rz   c                       sP   e Zd ZdZdZdZdZ‡ fdd„Zdd„ Z‡ fd	d
„Z	dd„ Z
dd„ Z‡  ZS )ÚBasicAuthTestsrd   re   Z	Incorrectrf   c                    sT   t t| ƒ ¡  dd„ }t|ƒ| _|  | j¡ d| jj | _| j 	¡  | jj
 ¡  d S )Nc                  _   s   t | i |¤ŽS r$   )rc   ©rh   ri   r   r   r   Ú#http_server_with_basic_auth_handler!  s   zABasicAuthTests.setUp.<locals>.http_server_with_basic_auth_handlerzhttp://127.0.0.1:%s)Úsuperr„   ÚsetUpr   r   Ú
addCleanupÚstop_serverr   Ú
server_urlÚstartr   Úwait)r   r†   ©Ú	__class__r   r   rˆ     s   

zBasicAuthTests.setUpc                 C   ó   | j  ¡  d | _ d S r$   ©r   r#   r"   r   r   r   rŠ   )  ó   

zBasicAuthTests.stop_serverc                    s   t t| ƒ ¡  d S r$   )r‡   r„   ÚtearDownr"   rŽ   r   r   r“   -  s   zBasicAuthTests.tearDownc                 C   sz   t j ¡ }| | j| j| j| j¡ t j t j 	|¡¡ z|  
t j | j¡¡ W d S  t jjy<   |  d| j ¡ Y d S w )Nz!Basic auth failed for the url: %s)r|   r   ÚHTTPBasicAuthHandlerÚadd_passwordrq   r‹   rv   rw   Úinstall_openerÚbuild_openerÚ
assertTrueÚurlopenÚerrorÚ	HTTPErrorÚfail©r   Zahr   r   r   Útest_basic_auth_success0  s   
ÿz&BasicAuthTests.test_basic_auth_successc                 C   sR   t j ¡ }| | j| j| j| j¡ t j t j 	|¡¡ |  
t jjt jj| j¡ d S r$   )r|   r   r”   r•   rq   r‹   rv   ÚINCORRECT_PASSWDr–   r—   ÚassertRaisesrš   r›   r™   r   r   r   r   Útest_basic_auth_httperror9  s   
z(BasicAuthTests.test_basic_auth_httperror)r   r   r   rv   rw   rŸ   rq   rˆ   rŠ   r“   rž   r¡   Ú__classcell__r   r   rŽ   r   r„     s    	r„   r7   T)Zopensslc                       sT   e Zd ZdZdZdZdZ‡ fdd„Zdd„ Zd	d
„ Z	dd„ Z
dd„ Zdd„ Z‡  ZS )ÚProxyAuthTestszhttp://localhostZtesterZtest123Z	TestRealmc                    sÜ   t tˆ ƒ ¡  dd„ }ˆ  |tj ¡ ¡ dtjd< dtjd< tƒ ˆ _ˆ j 	ˆ j
ˆ ji¡ ˆ j ˆ j¡ ‡ fdd„}t|ƒˆ _ˆ  ˆ j¡ ˆ j ¡  ˆ jj ¡  dˆ jj }tj d	|i¡}tj ¡ ˆ _tj |ˆ j¡ˆ _d S )
Nc                 S   ó   t j ¡  t j | ¡ d S r$   ©ÚosÚenvironÚclearrR   ©Zold_environr   r   r   Úrestore_environK  ó   
z-ProxyAuthTests.setUp.<locals>.restore_environrs   ÚNO_PROXYÚno_proxyc                     s   t ˆ jg| ¢R i |¤ŽS r$   )rz   r{   r…   r"   r   r   Úcreate_fake_proxy_handlerV  s   z7ProxyAuthTests.setUp.<locals>.create_fake_proxy_handlerzhttp://127.0.0.1:%dr   )r‡   r£   rˆ   r‰   r¦   r§   Úcopyr(   r{   r2   rv   rw   r3   rq   r   r   rŠ   rŒ   r   r   r   r|   r   ZProxyHandlerZProxyDigestAuthHandlerÚproxy_digest_handlerr—   Úopener)r   rª   r®   Z	proxy_urlÚhandlerrŽ   r"   r   rˆ   H  s&   




ÿzProxyAuthTests.setUpc                 C   r   r$   r‘   r"   r   r   r   rŠ   c  r’   zProxyAuthTests.stop_serverc                 C   sF   | j  | j| j| j| jd ¡ | j d¡ |  t	j
j| jj| j¡ d S )NZbadr)   )r°   r•   rq   ÚURLrv   rw   r{   r0   r    r|   rš   r›   r±   Úopenr"   r   r   r   Ú-test_proxy_with_bad_password_raises_httperrorg  s   ÿ
þz<ProxyAuthTests.test_proxy_with_bad_password_raises_httperrorc                 C   s(   | j  d¡ |  tjj| jj| j¡ d S ©Nr)   )	r{   r0   r    r|   rš   r›   r±   r´   r³   r"   r   r   r   Ú,test_proxy_with_no_password_raises_httperroro  s
   
þz;ProxyAuthTests.test_proxy_with_no_password_raises_httperrorc                 C   s|   | j  | j| j| j| j¡ | j d¡ | j 	| j¡}| 
¡ r,	 | 
¡ sW d   ƒ d S W d   ƒ d S 1 s7w   Y  d S r¶   )r°   r•   rq   r³   rv   rw   r{   r0   r±   r´   Úread©r   Úresultr   r   r   Útest_proxy_qop_auth_worksu  s   ÿ
ÿÿ"ÿz(ProxyAuthTests.test_proxy_qop_auth_worksc                 C   sž   | j  | j| j| j| j¡ | j d¡ z	| j 	| j¡}W n t
jjy(   Y d S w | | ¡ r=	 | ¡ s0W d   ƒ d S W d   ƒ d S 1 sHw   Y  d S )Nzauth-int)r°   r•   rq   r³   rv   rw   r{   r0   r±   r´   r|   rš   ÚURLErrorr¸   r¹   r   r   r   Ú0test_proxy_qop_auth_int_works_or_throws_urlerror}  s    ÿü
ÿÿ"ÿz?ProxyAuthTests.test_proxy_qop_auth_int_works_or_throws_urlerror)r   r   r   r³   rv   rw   rq   rˆ   rŠ   rµ   r·   r»   r½   r¢   r   r   rŽ   r   r£   @  s    r£   c                    s   G ‡ ‡fdd„dt jjƒ‰ ˆ S )Nc                       sB   e Zd ZdZg Zg ZdZdd„ Zdd„ Z‡ ‡fdd„Z	d	d
„ Z
dS )z1GetRequestHandler.<locals>.FakeHTTPRequestHandlerz	TestHTTP/éP   c                 S   s0   |   ¡ }|r| j |¡}||d … }|sd S d S r$   )Ú	send_headrY   rZ   )r   ÚbodyÚdoner   r   r   ru   —  s
   þz8GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_GETc                 S   s2   | j d }| j t|ƒ¡}|  ¡  | j |¡ d S )NzContent-Length)r^   Zrfiler¸   Úintru   Úrequestsr;   )r   Zcontent_lengthZ	post_datar   r   r   Údo_POST  s   
z9GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_POSTc                    s|   | j ˆ _| j | j¡ ˆ d¡\}}}|  |¡ |D ]\}}|  ||d| ji ¡ q|r8|  dd¡ |  	¡  |S |  	¡  d S )Nr   r   ro   z
text/plain)
r^   Úheaders_receivedrÃ   r;   ra   ÚpoprV   rW   r   rX   )r   Zresponse_coder^   rÀ   ÚheaderrJ   ©ÚFakeHTTPRequestHandlerÚ	responsesr   r   r¿   £  s   
z;GetRequestHandler.<locals>.FakeHTTPRequestHandler.send_headc                 W   rj   r$   r   )r   rh   r   r   r   rm   ²  s   z=GetRequestHandler.<locals>.FakeHTTPRequestHandler.log_messageN)r   r   r   Zserver_versionrÃ   rÅ   r   ru   rÄ   r¿   rm   r   rÈ   r   r   rÉ     s    rÉ   )r   r   rg   )rÊ   r   rÈ   r   ÚGetRequestHandlerŽ  s   &rË   c                       sÎ   e Zd ZdZ‡ fdd„Zd/dd„Zdd„ Zd/d	d
„Zd/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‡  ZS )0Ú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| ƒ ¡  |  tjj¡ dd„ }|  |tj 	¡ ¡ dtjd< dtjd< d S )Nc                 S   r¤   r$   r¥   r©   r   r   r   rª   É  r«   z*TestUrlopen.setUp.<locals>.restore_environÚ*r¬   r­   )
r‡   rÌ   rˆ   r‰   r|   r   Ú
urlcleanupr¦   r§   r¯   )r   rª   rŽ   r   r   rˆ   Â  s   
zTestUrlopen.setUpNc                 K   sx   g }t jj||fi |¤Ž}z$| | d¡¡ | | ¡ ¡ | | d¡¡ | | ¡ ¡ W | ¡  n| ¡  w d 	|¡S )Nrn   i   ó    )
r|   r   r™   ÚextendÚ	readlinesr;   Úreadliner¸   Úcloser!   )r   ÚurlÚdatari   ÚlÚfr   r   r   r™   Ð  s   
zTestUrlopen.urlopenc                 C   r   r$   r‘   r"   r   r   r   rŠ   Ý  r’   zTestUrlopen.stop_serverc                 C   sZ   |d u r
dg dfg}t |ƒ}t|ƒ| _|  | j¡ | j ¡  | jj ¡  | jj}||_|S )Nrn   s   we don't care)	rË   r   r   r‰   rŠ   rŒ   r   r   r   )r   rÊ   r²   r   r   r   r   Ústart_serverá  s   

zTestUrlopen.start_serverc                 K   s^   t tjdƒs|  d¡ ddlm} |d u rdg dfg}t|ƒ}|| fd|i|¤Ž}|j|_|S )NZHTTPSHandlerzssl support requiredr   )Úmake_https_serverrn   ó   we care a bitZhandler_class)Úhasattrr|   r   ÚskipTestZtest.ssl_serversrÙ   rË   r   )r   rÊ   ri   rÙ   r²   r   r   r   r   Ústart_https_serverî  s   
zTestUrlopen.start_https_serverc                 C   sV   d}ddgdfdg |fg}|   |¡}|  d|j ¡}|  ||¡ |  |jddg¡ d S )	Ns   We got here...i.  )ZLocationz(http://localhost:%(port)s/somewhere_elsers   rn   ú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}ddg|fg}|   |¡}|  d|j ¡}|  ||¡ d S )Ns   hello worlds   a
hello worl
1
d
0
rn   )zTransfer-EncodingZchunkedrÞ   )rØ   r™   r   rá   )r   râ   Zchunked_startrQ   r²   rÕ   r   r   r   Útest_chunked  s   ÿ
zTestUrlopen.test_chunkedc              
   C   sŽ   d}|   dg |fg¡}z
|  d|j ¡ W n tjjy1 } z| ¡ }| ¡  W Y d }~n
d }~ww |  d¡ |  	||¡ |  	|j
dg¡ d S )Ns   Bad bad bad...i”  zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble)rØ   r™   r   r|   rš   r¼   r¸   rÓ   rœ   rá   rÃ   )r   râ   r²   r×   rÕ   r   r   r   Útest_404  s   €þ
zTestUrlopen.test_404c                 C   sF   d}|   dg |fg¡}|  d|j ¡}|  ||¡ |  |jdg¡ d S )Nó   pycon 2008...rn   úhttp://localhost:%s/bizarreú/bizarrerà   ©r   râ   r²   rÕ   r   r   r   Útest_200#  s
   zTestUrlopen.test_200c                 C   sJ   d}|   dg |fg¡}|  d|j d¡}|  ||¡ |  |jddg¡ d S )Nræ   rn   rç   s   get=with_feelingrè   rà   ré   r   r   r   Útest_200_with_parameters*  s   ÿz$TestUrlopen.test_200_with_parametersc                 C   s8   |   ¡ }tjtd}| jd|j |d}|  |d¡ d S )N©Zcafileúhttps://localhost:%s/bizarre©ÚcontextrÚ   )rÝ   ÚsslÚcreate_default_contextÚCERT_localhostr™   r   rá   )r   r²   rï   rÕ   r   r   r   Ú
test_https2  s   zTestUrlopen.test_httpsc              	   C   s   | j td}t dtf¡i | jd|j td}|  |d¡ |  t	j
j¡}| jd|j td W d   ƒ n1 s:w   Y  | j td}|  t	j
j¡}| jd|j td W d   ƒ n1 saw   Y  W d   ƒ d S W d   ƒ d S 1 syw   Y  d S )N©Úcertfilers   rí   rì   rÚ   )rÝ   rò   r   Úcheck_warningsÚDeprecationWarningr™   r   rá   r    r|   rš   r¼   ÚCERT_fakehostname)r   r²   rÕ   Úcmr   r   r   Útest_https_with_cafile8  s(   ÿÿÿÿÿõ"õz"TestUrlopen.test_https_with_cafilec              	   C   s’   | j td}t dtf¡2 |  tjj¡}| j	d|j
 dd W d   ƒ n1 s*w   Y  W d   ƒ d S W d   ƒ d S 1 sBw   Y  d S )Nrô   rs   rí   T)Z	cadefault)rÝ   rò   r   rö   r÷   r    r|   rš   r¼   r™   r   )r   r²   rù   r   r   r   Útest_https_with_cadefaultI  s   ÿÿÿ"ÿz%TestUrlopen.test_https_with_cadefaultc                    s†   t d u r	|  d¡ t js|  d¡ d ‰ ‡ fdd„}t  t j¡}| |¡ | j|td}t jtd}| j	d|j
 |d |  ˆ d	¡ d S )
Nzssl module requiredzSNI support required in OpenSSLc                    s   |‰ d S r$   r   )Zssl_sockZserver_nameZinitial_context©Zsni_namer   r   Úcb_sniW  s   z*TestUrlopen.test_https_sni.<locals>.cb_sni)rï   rõ   rì   zhttps://localhost:%srî   Ú	localhost)rð   rÜ   ZHAS_SNIZ
SSLContextZPROTOCOL_TLS_SERVERZset_servername_callbackrÝ   rò   rñ   r™   r   rá   )r   rý   rï   r²   r   rü   r   Útest_https_sniQ  s   


zTestUrlopen.test_https_snic                 C   sd   |   ¡ }tjjd|j ddid}tj |¡ W d   ƒ n1 s"w   Y  |  |jd d¡ d S )NrÞ   ZRangezbytes=20-39©r^   )rØ   r|   r   ÚRequestr   r™   rá   rÅ   ©r   r²   Zreqr   r   r   Útest_sending_headersa  s   ÿÿz TestUrlopen.test_sending_headersc                 C   sv   |   ¡ }tjjd|j ddid}tj |¡ W d   ƒ n1 s"w   Y  |  d|j ¡ ¡ |  	d|j ¡ ¡ d S )NrÞ   zX-SoMe-hEaderZfoobarr   zX-Some-Header)
rØ   r|   r   r  r   r™   ZassertInrÅ   ÚkeysZassertNotInr  r   r   r   Útest_sending_headers_cameli  s   ÿÿz&TestUrlopen.test_sending_headers_camelc                 C   sp   |   ¡ }tj d|j ¡!}dD ]}|  t||ƒd| ¡ q|  | ¡ d¡ W d   ƒ d S 1 s1w   Y  d S )Núhttp://localhost:%s)r¸   rÓ   ÚinfoÚgeturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)rØ   r|   r   r™   r   r˜   rÛ   r¸   )r   r²   Úopen_urlÚattrr   r   r   Ú
test_basicr  s   ÿ"üzTestUrlopen.test_basicc                 C   sl   |   ¡ }tj d|j ¡}| | ¡ }W d   ƒ n1 sw   Y  |  |tjj	d¡ |  
| ¡ d¡ d S )Nr  zEobject returned by 'info' is not an instance of email.message.MessageZplain)rØ   r|   r   r™   r   r  ZassertIsInstanceÚemailÚmessageZMessagerá   Zget_content_subtype)r   r²   r	  Zinfo_objr   r   r   Ú	test_infoz  s   ÿ
ÿÿzTestUrlopen.test_infoc                 C   s\   |   ¡ }tj d|j ¡}| | ¡ }W d   ƒ n1 sw   Y  |  |d|j ¡ d S )Nr  )rØ   r|   r   r™   r   r  rá   )r   r²   r	  rÔ   r   r   r   Útest_geturl…  s   
ÿzTestUrlopen.test_geturlc                 C   sB   d}|   dg |fg¡}tj d|j ¡}|D ]}|  ||¡ qd S )Nræ   rn   r  )rØ   r|   r   r™   r   rá   )r   râ   r²   rÕ   Úliner   r   r   Útest_iteration  s   ÿzTestUrlopen.test_iterationc              
   C   s”   ddddd d g}d  |¡}|  dg |fg¡}tj d	|j ¡}t|ƒD ]\}}|  ||| d
|t|| ƒt|ƒf ¡ q%|  |d t|ƒ¡ d S )Ns   We
s   got
s   here
s	   verylong i    ó   
rÏ   rn   r  zQFetched line number %s doesn't match expected:
    Expected length was %s, got %sr4   )	r!   rØ   r|   r   r™   r   Ú	enumeraterá   r]   )r   Úlinesrâ   r²   rÕ   Úindexr  r   r   r   Útest_line_iteration”  s   
þÿzTestUrlopen.test_line_iterationc                 C   s¬   |   dg dfdg dfg¡}tj ¡ }tj d|j ¡}|  d |j¡ | |d 	d¡¡ |  d|j¡ |  d| 
d¡¡ | |d 	d¡¡ |  d	|j¡ |  d
| 
d¡¡ d S )Nrn   s   anyr  Ú1zus-asciió   1zContent-lengthZ
1234567890s
   1234567890Z10)rØ   r|   r   r—   r  r   rá   rÕ   r´   r9   Z
get_header)r   r²   r±   r   r   r   r   Útest_issue16464   s   þ
zTestUrlopen.test_issue16464r$   )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Ž   r   rÌ   ¹  s0    


	rÌ   c                  C   s    t  ¡ } tjt jg| ¢R Ž  d S r$   )r   Zthreading_setupÚunittestZaddModuleCleanupZthreading_cleanup)Úthread_infor   r   r   ÚsetUpModule³  s   r  Ú__main__)*rx   r¦   r  Úurllib.parser|   Zurllib.requestZhttp.serverr   r   r  r6   Ztest.supportr   r   r   rð   ÚImportErrorra   ÚdirnameÚ__file__Úherer!   rò   rø   r   r   r   r   r   r(   rg   rc   rz   ZTestCaser„   Zrequires_hashdigestr£   rË   rÌ   r  r   Úmainr   r   r   r   Ú<module>   sF    ÿy,!(M+ {ÿ