File: //home/arjun/projects/env/lib/python3.10/site-packages/flask/__pycache__/testing.cpython-310.pyc
o
'we!' � @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m Z d dl
m Z d dl m
Z
d dlZd dlmZ d d lmZ d d
lmZ d d lmZ d d
lmZ ejr]d dlmZ d dl m Z G dd� dej j �Z da ddd�Z!G dd� de�Z"G dd� de�Z#dS ) � )� annotationsN)�contextmanager)� ExitStack)�copy)�
TracebackType)�urlsplit)� CliRunner)�Client)�Request� )�
ScriptInfo)� SessionMixin)� TestResponse)�Flaskc s6 e Zd ZdZ dd� fdd�
Zddd�Z� 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`.
�/N�appr �path�str�base_url�
str | None� subdomain�
url_scheme�args�t.Any�kwargs�return�Nonec s� |s|s|r|d ut |p
|�ksJ d��|d u re|j�d�p d}|jd } |r.|� d|� �}|d u r7|jd }t|�}
|
jp?|� d|
jpE|� d| �d�� �}|
j}|
jret |
jt
�r\d nd
} || |
j 7 }|| _ t � j
||g|�R i |�� d S ) Nz8Cannot pass "subdomain" or "url_scheme" with "base_url".� SERVER_NAME� localhost�APPLICATION_ROOT�.�PREFERRED_URL_SCHEMEz://r � ?�?)�bool�config�getr �scheme�netloc�lstripr �query�
isinstance�bytesr �super�__init__) �selfr r r r r r r � http_host�app_root�url�sep�� __class__� �F/home/arjun/projects/env/lib/python3.10/site-packages/flask/testing.pyr/ / s4
���
�� zEnvironBuilder.__init__�objc K s | j jj|fi |��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)r0 r9 r r7 r7 r8 �
json_dumpsW s zEnvironBuilder.json_dumps)r NNN)r r r r r r r r r r r r r r r r )r9 r r r r r )�__name__�
__module__� __qualname__�__doc__r/ r< �
__classcell__r7 r7 r5 r8 r s �(r � r r c C s t stj�d�a t S )N�werkzeug)�_werkzeug_version� importlib�metadata�versionr7 r7 r7 r8 �_get_werkzeug_versionc s rH c st e Zd ZU dZded< d$� fd d
� Zed%d d
��Zdd� Zdd� Z ddd�d&� fdd�Z
d'dd�Z d(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|i |�� d| _g | _t� | _ddt� � �d�| _d S )NFz 127.0.0.1z Werkzeug/)� REMOTE_ADDR�HTTP_USER_AGENT)r. r/ �preserve_context�
_new_contextsr �_context_stackrH � environ_base)r0 r r r5 r7 r8 r/ | s
�zFlaskClient.__init__�%t.Generator[SessionMixin, None, None]c o s� � | j du r
td��| j}|j|i |��}| �|jj� |� |j�||j�}W d � n1 s1w Y |du r>t d��|V |�
� }|j� |�rMdS |� |j� |||� W d � n1 sbw Y | �
|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� TypeErrorrJ �test_request_context�_add_cookies_to_wsgi�request�environ�session_interface� open_session� RuntimeError�response_class�is_null_session� save_session� _update_cookies_from_response�host� partitionr �headers�getlist)r0 r r r �ctx�sess�respr7 r7 r8 �session_transaction� s2 �
� � �
�z FlaskClient.session_transactionc C s$ i | j �|�}| jr| jj|d< |S )Nz werkzeug.debug.preserve_context)rP rM rN �append)r0 �other�outr7 r7 r8 �
_copy_environ� s zFlaskClient._copy_environc C sL | � |�di ��|d< t| jg|�R i |��}z |�� W |�� S |�� w )NrP )rk r' r rJ � get_request�close)r0 r r �builderr7 r7 r8 �_request_from_builder_args� s
z&FlaskClient._request_from_builder_argsF��buffered�follow_redirectsrq r% rr r c s� |rQt |d tjjttf�rQt |d tjj�r+t|d �}| �|jp#i �|_|� � }n,t |d t�rCtj
|d | j | �i �d�� � }nt|d �}| �|j �|_ n| �
||�}| j�� t� j|||d�}| j j|_| jr{| j�� }| j�|� | jsm|S )Nr )r rP rp )r, rC �testr �dict� BaseRequestr rk rP rl � from_environrJ rX ro rO rm r. �openr: � json_modulerN �pop�
enter_context) r0 rq rr r r rn rW �response�cmr5 r7 r8 rw � s8 �
��
�
�zFlaskClient.openc C s | j rtd��d| _ | S )Nz Cannot nest client invocationsT)rM r[ �r0 r7 r7 r8 � __enter__� s zFlaskClient.__enter__�exc_type� type | None� exc_value�BaseException | None�tb�TracebackType | Nonec C s d| _ | j�� d S )NF)rM rO rm )r0 r r� r� r7 r7 r8 �__exit__� s zFlaskClient.__exit__)r r r r r r )r r r r r rQ )
r r rq r% rr r% r r r r )r rI )r r� r� r� r� r� r r )r= r> r? r@ �__annotations__r/ r rg rk ro rw r~ r� rA r7 r7 r5 r8 rI l s
1 �
-rI c s4 e Zd ZdZd� fdd � Z
dd� 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� jdi |�� d S )Nr7 )r r. r/ )r0 r r r5 r7 r8 r/ s zFlaskCliRunner.__init__N�clir c sD |du r� j j}d|vrt� fdd�d�|d< t� j||fi |��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.
Nr9 c s � j S )N)r r7 r} r7 r8 �<lambda>% s z'FlaskCliRunner.invoke.<locals>.<lambda>)�
create_app)r r� r r. �invoke)r0 r� r r r5 r} r8 r� s
zFlaskCliRunner.invoke)r r r r r r )NN)r� r r r r r r r )r= r> r? r@ r/ r� rA r7 r7 r5 r8 r� s
�r� )r r )$�
__future__r �importlib.metadatarE �typing�t�
contextlibr r r �typesr � urllib.parser �
werkzeug.testrC �
click.testingr r �werkzeug.wrappersr
ru r� r �sessionsr
�
TYPE_CHECKINGr r r rs r rD rH rI r� r7 r7 r7 r8 �<module> s. G