a 촫j'@sddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z ddlZddlmZdd lmZdd lmZdd lmZd d lmZd dlmZejrddlmZddlmZd dlm Z Gdddej!j"Z"da#ddddZ$GdddeZ%GdddeZ&dS)) annotationsN)contextmanager) ExitStack)copy) TracebackType)urlsplit) CliRunner)Result)Client)Request) ScriptInfo) SessionMixin)WSGIEnvironment) TestResponse)Flaskc sHeZdZdZdddddddddd fd d Zdddd d dZZS)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`. /Nrstrz str | Nonet.AnyNone)apppathbase_url subdomain url_schemeargskwargsreturnc s|s |s |r(|dut|p|ks(Jd|dur|jdp>d}|jd} |r\|d|}|durn|jd}t|} | jp~|d| jp|d| d}| j}| jr|d | j}||_ t j ||g|Ri|dS) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".Z SERVER_NAME localhostZAPPLICATION_ROOT.ZPREFERRED_URL_SCHEMEz://r?) boolconfiggetrschemenetloclstriprqueryrsuper__init__) selfrrrrrrrZ http_hostZapp_rooturl __class__L/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/flask/testing.pyr*1s.    zEnvironBuilder.__init__)objrrcKs|jjj|fi|S)zSerialize ``obj`` to a JSON-formatted string. The serialization will be configured according to the config associated with this EnvironBuilder's ``app``. )rjsondumps)r+r1rr/r/r0 json_dumpsXszEnvironBuilder.json_dumps)rNNN)__name__ __module__ __qualname____doc__r*r4 __classcell__r/r/r-r0rs"'rrrcCststjdatS)Nwerkzeug)_werkzeug_version importlibmetadataversionr/r/r/r0_get_werkzeug_versionds rAcseZdZUdZded<ddddfdd Zeddd dd d Zd d d ddZddddddZ dddddddddfddZ ddddZ dd d!dd"d#d$Z Z S)% FlaskClientaWorks like a regular Werkzeug test client but has knowledge about Flask's contexts to defer the cleanup of the request context until the end of a ``with`` block. 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 :doc:`/testing` chapter. r applicationrr)rrrcs>tj|i|d|_g|_t|_ddtd|_dS)NFz 127.0.0.1z Werkzeug/)Z REMOTE_ADDRZHTTP_USER_AGENT)r)r*preserve_context _new_contextsr_context_stackrA environ_base)r+rrr-r/r0r*}s zFlaskClient.__init__zt.Iterator[SessionMixin]cos|jdurtd|j}|j|i|}||jj| |j||j}Wdn1s`0Y|durzt d|V| }|j |rdS| |j |||Wdn1s0Y| |jjdd|jj|jddS)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>Cookies are disabled. Create a client with 'use_cookies=True'.z'Session backend did not open a session.:rz Set-Cookie)_cookies TypeErrorrCZtest_request_contextZ_add_cookies_to_wsgirequestenvironZsession_interfaceZ open_session RuntimeErrorresponse_classZis_null_sessionZ save_sessionZ_update_cookies_from_responsehost partitionrheadersgetlist)r+rrrctxsessrespr/r/r0session_transactions, . . zFlaskClient.session_transactionr)otherrcCs$i|j|}|jr |jj|d<|S)Nzwerkzeug.debug.preserve_context)rGrDrEappend)r+rWoutr/r/r0 _copy_environs zFlaskClient._copy_environztuple[t.Any, ...]zdict[str, t.Any] BaseRequestcCsP||di|d<t|jg|Ri|}z|W|S|0dS)NrG)rZr$rrC get_requestclose)r+rrbuilderr/r/r0_request_from_builder_argssz&FlaskClient._request_from_builder_argsFbufferedfollow_redirectsr"r)rrarbrrc s|rt|dtjjttfrt|dtjjrVt|d}||jpFi|_| }qt|dtrtj |d|j |id }qt|d}||j |_ n | ||}|jtj|||d}|j j|_|jD]}|j|q|j|S)Nr)rrGr`) isinstancer<testrdictr[rrZrGr\Z from_environrCrLr_rFr]r)openr2Z json_modulerE enter_contextclear) r+rarbrrr^rKresponsecmr-r/r0rfs2         zFlaskClient.openr;cCs|jrtdd|_|S)NzCannot nest client invocationsT)rDrMr+r/r/r0 __enter__szFlaskClient.__enter__z type | NonezBaseException | NonezTracebackType | None)exc_type exc_valuetbrcCsd|_|jdS)NF)rDrFr])r+rmrnror/r/r0__exit__szFlaskClient.__exit__)r5r6r7r8__annotations__r*rrVrZr_rfrlrpr9r/r/r-r0rBms   1-rBcsDeZdZdZddddfdd Zd dddd d fd d 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`. rrr)rrrc s||_tjfi|dSN)rr)r*)r+rrr-r/r0r*szFlaskCliRunner.__init__Nr )clirrrc sD|durjj}d|vr.tfddd|d<tj||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. Nr1csjSrs)rr/rkr/r0(z'FlaskCliRunner.invoke..)Z create_app)rrtr r)invoke)r+rtrrr-rkr0rws zFlaskCliRunner.invoke)NN)r5r6r7r8r*rwr9r/r/r-r0rr srr)' __future__rimportlib.metadatar>typingt contextlibrrrtypesr urllib.parserZ werkzeug.testr<Z click.testingrr r Zwerkzeug.wrappersr r[rtr sessionsr TYPE_CHECKINGZ_typeshed.wsgirrrrrdrr=rArBrrr/r/r/r0s0               F