B lbd)@sddlZddlmZddlmZddlmZddlmZddlZ ddl m Z ddlm Z ddl mZdd lmZd d lmZd d lmZd d lmZejrddlmZd dlmZGddde jjZGddde ZGddde ZdS)N)contextmanager) ExitStack)copy) TracebackType) CliRunner)Client) url_parse)Request) ScriptInfo) _cv_request) SessionMixin) TestResponse)Flaskc sbeZdZdZd deejeejeejeejejddfdd Zejejedd d Z 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`. /Nr)apppathbase_url subdomain url_schemeargskwargsreturnc s|s |s |r(|dk t|p|ks(td|dkr|jdp>d}|jd} |r\|d|}|dkrn|jd}t|} | jp~|d| jp|d| d}| j}| j rt | j t rd nd } || | j 7}||_ t j||f||dS) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".Z SERVER_NAMEZ localhostZAPPLICATION_ROOT.ZPREFERRED_URL_SCHEMEz://r??)boolAssertionErrorZconfiggetrZschemeZnetloclstriprZquery isinstancebytesrsuper__init__) selfrrrrrrrZ http_hostZapp_rootZurlsep) __class__,lib/python3.7/site-packages/flask/testing.pyr$-s(   &zEnvironBuilder.__init__)objrrcKs|jjj|f|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%r*rr(r(r) json_dumpsUszEnvironBuilder.json_dumps)rNNN) __name__ __module__ __qualname____doc__strtOptionalAnyr$r- __classcell__r(r()r'r)rsrcseZdZUdZded<ejejddfdd Zeejejej e ddfddd Z d d Z d d Z dddejeeejddfddZddddZejeejeejeddddZZS) 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 applicationN)rrrcs:tj||d|_g|_t|_ddtjd|_dS)NFz 127.0.0.1z werkzeug/)Z REMOTE_ADDRZHTTP_USER_AGENT) r#r$preserve_context _new_contextsr_context_stackwerkzeug __version__ environ_base)r%rr)r'r(r)r$ns zFlaskClient.__init__c os|jdkrtd|j}|di}|j|td}|j||}|j}| ||j }|dkrltdt |} z |VWdt | X| } ||s|||| | |j j} |j|j j| WdQRXdS)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)Z cookie_jar RuntimeErrorr8 setdefaultZ inject_wsgir rZtest_request_contextsession_interfaceZ open_sessionrequestsetresetZresponse_classZis_null_sessionZ save_sessionZget_wsgi_headersenvironZ extract_wsgi) r%rrrr?Z outer_reqctxcrBZsesstokenZrespZheadersr(r(r)session_transactionxs,        zFlaskClient.session_transactioncCs |j|}|jr|jj|d<|S)Nzwerkzeug.debug.preserve_context)r>r9r:append)r%otheroutr(r(r) _copy_environs  zFlaskClient._copy_environcCs@||di|d<t|jf||}z|S|XdS)Nr>)rMrrr8 get_requestclose)r%rrbuilderr(r(r)_request_from_builder_argss z&FlaskClient._request_from_builder_argsF)bufferedfollow_redirectsr)rrRrSrrc s|rt|dtjjttfrt|dtjjrVt|d}||jpFi|_| }qt|dtrtj |d|j |id }qt|d}||j |_ n | ||}|jtj|||d}|j j|_x |jr|j}|j|qW|S)Nr)rr>)rRrS)r!r<testrdict BaseRequestrrMr>rNZ from_environr8rFrQr;rOr#openr+Z json_moduler:pop enter_context) r%rRrSrrrPrCZresponsecm)r'r(r)rWs,       zFlaskClient.open)rcCs|jrtdd|_|S)NzCannot nest client invocationsT)r9r@)r%r(r(r) __enter__szFlaskClient.__enter__)exc_type exc_valuetbrcCsd|_|jdS)NF)r9r;rO)r%r\r]r^r(r(r)__exit__szFlaskClient.__exit__)r.r/r0r1__annotations__r3r5r$r Generatorr rIrMrQrrWr[r4type BaseExceptionrr_r6r(r()r'r)r7^s"  4 'r7csNeZdZdZdejddfdd Zd ejejejejdfdd 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`. rN)rrrc s||_tjf|dS)N)rr#r$)r%rr)r'r(r)r$szFlaskCliRunner.__init__)clirrrc s@|dkrjj}d|kr.tfddd|d<tj||f|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*csjS)N)rr()r%r(r)z'FlaskCliRunner.invoke..)Z create_app)rrer r#invoke)r%rerr)r')r%r)rhs zFlaskCliRunner.invoke)NN) r.r/r0r1r3r5r$rhr6r(r()r'r)rdsrd) typingr3 contextlibrrrtypesrZ werkzeug.testr<Z click.testingrrZ werkzeug.urlsrZwerkzeug.wrappersr rVrer globalsr Zsessionsr Z TYPE_CHECKINGrrrrTrr7rdr(r(r(r)s&             G