o 2g)(@sddlZddlmZddlmZddlmZddlZddl m Z ddlm Z ddl m Z ddlmZd d lmZd d lmZd d lmZd d lmZejrYd dlmZd dlmZGdddejj Z Gddde Z!Gddde Z"dS)N)contextmanager)copy) TracebackType) CliRunner)Client) url_parse)Request)_request_ctx_stack) ScriptInfo)dumps) SessionMixin)Flask)Responsecs|eZdZdZ    ddddedejedejed ejed ejd ejd dffd d Zdejd ejd efddZ Z S)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`. /Napprpathbase_url subdomain url_schemeargskwargsreturnc 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__\/var/www/mastermindingenieria.com/MONITOR/venv/lib/python3.10/site-packages/flask/testing.pyr+,s4    zEnvironBuilder.__init__objcKs|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,r5rr3r3r4r7TszEnvironBuilder.json_dumps)rNNN) __name__ __module__ __qualname____doc__strtOptionalAnyr+r7 __classcell__r3r3r1r4rs0 "(rc seZdZUdZded<dZdejdejddffd d Ze dejdejdej e ddffd d Z dddd dejde de de dejddf fddZdddZdedededdfddZZS) FlaskClientaEWorks 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 :doc:`/testing` chapter. r applicationFrrrNcs*tj|i|ddtjd|_dS)Nz 127.0.0.1z werkzeug/) REMOTE_ADDRHTTP_USER_AGENT)r*r+werkzeug __version__ environ_base)r,rrr1r3r4r+ps  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 RuntimeErrorrBr6 inject_wsgir toptest_request_contextsession_interface open_sessionrequestpushpopresponse_classis_null_session save_sessionget_wsgi_headersenviron extract_wsgi) r,rrrrH outer_reqctxcrNsessrespheadersr3r3r4session_transactionws4     "zFlaskClient.session_transactionas_tuplebufferedfollow_redirectsr`rarbrc sd}fdd}|sNt|dkrN|d}t|tjjr,t|} || jp$i| _| }n"t|tr?tj |j |id}nt|t rNt|}||j |_ |durx|| di|d<tj g|Ri|} z | }W| n| wtj||||dS)Ncsij|djiS)Nzflask._preserve_context)rGpreserve_context)otherr,r3r4 copy_environsz&FlaskClient.open..copy_environr r)rrGrGr_)lenr(rEtestrrrG get_requestdict from_environrB BaseRequestrWr#closer*open) r,r`rarbrrrPrfargbuilderr1rer4rns:       zFlaskClient.opencCs|jrtdd|_|S)NzCannot nest client invocationsT)rcrJrer3r3r4 __enter__szFlaskClient.__enter__exc_type exc_valuetbcCs,d|_ tj}|dur|jr|ndSq)NF)rcr rL preservedrR)r,rrrsrtrLr3r3r4__exit__s zFlaskClient.__exit__)rrA)r8r9r:r;__annotations__rcr=r?r+r Generatorr r^r!rnrqtype BaseExceptionrrvr@r3r3r1r4rA^sN  9 1rAc sZeZdZdZdddejddffdd Z d d ejd ejdejdejffd 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`. rrrrNc s||_tjdi|dS)Nr3)rr*r+)r,rrr1r3r4r+szFlaskCliRunner.__init__clirc 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. Nr5csjS)N)rr3rer3r4sz'FlaskCliRunner.invoke..) create_app)rr|r r*invoke)r,r|rrr1rer4rs zFlaskCliRunner.invoke)NN) r8r9r:r;r=r?r+rr@r3r3r1r4r{sr{)#typingr= contextlibrrtypesr werkzeug.testrE click.testingrr werkzeug.urlsrwerkzeug.wrappersrrlr r|r jsonr r7sessionsr TYPE_CHECKINGrrwrappersrrhrrAr{r3r3r3r4s(             H