o q>e!'@sddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z ddlZddlmZdd lmZdd lmZd d lmZd d lmZejr]ddlmZd dlmZGdddejjZda dddZ!GdddeZ"GdddeZ#dS)) annotationsN)contextmanager) ExitStack)copy) TracebackType)urlsplit) CliRunner)Client)Request) ScriptInfo) SessionMixin) TestResponse)Flaskcs6eZdZdZ    ddfdd ZdddZZS)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|s|r|dut|p |ksJd|dure|jdpd}|jd} |r.|d|}|dur7|jd}t|} | jp?|d| jpE|d| d}| j}| jret | jt r\d nd } || | j7}||_ t j ||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__N/var/www/bmteknikk.ddns.net/venv/lib/python3.10/site-packages/flask/testing.pyr//s4    zEnvironBuilder.__init__objcKs|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)r0r9rr7r7r8 json_dumpsWszEnvironBuilder.json_dumps)rNNN)rrrrrrrrrrrrrrrr)r9rrrrr)__name__ __module__ __qualname____doc__r/r< __classcell__r7r7r5r8rs(rrrcCststjdatS)Nwerkzeug)_werkzeug_version importlibmetadataversionr7r7r7r8_get_werkzeug_versioncs rHcsteZdZUdZded<d$fd d Zed%d d ZddZddZ dddd&fddZ d'ddZ d(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 applicationrrrrrcs>tj|i|d|_g|_t|_ddtd|_dS)NFz 127.0.0.1z Werkzeug/) REMOTE_ADDRHTTP_USER_AGENT)r.r/preserve_context _new_contextsr_context_stackrH environ_base)r0rrr5r7r8r/|s  zFlaskClient.__init__%t.Generator[SessionMixin, None, None]cos|jdur td|j}|j|i|}||jj||j||j}Wdn1s1wY|dur>t d|V| }|j |rMdS||j |||Wdn1sbwY| |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 TypeErrorrJtest_request_context_add_cookies_to_wsgirequestenvironsession_interface open_session RuntimeErrorresponse_classis_null_session save_session_update_cookies_from_responsehost partitionrheadersgetlist)r0rrrctxsessrespr7r7r8session_transactions2   zFlaskClient.session_transactioncCs$i|j|}|jr|jj|d<|S)Nzwerkzeug.debug.preserve_context)rPrMrNappend)r0otheroutr7r7r8 _copy_environs zFlaskClient._copy_environcCsL||di|d<t|jg|Ri|}z |W|S|w)NrP)rkr'rrJ get_requestclose)r0rrbuilderr7r7r8_request_from_builder_argss z&FlaskClient._request_from_builder_argsFbufferedfollow_redirectsrqr%rrrc s|rQt|dtjjttfrQt|dtjjr+t|d}||jp#i|_| }n,t|dtrCtj |d|j |id }nt|d}||j |_ n| ||}|jtj|||d}|j j|_|jr{|j}|j||jsm|S)Nr)rrPrp)r,rCtestrdict BaseRequestrrkrPrl from_environrJrXrorOrmr.openr: json_modulerNpop enter_context) r0rqrrrrrnrWresponsecmr5r7r8rws8        zFlaskClient.opencCs|jrtdd|_|S)NzCannot nest client invocationsT)rMr[r0r7r7r8 __enter__szFlaskClient.__enter__exc_type type | None exc_valueBaseException | NonetbTracebackType | NonecCsd|_|jdS)NF)rMrOrm)r0rrrr7r7r8__exit__szFlaskClient.__exit__)rrrrrr)rrrrrrQ) rrrqr%rrr%rrrr)rrI)rrrrrrrr)r=r>r?r@__annotations__r/rrgrkrorwr~rrAr7r7r5r8rIls   1  -rIcs4eZdZdZdfdd Z ddfd 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`. rrrrrrc s||_tjdi|dS)Nr7)rr.r/)r0rrr5r7r8r/ szFlaskCliRunner.__init__Nclirc sD|durjj}d|vrtfddd|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. Nr9csjS)N)rr7r}r7r8%sz'FlaskCliRunner.invoke..) create_app)rrr r.invoke)r0rrrr5r}r8rs zFlaskCliRunner.invoke)rrrrrr)NN)rrrrrrrr)r=r>r?r@r/rrAr7r7r5r8rs r)rr)$ __future__rimportlib.metadatarEtypingt contextlibrrrtypesr urllib.parser werkzeug.testrC click.testingrr werkzeug.wrappersr rurr sessionsr TYPE_CHECKINGrrrrsrrDrHrIrr7r7r7r8s.             G