o i'@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)dumpscs2eZdZdZ    dfdd ZddZZS) 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 s|s|s|r|dut|p |ksJd|durb|jdpd}|jd} |r-d||}|dur6|jd}t|} dj| jp@|| jpD|| dd }| j}| j rbt | j t rYd nd } || | j 7}||_ t t|j||g|Ri|dS) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url". SERVER_NAME localhostAPPLICATION_ROOTz{0}.{1}PREFERRED_URL_SCHEMEz{scheme}://{netloc}/{path}r )schemenetlocpath??)boolconfiggetformatrrrlstriprquery isinstancebytesappsuperr __init__) selfrrbase_url subdomain url_schemeargskwargs http_hostapp_rooturlsep __class__H/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/flask/testing.pyr/s6    $zEnvironBuilder.__init__cKs|d|jt|fi|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)r objr%r,r,r-r/XszEnvironBuilder.json_dumps)r NNN)__name__ __module__ __qualname____doc__rr/ __classcell__r,r,r*r-r s)r cOsttdt|i|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 r9csHeZdZdZdZfddZeddZddZd 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|i|ddtjd|_dS)Nz 127.0.0.1z werkzeug/) REMOTE_ADDRHTTP_USER_AGENT)rr:rwerkzeug __version__ environ_base)r r$r%r*r,r-rs zFlaskClient.__init__c os|jdur td|j}|di}|j|tj}|j|i|O}|j}| ||j }|dur7tdt |z |VWt nt w| } ||s[|||| | |j j} |j|j j| WddS1svwYdS)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) cookie_jar RuntimeError applicationr. inject_wsgirtoptest_request_contextsession_interface open_sessionrequestpushpopresponse_classis_null_session save_sessionget_wsgi_headersenviron extract_wsgi) r r$r%rr@ outer_reqctxcrGsessrespheadersr,r,r-session_transactions4     "zFlaskClient.session_transactioncOs|dd}|dd}|dd}|sJt|dkrJt|dtjjtfrJ|j}t|dtjjr=| |d n| |d|j |d<n,|j | did<| d |jt|j g|Ri|}z | }W|n|wtj|||||d S) Nas_tupleFbufferedfollow_redirectsrrzflask._preserve_contextr@r?)rXrYrZ)rKlenrr=testr dictr?copyupdate get_environpreserve_contextr.rCcloseropen)r r$r%rXrYrZrPbuilderr,r,r-rcs:        zFlaskClient.opencCs|jrtdd|_|S)NzCannot nest client invocationsT)rarBr r,r,r- __enter__szFlaskClient.__enter__cCs,d|_ tj}|dur|jr|ndSq)NF)rarrE preservedrK)r exc_type exc_valuetbrEr,r,r-__exit__s zFlaskClient.__exit__) r1r2r3r4rarrrWrcrfrkr5r,r,r*r-r:ss  4&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|jdi|dS)Nr,)rrrlr)r rr%r*r,r-rszFlaskCliRunner.__init__Nc sH|durjj}d|vrtfddd|d<ttj||fi|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. Nr0csjS)N)rr,rer,r-sz'FlaskCliRunner.invoke..) create_app)rclirrrlinvoke)r ror$r%r*rer-rps zFlaskCliRunner.invoke)NN)r1r2r3r4rrpr5r,r,r*r-rls rl)r4r6 contextlibr werkzeug.testr= click.testingrr werkzeug.urlsrrrorjsonr r/r\r r9r:rlr,r,r,r-s       I