File: //usr/local/lib/python3.9/site-packages/click/__pycache__/testing.cpython-39.pyc
a
X>h�? � @ s* d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ ddl mZ ejr�dd lmZ G d
d � d �Ze jeje ejd d �d
d��ZG dd� dej�Zejeje e ej ej f e ej d�dd�Z!G dd� d�Z"G dd� d�Z#dS )� N)�
TracebackType� )�_compat)�
formatting)�termui)�utils)�_find_binary_reader)� BaseCommandc @ s� e Zd Zejejdd�dd�Zeejd�dd�Ze e d�d d
�Z
de e d �d
d�Z de e d �dd�Z
d e e d �dd�Zeje d�dd�Zeje d�dd�Zed�dd�ZdS ) � EchoingStdinN)�input�output�returnc C s || _ || _d| _d S )NF)�_input�_output�_paused)�selfr r � r �7/usr/local/lib/python3.9/site-packages/click/testing.py�__init__ s zEchoingStdin.__init__)�xr
c C s t | j|�S �N)�getattrr )r r r r r � __getattr__ s zEchoingStdin.__getattr__)�rvr
c C s | j s| j�|� |S r )r r �write)r r r r r �_echo s zEchoingStdin._echo���)�nr
c C s | � | j�|��S r )r r �read�r r r r r r $ s zEchoingStdin.readc C s | � | j�|��S r )r r �read1r r r r r ' s zEchoingStdin.read1c C s | � | j�|��S r )r r �readliner r r r r! * s zEchoingStdin.readline�r
c s � fdd�� j �� D �S )Nc s g | ]}� � |��qS r �r ��.0r �r r r �
<listcomp>. � z*EchoingStdin.readlines.<locals>.<listcomp>)r � readlinesr& r r& r r) - s zEchoingStdin.readlinesc s t � fdd�� jD ��S )Nc 3 s | ]}� � |�V qd S r r# r$ r& r r � <genexpr>1 r( z(EchoingStdin.__iter__.<locals>.<genexpr>)�iterr r& r r& r �__iter__0 s zEchoingStdin.__iter__c C s
t | j�S r )�reprr r& r r r �__repr__3 s zEchoingStdin.__repr__)r )r )r )�__name__�
__module__� __qualname__�t�BinaryIOr �str�Anyr �bytesr �intr r r! �Listr) �Iteratorr, r. r r r r r
s r
)�streamr
c c s& | d u rd V nd| _ d V d| _ d S )NTF)r )r: r r r � _pause_echo7 s
r; c sR e Zd Zejeeejdd�� fdd� Zeed�dd��Z eed�dd ��Z
� Z S )
�_NamedTextIOWrapperN)�buffer�name�mode�kwargsr
c s$ t � j|fi |�� || _|| _d S r )�superr �_name�_mode)r r= r> r? r@ �� __class__r r r B s z _NamedTextIOWrapper.__init__r" c C s | j S r )rB r& r r r r> I s z_NamedTextIOWrapper.namec C s | j S r )rC r& r r r r? M s z_NamedTextIOWrapper.mode) r/ r0 r1 r2 r3 r4 r5 r �propertyr> r? �
__classcell__r r rD r r< A s �r< )r �charsetr
c C sb t | d�r6tt�tjtj | ��}|d ur.|S td��| d u rDd} nt| t�rX| � |�} t
� | �S )Nr z.Could not find binary reader for input stream.r( ) �hasattrr r2 �cast�IOr5 � TypeError�
isinstancer4 �encode�io�BytesIO)r rH r r r r �make_input_streamR s
rQ c @ s� e Zd ZdZddeeje ejeeje ejej
ej e e e f d�dd�Z
eed�dd ��Zeed�d
d ��Zeed�d d
��Zed�dd�ZdS )�Resultz3Holds the captured result of an invoked CLI script.N� CliRunner��runner� stdout_bytes� stderr_bytes� return_value� exit_code� exception�exc_infoc C s. || _ || _|| _|| _|| _|| _|| _d S r rT )r rU rV rW rX rY rZ r[ r r r r i s
zResult.__init__r" c C s | j S )z(The (standard) output as unicode string.)�stdoutr& r r r r � s z
Result.outputc C s | j �| jjd��dd�S )z&The standard output as unicode string.�replace�
�
)rV �decoderU rH r] r& r r r r\ � s �z
Result.stdoutc C s, | j du rtd��| j �| jjd��dd�S )z%The standard error as unicode string.Nz stderr not separately capturedr] r^ r_ )rW �
ValueErrorr` rU rH r] r& r r r �stderr� s
�z
Result.stderrc C s, | j rt| j �nd}dt| �j� d|� d�S )N�okay�<� �>)rZ r- �typer/ )r Zexc_strr r r r. � s zResult.__repr__)N)r/ r0 r1 �__doc__r6 r2 �Optionalr5 r7 �
BaseException�Tuple�Typer r rF r4 r r\ rb r. r r r r rR f s( ��� rR c
@ s� e Zd ZdZdeejejeeje f eedd�dd�Z d ed
�d d �Z
dejejeeje f ejeeje f d
�dd�Z e j
d ejejeeejej f ejejeeje f eejejejejej f d�dd��Zd d ejejeeje f ejejeeejej f ejejeeje f eeejed�dd�Ze j
d ejejedf eje d�dd��ZdS ) rS a� The CLI runner provides functionality to invoke a Click command line
script for unittesting purposes in a isolated environment. This only
works in single-threaded systems without any concurrency as it changes the
global interpreter state.
:param charset: the character set for the input and output data.
:param env: a dictionary with environment variables for overriding.
:param echo_stdin: if this is set to `True`, then reading from stdin writes
to stdout. This is useful for showing examples in
some circumstances. Note that regular prompts
will automatically echo the input.
:param mix_stderr: if this is set to `False`, then stdout and stderr are
preserved as independent streams. This is useful for
Unix-philosophy apps that have predictable stdout and
noisy stderr, such that each may be measured
independently
�utf-8NFT)rH �env�
echo_stdin�
mix_stderrr
c C s || _ |p i | _|| _|| _d S r )rH rn ro rp )r rH rn ro rp r r r r � s
zCliRunner.__init__r )�clir
c C s
|j pdS )z�Given a command object it will return the default program name
for it. The default is the `name` attribute or ``"root"`` if not
set.
�root)r> )r rq r r r �get_default_prog_name� s z CliRunner.get_default_prog_name)� overridesr
c C s t | j�}|r|�|� |S )z8Returns the environment overrides for invoking a script.)�dictrn �update)r rt r r r r �make_env� s
zCliRunner.make_env)r rn �colorr
c # s t || j�}d}tj}tj}tj}tj} dt_| �|�}t �
� }
| j r^t �
t jt||
�� }}t|| jddd� t_�| j r�d�_t|
| jddd�t_d} | jr�tjt_n t �
� } t| | jd dd
d �t_t|�dt jt td ��fd
d�
�} t|�dt jt td ��fdd�
�}
t|�ttd�dd��}|� dt jt jt j t jt td�� fdd�
}tj}tj}tj }t j }t j }| t_|
t_|t_ |t _ |t _ i }z�|� � D ]R\}}t!j"�#|�||< |du �r�z t!j"|= W n t$�y� Y n0 n
|t!j"|< �q�|
| fV W |� � D ]B\}}|du �r@z t!j"|= W n t$�y< Y n0 n
|t!j"|< �q
|t_|t_|t_|t_|t_|t_ |t _ |t _ | t_n�|� � D ]B\}}|du �r�z t!j"|= W n t$�y� Y n0 n
|t!j"|< �q�|t_|t_|t_|t_|t_|t_ |t _ |t _ | t_0 dS )an A context manager that sets up the isolation for invoking of a
command line tool. This sets up stdin with the given input data
and `os.environ` with the overrides from the given dictionary.
This also rebinds some internals in Click to be mocked (like the
prompt functionality).
This is automatically done in the :meth:`invoke` method.
:param input: the input stream to put into sys.stdin.
:param env: the environment overrides as dictionary.
:param color: whether the output should contain color codes. The
application can still override this explicitly.
.. versionchanged:: 8.0
``stderr`` is opened with ``errors="backslashreplace"``
instead of the default ``"strict"``.
.. versionchanged:: 4.0
Added the ``color`` parameter.
N�P z<stdin>�r)�encodingr> r? r z<stdout>�wz<stderr>�backslashreplace)r{ r> r? �errors)�promptr
c s> t j�| p d� � �� �d�}t j�|� d�� t j�� |S )N� r^ r_ )�sysr\ r r! �rstrip�flush)r �val��
text_inputr r �
visible_input s
z*CliRunner.isolation.<locals>.visible_inputc s. t j�| p d� d�� t j�� � �� �d�S )Nr� r_ r^ )r� r\ r r� r! r� )r r� r r � hidden_input s
z)CliRunner.isolation.<locals>.hidden_input)�echor
c S s* t j�d�}| r t j�|� t j�� |S )Nr )r� �stdinr r\ r r� )r� �charr r r �_getchar$ s
z%CliRunner.isolation.<locals>._getchar)r: rx r
c s |d u r� S | S r r )r: rx )�
default_colorr r �should_strip_ansi0 s z.CliRunner.isolation.<locals>.should_strip_ansi)N)N)NN)%rQ rH r� r� r\ rb r Z FORCED_WIDTHrw rO rP ro r2 rJ r3 r
r< � _CHUNK_SIZErp r; ri r4 �boolrK r5 r Zvisible_prompt_funcZhidden_prompt_funcr� r r� r �items�os�environ�get� Exception)r r rn rx Z bytes_inputZ
echo_inputZ old_stdinZ
old_stdoutZ
old_stderrZold_forced_widthZ bytes_outputZ bytes_errorr� r� r� r� Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZ old__compat_should_strip_ansiZold_env�key�valuer )r� r� r � isolation� s�
�
�
�
� ��
�
zCliRunner.isolation)rq �argsr rn �catch_exceptionsrx �extrar
c K s� d}| j |||d����} d}
d} d} t|t�r8t�|�}z|�d�}
W n tyb | �|�}
Y n0 �z&z |jf |pvd|
d�|��}
W n� t �y } z|t
� � }t �
t jt jtt jf |j�}|du r�d}|dkr�|} t|t��st
j�t|�� t
j�d� d}|} W Y d}~nFd}~0 t�yX } z$|�s4� |} d} t
� � }W Y d}~n
d}~0 0 W t
j�� | d �� }| j�r�d}n | d �� }n2t
j�� | d �� }| j�r�d}n | d �� }0 W d � n1 �s�0 Y t| |||
| | |d �S )
a� Invokes a command in an isolated environment. The arguments are
forwarded directly to the command line script, the `extra` keyword
arguments are passed to the :meth:`~clickpkg.Command.main` function of
the command.
This returns a :class:`Result` object.
:param cli: the command to invoke
:param args: the arguments to invoke. It may be given as an iterable
or a string. When given as string it will be interpreted
as a Unix shell command. More details at
:func:`shlex.split`.
:param input: the input data for `sys.stdin`.
:param env: the environment overrides.
:param catch_exceptions: Whether to catch any other exceptions than
``SystemExit``.
:param extra: the keyword arguments to pass to :meth:`main`.
:param color: whether the output should contain color codes. The
application can still override this explicitly.
.. versionchanged:: 8.0
The result object has the ``return_value`` attribute with
the value returned from the invoked command.
.. versionchanged:: 4.0
Added the ``color`` parameter.
.. versionchanged:: 3.0
Added the ``catch_exceptions`` parameter.
.. versionchanged:: 3.0
The result object has the ``exc_info`` attribute with the
traceback if available.
N)r rn rx r � prog_namer )r� r� r_ r rT )r� rM r4 �shlex�split�pop�KeyErrorrs �main�
SystemExitr� r[ r2 rJ ri �Unionr7 r5 �coder\ r r� r� �getvaluerp rR )r rq r� r rn r� rx r� r[ Z
outstreamsrX rZ rY r� �eZe_coder\ rb r r r �invokea sd ,
�
.�zCliRunner.invokezos.PathLike[str])�temp_dirr
c c s� t �� }tj|d�}t �|� z>|V W t �|� |du r�zt�|� W q� tyZ Y q�0 n6t �|� |du r�zt�|� W n ty� Y n0 0 dS )a� A context manager that creates a temporary directory and
changes the current working directory to it. This isolates tests
that affect the contents of the CWD to prevent them from
interfering with each other.
:param temp_dir: Create the temporary directory under this
directory. If given, the created directory is not removed
when exiting.
.. versionchanged:: 8.0
Added the ``temp_dir`` parameter.
)�dirN)r� �getcwd�tempfile�mkdtemp�chdir�shutil�rmtree�OSError)r r� �cwd�dtr r r �isolated_filesystem� s"
�
z CliRunner.isolated_filesystem)rm NFT)N)NNF)NNNTF)N)r/ r0 r1 rh r4 r2 ri �Mappingr� r rs rw �
contextlib�contextmanagerr� r6 rK r5 r9 rk rO rP r� �SequencerR r� r� r r r r rS � sd �� �� � � � � e ��rS )$r� rO r� r� r� r� r� �typingr2 �typesr r� r r r r r �
TYPE_CHECKING�corer r
r� ri r9 r; �
TextIOWrapperr< r� r4 r6 rK r5 r3 rQ rR rS r r r r �<module> s0 " � :