U
|
ß=®d(& ã @ sê d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
|
m Z d dl Z d dlmZ d d l mZ d d
|
lmZ d d lmZ d d lmZ ejr²d dl mZ d dlmZ G dd de j j Z G dd deZ G dd deZ dS )é )Ú annotationsN)Úcontextmanager)Ú ExitStack)Úcopy)Ú TracebackType)Úurlsplit)Ú CliRunner)ÚClient)ÚRequesté )Ú
|
ScriptInfo)Ú SessionMixin)Ú TestResponse)ÚFlaskc
|
sH e Zd ZdZdddddddddd fd
|
d Zdddd d dZ ZS )ÚEnvironBuildera½ An :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 Ústrz
|
str | Noneút.AnyÚNone)ÚappÚpathÚbase_urlÚ subdomainÚ
|
url_schemeÚargsÚkwargsÚreturnc sê |s |s |r(|d k t |p|ks(td|d krÊ|j d¡p>d}|jd } |r\| d| }|d krn|jd }t|}
|
|
|
jp~| d|
|
jp| d| d¡ }|
|
j}|
|
j rÊt
|
|
|
j t r¸d nd
|
} || |
|
j 7 }|| _ t j||f|| d S ) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".Z SERVER_NAMEÚ localhostZAPPLICATION_ROOTÚ.ZPREFERRED_URL_SCHEMEz://r ó ?ú?)ÚboolÚAssertionErrorÚconfigÚgetr ÚschemeÚnetlocÚlstripr ÚqueryÚ
|
isinstanceÚbytesr ÚsuperÚ__init__) Úselfr r r r r r r Z http_hostZapp_rootÚurlÚsep©Ú __class__© úDd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\flask/testing.pyr, . s0
|
ÿþü
|
|
$ÿzEnvironBuilder.__init__)Úobjr r c K s | j jj|f|S )z³Serialize ``obj`` to a JSON-formatted string.
|
|
The serialization will be configured according to the config associated
|
with this EnvironBuilder's ``app``.
|
)r ÚjsonÚdumps)r- r4 r r2 r2 r3 Ú
|
json_dumpsV s zEnvironBuilder.json_dumps)r NNN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r, r7 Ú __classcell__r2 r2 r0 r3 r s ú"(r c s e Zd ZU dZded< dddd fdd Zeddd dd
|
d Zd d Zdd Z ddddddddd fddZ
|
d dddZ ddd dd d d Z Z S ) Ú FlaskClienta Works 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 Ú applicationr r )r r r c s: t j|| d| _g | _t | _ddtj d| _d S )NFz 127.0.0.1z werkzeug/)Z REMOTE_ADDRZHTTP_USER_AGENT) r+ r, Úpreserve_contextÚ _new_contextsr Ú_context_stackÚwerkzeugÚ __version__Ú environ_base)r- r r r0 r2 r3 r, o s
|
þzFlaskClient.__init__z%t.Generator[SessionMixin, None, None]c o sÌ | j dkrtd| j}|j||}| |jj¡ | |j ||j¡}W 5 Q R X |dkrbt d|V |
|
¡ }|j |¡rdS | |j |||¡ W 5 Q R X | |jj d¡d |jj|j d¡¡ dS )a When 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.ú:r z
|
Set-Cookie)Ú_cookiesÚ TypeErrorr> Ztest_request_contextZ_add_cookies_to_wsgiÚrequestÚenvironZsession_interfaceZ open_sessionÚ RuntimeErrorÚresponse_classZis_null_sessionZ save_sessionZ _update_cookies_from_responseÚhostÚ partitionr ÚheadersÚgetlist)r- r r r ÚctxÚsessÚrespr2 r2 r3 Úsession_transactiony s,
|
ÿ
|
ýz FlaskClient.session_transactionc C s | j |}| jr | jj|d< |S )Nz werkzeug.debug.preserve_context)rD r? r@ Úappend)r- ÚotherÚoutr2 r2 r3 Ú _copy_environ« s
|
zFlaskClient._copy_environc C sD | | di ¡¡|d< t| jf||}z | ¡ W ¢S | ¡ X d S )NrD )rW r$ r r> ÚcloseÚ get_request)r- r r Úbuilderr2 r2 r3 Ú_request_from_builder_args³ s
|
z&FlaskClient._request_from_builder_argsF©ÚbufferedÚfollow_redirectsr! r )r r] r^ r r c sö |r¢t |d tjjttfr¢t |d tjjrVt|d }| |jpFi ¡|_| ¡ }q®t |d trtj
|
|d | j | i ¡d ¡ }q®t|d }| |j ¡|_ n | ||¡}| j ¡ t j|||d}| j j|_| jrò| j ¡ }| j |¡ qÔ|S )Nr )r rD r\ )r) rB Útestr ÚdictÚ BaseRequestr rW rD rY Z from_environr> rI r[ rA rX r+ Úopenr5 Z json_moduler@ ÚpopÚ enter_context) r- r] r^ r r rZ rH ÚresponseÚcmr0 r2 r3 rb ¼ s8 ÿ
|
ÿ
|
ý
|
|
zFlaskClient.open)r c C s | j rtdd| _ | S )Nz Cannot nest client invocationsT)r? rJ ©r- r2 r2 r3 Ú __enter__é s zFlaskClient.__enter__z type | NonezBaseException | NonezTracebackType | None)Úexc_typeÚ exc_valueÚtbr c C s d| _ | j ¡ d S )NF)r? rA rX )r- ri rj rk r2 r2 r3 Ú__exit__ï s zFlaskClient.__exit__)r8 r9 r: r; Ú__annotations__r, r rS rW r[ rb rh rl r< r2 r2 r0 r3 r= _ s
|
|
1 ü -r= c sD e Zd ZdZdddd fdd Zd ddddd fd
|
d Z ZS ) ÚFlaskCliRunnerz®A :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`.
|
r r r )r r r c s || _ t jf | d S ©N)r r+ r, )r- r r r0 r2 r3 r, ÿ s zFlaskCliRunner.__init__N)Úclir r r c s@ |dkr j j}d|kr.t fddd|d< t j||f|S )ac Invokes a CLI command in an isolated environment. See
|
:meth:`CliRunner.invoke <click.testing.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.
|
Nr4 c s j S ro )r r2 rg r2 r3 Ú<lambda> ó z'FlaskCliRunner.invoke.<locals>.<lambda>)Z
|
create_app)r rp r r+ Úinvoke)r- rp r r r0 rg r3 rs s
|
zFlaskCliRunner.invoke)NN)r8 r9 r: r; r, rs r< r2 r2 r0 r3 rn ù s
|
ÿrn ) Ú
|
__future__r ÚtypingÚtÚ
|
contextlibr r r Útypesr Ú urllib.parser Z werkzeug.testrB Z click.testingr r Zwerkzeug.wrappersr
|
ra rp r Úsessionsr Ú TYPE_CHECKINGr r r r_ r r= rn r2 r2 r2 r3 Ú<module> s& G
|
