_'@sdZddlZddlmZddlZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZGd d d ejjZd dZGdddeZGdddeZdS)z flask.testing ~~~~~~~~~~~~~ Implements test support helpers. This module is lazily imported and usually not used in production environments. :copyright: 2010 Pallets :license: BSD-3-Clause N)contextmanager) CliRunner)Client) url_parse)_request_ctx_stack) ScriptInfo)dumpscs@eZdZdZddddfddZddZS)EnvironBuilderaAn :class:`~werkzeug.test.EnvironBuilder`, that takes defaults from the application. :param app: The Flask application to configure the environment from. :param path: URL path being requested. :param base_url: Base URL where the app is being served, which ``path`` is relative to. If not given, built from :data:`PREFERRED_URL_SCHEME`, ``subdomain``, :data:`SERVER_NAME`, and :data:`APPLICATION_ROOT`. :param subdomain: Subdomain name to append to :data:`SERVER_NAME`. :param url_scheme: Scheme to use instead of :data:`PREFERRED_URL_SCHEME`. :param json: If given, this is serialized as JSON and passed as ``data``. Also defaults ``content_type`` to ``application/json``. :param args: other positional arguments passed to :class:`~werkzeug.test.EnvironBuilder`. :param kwargs: other keyword arguments passed to :class:`~werkzeug.test.EnvironBuilder`. /Nc sQ|p|p| s=|dk t|p(|ks=td|dkr%|jjdp^d}|jd} |rdj||}|dkr|jd}t|} djd| jp|d | jp|d | jd }| j }| j r%t | j t rd nd } || | j 7}||_ tt|j||||dS)Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".Z SERVER_NAME localhostZAPPLICATION_ROOTz{0}.{1}ZPREFERRED_URL_SCHEMEz{scheme}://{netloc}/{path}schemenetlocpathr ??)boolAssertionErrorconfiggetformatrr rlstriprquery isinstancebytesappsuperr __init__) selfrrbase_urlZ subdomainZ url_schemeargskwargsZ http_hostZapp_rooturlsep) __class__./tmp/pip-build-5gj8f0j9/flask/flask/testing.pyr/s.            zEnvironBuilder.__init__cKs |jd|jt||S)zSerialize ``obj`` to a JSON-formatted string. The serialization will be configured according to the config associated with this EnvironBuilder's ``app``. r) setdefaultr json_dumps)robjr!r%r%r&r(XszEnvironBuilder.json_dumps)__name__ __module__ __qualname____doc__rr(r%r%)r$r&r s #r cOs tjtdt||S)zCreate a :class:`flask.testing.EnvironBuilder`. .. deprecated: 1.1 Will be removed in 2.0. Construct ``flask.testing.EnvironBuilder`` directly instead. z"make_test_environ_builder()" is deprecated and will be removed in 2.0. Construct "flask.testing.EnvironBuilder" directly instead.)warningswarnDeprecationWarningr )r r!r%r%r&make_test_environ_builderbs r1csdeZdZdZdZfddZeddZddZd d Z d d Z S) FlaskClientaDWorks 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`. .. versionchanged:: 0.12 `app.test_client()` includes preset default environment, which can be set after instantiation of the `app.test_client()` object in `client.environ_base`. Basic usage is outlined in the :ref:`testing` chapter. Fcs9tt|j||ddddtji|_dS)NZ REMOTE_ADDRz 127.0.0.1ZHTTP_USER_AGENTz werkzeug/)rr2rwerkzeug __version__ environ_base)rr r!)r$r%r&rszFlaskClient.__init__c os%|jdkrtd|j}|jdi}|jj|tj}|j||}|j}|j ||j }|dkrtdtj |z |VWdtj X|j } |j|s|j||| | j|j j} |jj|j j| WdQRXdS)aWhen 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. Nz:Session transactions only make sense with cookies enabled.environ_overridesz?Session backend did not open a session. Check the configuration)Z cookie_jar RuntimeError applicationr'Z inject_wsgirtopZtest_request_contextsession_interfaceZ open_sessionrequestpushpopresponse_classZis_null_sessionZ save_sessionZget_wsgi_headersenvironZ extract_wsgi) rr r!rr6Z outer_reqctxcr:sessrespheadersr%r%r&session_transactions,         zFlaskClient.session_transactioncOsR|jdd}|jdd}|jdd}| rt|dkrt|dtjjtfr|jj}t|dtjjr|j |dj n|j |d|j |d` for full method documentation. See :ref:`testing-cli` for examples. If the ``obj`` argument is not given, passes an instance of :class:`~flask.cli.ScriptInfo` that knows how to load the Flask app being tested. :param cli: Command object to invoke. Default is the app's :attr:`~flask.app.Flask.cli` group. :param args: List of strings to invoke the command with. :return: a :class:`~click.testing.Result` object. Nr)Z create_appcsjS)N)rr%)rr%r&sz'FlaskCliRunner.invoke..)rclirrrUinvoke)rrWr r!)r$)rr&rXs    zFlaskCliRunner.invoke)r*r+r,r-rrXr%r%)r$r&rUs rU)r-r. contextlibrZ werkzeug.testr3Z click.testingrrZ werkzeug.urlsrrrWrjsonr r(rIr r1r2rUr%r%r%r& s  I