o ]Ocd)@sddlZddlmZddlmZddlmZddlmZddlZ ddl m Z ddlm Z ddl mZdd lmZd d lmZd d lmZd d lmZejrYddlmZd dlmZGddde jjZGddde ZGddde ZdS)N)contextmanager) ExitStack)copy) TracebackType) CliRunner)Client) url_parse)Request) ScriptInfo) _cv_request) SessionMixin) TestResponse)Flaskcs|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".Z SERVER_NAME localhostZAPPLICATION_ROOT.ZPREFERRED_URL_SCHEMEz://r??)boolconfiggetrschemenetloclstriprquery isinstancebytesrsuper__init__) selfrrrrrrrZ http_hostZapp_rooturlsep __class__/D:\Flask\env\Lib\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)r)r0rr.r.r/ json_dumpsUszEnvironBuilder.json_dumps)rNNN) __name__ __module__ __qualname____doc__strtOptionalAnyr(r3 __classcell__r.r.r,r/rs0 "(rc seZdZUdZded<dejdejddffdd Zedejdejdej e ddffd d Z d d Z ddZ ddddejdededejddf fddZdddZdejedejedejeddfddZZS) 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 applicationrrrNcs>tj|i|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)rrr,r.r/r(ns  zFlaskClient.__init__c os|jdur td|j}|di}|j|td}|j|i|Q}|j}| ||j }|dur9tdt |} z |VWt | nt | w| } ||s_|||| | |j j} |j|j j| WddS1szwYdS)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 RuntimeErrorr> setdefaultZ inject_wsgir r Ztest_request_contextsession_interfaceZ open_sessionrequestsetresetresponse_classZis_null_sessionZ save_sessionZget_wsgi_headersenvironZ extract_wsgi) r)rrrrEZ outer_reqctxcrHsesstokenrespheadersr.r.r/session_transactionxs4      "zFlaskClient.session_transactioncCs$i|j|}|jr|jj|d<|S)Nzwerkzeug.debug.preserve_context)rDr?r@append)r)otheroutr.r.r/ _copy_environs zFlaskClient._copy_environcCsL||di|d<t|jg|Ri|}z |W|S|w)NrD)rWr rr> get_requestclose)r)rrbuilderr.r.r/_request_from_builder_argss z&FlaskClient._request_from_builder_argsFbufferedfollow_redirectsr]r^rc 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)rrDr\)r%rBtestrdict BaseRequestrrWrDrXZ from_environr>rMr[rArYr'openr1Z json_moduler@pop enter_context) r)r]r^rrrZrIresponsecmr,r.r/rbs8        zFlaskClient.opencCs|jrtdd|_|S)NzCannot nest client invocationsT)r?rFr)r.r.r/ __enter__szFlaskClient.__enter__exc_type exc_valuetbcCsd|_|jdS)NF)r?rArY)r)rirjrkr.r.r/__exit__szFlaskClient.__exit__)rr=)r4r5r6r7__annotations__r9r;r(r Generatorr rSrWr[rrbrhr:type BaseExceptionrrlr<r.r.r,r/r=^sJ    6  -r=c 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)Nr.)rr'r()r)rrr,r.r/r(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. Nr0csjS)N)rr.rgr.r/sz'FlaskCliRunner.invoke..)Z create_app)rrrr r'invoke)r)rrrrr,rgr/rts zFlaskCliRunner.invoke)NN) r4r5r6r7r9r;r(rtr<r.r.r,r/rqsrq) typingr9 contextlibrrrtypesrZ werkzeug.testrBZ click.testingrrZ werkzeug.urlsrZwerkzeug.wrappersr rarrr globalsr sessionsr TYPE_CHECKINGrrrr_rr=rqr.r.r.r/s(             G