h!'<ddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z ddlZddlmZdd lmZdd lmZd d lmZd d lmZejr ddlmZd dlmZGddejjZda ddZ!GddeZ"GddeZ#dS)) annotationsN)contextmanager) ExitStack)copy) TracebackType)urlsplit) CliRunner)Client)Request) ScriptInfo) SessionMixin) TestResponse)Flaskc6eZdZdZ ddfd ZddZxZS)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`. /Napprpathstrbase_url str | None subdomain url_schemeargst.AnykwargsreturnNonec |s|s|r!|dut|p|ks Jd||jdpd}|jd} |r|d|}| |jd}t|} | jp|d| jp|d| d}| j}| jr+t| jtrd nd } || | jzz }||_ tj ||g|Ri|dS) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url". SERVER_NAME localhostAPPLICATION_ROOT.PREFERRED_URL_SCHEMEz://r??)boolconfiggetrschemenetloclstriprquery isinstancebytesrsuper__init__) selfrrrrrrr http_hostapp_rooturlsep __class__s U/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/flask/testing.pyr2zEnvironBuilder.__init__/s} F  FZ F D   #  = = =  F = = =   }55DIz"45H 7(66966 ! Z(>? 4..C:+++ 0Gi++OOC((++ 8Dy ((E::Cddci'x9$999&99999objc 2|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)r3r;rs r9 json_dumpszEnvironBuilder.json_dumpsWs# #tx}"311&111r:)rNNN)rrrrrrrrrrrrrrrr)r;rrrrr)__name__ __module__ __qualname____doc__r2r? __classcell__r8s@r9rrso0# $!% &:&:&:&:&:&:&:P22222222r:rrrc\tstjdatS)Nwerkzeug)_werkzeug_version importlibmetadataversionr:r9_get_werkzeug_versionrNcs* C%.66zBB r:cxeZdZUdZded<dfd Zedd Zd Zd Z ddddfdZ d dZ d!dZ xZ 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 applicationrrrrrctj|i|d|_g|_t |_ddt d|_dS)NFz 127.0.0.1z Werkzeug/) REMOTE_ADDRHTTP_USER_AGENT)r1r2preserve_context _new_contextsr_context_stackrN environ_base)r3rrr8s r9r2zFlaskClient.__init__|se$)&))) %<>'kk&D+@+B+BDD  r:%t.Generator[SessionMixin, None, None]c/K|jtd|j}|j|i|}||jj|5|j||j}dddn #1swxYwY|td|V| }|j |rdS|5|j |||dddn #1swxYwY| |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 TypeErrorrQtest_request_context_add_cookies_to_wsgirequestenvironsession_interface open_session RuntimeErrorresponse_classis_null_session save_session_update_cookies_from_responsehost partitionrheadersgetlist)r3rrrctxsessresps r9session_transactionzFlaskClient.session_transactions( = P &c&777 !!#+"5666  H H(55c3;GGD H H H H H H H H H H H H H H H <HII I !!##  0 0 6 6  F  @ @  ! . .sD$ ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ ** K  & &s + +A . K  L  . .     s$!A;;A?A? C66C:=C:cHi|j|}|jr|jj|d<|S)Nzwerkzeug.debug.preserve_context)rXrUrVappend)r3otherouts r9 _copy_environzFlaskClient._copy_environs6,",e,   O595G5NC1 2 r:c||di|d<t|jg|Ri|} ||S#|wxYw)NrX)rur*rrQ get_requestclose)r3rrbuilders r9_request_from_builder_argsz&FlaskClient._request_from_builder_argss}!%!3!3FJJ~r4R4R!S!S~ !1CDCCCFCC &&(( MMOOOOGMMOOOOs A++BFbufferedfollow_redirectsr|r(r}rc|rAt|dtjjtt frt|dtjjrKt |d}||jpi|_| }nt|dtrNt |d|j |i }nKt |d}||j |_ n| ||}|jt!|||}|j j|_|jr:|j}|j||j:|S)Nr)rrXr{)r/rHtestrdict BaseRequestrrurXrw from_environrQrarzrWrxr1openr= json_modulerVpop enter_context) r3r|r}rrryr`responsecmr8s r9rzFlaskClient.opens  DJ Ghm2D+F   D$q'8=#?@@ FtAw--'+'9'9':N:TRT'U'U$!--//DGT** F(55G!1@R@RSU@V@V6+-- tAw--"&"4"4W_"E"E55dFCCG !!###77<< -    $/4  2#''))B   - -b 1 1 1  2r:c@|jrtdd|_|S)NzCannot nest client invocationsT)rUrdr3s r9 __enter__zFlaskClient.__enter__s*   A?@@ @ $ r:exc_type type | None exc_valueBaseException | NonetbTracebackType | NonecFd|_|jdS)NF)rUrWrx)r3rrrs r9__exit__zFlaskClient.__exit__s' !& !!#####r:)rrrrrr)rrrrrrY) rrr|r(r}r(rrrr)rrP)rrrrrrrr)r@rArBrC__annotations__r2rrprurzrrrrDrEs@r9rPrPls        / / / ^/ b!& ++++++++Z $$$$$$$$r:rPc4eZdZdZd fd Z ddfd ZxZS)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`. rrrrrrc H||_tjdi|dS)NrM)rr1r2)r3rrr8s r9r2zFlaskCliRunner.__init__ s-""6"""""r:Nclirc | jj}d|vrtfd|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. Nr;cjS)N)rrsr9z'FlaskCliRunner.invoke..%s $(r:) create_app)rrr r1invoke)r3rrrr8s` r9rzFlaskCliRunner.invokes]" ;(,C   &2B2B2B2BCCCF5Muww~c4226222r:)rrrrrr)NN)rrrrrrrr)r@rArBrCr2rrDrEs@r9rrso ###### 0433333333333r:r)rr)$ __future__rimportlib.metadatarJtypingt contextlibrrrtypesr urllib.parser werkzeug.testrH click.testingr r werkzeug.wrappersr rrr sessionsr TYPE_CHECKINGrrrrrrIrNrPrrMr:r9rs""""""%%%%%% !!!!!!###### 444444""""""?******D2D2D2D2D2X]1D2D2D2NW$W$W$W$W$&W$W$W$t!3!3!3!3!3Y!3!3!3!3!3r: