File: //usr/local/lib/python3.7/test/__pycache__/test_urllib2.cpython-37.pyc
B
��g�/ � @ s d dl Z d dlmZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZmZmZ d dlmZ d dlZ d dlZG dd� de j�ZG dd � d e j�ZG d
d � d �Z G d d
� d
�Z G dd� de �Z G dd� dej �Z!G dd� d�Z"G dd� d�Z#G dd� dej$�Z%G dd� d�Z&G dd� d�Z'd d � Z(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*j+�Z/G d&d'� d'�Z0G d(d)� d)e j�Z1d*d+� Z2G d,d-� d-e j�Z3G d.d/� d/e j�Z4G d0d1� d1e j�Z5e6d2k�re �7� dS )3� N)�support)� test_urllib)�Request�OpenerDirector�HTTPBasicAuthHandler� HTTPPasswordMgrWithPriorAuth� _parse_proxy� _proxy_bypass_macosx_sysconf�AbstractDigestAuthHandler)�urlparsec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TrivialTestsc
C s� x�dD ]�}i }t d| |� |d= |dkrftjdkrf|�d�|�d� }}| �|jd� | �|jd� x4|�� D ](\}}| �|jd | d
|||jf � qpW qW d S ) N)�request�response�parse�errorZ robotparserzfrom urllib.%s import *� __builtins__r
�ntZ url2pathname� pathname2urlZ
nturl2pathz urllib.%sz.%r is exposed in 'urllib.%s' but defined in %r)�exec�os�name�pop� assertEqual�
__module__�items)�self�module�context�u�p�k�v� r" �-/usr/local/lib/python3.7/test/test_urllib2.py� test___all__ s
zTrivialTests.test___all__c C sh | � ttjjd� tj�tjj�� tj
d�}tj dkr@d| }nd| }tj�|�}|� � |�
� 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 sd ddddgfdddgfdddd d
d d d
gfddddgfg}x$|D ] \}}| � tj�|�|� q@W d 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_listA s
z!TrivialTests.test_parse_http_listc C s" t j�d�}| �|jt|�� d S )N�reason)r( r �URLError�assertInr< �str)r �errr" r" r# �test_URLError_reasonstrL s z$TrivialTests.test_URLError_reasonstrN)�__name__r � __qualname__r$ r2 r; rA r" r" r" r# r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RequestHdrsTestsc C sD d}| � t|ddid�jd d� | � t|ddid�jd d� dS )a(
The Request.headers dictionary is not a documented interface. It
should stay that way, because the complete set of headers are only
accessible through the .get_header(), .has_header(), .header_items()
interface. However, .headers pre-dates those methods, and so real code
will be using the dictionary.
The introduction in 2.4 of those methods was a mistake for the same
reason: code that previously saw all (urllib2 user)-provided headers in
.headers now sees only a subset.
zhttp://example.comz Spam-eggs�blah)�headersz spam-EggSN)r r rF )r �urlr" r" r# �test_request_headers_dictS s
z*RequestHdrsTests.test_request_headers_dictc C s� d}t |ddid�}| �|�d�� | �|�� dg� |�dd� | �t|�� �ddg� | �|�d �� | �|� d �� | �|� d d
�d
� |�
d� | �|�d�� |� d d � | �|�d �� |�
d � | �|�d �� d
S )a;
Note the case normalization of header names here, to
.capitalize()-case. This should be preserved for
backwards-compatibility. (In the HTTP case, normalization to
.title()-case is done by urllib2 before sending headers to
http.client).
Note that e.g. r.has_header("spam-EggS") is currently False, and
r.get_header("spam-EggS") returns None, but that could be changed in
future.
Method r.remove_header should remove items both from r.headers and
r.unredirected_hdrs dictionaries
zhttp://example.comz Spam-eggsrE )rF )z Spam-eggsrE zFoo-Bar�baz)zFoo-barrI z Not-there�defaultzUnredirected-spamZEggsN) r �
assertTrue�
has_headerr Z header_items�
add_header�sorted� assertFalse� assertIsNone�
get_headerZ
remove_header�add_unredirected_header)r rG �reqr" r" r# � test_request_headers_methodsh s
z-RequestHdrsTests.test_request_headers_methodsc C s� t j�� }|j}|j}|dddd� |dddd� |dddd � |dd
d d � |dd
dd� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� |d ddd� |d ddd� | �|d d�d� | �|d d�d� |d dd d � |d dd d � | �|d d�d!� |d"dd#d"� | �|d"d�d$� | �|d"d%�d&� | �|dd'�d&� | �|dd(�d)� | �|dd�d)� | �|dd
�d*� | �|dd
�d+� d S ),Nz
Some Realmzhttp://example.com/�joe�passwordzhttp://example.com/ni�nizhttp://c.example.com:3128�3r5 z
d.example.com�4�dze.example.com:3128�5�ez example.com)rU rV zhttp://example.comzhttp://example.com/spamz http://example.com/spam/spamzhttp://example.com/foo�foozhttp://example.com/bar�bar�nini)r] rW )r^ r_ r4 �firstrE �second�spam)ra rb r3 �1)rc r3 zhttp://a.example.com/)NNz
c.example.comzc.example.com:3128)rX r5 )rY rZ )r[ r\ )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 sr t j�� }|j}|j}|dddd� |dddd� |d d
d d � |d
ddd� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|dd�d� | �|d d �d� | �|d d
�d � | �|d d �d � | �|d
d�d � | �|d
d �d� | �|d
d!�d � | �|d
d"�d� d#S )$z�
The point to note here is that we can't guess the default port if
there's no scheme. This applies to both add_password and
find_user_password.
r1 zhttp://g.example.com:80�10�jr6 zhttp://h.example.com�11r r7 zi.example.com:80�12�l�iz
j.example.com�13�mzg.example.com:100)NNzg.example.com:80)rk rl z
g.example.comzhttp://g.example.com:100zhttp://g.example.comz
h.example.com)rm r zh.example.com:80zhttp://h.example.com:80z
i.example.com)rn ro zhttp://i.example.com:80)rq rr zj.example.com:80zhttp://j.example.comzhttp://j.example.com:80N)r( r
rd re rf r )r rg rh ri r" r" r# �"test_password_manager_default_port� sD
z3RequestHdrsTests.test_password_manager_default_portN)rB r rC rH rT rj rs r" r" r" r# rD Q s $FrD c @ s( e Zd Zg Zdejfdd�Zdd� ZdS )�
MockOpenerNc C s ||| | _ | _| _d S )N)rS �data�timeout)r rS ru rv r" r" r# �open� s zMockOpener.openc G s || | _ | _d S )N)�proto�args)r rx ry r" r" r# r s zMockOpener.error)rB r rC �
addheaders�socket�_GLOBAL_DEFAULT_TIMEOUTrw r r" r" r" r# rt � s rt c @ s( e Zd Zddd�Zd dd�Zdd� ZdS )
�MockFileNc C s d S )Nr" )r �countr" r" r# r/ 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# r0
s zMockFile.close)N)N)rB r rC r/ r r0 r" r" r" r# r} s
r} c @ s e Zd Zdd� ZdS )� MockHeadersc C s t | �� �S )N)r: �values)r r r" r" r# �
getheaders s zMockHeaders.getheadersN)rB r rC r� r" r" r" r# r� s r� c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) � MockResponseNc C s. t j�| |� ||||f\| _| _| _| _d S )N)�io�StringIO�__init__�code�msgrF rG )r r� r� rF ru rG r" r" r# r� s zMockResponse.__init__c C s | j S )N)rF )r r" r" r# �info s zMockResponse.infoc C s | j S )N)rG )r r" r" r# �geturl s zMockResponse.geturl)N)rB r rC r� r� r� r" r" r" r# r� s
r� c @ s e Zd Zdd� Zdd� ZdS )�
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)rB r rC r� r� r" r" r" r# r� " s r� c @ s e Zd Zdd� Zdd� ZdS )�
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| jf|�� S )N)r� r� r� )r ry r" r" r# �__call__0 s zFakeMethod.__call__N)rB r rC r� r� r" r" r" r# r� * s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�MockHTTPResponsec C s" || _ || _|| _|| _d| _d S )N�� )�fpr� �statusr< r� )r r� r� r� r< r" r" r# r� 5 s
zMockHTTPResponse.__init__c C s dS )N� r" )r r" r" r# r/ < s zMockHTTPResponse.readc C s i S )Nr" )r r" r" r# r� ? s zMockHTTPResponse.infoc C s | j S )N)rG )r r" r" r# r� B s zMockHTTPResponse.geturlN)rB r rC r� r/ r� r� r" r" r" r# r� 4 s r� c @ sT e Zd Zdd� Zejfdd�Zdd� Zddd �Zdd
d �d d
�Z dd� Z
dd� Z dS )�
MockHTTPClassc C s( d| _ g | _d | _d| _d | _i | _d S )Nr F)�level� req_headersru �raise_on_endheadersZsock�_tunnel_headers)r r" r" r# r� G s zMockHTTPClass.__init__c C s || _ || _| S )N)�hostrv )r r� rv r" r" r# r� O s zMockHTTPClass.__call__c C s
|| _ d S )N)r� )r r� r" r" r# �set_debuglevelT s z MockHTTPClass.set_debuglevelNc C s& || _ || _|r|| _n
| j�� d S )N)� _tunnel_hostZ _tunnel_portr� �clear)r r� �portrF r" r" r# �
set_tunnelW s
zMockHTTPClass.set_tunnelF)�encode_chunkedc C sP || _ || _|d k r&| j|�� 7 _| j�� |r:|| _|| _| jrLt� �d S )N) �method�selectorr� r �sortru r� r� �OSError)r r� rG �bodyrF r� r" r" r# r
_ s
zMockHTTPClass.requestc C s t t� i dd�S )Nr� �OK)r� r} )r r" r" r# � getresponsel s zMockHTTPClass.getresponsec C s d S )Nr" )r r" r" r# r0 o s zMockHTTPClass.close)NN)NN) rB r rC r� r{ r| r� r� r� r
r� r0 r" r" r" r# r� F s
r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� MockHandleri� c C s | � |� d S )N)�_define_methods)r �methodsr" r" r# r� x s zMockHandler.__init__c C sN xH|D ]@}t |�dkr |\}}n
|d }}t||| j�}t| j||� qW d S )N� )�lenr� r� �setattr� __class__)r r� �specr r� Zmethr" r" r# r� { s
zMockHandler._define_methodsc O s� | j j�| |||f� |d kr"d S |dkr.| S |dkrHtddi d�}|S |dkrXtd�S |�d�r�||�d �d
d � }y t|�}W n tk
r� Y nX tddi d�}| j � d |d ||di �S |d
kr�t
j � d��ds�t �d S )Nz return selfzreturn responser� r� r� zreturn requestz http://blah/r � � �httpr �raiserE F)
�parent�calls�appendr� r �
startswith�rfind�intr' r r( r= �AssertionError)r Zfn_namer� ry �kwdsZresr� r" r" r# r� � s*
zMockHandler.handlec C s d S )Nr" )r r" r" r# r0 � 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�sdS | j|jk S )N�
handler_orderT)�hasattrr� )r �otherr" r" r# �__lt__� s
zMockHandler.__lt__N)
rB r rC r� r� r� r� r0 r� r� r" r" r" r# r� s s r� c C sf g }d}xX|D ]P}G dd� dt �}||�}| j|7 _|�| � |d }|�|� | �|� qW |S )a� Create MockHandlers and add them to an OpenerDirector.
meth_spec: list of lists of tuples and strings defining methods to define
on handlers. eg:
[["http_error", "ftp_open"], ["http_open"]]
defines methods .http_error() and .ftp_open() on one handler, and
.http_open() on another. These methods just record their arguments and
return None. Using a tuple instead of a string causes the method to
perform some action (see MockHandler.handle()), eg:
[["http_error"], [("http_open", "return request")]]
defines .http_error() on one handler (which simply returns None), and
.http_open() on another handler, which returns a Request object.
r c @ s e Zd ZdS )z6add_ordered_mock_handlers.<locals>.MockHandlerSubclassN)rB r rC r" r" r" r# �MockHandlerSubclass� s r� r� )r� r� r� r� � add_handler)�opener� meth_spec�handlersr~ �methsr� r7 r" r" r# �add_ordered_mock_handlers� s
r� c G s" t � }x| D ]}|�|� q W |S )N)r r� )Zhandler_instancesr� r7 r" r" r# �build_test_opener� s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�MockHTTPHandlerc C s || _ || _| �� d S )N)r� rF �reset)r r� rF 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� dd l }dd l}| j�|�|�� | jdkrl| jd | _tjj| j }|�
| j �}| j �
d|t� | j ||�S || _|�
d�}tdd|d|�� �S d S )Nr r� r� z
r� r� r� )�email�copyr� r� Zdeepcopyr� r� �client� responsesr� Zmessage_from_stringrF r� r r} rS r� � get_full_url)r rS r� r� r r� r" r" r# � http_open� s
zMockHTTPHandler.http_openN)rB r rC r� r� r� r" r" r" r# r� � s r� c @ s e Zd Zddd�Zdd� ZdS )�MockHTTPSHandlerr c C s t jjj| |d� t� | _d S )N)�
debuglevel)r( r
�AbstractHTTPHandlerr� r� �httpconn)r r� r" r" r# r� � s zMockHTTPSHandler.__init__c C s | � | j|�S )N)�do_openr� )r rS r" r" r# �
https_open� s zMockHTTPSHandler.https_openN)r )rB r rC r� r� r" r" r" r# r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�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 s6 |� d�rd| _tjj| j }t| j|t� d|�� �S )N�
AuthorizationTr� ) rL r� r� r� r� r� r� r} r� )r rS r r" r" r# r� s
z"MockHTTPHandlerCheckAuth.http_openN)rB r rC r� r� r� r" r" r" r# r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MockPasswordManagerc C s || _ || _|| _|| _d S )N)�realmrG �userrV )r r� Zurir� rV r" r" r# re s z MockPasswordManager.add_passwordc C s || _ || _| j| jfS )N)� target_realm�
target_urlr� rV )r r� Zauthurir" r" r# rf s z&MockPasswordManager.find_user_passwordN)rB r rC re rf r" r" r" r# r�
s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�OpenerDirectorTestsc C s( G dd� dt �}| �tt� j|� � d S )Nc @ s e Zd ZdS )z<OpenerDirectorTests.test_add_non_handler.<locals>.NonHandlerN)rB r rC 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` ddl m} t� }ddgdgg}t||� |�tj�� � x dD ]}| �||j |d � q@W d S )Nr )r= )r� z return self)Z
proxy_openz return self)Zredirect_requestz return self)Zdo�proxy�redirectz://example.com/)
� urllib.errorr= r r� r� r( r
ZUnknownHandlerr& rw )r r= �or� �schemer" r" r# �test_badly_named_methods" s
z,OpenerDirectorTests.test_badly_named_methodsc
C s� t � }dddgdgdgdgg}t||�}td�}|�|�}| �||d � |d df|d dfg}xBt||j�D ]2\}}|\} }
} } | �| |
f|� | �| |f� qpW d S )Nr� �ftp_open�http_error_302)r� z return selfzhttp://example.com/r� r )r r� r rw r �zipr� )
r r� r� r� rS �rr� �expected�got�handlerr ry r� r" r" r# � test_handled7 s
z OpenerDirectorTests.test_handledc C s� t � }g }xRdgdfdgdfgD ]:\}}G dd� dt�}||�}||_|�|� |�|� q W |�d� | �|jd d |d � | �|jd d |d � d S ) N)r� z return selfi� r� r c @ s e Zd ZdS )zCOpenerDirectorTests.test_handler_order.<locals>.MockHandlerSubclassN)rB r rC r" r" r" r# r� V s r� zhttp://example.com/r� )r r� r� r� r� rw r r� )r r� r� r� r� r� r7 r" r" r# �test_handler_orderQ s
z&OpenerDirectorTests.test_handler_orderc C sZ t � }dgdgg}t||�}td�}| �tjj|j|� | �|j |d d|fi fg� d S )N)r� r� )r� z return selfzhttp://example.com/r r� )
r r� r r& r( r r= rw r r� )r r� r� r� rS r" r" r# �
test_raisec s
z OpenerDirectorTests.test_raisec C s� t � }dgddgdddgdgg}t||�}G dd � d �}td
�}|�|� t|j�d ks\t�|d d|ff|d d||� d
di ffg}xJt||j�D ]:\}}|\} }
} | �| |
f|d d � � | �| |d � q�W d S )N)r� z error 302)Zhttp_error_400r� r� )r� zreturn responseZhttp_error_303Z
http_errorr� c @ s e Zd Zdd� ZdS )z4OpenerDirectorTests.test_http_error.<locals>.Unknownc S s dS )NTr" )r r� r" r" r# �__eq__~ s z;OpenerDirectorTests.test_http_error.<locals>.Unknown.__eq__N)rB r rC r� r" r" r" r# �Unknown} s r� zhttp://example.com/r� r i. r� ) r r� r rw r� r� r� r� r ) r r� r� r� r� rS r� r� r� r� Z method_namery r" r" r# �test_http_errorp s$
z#OpenerDirectorTests.test_http_errorc C s t � }ddgddgg}t||�}td�}|�|� |d df|d df|d df|d dfg}x�t|j�D ]�\}\}}} }
|dk r�| �||f|| � | �t| �d� | �| d t� qj| �||f|| � | �t| �d� | �| d t� | d d k rj| �| d t � qjW d S ) N)� http_requestzreturn request)�
http_responsezreturn responsezhttp://example.com/r r� r� r r� )
r r� r rw � enumerater� r r� ZassertIsInstancer� ) r r� r� r� rS r� rp r� r ry r� r" r" r# �test_processors� s(
z#OpenerDirectorTests.test_processorsN)
rB r rC r� r� r� r� r� r� r r" r" r" r# r� s
r� c C s^ y| � d� W n tk
r, t�d��Y nX tj�| �}tjdkrZ|� d�rZ|dd � }|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 Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd 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*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z e � e!j"d9kd:�d;d<� �Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGdH� Z)dIdJ� Z*dKdL� Z+dMdN� Z,dOdP� Z-dQdR� Z.dSS )T� HandlerTestsc s� G dd� d�� G � fdd�dt jj�}dd l}d}||�}t� |_�xjdd|jd d d
d d gd
df dd|jdd d
d d gd
df dd|jdd d
d d gd
df dd|jdd d
d d gd
df dddd d dd d gd d f dd|jd d dg dd f gD ]�\ }}}}} }
} } }
t|�}d |_|� |�}| �
|j |� | �
|j | � | �
|j
t�|�� | �
|j|� | �
|j| � | �
|jj| � | �
|jj|
� |�� }| �
|�d�|
� | �
t|d �t|�� q�W d S ) Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z-HandlerTests.test_ftp.<locals>.MockFTPWrapperc S s
|| _ d S )N)ru )r ru r" r" r# r� � s z6HandlerTests.test_ftp.<locals>.MockFTPWrapper.__init__c S s$ || | _ | _t�| j�t| j�fS )N)�filename�filetyper� r� ru r� )r r r
r" r" r# �retrfile� s z6HandlerTests.test_ftp.<locals>.MockFTPWrapper.retrfilec S s d S )Nr" )r r" r" r# r0 � s z3HandlerTests.test_ftp.<locals>.MockFTPWrapper.closeN)rB r rC r� r r0 r" r" r" r# �MockFTPWrapper� s r c s& e Zd Zdd� Zejf� fdd� ZdS )z-HandlerTests.test_ftp.<locals>.NullFTPHandlerc S s
|| _ d S )N)ru )r ru r" r" r# r� � s z6HandlerTests.test_ftp.<locals>.NullFTPHandler.__init__c s4 || | _ | _|| | _| _|| _� | j�| _| jS )N)r� �passwdr� r� �dirsru �
ftpwrapper)r r� r
r� r� r rv )r r" r# � connect_ftp� s
z9HandlerTests.test_ftp.<locals>.NullFTPHandler.connect_ftpN)rB r rC 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�ftplibrt r� ZFTP_PORTr rv r� r r� r
r� r{ �
gethostbynamer� r r r r
r� �getr� r� )r r r ru r7 rG r� r� r� r
Ztype_r r ZmimetyperS r� rF r" )r r# �test_ftp� sL
"
zHandlerTests.test_ftpc
C s� dd l }tj�� }t� }|_tj}tt j
� |��}d}d| d| dt �
d�|f g}yt �
t �� �}W n t jk
r� d}Y nX |r�|�d||f � x�|D ]�} t|d�}
zpz|
�|� W d |
�� X |�t| ��} z | �� } | �� }
| �� }W d | �� X t �|�}|jj|j d d
�}W d t � |� X | � | |� | � |
d d � | � |
d
d� | � |
d |� | � || � q�W x�d| dddt �
d�t � � |f dt � � |f gD ]T} z@t|d�}
z|
�|� W d |
�� X | � tj!j"|jt| �� W d t � |� X �q�W tj�� }t� }|_x|dD ]t\} }t| �}y|�|� W n$ tj!j"k
�rd | �#|� Y n X | �$|j%|� | � |j&d� | � |j&dk|� �q"W d S )Nr s
hello, world
zfile://localhost%sz file://%sz file://%s%sr r� �wbT)Zusegmtz Content-typez
text/plainzContent-lengthrq 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))z file://ftp.example.com//foo.txtF)z file://ftp.example.com///foo.txtF)z file://ftp.example.com/foo.txtF)z"file://somehost//foo/something.txtF)z#file://localhost//foo/something.txtF�ftp)'Z email.utilsr( r
Z FileHandlerrt r� r �TESTFNr r r* r+ r{ r Z gethostnameZgaierrorr� rw �writer0 Z file_openr r/ r� r� �statZutilsZ
formatdate�st_mtime�remover �getcwdr& r r= rO �assertIsrS �type)r r� r7 r� r r Ztowrite�urlsZ localaddrrG r1 r� ru rF ZrespurlZstatsZmodifiedr rS r" r" r# � test_file� s|
zHandlerTests.test_filec C sr t j�� }t� }|_d}x�dD ]�\}}t||ddi�}d |_|�dd� t� }|� ||�}|j
|j |j |j
|j|jdkdf |� � } | j | j | �|�
� |� | �|jd � | �|jd
� | �|j|� | �|jd � | �|jd d
dg� | �|j|� q W d|_| �t jj |j ||� tdd�}| �t |j |� dg|_ �x$dD �]}td|�}t ddi d�}|� |�}
|d k�r�| �!d|j"� | �!d|j"� n$| �|j"d d� | �|j"d d� | �|j"d d � | �|j"d d� |�dd� |�dd� |�dd� |�dd� |� |�}
| �|j"d d� | �|j"d d� | �|j"d d� | �|j"d d� �qNW d S )Nzhttp://example.com/))�GETN)�POSTs blah�Foor^ �Spam�eggsr� r� z example.comr r% )Z
Connectionr0 )r) r^ )r* r+ TZbadpost)� Nr� zContent-lengthz Content-type�0z!application/x-www-form-urlencoded�Hostr] rI )#r( r
r� rt r� r rv rR r� r� r/ r r� r� r� r� r � __contains__r r� r� r� r� r� ru r� r& r r= r� � do_request_rz r� � assertNotIn�unredirected_hdrs) r r7 r� rG r� ru rS r� r� �hdrs�newreqr" r" r# � test_httpZ sh
zHandlerTests.test_httpc C s� t j�� }t� }|_tjddd�}|j}|�� | � t
j |� t |d��B}t
d|i �}|�|�}|�d�}| �|d� | �|�d�� W d Q R X t |d��F}t
d|d d
i�}|�|�}| �t|�d��d
� | �|�d�� W d Q R X d S ) Nzw+bF)�mode�delete�rbzhttp://example.com/zTransfer-encoding�chunkedzContent-lengthzContent-Length� )r( r
r� rt r� �tempfileZNamedTemporaryFiler r0 �
addCleanupr �unlinkrw r r0 rQ r rO rL r� ) r r7 r� �file_objZ file_pathr1 rS r4 �ter" r" r# �test_http_body_file� s"
z HandlerTests.test_http_body_filec C s� t j�� }t� }|_t�� }td|i �}|�|�}| � |�
d�d� | � |� d�� ddi}td||�}|�|�}| � t
|�
d��d� | � |� d�� |�� d S )Nzhttp://example.com/zTransfer-encodingr9 zContent-lengthzContent-Lengthr: )r( r
r� rt r� r� �BytesIOr r0 r rQ rO rL r� r0 )r r7 r� r> rS r4 rF r" r" r# �test_http_body_fileobj� s
z#HandlerTests.test_http_body_fileobjc
C s� t j�� }t� }|_tjddg}x�i ddifD ]x}tj|tj d��^}t
d|j |�}|� |�}|s�| �
|�d�d � | �
|�d�d � n| �
t|�d��d� W d Q R X q0W d S )
Nz-c�passzContent-Lengthr: )�stdoutzhttp://example.com/zContent-lengthzTransfer-encodingr9 )r( r
r� rt r� �sys�
executable�
subprocess�Popen�PIPEr rD r0 r rQ r� )r r7 r� �cmdrF �procrS r4 r" r" r# �test_http_body_pipe� s
z HandlerTests.test_http_body_pipec C s� t j�� }t� }|_dd� }xli ddifD ]\}td|� |�}|�|�}|sr| �|�d�d � | �|�d�d� q,| �t |�d��d� q,W d S ) Nc s s
dV d S )Ns oner" r" r" r" r# �
iterable_body� s z;HandlerTests.test_http_body_iterable.<locals>.iterable_bodyzContent-Length� zhttp://example.com/zContent-lengthzTransfer-encodingr9 )
r( r
r� rt r� r r0 r rQ r� )r r7 r� rM rF rS r4 r" r" r# �test_http_body_iterable� s
z$HandlerTests.test_http_body_iterablec C sH t j�� }t� |_|�tdd��}| �|�d�d� | � |�
d�� d S )Nzhttp://example.com/r" zTransfer-encodingr9 zContent-length) r( r
r� rt r� r0 r r rQ rO rL )r r7 rS r" r" r# �test_http_body_empty_seq� s
z%HandlerTests.test_http_body_empty_seqc C sp t j�� }t� }|_t�dddddg�}x@i ddifD ]0}td||�}|�|�}| �t |�
d ��d� q8W d S )
Nr r� r� � � zContent-Length� zhttp://example.com/zContent-length) r( r
r� rt r� �arrayr r0 r r� rQ )r r7 r� Ziterable_arrayrF rS r4 r" r" r# �test_http_body_array� s
z!HandlerTests.test_http_body_arrayc C s6 t � }tdd�}|�|� |�d� | �|jd� d S )Nr� )r� zhttps://www.example.com)r r� r� rw r Z _debuglevel)r r� r7 r" r" r# � test_http_handler_debuglevel s
z)HandlerTests.test_http_handler_debuglevelc C s� t j�� }t� |_d}ddddg}xZ|D ]R}t||�}|�|�}| �|jd d� |� dd � |�|�}| �|jd d� q(W d S ) Nr, 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.htmlr. z example.comzsomeproxy:3128)
r( r
r� rt r� r r0 r r2 � set_proxy)r r7 ru Zds_urlsZds_urlZds_reqZ np_ds_reqZp_ds_reqr" r" r# �test_http_doubleslash s
z"HandlerTests.test_http_doubleslashc C sn dddg}t d�}xV|D ]N}||_t|�}| �|�� |� | �|jpFd|j� | �t|�� �j|j� qW d 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� rG Zparsedr" r" r# �test_full_url_setter- s
z!HandlerTests.test_full_url_setterc C s6 t d�}|`| �|j� | �|j� | �|jd� d S )Nzhttp://www.example.comr� )r rY rP rZ r r� )r r� r" r" r# �test_full_url_deleterD s
z"HandlerTests.test_full_url_deleterc C sz t j�� }t� |_d}t|�}|�|�}| �|jd� | �|j d� d}t|�}|�|�}| �|jd� | �|j d� d S )Nz http://www.python.org?getspamzwww.python.orgz /?getspamzhttp://www.python.orgr� )
r( r
r� rt r� r r0 r r� r� )r r7 Z weird_urlrS r4 Zurl_without_pathr" r" r# �test_fixpath_in_weirdurlsK s
z&HandlerTests.test_fixpath_in_weirdurlsc C s t j�� }t� }|_d}t|�}tddi d|�}|�||�}| �||� | � t
|d�� tddi d|�}|�||�}| �||� | � t
|d�� tdd i d|�}|�||�}| �||� | � t
|d�� td
d i d|�}| � |�||�� | � |j
d � | � |j||d
d i f� d S )
Nzhttp://example.com/r� r� r� rx �� ZAccepted�� zPartial contenti� z Bad gatewayr� )r( r
ZHTTPErrorProcessorrt r� r r� r r# rO r� rP r rx ry )r r7 r� rG rS r� �newrr" r" r# � test_errors^ s(
zHandlerTests.test_errorsc C s� t � }tj�|�}t� |_td�}tddi d�}|�|�}| � |j
|� | � |j
|� | � |j d� | �
|j� |�||�}| � |j|� | � |j|� | � ||� d S )Nzhttp://example.com/r� r� r� z example.com)r� r( r
�HTTPCookieProcessorrt r� r r� r� r# r� r �origin_req_hostrO Z unverifiabler r� r� )r �cjr7 rS r� r4 r` r" r" r# � test_cookiesw s
zHandlerTests.test_cookiesc
C s� d}d}t j�� }t� }|_�xndD �]d}�x\dD �]R}t|d| �}t||�}tj|_ |�
dd� |d k r�|�
dt t |��� |�
d d
� y ||t� |d td |i�� W n. t jjk
r� | �|d
� | �|� Y nX | �|j�� |� y| �|j�� d� W n$ tk
�r( | �|jj� Y nX dd� |jjD �} | �d| � | �d| � | �|jjd d� | �d |jj� | �d |jj � q4W q&W t|�}tj|_ |fdd�}
t|dd�}d} tj|_ y x|
||d� | d } �q�W W n, t jjk
�r | �| t jjj � Y nX t|dd�}d} tj|_ y$x |
||d| � | d } �q.W W n, t jjk
�rz | �| t jjj � Y nX d S )Nzhttp://example.com/a.htmlzhttp://example.com/b.html)i- i. i/ i3 )Nz
blah
blah
z
http_error_%sZNonsensezviking=withholdzContent-Lengthr* rb �Blah�locationi3 r' c S s g | ] }|� � �qS r" )�lower)�.0�xr" r" r# �
<listcomp>� s z.HandlerTests.test_redirect.<locals>.<listcomp>zcontent-lengthz content-typec S s | � |t� ddtd|i�� d S )Ni. rf rg )r� r} r� )r7 rS rG r" r" r# r� � s z,HandlerTests.test_redirect.<locals>.redirectz example.com)rc r zhttp://example.com/r� zhttp://example.com/%d) r( r
�HTTPRedirectHandlerrt r� �getattrr r{ r| rv rM r? r� rR r} r� r � HTTPErrorr �assertIsNotNonerS r� �
get_method�AttributeErrorrO ru rF r1 r2 Z max_repeatsZmax_redirections) r �from_urlZto_urlr7 r� r� ru r� rS rF r� r~ r" r" r# �
test_redirect� sj
zHandlerTests.test_redirectc C s� d}dddg}dddg}d}t j�� }t� }|_t|�}tj|_x>|D ]6}|d | } | � t j
j |j |t
� d
d td | i�� qHW xF|D ]>}|d | }
|� |t
� d
d
td |
i�� | �|j�� |
� q�W d S )Nzhttp://example.com/a.htmlr� �httpsr �fileZimapZldapzexample.com/b.htmlz://i. zSecurity Loopholerg z That's fine)r( r
rl rt r� r r{ r| rv r& r rn r� r} r� r rS r� ) r rr Z
valid_schemesZinvalid_schemesZschemeless_urlr7 r� rS r� Z invalid_url� valid_urlr" r" r# �test_invalid_redirect� s$
z"HandlerTests.test_invalid_redirectc C sn d}d}t j�� }t� }|_t|�}tj|_t j �
||�}|� |t � ddt
d|i�� | �|j�� |� d S )Nzhttp://example.com/a.htmlz/b.htmli. z That's finerg )r( r
rl rt r� r r{ r| rv r Zurljoinr� r} r� r rS r� )r rr Z relative_urlr7 r� rS rv r" r" r# �test_relative_redirect� s
z#HandlerTests.test_relative_redirectc C s� ddl m} ddlm} |� }||dd� tdd�}tj�� }tj�� }tj� |�}t
||||�}|� d� | � |j
�d�� d S ) Nr )� CookieJar)�interact_netscapezhttp://www.example.com/z spam=eggsi. z%Location: http://www.cracker.com/
ZCookie)Zhttp.cookiejarry Ztest.test_http_cookiejarrz r� r( r
�HTTPDefaultErrorHandlerrl rb r� rw rO rS rL ) r ry rz rd �hh�hdeh�hrhZcpr� r" r" r# �test_cookie_redirect� s
z!HandlerTests.test_cookie_redirectc C sT d}t dd| �}tj�� }tj�� }t|||�}|�d�}| �|�� |� � � d S )Nz(http://www.example.com/index.html#OK
i. z
Location: zhttp://www.example.com)
r� r( r
r{ rl r� rw r r� �strip)r Zredirected_urlr| r} r~ r� r� r" r" r# �test_redirect_fragment s
z#HandlerTests.test_redirect_fragmentc sn t jj� d}t�|�t j_��tt jd� � td��� ��fdd�}|t jj_t j�
d�}�� |� � d� d S )Ns( HTTP/1.1 302 Found
Location: ?query
�HTTPConnection)z/pathz /path?queryc s2 �� |t��� � j| ||f|�|� d| j_d S )Ns HTTP/1.1 200 OK
Hello!)r �nextr
r� Zfakedata)�connr� rG �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 �fakehttpr< r� �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� G dd� dt jj�}|� }t j�|�}d}xp|D ]h\}}| �t|���L t�d| d �|_|� d�}d| d }|j
} | � | � |�t| �� W d Q R X q0W d S ) Nc @ s e Zd Zdd� ZdS )z4HandlerTests.test_redirect_encoding.<locals>.Handlerc S s( | � | j|�}| jj| _t�d�| _|S )Ns) HTTP/1.1 200 OK
Content-Length: 3
123)r� �
connection�buf�last_bufr r� )r rS �resultr" r" r# r� ! s
z>HandlerTests.test_redirect_encoding.<locals>.Handler.http_openN)rB r rC r� r" r" r" r# �Handler s r� ))s /på-dansk/s /p%C3%A5-dansk/)s /spaced%20path/s /spaced%20path/)s
/spaced path/s /spaced%20path/)s /?på-dansks /?p%C3%A5-dansks! HTTP/1.1 302 Redirect
Location: s
zhttp://example.com/s GET � )
r( r
� HTTPHandler� build_opener�subTest�reprr r� r� rw r� rK r� )
r r� r� r� r8 rg r� r r� r
r" r" r# �test_redirect_encoding s
z#HandlerTests.test_redirect_encodingc C s� t � }tj�tdd��}|�|� dgg}t||�}td�}| �|j d� |�
|� | �|j d� | �|d dfgdd � |j D �� d S )
Nzproxy.example.com:3128)r� )r� zreturn responsezhttp://acme.example.com/zacme.example.comr r� c S s g | ]}|d d� �qS )r r� r" )ri �tupr" r" r# rk O s z+HandlerTests.test_proxy.<locals>.<listcomp>) r r( r
� ProxyHandler�dictr� r� r r r� rw r� )r r� �phr� r� rS r" r" r# �
test_proxy@ s
zHandlerTests.test_proxyc C s� dt jd< t� }tj�tdd��}|�|� td�}| � |j
d� |� |� | � |j
d� td�}| � |j
d� |� |� | � |j
d� t jd= d S ) Nz
python.org�no_proxyzproxy.example.com)r� zhttp://www.perl.org/z www.perl.orgzhttp://www.python.orgzwww.python.org) r �environr r( r
r� r� r� r r r� rw )r r� r� rS r" r" r# �test_proxy_no_proxyQ s
z HandlerTests.test_proxy_no_proxyc C sf dt jd< t� }tj�tdd��}|�|� td�}| � |j
d� |� |� | � |j
d� t jd= d S )N�*r� zproxy.example.com)r� zhttp://www.python.orgzwww.python.org) r r� r r( r
r� r� r� r r r� rw )r r� r� rS r" r" r# �test_proxy_no_proxy_all` s
z$HandlerTests.test_proxy_no_proxy_allc C s� t � }tj�tdd��}|�|� dgg}t||�}td�}| �|j d� |�
|� | �|j d� | �|d dfgdd � |j D �� d S )
Nzproxy.example.com:3128)rt )r� zreturn responsezhttps://www.example.com/zwww.example.comr r� c S s g | ]}|d d� �qS )r r� r" )ri r� r" r" r# rk y s z1HandlerTests.test_proxy_https.<locals>.<listcomp>) r r( r
r� r� r� r� r r r� rw r� )r r� r� r� r� rS r" r" r# �test_proxy_httpsk s
z HandlerTests.test_proxy_httpsc C s� t � }tj�tdd��}|�|� t� }|�|� td�}|�dd� |�dd� | � |j
d� | � |j � |�
|� | �d |jj� | �d
|jj� | �|j � | � |j
d� | � |�d �d� d S ) Nzproxy.example.com:3128)rt zhttps://www.example.com/zProxy-Authorization�FooBarz
User-Agent�Grailzwww.example.com)zProxy-Authorizationr� )z
User-Agentr� zProxy-authorization)r r( r
r� r� r� r� r rM r r� rP r� rw r1 r� r� r> ro rQ )r r� r� Z
https_handlerrS r" r" r# �$test_proxy_https_proxy_authorization{ s$
z1HandlerTests.test_proxy_https_proxy_authorization�darwinzonly relevant for OSXc C sz ddddddgd�}x"dD ]}| � t||�d | � qW x"d
D ]}| �t||�d | � q>W d g d�}| � td
|�� d S )NFzfoo.barz *.bar.comz 127.0.0.1z10.10z10.0/16)Zexclude_simple�
exceptions)zfoo.barz www.bar.comz 127.0.0.1z 10.10.0.1z10.0.0.1z expected bypass of %s to be True)z abc.foo.barzbar.comz 127.0.0.2z 10.11.0.1Z notinbypassz!expected bypass of %s to be FalseT�test)rK r rO )r Zbypassr� r" r" r# �test_osx_proxy_bypass� s
z"HandlerTests.test_osx_proxy_bypassc C sz | j ||d��b t� }t� }tj�|�}d�|�d }td|�}|�|� |�|� | � ||d|||dd� W d Q R X d S )N)r� rF z
z
i� r� z!http://acme.example.com/protected)
r� r r� r( r
r �joinr� r� �_test_basic_auth)r rF r� r� �password_manager� auth_handlerr� � http_handlerr" r" r# �check_basic_auth� s
z HandlerTests.check_basic_authc
C s d}d}d|� d�}d|� d�}d}d|� d�}x~d|� d�d|� d�d|� d �|� d
|� �|� d
|� �|� d
|� �|� d
|� �|� d
|� �fD ] }d |� �g}| � ||� q�W t�d tf�� d
|� �g}| � ||� W d Q R X x8||g||g||gfD ] } dd� | D �}| � ||� q�W d S )Nzrealm2@example.comz
Basic realm="�"zOtherscheme xxxzDigest realm="zn", qop="auth, auth-int", nonce="dcd98b7102dd2f0e8b11d0f600bfb0c093", opaque="5ccc069c403ebaf9f0171e9517f40e41"z
Basic realm='�'z", charset="UTF-8"z, zWWW-Authenticate: z Basic Auth Realm was unquotedz WWW-Authenticate: Basic realm=c S s g | ]}d |� ��qS )zWWW-Authenticate: r" )ri Z challenger" r" r# rk � s z0HandlerTests.test_basic_auth.<locals>.<listcomp>)r� r Zcheck_warnings� UserWarning)
r r� Zrealm2�basicZbasic2Zother_no_realm�digestZ realm_strrF Z
challengesr" r" r# �test_basic_auth� s8
z HandlerTests.test_basic_authc
C sv t � }tj�tdd��}|�|� t� }tj�|�}d}tdd| �}|�|� |�|� | � ||d|||dd� d S )Nzproxy.example.com:3128)r� z
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_auth� s
z"HandlerTests.test_proxy_basic_authc
C s� G dd� dt �}G dd� dtjj�}G dd� dtjj�}|� }t� }||�}||�}d}tdd | �} |�|� |�|� |�| � | �||d
|| |d d � | � |j
d d
gd � d S )Nc @ s e Zd Zdd� Zdd� ZdS )zQHandlerTests.test_basic_and_digest_auth_handlers.<locals>.RecordingOpenerDirectorc S s t �| � 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 �|� d S )N)r� r� )r r� r" r" r# �record s zXHandlerTests.test_basic_and_digest_auth_handlers.<locals>.RecordingOpenerDirector.recordN)rB r rC r� r� r" r" r" r# �RecordingOpenerDirector� s r� c @ s e Zd Zdd� ZdS )zOHandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestDigestAuthHandlerc _ s& | j �d� tjjj| f|�|� d S )Nr� )r� r� r( r
�HTTPDigestAuthHandler�http_error_401)r ry r� r" r" r# r� s z^HandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestDigestAuthHandler.http_error_401N)rB r rC r� r" r" r" r# �TestDigestAuthHandler s r� c @ s e Zd Zdd� ZdS )zNHandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestBasicAuthHandlerc _ s& | j �d� tjjj| f|�|� d S )Nr� )r� r� r( r
r r� )r ry r� r" r" r# r� s z]HandlerTests.test_basic_and_digest_auth_handlers.<locals>.TestBasicAuthHandler.http_error_401N)rB r rC 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 sD t � }tj�d �}tdd�}|�|� |�|� | �t|jd� d S )Ni� z WWW-Authenticate: Kerberos
zhttp://www.example.com) r r( r
r� r� r� r& r' rw )r r� Zdigest_auth_handlerr� r" r" r# �$test_unsupported_auth_digest_handler& s
z1HandlerTests.test_unsupported_auth_digest_handlerc C sD t � }tj�d �}tdd�}|�|� |�|� | �t|jd� d S )Ni� zWWW-Authenticate: NTLM
zhttp://www.example.com) r r( r
r r� r� r& r' rw )r r� Zbasic_auth_handlerr� r" r" r# �#test_unsupported_auth_basic_handler0 s
z0HandlerTests.test_unsupported_auth_basic_handlerc C sH dd l } d\}
} |�|||
| � | �||j� | �||j� | �|
|j� | �| |j� |�|� | �|j|� | �|j |� | �t
|j �d� | � |j d �
|�� td|
| f d�} d| �| ��� �� }
| �|j d �|�|
� | �|j d j| |
� d |_|_|�� |�|� | �t
|j �d� | � |j d �
|�� d S )Nr )�wile�coyoter� z%s:%s�asciizBasic r� )�base64re r r� rG r� rV rw r� r� r� r� rO rL �bytesZ encodebytesr� �decoderQ r2 r� )r r� r� Z auth_headerr� r� r� � request_urlZ
protected_urlr� r� rV ZuserpassZauth_hdr_valuer" r" r# r� : s0
z HandlerTests._test_basic_authc C s~ d\}}d}t d�}t� }t|�}|jd |||dd� |�|�}| �|� t� }|�|� |�|� |�|� | �|j � d S )N)r� r� z!http://acme.example.com/protectedr� T)�is_authenticated)
r� r r re r� rK r r� rw r� ) r r� rV 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|�}|�||||� |�|�}| �|� t� }|�|� tddd �} |�| � |�|� |�|�}| � |� t
d�} | �| j � t� }|�|� |�| � |�|� | � | j � d S )N)r� r� z!http://acme.example.com/protectedZACMEi� z&WWW-Authenticate: Basic realm="%s"
r� ) r r re r� rO r r� r� rw rK r� r� )
r r� rV r� r� r� r� r� r� r� r" r" r# �.test_basic_prior_auth_send_after_first_successy s.
z;HandlerTests.test_basic_prior_auth_send_after_first_successc
C s� xzdD ]r\}}d� |�}t�|�� | �}tj�� }td�}d|_|� ||��}|�
� W dQ R X | � |j j
d� |�� qW dS )z=Test the connection is cleaned up when the response is closed))zConnection: closes data)zTransfer-Encoding: chunkeds 4
data
0
)zContent-Length: 4s datazHTTP/1.1 200 OK
{}
z
http://dummy/Nz Connection not closed with {!r})�formatr r� r r( r
r� r rv r� r/ rK �fakesock�closed)r Ztransferru �headerr� r� rS Zrespr" r" r# �test_http_closed� s
z HandlerTests.test_http_closedc C s\ t �d�}tj�� }td�}d|_| �tj j
�� |� ||� W dQ R X | � |j
jd� dS )z;Test the connection is cleaned up after an invalid responser, z
http://dummy/NzConnection not closed)r r� r( r
r� r rv r& r� r� Z
BadStatusLiner� rK r� r� )r r� r� rS r" r" r# �test_invalid_closed� s
z HandlerTests.test_invalid_closedN)/rB r rC r r& r5 r@ rB rL rO rP rU rV rX r[ r\ r] ra re rs rw rx r r� r� r� r� r� r� r� r� r �
skipUnlessrE �platformr� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# r � sP @\B
J
# -2
%(r c @ sN e Zd Zdd� Zdd� Ze�e�d�d�dd� �Z d d
� Z
d d � Z d
d� Z dS )� MiscTestsc s" | � t� fdd�|jD ��� d S )Nc 3 s | ]}|j � kV qd S )N)r� )ri r7 )�
handler_classr" r# � <genexpr>� s z/MiscTests.opener_has_handler.<locals>.<genexpr>)rK �anyr� )r r� r� r" )r� r# �opener_has_handler� s z MiscTests.opener_has_handlerc C s( G dd� dt jj�}G dd� dt jj�}G dd� dt jj�}t jj}|||�}| �||� | �||� |||� �}| �||� | �||� ||�}| �||� |� }| �|t jj� |t jj�}| �|t jj� |t j�� �}| �|t jj� G dd� dt jj�}|||�}| �||� | �||� d S ) Nc @ s e Zd ZdS )z2MiscTests.test_build_opener.<locals>.MyHTTPHandlerN)rB r rC r" r" r" r# �
MyHTTPHandler� s r� c @ s e Zd Zdd� ZdS )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)rB r rC r� r" r" r" r# �
FooHandler� s r� c @ s e Zd Zdd� ZdS )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)rB r rC r� r" r" r" r# �
BarHandler� s r� c @ s e Zd ZdS )z7MiscTests.test_build_opener.<locals>.MyOtherHTTPHandlerN)rB r rC r" r" r" r# �MyOtherHTTPHandler� s r� )r( r
r� � BaseHandlerr� r� )r r� r� r� r� r� r� r" r" r# �test_build_opener� s,
zMiscTests.test_build_openerZnetworkz test requires network accessc C s� t �d��� tj�� }tj�d�}| �d |j� |�|d� d�� | �d|j� | �d|�
d�� |�|d� d�� | �d|j� | �d|�
d�� W d Q R X d S ) Nzhttp://www.example.com/rc zus-ascii� 1zContent-lengthZ
1234567890s
1234567890rk ) r Ztransient_internetr( r
r� r r ru rw r rQ )r r� r
r" r" r# �test_issue16464� s
zMiscTests.test_issue16464c C s� d}d } }}d}t j�|||||�}| �t|d�� | �|jd� | �t|d�� | �|jd� d|j|j f }| �t
|�|� d|j|j f }| �t |�|� dS )z�
Issue 13211 reveals that HTTPError didn't implement the URLError
interface even though HTTPError is a subclass of URLError.
zsomething bad happenedNzContent-Length: 42r< rF zHTTP Error %s: %sz<HTTPError %s: %r>) r( r rn rK r� r r< rF r� r� r? r� )r r� rG r� r� r3 r@ Zexpected_errmsgr" r" r# �test_HTTPError_interface� s z"MiscTests.test_HTTPError_interfacec C sR ddddddddddd d
g }x |D ]\}}| � t|�|� q"W | �ttd �f d S ) N)zproxy.example.com)NNNzproxy.example.com)zproxy.example.com:3128)NNNzproxy.example.com:3128)z joe:password@proxy.example.com)NrU rV zproxy.example.com)z#joe:password@proxy.example.com:3128)NrU rV zproxy.example.com:3128)zhttp://proxy.example.com/)r� NNzproxy.example.com)z http://proxy.example.com:3128/)r� NNzproxy.example.com:3128)z&http://joe:password@proxy.example.com/)r� rU rV zproxy.example.com)z*http://joe:password@proxy.example.com:3128)r� rU rV zproxy.example.com:3128)z1ftp://joe:password@proxy.example.com/rubbish:3128)r rU rV zproxy.example.com)z%http://joe:password@proxy.example.com)r� rU rV zproxy.example.comzfile:/ftp.example.com)r r r& r' )r Zparse_proxy_test_casesZtcr� r" r" r# �test_parse_proxy s zMiscTests.test_parse_proxyc C s< t � }| �t��}|�d� W d Q R X | �t|j�d� d S )NZinvalidz5Unsupported digest authentication algorithm 'invalid')r
r& r' Zget_algorithm_implsr r? Z exception)r r� �excr" r" r# �test_unsupported_algorithm6 s z$MiscTests.test_unsupported_algorithmN)
rB r rC r� r� r r� r Zis_resource_enabledr� r� r� r� r" r" r" r# r� � s , %r� c @ s� e Zd ZG dd� de�Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd S ) � RequestTestsc @ s e Zd ZdZdS )zRequestTests.PutRequest�PUTN)rB r rC r� r" r" r" r# �
PutRequestA s r� c C sL t d�| _t ddddid�| _t ddd�| _| �d�| _| jddd�| _d S ) Nz http://www.python.org/~jeremy/ru zX-Testr� )rF �HEAD)r� r( )r r �post�headr� �put�
force_post)r r" r" r# �setUpD s
zRequestTests.setUpc C s^ | � d| j�� � | � d| j�� � | � d| j�� � | � d| j�� � | � d| j�� � d S )Nr( r' r� r� )r r� rp r r� r� r� )r r" r" r# � test_methodN s
zRequestTests.test_methodc C sL | � | jj� | �d| j�� � d| j_| �| jj� | �d| j�� � d S )Nr' rb r( )rO r ru r rp rK )r r" r" r# � test_dataU s
zRequestTests.test_datac C sN | � d| jj� | j�dd� | �d| jjd � d| j_| � d| jj� d S )NzContent-length�* rb )r1 r r2 rR r ru )r r" r" r# �.test_setting_data_should_remove_content_length_ s
z;RequestTests.test_setting_data_should_remove_content_lengthc C sT | � d| jj� d| j_| j�dd� | �d| jjd � | j`| � d| jj� d S )NzContent-lengthr] rQ )r1 r r2 ru rR r )r r" r" r# �/test_deleting_data_should_remove_content_lengthg s z<RequestTests.test_deleting_data_should_remove_content_lengthc C s | � d| j�� � d S )Nz http://www.python.org/~jeremy/)r r r� )r r" r" r# �test_get_full_urlo s z RequestTests.test_get_full_urlc C s* | � d| jj� td�}| � d|j� d S )Nz /~jeremy/zhttp://www.python.org/r% )r r r� r )r rS r" r" r# �
test_selectors s zRequestTests.test_selectorc C s | � d| jj� d S )Nr� )r r r$ )r r" r" r# �
test_get_typex s zRequestTests.test_get_typec C s | � d| jj� d S )Nzwww.python.org)r r r� )r r" r" r# �
test_get_host{ s zRequestTests.test_get_hostc C s t d�}| �d|j� d S )Nzhttp://www.%70ython.org/zwww.python.org)r r r� )r rS r" r" r# �test_get_host_unquote~ s z"RequestTests.test_get_host_unquotec C sR | � | j�� � | j�dd� | �| j�� � | �d| jj� | �d| jj� d S )Nz www.perl.orgr� zwww.python.org)rO r Z has_proxyrW rK r rc r� )r r" r" r# r� � s
zRequestTests.test_proxyc C s t d�}| �d|j� d S )Nz<URL:http://www.python.org>zwww.python.org)r r r� )r rS r" r" r# �test_wrapped_url� s z RequestTests.test_wrapped_urlc C sL t d�}| �d|j� t d�}| �d|j� d}t |�}| �|�� |� 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 rS rG r" r" r# �test_url_fragment� s z RequestTests.test_url_fragmentc C s6 dddg}x&|D ] }t |�}| �|�� |j� qW d S )Nzhttp://docs.python.orgz.http://docs.python.org/library/urllib2.html#OKz-http://www.python.org/?qs=query#fragment=true)r r r� rY )r r% rG rS r" r" r# � test_url_fullurl_get_full_url� s
z*RequestTests.test_url_fullurl_get_full_urlN)rB r rC r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# r� @ s
r� �__main__)8r r� r r r r� r{ rT rE r; rG Zurllib.requestr( r r r r r r r
Z urllib.parser r� Z http.clientr� ZTestCaser rD rt r} r� r� r� r� r� r� �IOBaser� r� r� r� r� r
r� r� r� r� r� r� r� r r r� r� rB �mainr" r" r" r# �<module> sb $ 3 ,
-6"
a