B ;=[$@sdZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZdd d ZGd dde ZGdddeZdS)u flask.testing ~~~~~~~~~~~~~ Implements test support helpers. This module is lazily imported and usually not used in production environments. :copyright: © 2010 by the Pallets team. :license: BSD, see LICENSE for more details. N)contextmanager) CliRunner) ScriptInfo)ClientEnvironBuilder)_request_ctx_stack)dumps) url_parse/c Os(|s |s |r(|dk t|p|ks(td|dkr|jdp>d}|jd}|rZd||}|dkrl|jd}t|} dj| jp|| jp||d d }| j }| j rt | j t rd nd } || | j 7}d |krd|kstd| t|d |d<WdQRXd|krd|d<t||f||S)aCreate a :class:`~werkzeug.test.EnvironBuilder`, taking some 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`. Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".Z SERVER_NAMEZ localhostZAPPLICATION_ROOTz{0}.{1}ZPREFERRED_URL_SCHEMEz{scheme}://{netloc}/{path}r )schemenetlocpath??Zjsondataz-Client cannot provide both 'json' and 'data'.Z content_typezapplication/json)boolAssertionErrorZconfiggetformatr r r lstripr Zquery isinstancebytesZ app_context json_dumpspopr) appr Zbase_urlZ subdomainZ url_schemeargskwargsZ http_hostZapp_rootZurlsepr,lib/python3.7/site-packages/flask/testing.pymake_test_environ_builders6        r csHeZdZdZdZfddZeddZddZd d Z d d Z 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. Fcs(tt|j||ddtjd|_dS)Nz 127.0.0.1z werkzeug/)Z REMOTE_ADDRZHTTP_USER_AGENT)superr!__init__werkzeug __version__ environ_base)selfrr) __class__rrr#lszFlaskClient.__init__c os|jdkrtd|j}|di}|j|tj}|j||}|j}| ||j }|dkrhtdt |z |VWdt X| } ||s|||| | |j j} |j|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 application setdefaultZ inject_wsgirtopZtest_request_contextsession_interfaceZ open_sessionZrequestpushrZresponse_classZis_null_sessionZ save_sessionZget_wsgi_headersenvironZ extract_wsgi) r'rrrr)Z outer_reqctxcr.ZsessZrespZheadersrrrsession_transactionss(       zFlaskClient.session_transactioncOs|dd}|dd}|dd}|st|dkrt|dttfr|j}t|dtrr||dn||d|j |d<nJ|j | did<| d |jt |j f||}z |}Wd| Xtj|||||d S) Nas_tupleFbufferedfollow_redirectsrzflask._preserve_contextr)r&)r3r4r5)rlenrrdictr&copyupdateZ get_environpreserve_contextr,r r+closeropen)r'rrr3r4r5r0Zbuilderrrrr=s,       zFlaskClient.opencCs|jrtdd|_|S)NzCannot nest client invocationsT)r;r*)r'rrr __enter__szFlaskClient.__enter__cCs&d|_tj}|dk r"|jr"|dS)NF)r;rr-Z preservedr)r'exc_type exc_valuetbr-rrr__exit__szFlaskClient.__exit__) __name__ __module__ __qualname____doc__r;r#rr2r=r>rB __classcell__rr)r(rr![s   3%r!cs.eZdZdZfddZdfdd ZZS)FlaskCliRunnerzA :class:`~click.testing.CliRunner` for testing a Flask app's CLI commands. Typically created using :meth:`~flask.Flask.test_cli_runner`. See :ref:`testing-cli`. c s||_tt|jf|dS)N)rr"rHr#)r'rr)r(rrr#szFlaskCliRunner.__init__Nc sD|dkrjj}d|kr.tfddd|d<ttj||f|S)acInvokes a CLI command in an isolated environment. See :meth:`CliRunner.invoke ` 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. NobjcsjS)N)rr)r'rrsz'FlaskCliRunner.invoke..)Z create_app)rclirr"rHinvoke)r'rKrr)r()r'rrLs zFlaskCliRunner.invoke)NN)rCrDrErFr#rLrGrr)r(rrHs rH)r NNN)rFr$ contextlibrZ click.testingrZ flask.clirZ werkzeug.testrrZflaskrZ flask.jsonrrZ werkzeug.urlsr r r!rHrrrr s       B