Your IP : 216.73.216.196
B
j��OB� � @ s( d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
mZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl m Z m Z e �rHddl m!Z" n ddl m"Z" ddgZ#G dd � d e$�Z%G d d � d e&�Z'G d d � d e �Z(G d!d� de �Z)G d"d� de$�Z*e�+d#ej,�Z-d$d%� Z.G d&d'� d'e$�Z/G d(d)� d)e$�Z0G d*d+� d+e/�Z1e1e/j2d,< G d-d.� d.e/�Z3e3e/j2d/< G d0d1� d1e1�Z4e4e/j2d2< G d3d4� d4e/�Z5e5e/j2d5< G d6d7� d7e/�Z6e6e/j2d8< G d9d:� d:e/�Z7e7e/j2d;< G d<d=� d=e6�Z8e8e/j2d>< G d?d@� d@e6�Z9e9e/j2dA< G dBdC� dCe/�Z:e:e/j2dD< e:e/j2dE< e:e/j2dF< G dGdH� dHe$�Z;dIdJ� Z<dVdKdL�Z=dWdNdO�Z>dPdQ� Z?dRdS� Z@dTdU� ZAdS )XzF
Routines for testing WSGI applications.
Most interesting is TestApp
� N)�urlparse)� print_stderr)�StringIO)�BytesIO)� SimpleCookie� CookieError)� cookie_quote)� urlencode)� splittype)� splithost)� string_types)� binary_type)� text_type)� to_string)�to_bytes)�
join_bytes)� OrderedDict)�dumps)�loads)�PY3)�Request�Response)�lint3)�lint�TestApp� TestRequestc @ s e Zd ZdS )� NoDefaultN)�__name__�
__module__� __qualname__� r r �</opt/alt/python37/lib/python3.7/site-packages/webtest/app.pyr / s r c @ s e Zd Zdd� ZdS )�AppErrorc G s� t |�}d}x�|D ]z}t|t�rN|j}t|t�r�|jrD|�|j�}q�t|�}n4t|t�r�y t |�}W n tk
r� t|�}Y nX ||f7 }qW || }t �
| |� d S )Nr ) r �
isinstancer �bodyr
�charset�decode�repr�UnicodeDecodeError� Exception�__init__)�self�message�argsZstr_args�argr$ r r r! r* 5 s"
zAppError.__init__N)r r r r* r r r r! r" 3 s r" c @ sv e Zd ZdZdZdZdd� Zeedd�Zdd� Z ee d d�Z
ed
d � �Z e �
d e je jB �Zd
d� Zdd� Zd8dd�Zd9dd�Zdd� Zd:dd�Ze �
ed��ZdZd d � Zeed �� d�Z d d � Z ee d!�� d�Z d"d#� Z d$d%� Z d&d'� Z!d(d)� Z"d*d+� Z#d,d-� Z$ee$e$jd�Z$d.d/� Z%ee%e%jd�Z%d0d1� Z&ee&e&jd�Z&d2d3� Z'ee'e'jd�Z'd4d5� Z(ee(e(jd�Z(d6d7� Z)dS );� TestResponsez;
Instances of this class are return by ``TestApp``
Nc C s | j dkr| �� | j S )z�
Returns a dictionary of :class:`~webtest.Form` objects. Indexes are
both in order (from zero) and by form id (if the form is given an id).
N)�_forms_indexed� _parse_forms)r+ r r r! �
forms__getS s
zTestResponse.forms__getz_
A list of :class:`~webtest.Form`s found on the page
)�docc C s* | j }|std��d|kr"td��|d S )Nz*You used response.form, but no forms exist� z5You used response.form, but more than one form existsr )�forms� TypeError)r+ r5 r r r! � form__geta s zTestResponse.form__getz�
Returns a single :class:`~webtest.Form` instance; it is an
error if there are multiple forms on the page.
c C s, t | dd�r| jr| jS tr&t| j�S | jS )N� _use_unicodeT)�getattrr% � unicode_bodyr r r$ )r+ r r r! �testbodyr s
zTestResponse.testbodyz<(/?)([:a-z0-9_\-]*)(.*?)>c
C s� i }| _ g }d }x�| j�| j�D ]|}|�d�dk}|�d��� }|dkrLq"|r�|sdtd|�� ��|�| j||� � � � d }q"|r�td|�� ��|�� }q"W |r�td| j|d � ��x6t
|�D ]*\}}t | |�} | ||< | j r�| || j < q�W d S )Nr4 �/� �formz</form> unexpected at %szNested form tags at %szDanging form: %r)
r0 �_tag_re�finditerr; �group�lower�AssertionError�start�append�end� enumerate�Form�id)
r+ r5 Z
form_texts�started�matchrF �tag�i�textr> r r r! r1 | s.
zTestResponse._parse_formsc K sT | j dkr| j dk s"td| j ��| jd }t|�\}}t|�\}}| jj|f|�S )z�
If this request is a redirect, follow that redirect. It
is an error if this is not a redirect response. Returns
another response object.
i, i� z/You can only follow redirect responses (not %s)�location)�
status_intrC �status�headersr
r �test_app�get)r+ �kwrO �type�rest�host�pathr r r! �follow� s
zTestResponse.followFc C s8 d}| j ddd||||||d� \} }
} | j| d |d�S )a�
Click the link as described. Each of ``description``,
``linkid``, and ``url`` are *patterns*, meaning that they are
either strings (regular expressions), compiled regular
expressions (objects with a ``search`` method), or callables
returning true or false.
All the given patterns are ANDed together:
* ``description`` is a pattern that matches the contents of the
anchor (HTML and all -- everything between ``<a...>`` and
``</a>``)
* ``linkid`` is a pattern that matches the ``id`` attribute of
the anchor. It will receive the empty string if no id is
given.
* ``href`` is a pattern that matches the ``href`` of the anchor;
the literal content of that attribute, not the fully qualified
attribute.
* ``anchor`` is a pattern that matches the entire anchor, with
its contents.
If more than one link matches, then the ``index`` link is
followed. If ``index`` is not given and more than one link
matches, or if no link matches, then ``IndexError`` will be
raised.
If you give ``verbose`` then messages will be printed about
each link, and why it does or doesn't match. If you use
``app.click(verbose=True)`` you'll see a list of all the
links.
You can use multiple criteria to essentially assert multiple
aspects about the link, e.g., where the link's destination is.
T�a�hrefN) rL � href_attr� href_extract�contentrI � href_pattern� html_pattern�index�verbose�uri)�
extra_environ)�
_find_element�goto) r+ � descriptionZlinkidr\ Zanchorrb rc re �__tracebackhide__�
found_html�
found_desc� found_attrsr r r! �click� s (zTestResponse.clickc C s: d}| j ddt�d�||||||d� \}} }
| �|
d �S )z�
Like ``.click()``, except looks for link-like buttons.
This kind of button should look like
``<button onclick="...location.href='url'...">``.
T�buttonZonclickzlocation\.href='(.*?)') rL r] r^ r_ rI r` ra rb rc rd )rf �re�compilerg ) r+ rh Zbuttonidr\ rn rb rc ri rj rk rl r r r! � clickbutton� s zTestResponse.clickbuttonc
s� t |�}
t |�} t |�} t |�}
| j}t�d||f tjtj �}t�dtjtjB �}g }x(|�|�D ]}|�|�� |� � f� qfW � fdd�}g }d}�x�|�|�D �]�}d}x.|D ]&\}}|�� |kr�|� � |k r�d}P q�W |r�q�|�
d�}|�
d�}|�
d �}t |�} � �r |d
| � | � |��s8|d | � q�| | } |�rj|�
| �} | �s`|d � q�| �
d�} | | d
< | �d��r�|d� q�| �d��r�|d� q�|d7 }|
�r�|
|��s�|d� q�| �r�| | � dd���s�|d� q�| �r�| | ��s�|d� q�|
�r|
|��s|d� q�|d� |�||| f� q�W |�sDtd| ��|d k�r�t|�dk�rxtdd�d d � |D �� ��|d } n:y || } W n, tk
�r� td t|�||f ��Y nX | S ) Nz<%s\s+(.*?)>(.*?)</%s>z<script.*?>.*?</script>c s � r t | � d S )N)�print)�s)rc r r! �printlog� s z,TestResponse._find_element.<locals>.printlogr FTr4 r= z Element: %rz Skipped: no %s attributez( Skipped: doesn't match extract patternrd �#z& Skipped: only internal fragment hrefz javascript:z$ Skipped: cannot follow javascript:z$ Skipped: doesn't match descriptionrI � z Skipped: doesn't match idz Skipped: doesn't match hrefz Skipped: doesn't match htmlz
Acceptedz-No matching elements found (from %s possible)zMultiple links match: %sz, c S s g | ]\}}}t |��qS r )r' )�.0Zanc�d�attrr r r! �
<listcomp>: s z.TestResponse._find_element.<locals>.<listcomp>z6Only %s (out of %s) links match; index %s out of range)�
_make_patternr; ro rp �I�Sr@ rE rD rF rA � _parse_attrsrT �search�
startswith�
IndexError�len�join) r+ rL r] r^ r_ rI r` ra rb rc Z content_patZid_patZhref_patZhtml_patr$ r? Z
_script_reZ bad_spansrK rt Z found_linksZ total_linksZ found_badZ bad_startZbad_endZel_htmlZel_attrZ
el_content�attrsZel_href�mZ
found_linkr )rc r! rf � s�
zTestResponse._find_elementrT c s t �|�\}}}}}d } }}t �|||||f�}t �� jj|�}|�� }|dks`td| ��ts�t � dd�r� fdd���|�}d|kr��fd d
�|d D �|d< d |krȇfd d
�|d D �|d < d
|kr�|d
�|d
< |dkr� j
j }n� j
j }||f|�S )a
Go to the (potentially relative) link ``href``, using the
given method (``'get'`` or ``'post'``) and any extra arguments
you want to pass to the ``app.get()`` or ``app.post()``
methods.
All hostnames and schemes will be ignored.
rv )rT �postz9Only "get" or "post" are allowed for method (you gave %r)r8 Fc s t | t�r| �� j�S | S )N)r# r �encoder% )rs )r+ r r! �to_strZ s
z!TestResponse.goto.<locals>.to_str�paramsc s g | ]}t t� |���qS r )�tuple�map)rw �p)r� r r! rz b s z%TestResponse.goto.<locals>.<listcomp>� upload_filesc s g | ]}t � |��qS r )r� )rw �f)r� r r! rz f s � content_typerT )
r �urlsplit�
urlunsplitZurljoin�request�urlrB rC r r9 rS rT r� ) r+ r\ �methodr- �schemerX rY �query�fragmentr )r+ r� r! rg E s.
zTestResponse.gotoz
[ \n\r\t]+c C s& | j d kr | j�td�| j�| _ | j S )N� )� _normal_body�_normal_body_regex�subr r$ )r+ r r r! �normal_body__getv s
z TestResponse.normal_body__getz]
Return the whitespace-normalized body
c C s | j std��| j�| j �S )NzDYou cannot access Response.unicode_normal_body unless charset is set)r% �AttributeError� normal_bodyr& )r+ r r r! �unicode_normal_body__get� s z%TestResponse.unicode_normal_body__getzC
Return the whitespace-normalized body, as unicode
c C sr t |t�s&t|d�r |�� }nt|�}t |t�r>| js>t|�}t |t�rV| j}| j }n | j
}| j }||kpp||kS )z�
A response 'contains' a string if it is present in the body
of the response. Whitespace is normalized when searching
for a string.
� __unicode__) r# r �hasattrr� �strr r% r r: �unicode_normal_bodyr$ r� )r+ rs r$ r� r r r! � __contains__� s
zTestResponse.__contains__c O s� d|kr(|d }|d= t |t�r,|g}ng }|r8td��x8|D ]0}|| kr>td| � tt| �� td| ��q>W x8|D ]0}|| krxtd| � tt| �� td| ��qxW dS )z�
Assert that the response contains all of the strings passed
in as arguments.
Equivalent to::
assert string in res
�noz)The only keyword argument allowed is 'no'zActual response (no %r):z Body does not contain string %rzActual response (has %r)zBody contains bad string %rN)r# r r6 r r� r� )r+ ZstringsrU r� rs Zno_sr r r! � mustcontain� s*
zTestResponse.mustcontainc sZ d� dd� � j�� D ��}� fdd�� jD �}|�� dt� j�d� dd� |D ��|f S )N�
c S s g | ]}|� � r|�qS r )�strip)rw �lr r r! rz � s z(TestResponse.__str__.<locals>.<listcomp>c s* g | ]"\}}|� � d kr� �|�|f�qS )zcontent-length)rB �_normalize_header_name)rw �n�v)r+ r r! rz � s zResponse: %s
%s
%sc S s g | ]\}}d ||f �qS )z%s: %sr )rw r� r� r r r! rz � s )r� r; �
splitlinesZ
headerlist�sortr rQ )r+ Z simple_bodyrR r )r+ r! �__str__� s
zTestResponse.__str__c C s |� dd��� � dd�}|S )N�-r� )�replace�title)r+ �namer r r! r� � s z#TestResponse._normalize_header_namec C s� | j rd| j }nd}| jrjt| j�}t|�dkr`|d d� d |dd � }|dt| j� 7 }d| }nd }| jr�d
| j }nd}d t| j� | | | d S )
Nz %srv � �
z...���z/%sz body=%sz no bodyz
location: %s�<�>)r� r$ r' r� rO r rQ )r+ Zct�brr$ rO r r r! �__repr__� s
zTestResponse.__repr__c C sT d| j krtd| j ��yddlm} W n tk
rD td��Y nX || j�}|S )z�
Returns the response as a `BeautifulSoup
<http://www.crummy.com/software/BeautifulSoup/documentation.html>`_
object.
Only works with HTML responses; other content-types raise
AttributeError.
�htmlz,Not an HTML response body (content-type: %s)r )�
BeautifulSoupz:You must have BeautifulSoup installed to use response.html)r� r� r� � ImportErrorr; )r+ r� Zsoupr r r! r� � s
zTestResponse.htmlc C s� d| j krtd| j ��yddlm} W nd tk
r� y ddl}W nB tk
r� yddlm} W n tk
r� td��Y nX Y nX Y nX |�| j�S )z�
Returns the response as an `ElementTree
<http://python.org/doc/current/lib/module-xml.etree.ElementTree.html>`_
object.
Only works with XML responses; other content-types raise
AttributeError
�xmlz+Not an XML response body (content-type: %s)r )� ElementTreeNzKYou must have ElementTree installed (or use Python 2.5) to use response.xml)r� r� Z xml.etreer� r� Z elementtree�XMLr$ )r+ r� r r r! r� � s
zTestResponse.xmlc C s� d| j kr"d| j kr"td| j ��yddlm} W n tk
rN td��Y nX yddlm} W n tk
rz |j}Y nX | j dkr�|| j| j j
d �S |j | j| j j
d �S d
S ) a{
Returns the response as an `lxml object
<http://codespeak.net/lxml/>`_. You must have lxml installed
to use this.
If this is an HTML response and you have lxml 2.x installed,
then an ``lxml.html.HTML`` object will be returned; if you
have an earlier version of lxml then a ``lxml.HTML`` object
will be returned.
r� r� z3Not an XML or HTML response body (content-type: %s)r )�etreez1You must have lxml installed to use response.lxml)�
fromstringz text/html)Zbase_urlN) r� r� �lxmlr� r� Z lxml.htmlr� ZHTMLr; r� r� r� )r+ r� r� r r r! r� s"
zTestResponse.lxmlc C s2 | j dkrtd| j ��tdkr(td��t| j�S )a
Return the response as a JSON response. You must have `simplejson
<http://goo.gl/B9g6s>`_ installed to use this, or be using a Python
version with the json module.
The content type must be application/json to use this.
zapplication/jsonz+Not a JSON response body (content-type: %s)Nz7You must have simplejson installed to use response.json)r� r� r r� r; )r+ r r r! �json> s
zTestResponse.jsonc C s^ d| j kr"d| j kr"td| j ��yddlm} W n tk
rN td��Y nX || j�}|S )z�
Returns the response as a `PyQuery <http://pyquery.org/>`_ object.
Only works with HTML and XML responses; other content-types raise
AttributeError.
r� r� z3Not an HTML or XML response body (content-type: %s)r )�PyQueryz7You must have PyQuery installed to use response.pyquery)r� r� �pyqueryr� r� r; )r+ r� rx r r r! r� Q s
zTestResponse.pyqueryc C sz ddl }ddl}|jddd�}|j}|�� t|d�}|�t| j�� |�� |d dkrdd| }nd | }|� |� dS )
z{
Show this response in a browser window (for debugging purposes,
when it's hard to read the HTML).
r Nz webtest-pagez.html)�prefix�suffix�wr< zfile:///zfile://)
�
webbrowser�tempfileZNamedTemporaryFiler� �close�open�writer r$ Zopen_new)r+ r� r� r� r� r� r r r! � showbrowserf s
zTestResponse.showbrowser)NNNNNFN)NNNNNF)rT )*r r r �__doc__r� r0 r2 �propertyr5 r7 r> r; ro rp r} r| r? r1 rZ rm rq rf rg r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r/ J sZ
1
[
- r/ c @ s e Zd ZdZeZdS )r TN)r r r �disabledr/ Z
ResponseClassr r r r! r | s c @ s� e Zd ZdZdZeZd.dd�Zdd� Zd/dd �Z d
d � Z
d0d
d�Z d1dd�Z d2dd�Z
edddd fdd�Zd3dd�Zedddd fdd�Zd4dd�Zedddd fd d �Zd5d d �Zd6d d!�Zd"d#� Zd$d%� Zd7d&d'�Zd(d)� Zd*d+� Zd,d-� ZdS )8r a
Wraps a WSGI application in a more convenient interface for
testing.
``app`` may be an application, or a Paste Deploy app
URI, like ``'config:filename.ini#test'``.
``extra_environ`` is a dictionary of values that should go
into the environment for each request. These can provide a
communication channel with the application.
``relative_to`` is a directory, and filenames used for file
uploads are calculated relative to this. Also ``config:``
URIs that aren't absolute.
TNc C sR t |t�r"ddlm} |||d�}|| _|| _|d kr:i }|| _|| _| �� d S )Nr )�loadapp)� relative_to) r# r Z paste.deployr� �appr� re � use_unicode�reset)r+ r� re r� r� r� r r r! r* � s
zTestApp.__init__c C s
i | _ dS )zc
Resets the state of the application; currently just clears
saved cookies.
N)�cookies)r+ r r r! r� � s z
TestApp.resetc C s$ | j �� }d|d< |r |�|� |S )NTzpaste.throw_errors)re �copy�update)r+ re �environr r r! �
_make_environ� s
zTestApp._make_environc C s( t �|�\}}}}}t �||||df�S )Nrv )r r� r� )r+ r� r� ZnetlocrY r� r� r r r! �_remove_fragment� s zTestApp._remove_fragmentFc
C s� | � |�}d}t|�}| �|�}|r\t|t�s:t|dd�}d|krL|d7 }n|d7 }||7 }d|krz|�dd�\}|d< nd|d< | j�||�} |r�| j �
|� | j | ||d�S ) a�
Get the given url (well, actually a path like
``'/page.html'``).
``params``:
A query string, or a dictionary that will be encoded
into a query string. You may also include a query
string on the ``url``.
``headers``:
A dictionary of extra headers to send.
``extra_environ``:
A dictionary of environmental variables that should
be added to the request.
``status``:
The integer status code you expect (if not 200 or 3xx).
If you expect a 404 response, for instance, you must give
``status=404`` or it will be an error. You can also give
a wildcard, like ``'3*'`` or ``'*'``.
``expect_errors``:
If this is not true, then if anything is written to
``wsgi.errors`` it will be an error. If it is true, then
non-200/3xx responses are also okay.
Returns a :class:`webtest.TestResponse` object.
T)�doseq�?�&r4 � QUERY_STRINGrv )rQ �
expect_errors) r� r� r� r# r r �split� RequestClass�blankrR r� �
do_request)
r+ r� r� rR re rQ r� r� ri �reqr r r! rT � s&
z TestApp.getrv c
C s� | � |�}
t|| �}|s*| rVt| ��d�rVtj|dd�}| �||pDd�\} }| |
d< n|rf|
�dd� d|kr�|�dd�\}|
d < nd
|
d < | d k r�| |
d< t t
|��|
d < ||
d
< t t |��|
d< | �
|�}| j�||
�} |r�| j�|� | j| ||d�S )z'
Do a generic request.
Z multipartT)Zkeep_blank_valuesr Z CONTENT_TYPEz!application/x-www-form-urlencodedr� r4 r� rv NZCONTENT_LENGTHZREQUEST_METHODz
wsgi.input)rQ r� )r� �
encode_paramsr r� �cgiZ parse_qsl�encode_multipart�
setdefaultr� r� r� r r r� r� r� rR r� r� ) r+ r� r� r� rR re rQ r� r� r� r� r� r r r! � _gen_request� s2
zTestApp._gen_requestc C s | j d||||||||d� S )a�
Do a POST request. Very like the ``.get()`` method.
``params`` are put in the body of the request.
``upload_files`` is for file uploads. It should be a list of
``[(fieldname, filename, file_content)]``. You can also use
just ``[(fieldname, filename)]`` and the file content will be
read from disk.
Returns a ``webob.Response`` object.
�POST)r� rR re rQ r� r� r� )r� ) r+ r� r� rR re rQ r� r� r� r r r! r� s
z TestApp.postc C s0 d}|t k rt|�}| jd|||||d||d� S )z�
Do a POST request. Very like the ``.get()`` method.
``params`` are dumps to json and put in the body of the request.
Content-Type is set to ``application/json``.
Returns a ``webob.Response`` object.
zapplication/jsonr� N)r� rR re rQ r� r� r� )r r r� )r+ r� r� rR re rQ r� r� r r r! � post_json% s zTestApp.post_jsonc C s | j d||||||||d� S )a}
Do a PUT request. Very like the ``.post()`` method.
``params`` are put in the body of the request, if params is a
tuple, dictionary, list, or iterator it will be urlencoded and
placed in the body as with a POST, if it is string it will not
be encoded, but placed in the body directly.
Returns a ``webob.Response`` object.
�PUT)r� rR re rQ r� r� r� )r� ) r+ r� r� rR re rQ r� r� r� r r r! �put7 s
z TestApp.putc C s0 d}|t k rt|�}| jd|||||d||d� S )z�
Do a PUT request. Very like the ``.post()`` method.
``params`` are dumps to json and put in the body of the request.
Content-Type is set to ``application/json``.
Returns a ``webob.Response`` object.
zapplication/jsonr� N)r� rR re rQ r� r� r� )r r r� )r+ r� r� rR re rQ r� r� r r r! �put_jsonI s zTestApp.put_jsonc C s. |rt �dtj� | jd|||||d||d� S )zv
Do a DELETE request. Very like the ``.get()`` method.
Returns a ``webob.Response`` object.
zXYou are not supposed to send a body in a DELETE request. Most web servers will ignore it�DELETEN)r� rR re rQ r� r� r� )�warnings�warnr � WSGIWarningr� )r+ r� r� rR re rQ r� r� r r r! �delete[ s zTestApp.deletec C sB |rt �dtj� d}|tk r&t|�}| jd|||||d||d� S )z�
Do a DELETE request. Very like the ``.get()`` method.
Content-Type is set to ``application/json``.
Returns a ``webob.Response`` object.
zXYou are not supposed to send a body in a DELETE request. Most web servers will ignore itzapplication/jsonr� N)r� rR re rQ r� r� r� )r� r� r r� r r r� )r+ r� r� rR re rQ r� r� r r r! � delete_jsonl s zTestApp.delete_jsonc C s | j d||||d|d�S )zw
Do a OPTIONS request. Very like the ``.get()`` method.
Returns a ``webob.Response`` object.
ZOPTIONSN)rR re rQ r� r� )r� )r+ r� rR re rQ r� r r r! �options� s
zTestApp.optionsc C s | j d||||d|d�S )zt
Do a HEAD request. Very like the ``.get()`` method.
Returns a ``webob.Response`` object.
ZHEADN)rR re rQ r� r� )r� )r+ r� rR re rQ r� r r r! �head� s
z TestApp.headc C s� dt � � }g }x@|D ]8\}}|�d| � |�d| � |�d� |�|� qW xp|D ]h}| �|�\}}}|�d| � |�d||f � t�|�d } |�d| p�d � |�d� |�|� qXW |�d| d � |�d� td |�}
d
| } | |
fS ) z�
Encodes a set of parameters (typically a name/value list) and
a set of files (a list of (name, filename, file_body)) into a
typical POST body, returning the (content_type, body).
z----------a_BoUnDaRy%s$z--z)Content-Disposition: form-data; name="%s"rv z8Content-Disposition: form-data; name="%s"; filename="%s"r zContent-Type: %szapplication/octet-streamz
z multipart/form-data; boundary=%s)�randomrE �_get_file_info� mimetypesZ
guess_typer ) r+ r� �files�boundary�lines�key�value� file_info�filenameZfcontentr$ r� r r r! r� � s.
zTestApp.encode_multipartc C s� t |�dkrl|d }| jr*tj�| j|�}t|d�}|�� }trVt|t �rV|�
|j �}|� � |d ||fS t |�dkr�|d }t|t
�s�tdt
t|�f ��|S tdt|�d d� ��d S ) Nr= r4 �rbr � z File content must be %s not %szuupload_files need to be a list of tuples of (fieldname, filename, filecontent) or (fieldname, filename); you gave: %r�d )r� r� �osrY r� r� �readr r# r r� �encodingr� r
�
ValueErrorrV r' )r+ r r r� r_ r r r! r� � s&
zTestApp._get_file_infoc K s� t |t�r | jj|f|�}n@|�� }x |�� D ]\}}t|||� q.W |jdkr\t|j �|_d|j
d< x$| j �� D ]\}}|j
� ||� qrW | j
|||d�S )a
Creates and executes a request. You may either pass in an
instantiated :class:`TestRequest` object, or you may pass in a
URL and keyword arguments to be passed to
:meth:`TestRequest.blank`.
You can use this to run a request without the intermediary
functioning of :meth:`TestApp.get` etc. For instance, to
test a WebDAV method::
resp = app.request('/new-col', method='MKCOL')
Note that the request won't have a body unless you specify it,
like::
resp = app.request('/test.txt', method='PUT', body='test')
You can use ``POST={args}`` to set the request body to the
serialized arguments, and simultaneously set the request
method to ``POST``
���Tzpaste.throw_errors)rQ r� )r# r r� r� r� �items�setattrZcontent_lengthr� r$ r� re r� r� )r+ Z
url_or_reqrQ r� Z
req_paramsr� r� r r r r! r� � s
zTestApp.requestc C s� d}t � }||jd< |j�dd�}|rF|j�|�rF|jt|�d� |_| jrpd�dd� | j�� D ��}||jd< d|jd < i |jd
< t �
| j �}|j |dd �} | j
| _|| _|| _ | | _y
| j W n tk
r� Y nX |�� | _x>|jd
�� D ],\}
} t| |
��rtd |
��t| |
| � q�W |�s>| �|| � | �| � i | _xx| j� d
�D ]h} y t | �}
W n$ t k
�r� t d| f ��Y nX x.|
�� D ]"\}}|j | j|< |j | j|< �q�W �qRW | S )a�
Executes the given request (``req``), with the expected
``status``. Generally ``.get()`` and ``.post()`` are used
instead.
To use this::
resp = app.do_request(webtest.TestRequest.blank(
'url', ...args...))
Note you can pass any keyword arguments to
``TestRequest.blank()``, which will be set on the request.
These can be arguments like ``content_type``, ``accept``, etc.
Tz wsgi.errorsZ SCRIPT_NAMErv Nc S s g | ]\}}d |t |�f �qS )z%s=%s; )r )rw r� r r r r! rz s z&TestApp.do_request.<locals>.<listcomp>Z HTTP_COOKIEz
paste.testingzpaste.testing_variables)Zcatch_exc_infozopaste.testing_variables contains the variable %r, but the response object already has an attribute by that namez
set-cookiez Could not parse cookie header %r) r r� rT Z path_infor� r� r� r� r r Z
middlewarer� Z get_responser� r8 r� rS r$ r6 �getvalue�errorsr� r
r
�
_check_status�
_check_errorsZ cookies_setrR Zgetallr r r )r+ r� rQ r� ri r � script_nameZ
cookie_headerr� �resr� r �header�cr Zmorselr r r! r� � sX
zTestApp.do_requestc C s� d}|dkrd S t |j�}t|t�rFd|krFt�t�|�|tj�rFd S t|t t
f�r�|j |kr~t d|d�
tt|��|jj|��d S |d kr�|j dkr�|j dk r�d S t d||jj|��||j kr�t d||��d S ) NT�*z*Bad response: %s (not one of %s for %s)
%sz, �� i� z7Bad response: %s (not 200 OK or 3xx redirect for %s)
%szBad response: %s (not %s))r rQ r# r ro rK �fnmatch� translater| �listr� rP r" r� r� r� r� r� )r+ rQ r ri Z
res_statusr r r! r 4 s2
zTestApp._check_statusc C s |j }|rtd|��d S )Nz!Application had errors logged:
%s)r r" )r+ r r r r r! r O s zTestApp._check_errors)NNT)N)NNNNF)rv NNNNFN)rv NNNNFN)rv NNNNFN)rv NNNFN)NNNF)NNNF)NF) r r r r� r� r r� r* r� r� r� rT r� r� r r� r� r� r� r� r� r� r� r� r� r� r r r r r r! r � sJ
5
$?zW([^= \n\r\t]+)[ \n\r\t]*(?:=[ \n\r\t]*(?:"([^"]*)"|\'([^\']*)\'|([^"\'][^ \n\r\t>]*)))?c C sV i }xLt �| �D ]>}|�d��� }|�d�p4|�d�}t|p>d�}||t|�< qW |S )Nr4 r= r rv )�_attr_rer@ rA rB � html_unquoter� )rN r� rK Z attr_nameZ attr_bodyr r r! r~ ` s r~ c @ sL e Zd ZdZi ZdZddd�Zdd� Zdd � Zd
d � Z e
e e�Z d d
� Z dS )�Fieldz
Field object.
TNc K s. || _ || _|| _|| _|| _|| _|| _d S )N)r> rL r� �pos�_valuerI r� )r+ r> rL r� r r rI r� r r r! r* x s zField.__init__c C s$ | j std| j| jf ��|| _d S )Nz-You cannot set the value of the <%s> field %r)�settabler� rL r� r )r+ r r r r! �
value__set� s
zField.value__setc C s
|| _ dS )z^
Like setting a value, except forces it even for, say, hidden
fields.
N)r )r+ r r r r! � force_value� s zField.force_valuec C s | j S )N)r )r+ r r r! �
value__get� s zField.value__getc C s. d| j j| jf }| jr&|d| j 7 }|d S )Nz
<%s name="%s"z id="%s"r� )� __class__r r� rI )r+ r r r r! r� � s zField.__repr__)NN)
r r r r� �classesr r* r! r" r# r� r r� r r r r! r m s
r c @ s e Zd ZdS )�NoValueN)r r r r r r r! r& � s r& c sB e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zeee�Z � Z
S )
�Selectz)
Field representing ``<select>``
c s( t t| �j||� g | _d | _t| _d S )N)�superr' r* r� �
selectedIndexr& �
_forced_value)r+ r- r� )r$ r r! r* � s zSelect.__init__c C s
|| _ d S )N)r* )r+ r r r r! r" � s zSelect.force_valuec C sf | j tk rt| _ xPt| j�D ] \}\}}|t|�kr || _P q W td|d�dd� | jD ��f ��d S )Nz Option %r not found (from %s)z, c S s g | ]\}}t |��qS r )r' )rw �or r r r! rz � s z%Select.value__set.<locals>.<listcomp>)r* r& rG r� �
_stringifyr) r
r� )r+ r rM �option�checkedr r r! r! � s
zSelect.value__setc C sb | j tk r| j S | jd k r*| j| j d S x2| jD ]\}}|r2|S q2W | jrZ| jd d S d S d S )Nr )r* r& r) r� )r+ r- r. r r r! r# � s
zSelect.value__get) r r r r� r* r" r! r# r� r �
__classcell__r r )r$ r! r' � s
r' �selectc sB e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zeee�Z � Z
S )
�MultipleSelectz=
Field representing ``<select multiple="multiple">``
c s( t t| �j||� g | _g | _g | _d S )N)r( r1 r* r� �selectedIndices�_forced_values)r+ r- r� )r$ r r! r* � s zMultipleSelect.__init__c C s || _ g | _d S )N)r3 r2 )r+ �valuesr r r! r" � s zMultipleSelect.force_valuec C s~ dd� |D �}g | _ x8t| j�D ]*\}\}}||kr | j�|� |�|� q W |rztdd�|�d�dd� | jD ��f ��d S )Nc S s g | ] }t |��qS r )r, )rw r r r r! rz � s z-MultipleSelect.value__set.<locals>.<listcomp>z Option(s) %r not found (from %s)z, c S s g | ]\}}t |��qS r )r' )rw r+ r r r r! rz � s )ZselectedIndiciesrG r� r2 rE �remover
r� )r+ r4 Z
str_valuesrM r- r. r r r! r! � s zMultipleSelect.value__setc sn g }� j r � fdd�� j D �}n,� jsLg }x � jD ]\}}|r2|�|� q2W � jr\|� j7 }� jrj|sjd }|S )Nc s g | ]}� j | d �qS )r )r� )rw rM )r+ r r! rz � s z-MultipleSelect.value__get.<locals>.<listcomp>)r2 r3 r� rE )r+ Zselected_valuesr- r. r )r+ r! r# � s
zMultipleSelect.value__get) r r r r� r* r" r! r# r� r r/ r r )r$ r! r1 � s r1 �multiple_selectc @ s$ e Zd ZdZdd� Zeeej�ZdS )�Radioz5
Field representing ``<input type="radio">``
c C s> | j d k r| j| j d S x | jD ]\}}|r"|S q"W d S d S )Nr )r) r� )r+ r- r. r r r! r# s
zRadio.value__getN) r r r r� r# r� r' r! r r r r r! r7 s
r7 �radioc s: e Zd ZdZ� fdd�Zdd� Zdd� Zeee�Z� Z S )�Checkboxz8
Field representing ``<input type="checkbox">``
c s t t| �j||� d|k| _d S )Nr. )r( r9 r* r. )r+ r- r� )r$ r r! r* ! s zCheckbox.__init__c C s | | _ d S )N)r. )r+ r r r r! r! % s zCheckbox.value__setc C s$ | j r | jd krdS | jS nd S d S )N�on)r. r )r+ r r r! r# ( s
zCheckbox.value__get)
r r r r� r* r! r# r� r r/ r r )r$ r! r9 s
r9 Zcheckboxc @ s$ e Zd ZdZdd� Zeeej�ZdS )�Textz4
Field representing ``<input type="text">``
c C s | j d krdS | j S d S )Nrv )r )r+ r r r! r# ; s
zText.value__getN) r r r r� r# r� r r! r r r r r! r; 6 s r; rN c @ s$ e Zd ZdZdd� Zeeej�ZdS )�Filez4
Field representing ``<input type="file">``
c C s | j d krdS | j S d S )Nrv )r )r+ r r r! r# L s
zFile.value__getN) r r r r� r# r� r r! r r r r r! r< F s r< �filec @ s e Zd ZdZdS )�Textareaz+
Field representing ``<textarea>``
N)r r r r� r r r r! r> W s r> �textareac @ s e Zd ZdZdS )�Hiddenz6
Field representing ``<input type="hidden">``
N)r r r r� r r r r! r@ _ s r@ Zhiddenc @ s, e Zd ZdZdZdd� Zee�Zdd� ZdS )�SubmitzG
Field representing ``<input type="submit">`` and ``<button>``
Fc C s d S )Nr )r+ r r r! r# n s zSubmit.value__getc C s | j S )N)r )r+ r r r! �value_if_submitteds s zSubmit.value_if_submittedN) r r r r� r r# r� r rB r r r r! rA g s
rA �submitrn Zimagec @ s� e Zd ZdZe�dej�Ze�dej�Ze Z
dd� Z dd� Z dd � Z
d
d � Zd d
� Zdd� Zd dd�Zdefdd�Zd dd�Zd!dd�Zdd� Zd"dd �Zd d � ZdS )#rH a
This object represents a form that has been found in a page.
This has a couple useful attributes:
``text``:
the full HTML of the form.
``action``:
the relative URI of the action.
``method``:
the method (e.g., ``'GET'``).
``id``:
the id, or None if not given.
``fields``:
a dictionary of fields, each value is a list of fields by
that name. ``<input type="radio">`` and ``<select>`` are
both represented as single fields with multiple options.
z <(/?)([a-z0-9_\-]*)([^>]*?)>z<<label\s+(?:[^>]*)for=(?:"|')([a-z0-9_\-]+)(?:"|')(?:[^>]*)>c C s || _ || _| �� | �� d S )N)�responserN �
_parse_fields�
_parse_action)r+ rD rN r r r! r* � s z
Form.__init__c C s� d }d }t � }�xv| j�| j�D �]b}|�d�dk}|�d��� }|dkrLq |dkrt|rt|sntd|�d� ��d }q |dkr�|r�|s�td |�� ��t| j|d |�� � �|d _ d }q |r�q t
|�d
��}d |kr�|� d �}nd }|d k�r |j �
|�d
�d|kf� q |dk�r�|�d�dk�r�|�|�} | �sh| jjd | |||�� f|�} |�|g ��
| � n | d } t| | jjd ��s�t�| j �
|�d
�d|kf� q |}
|dk�r�|�dd��� }
|
dk�r�|�d��r�| jj�d| j�} n| jj�|
| j�} | | |||�� f|�} |dk�rH|�r:td||�d�f ��| |�� f}n*|dk�rr|�rntd||�d�f ��| }|�|g ��
| � q W || _d S )Nr4 r< r= )�inputr0 r- r? rn r0 z%r without starting selectr r? z$</textarea> with no <textarea> at %sr r� r- r ZselectedrG rV r8 r. rN Zmultipler6 zNested textareas: %r and %rzNested selects: %r and %r)r r? r@ rN rA rB rC rD r r r~ �popr� rE rT �
FieldClassr% r� r# rF �fields) r+ Z in_selectZ in_textarearJ rK rF rL r� r� �fieldZtag_typerI r r r! rE � s|
zForm._parse_fieldsc C s� d | _ x�| j�| j�D ]t}|�d�dk}|�d��� }|dkr@q|rFP t|�d��}|�dd�| _ |�dd �| _|�d
�| _ |�d d �| _
qW d
s�t d��| j d k s�t d��d S )Nr4 r< r= r> r �actionrv r� �GETrI �enctypez!application/x-www-form-urlencodedr zNo </form> tag foundzNo <form> tag found) rL r? r@ rN rA rB r~ rT r� rI rN rC )r+ rK rF rL r� r r r! rF � s" zForm._parse_actionc C sl | j �|�}|dk s6td|d�tt| j �� ��f ��t|�dks^td|d�tt|��f ��||d _dS )a�
Set the value of the named field. If there is 0 or multiple
fields by that name, it is an error.
Setting the value of a ``<select>`` selects the given option
(and confirms it is an option). Setting radio fields does the
same. Checkboxes get boolean values. You cannot set hidden
fields or buttons.
Use ``.set()`` if there is any ambiguity and you must provide
an index.
Nz*No field by the name %r found (fields: %s)z, r4 z Multiple fields match %r: %sr ) rJ rT rC r� r� r' �keysr� r )r+ r� r rJ r r r! � __setitem__� s
zForm.__setitem__c C sP | j �|�}|dk s td| ��t|�dksHtd|d�tt|��f ��|d S )zE
Get the named field object (ambiguity is an error).
Nz No field by the name %r foundr4 z Multiple fields match %r: %sz, r )rJ rT rC r� r� r� r' )r+ r� rJ r r r! � __getitem__ s
zForm.__getitem__c C sp | j �| j�}x\| j�� D ]N\}}xD|D ]<}t|ttf�s(|jsNt d| ��q(|j|kr(t d| ��q(W qW dS )zsCheck that the html is valid:
- each field must have an id
- each field must have a label
z%r as no id attributez%r as no associated labelN)
� _label_re�findallrN rJ r r# rA r@ rI r� )r+ �labelsr� rJ rK r r r! r s
z Form.lintNc C sD |dkr|| |< n.| j �|�}|dk s2td| ��|| }||_dS )zF
Set the given name, using ``index`` to disambiguate.
NzNo fields found matching %r)rJ rT rC r )r+ r� r rb rJ rK r r r! �set0 s
zForm.setc C s` | j �|�}|dkr |tk r |S |dkr0| | S | j �|�}|dk sPtd| ��|| }|S dS )zb
Get the named/indexed field object, or ``default`` if no field
is found.
NzNo fields found matching %r)rJ rT r rC )r+ r� rb �defaultrJ rK r r r! rT = s
zForm.getc C s& | j ||d�}t|t�s t�||_dS )z]
Like ``.set()``, except also confirms the target is a
``<select>``.
)rb N)rT r# r' rC r )r+ r� r rb rK r r r! r0 N s z Form.selectc K sX | j ||d�}| �� }|r"||d< | jdkr:|�d| j� | jj| jf| j|d�|��S )a
Submits the form. If ``name`` is given, then also select that
button (using ``index`` to disambiguate)``.
Any extra keyword arguments are passed to the ``.get()`` or
``.post()`` method.
Returns a :class:`webtest.TestResponse` object.
)rb r� rM r� )r� r� )�
submit_fields�
upload_fieldsr� r� rN rD rg rL )r+ r� rb r- rJ �uploadsr r r! rC W s
z Form.submitc C sT g }xJ| j �� D ]<\}}x2|D ]*}t|t�r |jr |�|gt|j� � q W qW |S )z�
Return a list of file field tuples of the form:
(field name, file name)
or
(field name, file name, file contents).
)rJ r r# r< r rE r )r+ rY r� rJ rK r r r! rX j s
zForm.upload_fieldsc C s� g }|dk r.| j ||d�}|�|j|�� f� x�| j�� D ]r\}}|dkrLq:x^|D ]V}|j}|dkrfqRt|t�rrqRt|t �r�x*|D ]}|�||f� q�W qR|�||f� qRW q:W |S )zf
Return a list of ``[(name, value), ...]`` for the current
state of the form.
N)rb )
rT rE r� rB rJ r r r# r< r )r+ r� rb rC rK rJ r �itemr r r! rW x s$
zForm.submit_fieldsc C s$ d}| j r |dt| j � 7 }|d S )Nz<Formz id=%rz />)rI r� )r+ r r r r! r� � s z
Form.__repr__)N)N)NN)NN)r r r r� ro rp r| r? rR r rI r* rE rF rP rQ r rU r rT r0 rC rX rW r� r r r r! rH } s& D
rH c C s t | t�r| S t| �S )N)r# r r� )r r r r! r, � s
r, c C s || kr| � |�S |S )z4
Pop the key if found (else return default)
)rH )rx r rV r r r! �_popget� s
r[ Tc C s� |dkrt jj}| �|�} |�|�}g }xf| r�|r�| d |d kr�|dkrj|�dt|d �t|� � n|�d| � |�d� | �d� q*W |r�|r�d�|�| |�|� S d�|�|�|� S n
|�|�S dS )zg
Anything shared by pref and full will be replaced with spaces
in full, and full returned.
Nr r� rv )r rY �sepr� rE r� rH r� )ZprefZfullr\ �indentZ include_sepZpaddingr r r! �
_space_prefix� s
r^ c C sR | d kr d S t | t�r t�| �} t| d�r0| jS t| d�r>| S dsNtd| ��d S )Nr �__call__r z-Cannot make callable pattern object out of %r)r# r ro rp r� r rC )Zpatr r r! r{ � s
r{ c C s" x dD ]\}}| � ||�} qW | S )z8
Unquote (some) entities in HTML. (incomplete)
))z r� )z>r� )z<r� )z"�")z&r� )r� )r� Zent�replr r r! r � s r c C s� | t kr dS t| t�s t| d�r,t| �� �} t| ttf�r�|r�|�� }d|kr�|�d�d }|� d��� }g }x2| D ]*\}}t|t
�r�|� |�}|� ||f� qtW |} t
| dd�} | S )Nrv r zcharset=r4 z; T)r� )r r# �dictr� r r r� rB r� r� r r� rE r )r� r� r% Zencoded_params�kr� r r r! r� � s$
r� )N)NNT)Br� r� r� r� r� r ro r Zwebtest.compatr r r r r r r r r
r r r
r r r r r r r r Zwebobr r Zwebtestr r �__all__�objectr r) r" r/ r r rp r} r r~ r r& r' r% r1 r7 r9 r; r< r> r@ rA rH r, r[ r^ r{ r r� r r r r! �<module> s� 6 [
//
1
#