2ec@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZd Zd efd YZdZ defdYZ!de"fdYZ#defdYZ$defdYZ%defdYZ&defdYZ'defdYZ(e j)de j*Z+dZ,defd YZ-d!e-fd"YZ.e.e-j/d#efd?YZ9e:d@Z;e:e:e<dAZ=dBZ>e:dCZ?dDZ@dS(Es Routines for testing WSGI applications. Most interesting is the `TestApp `_ for testing WSGI applications, and the `TestFileEnvironment `_ class for testing the effects of command-line scripts. i(tprint_functionN(t cStringIO(t urlencode(tparse(t BaseCookie(twsgilib(tlint(t HeaderDictcGs tj|S(s An os.tempnam with the warning turned off, because sometimes you just need to use this and don't care about the stupid security warning. (tosttempnam(targs((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyttempnam_no_warning"st NoDefaultcBseZRS((t__name__t __module__(((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR *scCst|}|j|S(N(tlisttsort(tl((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytsorted-s  t Dummy_smtplibcBsGeZdZdZdZdZdZeeZdZ RS(cCsWddl}|jdtd|j s5td||_t|_||j_dS(Nis1Dummy_smtplib is not maintained and is deprecatedisIsmtplib.SMTP() called again before Dummy_smtplib.existing.reset() called.( twarningstwarntDeprecationWarningtexistingtAssertionErrortservertTruetopent __class__(tselfRR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt__init__6s      cCs&|jstd|t|_dS(NsCalled %s.quit() twice(RRtFalse(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytquitBs  cCs||_||_||_dS(N(t from_addresst to_addressestmessage(RR!R"tmsg((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytsendmailGs  cCs |t_dS(N(tsmtplibtSMTP(tcls((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytinstallLscCs&|j stdd|j_dS(NsSMTP connection not quit(RRtNoneRR(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytresetQs  N( R RR*RRR R%R)t classmethodR+(((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR2s    tAppErrorcBseZRS((R R(((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR-WstTestAppcBs eZeZddddddZdZdZddddedZdddddedZ dddddedZ dddddedZ ddddedZ ddded Z d Zd Zd Zd ZdZdZdZRS(cCst|tjtjfr@ddlm}||d|}n||_||_||_|dkrpi}n||_ ||_ ||_ |j dS(s} 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'``. ``namespace`` is a dictionary that will be written to (if provided). This can be used with doctest or some other system, and the variable ``res`` will be assigned everytime you make a request (instead of returning the request). ``relative_to`` is a directory, and filenames used for file uploads are calculated relative to this. Also ``config:`` URIs that aren't absolute. ``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. ``pre_request_hook`` is a function to be called prior to making requests (such as ``post`` or ``get``). This function must take one argument (the instance of the TestApp). ``post_request_hook`` is a function, similar to ``pre_request_hook``, to be called after requests are made. i(tloadappt relative_toN(t isinstancetsixt binary_typet text_typet paste.deployR/tappt namespaceR0R*t extra_environtpre_request_hooktpost_request_hookR+(RR6R7R0R8R9R:R/((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR_s        cCs i|_dS(sc Resets the state of the application; currently just clears saved cookies. N(tcookies(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR+scCs|jj}t|d<|S(Nspaste.throw_errors(R8tcopyR(Rtenviron((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt _make_environs c Cs|dkri}nt}|rt|tjtjfsQt|dt}nd|krj|d7}n |d7}||7}n|j}t|}d|kr|j dd\}|d`_ tdoseqt?t&it QUERY_STRINGttstatusN(R*RR1R2R3R4RR>tstrtsplitt _set_headerstupdatet TestRequestt do_request( RturltparamstheadersR8RDt expect_errorst__tracebackhide__R=treq((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytgets&           RCc Cs|dkri}n|dkr*i}n|j} t|tttfr]t|}nt|drt|j}nt j rt|t j r|j d}n|rt j|dt}|j||\} }| | d|jdd\}| dR1RttupletdictRthasattrRRR2tPY3R4tencodeturlparset parse_qslRtencode_multipartt setdefaultRERFtlentBytesIORGRHRIRJ( RtmethodRKRLRMR8RDt upload_filesRNR=t content_typeRP((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt _gen_requests:           cCs4|jd|d|d|d|d|d|d|S(s 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 `response object `_ tPOSTRLRMR8RDRdRN(Rf(RRKRLRMR8RDRdRN((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytposts cCs4|jd|d|d|d|d|d|d|S(s Do a PUT 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 `response object `_ tPUTRLRMR8RDRdRN(Rf(RRKRLRMR8RDRdRN((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytputs cCs4|jd|d|d|d|d|ddd|S( s Do a DELETE request. Very like the ``.get()`` method. ``params`` are put in the body of the request. Returns a `response object `_ tDELETERLRMR8RDRdRNN(RfR*(RRKRLRMR8RDRN((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytdeletes  c Cs.|jd|d|d|d|ddd|S(s Do a HEAD request. Very like the ``.get()`` method. Returns a `response object `_ tHEADRMR8RDRdRNN(RfR*(RRKRMR8RDRN((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pythead's cCs|s dSxv|jD]h\}}|jdkr>d}n7|jdkrYd}nd|jddj}|||\}}t||rtd|nt|||qW|jdk r||jd RQRfRhRjRlRnRGR_R|RJRRR(((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR.[s2 +   5 %      $  <  RcBs5eZdZdZdZdZdZRS(cCst|_||_dS(N(tStringIOtcapturedtactual(RR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRs cCs$|jj||jj|dS(N(RRR(Rts((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRscCs|jjdS(N(Rtflush(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRscCs"x|D]}|j|qWdS(N(R(RRtitem((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt writeliness cCs |jjS(N(RR(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRs(R RRRRRR(((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRs     RcBs2eZeZdZdZeeddZdZeeddZ e j de j e j BZdZedZd Zd Zddddded Zddddded Zd ZddZe j dZdZeeddZdZdZdZdZdZ RS(cCsz||_t|jd|_||_||_tj|j|_||_ ||_ d|_ ||_ d|_dS(Ni(ttest_apptintRFRDRRMRtfromlistt header_dictRRR*t _normal_bodyRt_forms_indexed(RRRDRMRRR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRs       cCs#|jdkr|jn|jS(s Returns a dictionary of ``Form`` objects. Indexes are both in order (from zero) and by form id (if the form is given an id). N(RR*t _parse_forms(R((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt forms__gets tdocs A list of
s found on the page (instances of `Form `_) cCsA|j}|stdnd|kr9tdn|dS(Ns*You used response.form, but no forms existis5You used response.form, but more than one form existsi(tformst TypeError(RR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyt form__gets    s Returns a single `Form `_ instance; it is an error if there are multiple forms on the page. s<(/?)([:a-z0-9_\-]*)(.*?)>c Csui}|_g}d}|j}tjs@|jdd}nx|jj|D]}|jddk}|jdj }|dkrqSn|r|st d|j |j |||j !d}qS| st d|j |j }qSW| s"t d ||xLt|D]>\}} t|| } | ||<| jr/| || j unexpected at %ssNested form tags at %ssDangling form: %r(RR*RR2tPY2Rt_tag_retfinditertgroupRqRtstartRztendt enumeratetFormtid( RRt form_textststartedRtmatchRttagtittextR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR*s4          cCsd}xX|jD]M\}}|j|jkr| sTtd|||f|}qqW|dkr|tkrtd|djg|jD]\}}|^qfq|Sn|S(s Returns the named header; an error if there is not exactly one matching header (unless you give a default -- always an error if there is more than one header) s&Ambiguous header: %s matches %r and %rsNo header found: %r (from %s)s, N(R*RMRqRR tKeyErrorR(RRtdefaulttfoundtcur_nameRutntv((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyRtGs    8cCsLg}x?|jD]4\}}|j|jkr|j|qqW|S(sE Gets all headers by the ``name``, returns as a list (RMRqRz(RRRRRu((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pyR]s cKsS|jdkr|jdks1td|j|jd}|jj||S(s 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,is/You can only follow redirect responses (not %s)tlocation(RDRRRtRRQ(RtkwR((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytfollowgs ! c Csbt}|jdddddd d|d|d|d |d |d | \}} } |j| d S(s 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 ```` and ````) * ``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. Rtat href_attrthreft href_extractRRt href_patternt html_patterntindextverboseturiN(Rt _find_elementR*tgoto( Rt descriptiontlinkidRtanchorRRROt found_htmlt found_desct found_attrs((s7/usr/local/lib/python2.7/dist-packages/paste/fixture.pytclickts'  c Cskt}|jdddddtjdd|d|d |d |d |d | \}} } |j| d S(s Like ``.click()``, except looks for link-like buttons. This kind of button should look like ``