ó `¾Tc@sdZddlmZddlmZmZddlmZyddlm Z Wn!e k rsddl m Z nXdd d„Zd efd „ƒYZd S( sþ flask.testing ~~~~~~~~~~~~~ Implements test support helpers. This module is lazily imported and usually not used in production environments. :copyright: (c) 2011 by Armin Ronacher. :license: BSD, see LICENSE for more details. iÿÿÿÿ(tcontextmanager(tClienttEnvironBuilder(t_request_ctx_stack(t url_parse(turlsplitt/cOsœ|jjdƒ}|jjdƒ}|dkr‰t|ƒ}d|jpQ|pQd}|rq||jdƒ7}n|jr‰|j}q‰nt||||ŽS(sDCreates a new test builder with some application defaults thrown in.t SERVER_NAMEtAPPLICATION_ROOTs http://%s/t localhostRN(tconfigtgettNoneRtnetloctlstriptpathR(tappRtbase_urltargstkwargst http_hosttapp_rootturl((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pytmake_test_environ_builders   t FlaskClientcBs>eZdZeZed„ƒZd„Zd„Zd„Z RS(skWorks like a regular Werkzeug test client but has some knowledge about how Flask works to defer the cleanup of the request context stack to the end of a with body when used in a with statement. For general information about how to use this class refer to :class:`werkzeug.test.Client`. Basic usage is outlined in the :ref:`testing` chapter. c os!|jdkrtdƒ‚n|j}|jdiƒ}|jj|ƒtj}|j||Ž·}|j |j ƒ}|dkr”tdƒ‚ntj |ƒz |VWdtj ƒX|j ƒ}|jj|ƒsé|j||ƒn|j|j jƒ} |jj|j j| ƒWdQXdS(s~When used in combination with a with statement this opens a session transaction. This can be used to modify the session that the test client uses. Once the with block is left the session is stored back. with client.session_transaction() as session: session['value'] = 42 Internally this is implemented by going through a temporary test request context and since session handling could depend on request variables this function accepts the same arguments as :meth:`~flask.Flask.test_request_context` which are directly passed through. s:Session transactions only make sense with cookies enabled.tenviron_overridess?Session backend did not open a session. Check the configurationN(t cookie_jarR t RuntimeErrort applicationt setdefaultt inject_wsgiRttopttest_request_contextt open_sessiontrequesttpushtpoptresponse_classtsession_interfacetis_null_sessiont save_sessiontget_wsgi_headerstenviront extract_wsgi( tselfRRRRt outer_reqctxtctsesstresptheaders((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pytsession_transaction0s&       c Os†|j|jdiƒd<|jdtƒ}|jdtƒ}|jdtƒ}t|j||Ž}tj||d|d|d|ƒS(NRsflask._preserve_contexttas_tupletbufferedtfollow_redirects(tpreserve_contextRR$tFalseRRRtopen(R,RRR3R4R5tbuilder((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pyR8`scCs%|jrtdƒ‚nt|_|S(NsCannot nest client invocations(R6RtTrue(R,((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pyt __enter__ns  cCs8t|_tj}|dk r4|jr4|jƒndS(N(R7R6RRR t preservedR$(R,texc_typet exc_valuettbR((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pyt__exit__ts  ( t__name__t __module__t__doc__R7R6RR2R8R;R@(((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pyR%s 0  N(RCt contextlibRt werkzeug.testRRtflaskRt werkzeug.urlsRt ImportErrorturlparseRR RR(((sS/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/flask/testing.pyt s