CasperSecurity
o
�+ca�> � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
ddl m Z ddl m
Z
ddl mZ ddlmZ ejrGddlmZ G d d
� d
�Ze jd eje d ejd fd
d��ZG dd� dej�Zdejeje e ej f de d ej fdd�Z G dd� d�Z!G dd� d�Z"dS )� N)�
TracebackType� )�
formatting)�termui)�utils)�_find_binary_reader)� BaseCommandc @ s� e Zd Zdejdejddfdd�Zdedejfdd �Zd
e de fd d �Z
dde de fdd�Z dde de fdd�Z
dde de fdd�Zdeje fdd�Zdeje fdd�Zdefdd�ZdS ) � EchoingStdin�input�output�returnNc C s || _ || _d| _d S )NF)�_input�_output�_paused)�selfr
r � r �//usr/lib/python3/dist-packages/click/testing.py�__init__ s
zEchoingStdin.__init__�xc C s t | j|�S �N)�getattrr
)r r r r r � __getattr__ s zEchoingStdin.__getattr__�rvc C s | j s | j�|� |S r )r r �write)r r r r r �_echo s zEchoingStdin._echo����nc C � | � | j�|��S r )r r
�read�r r r r r r # � zEchoingStdin.readc C r r )r r
�read1r r r r r! & r zEchoingStdin.read1c C r r )r r
�readliner r r r r" ) r zEchoingStdin.readlinec s � fdd�� j �� D �S )Nc s g | ]}� � |��qS r �r ��.0r �r r r �
<listcomp>- s z*EchoingStdin.readlines.<locals>.<listcomp>)r
� readlinesr& r r& r r( , � zEchoingStdin.readlinesc s t � fdd�� jD ��S )Nc 3 s � | ]}� � |�V qd S r r# r$ r& r r � <genexpr>0 s � z(EchoingStdin.__iter__.<locals>.<genexpr>)�iterr
r& r r& r �__iter__/ r) zEchoingStdin.__iter__c C s
t | j�S r )�reprr
r& r r r �__repr__2 s
zEchoingStdin.__repr__)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 r
d V d S d| _ d V d| _ d S )NTF)r )r: r r r � _pause_echo6 s �
r; c
sZ e Zd Zdejdededejddf
� fdd� Zedefd d
��Z edefd d ��Z
� Z S )
�_NamedTextIOWrapper�buffer�name�mode�kwargsr Nc s$ t � j|fi |�� || _|| _d S r )�superr �_name�_mode)r r= r> r? r@ �� __class__r r r A s
z _NamedTextIOWrapper.__init__c C � | j S r )rB r& r r r r> H � z_NamedTextIOWrapper.namec C rF r )rC r& r r r r? L rG z_NamedTextIOWrapper.mode) r/ r0 r1 r2 r3 r4 r5 r �propertyr> r? �
__classcell__r r rD r r< @ s �����r< r
�charsetc C sd t | d�rtt�tj| ��}|d ur|S td��| d u r d} n
t| t�r)| �|�} t �
t�t | ��S )Nr z.Could not find binary reader for input stream.� ) �hasattrr r2 �cast�IO� TypeError�
isinstancer4 �encode�io�BytesIOr6 )r
rJ r r r r �make_input_streamQ s
rT c @ s� e Zd ZdZ ddddedeje dejded eje d
ejej
ej e e e f fd d �Z
ed
efdd��Zed
efdd��Zed
efdd��Zd
efdd�ZdS )�Resultz3Holds the captured result of an invoked CLI script.N�runner� CliRunner� stdout_bytes� stderr_bytes� return_value� exit_code� exception�exc_infoc C s. || _ || _|| _|| _|| _|| _|| _d S r �rV rX rY rZ r[ r\ r] )r rV rX rY rZ r[ r\ r] r r r r h s
zResult.__init__r c C rF )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�
�
)rX �decoderV rJ r` r& r r r r_ � s �z
Result.stdoutc C s, | j du r td��| j �| jjd��dd�S )z%The standard error as unicode string.Nz stderr not separately capturedr` ra rb )rY �
ValueErrorrc rV rJ r` r& r r r �stderr� s
�z
Result.stderrc C s, | j rt| j �nd}dt| �j� d|� d�S )N�okay�<� �>)r\ r- �typer/ )r �exc_strr r r r. � s zResult.__repr__r )r/ r0 r1 �__doc__r6 r2 �Optionalr5 r7 �
BaseException�Tuple�Typer r rH r4 r r_ re r. r r r r rU e s6 ��������
� rU c @ s� e Zd ZdZ d dedejejeeje f ded ed
df
d d �Z d
dd
efdd�Z
d!dejejeeje f d
ejeeje f fdd�Z e j
d"dejejeeejf dejejeeje f ded
ejejejejej f fdd��Z d#d
ddejejeeje f dejejeeejf dejejeeje f dededejd
efdd �Ze j
d!d ejejeejf d
eje fd d ��Z dS )$rW 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-8NFTrJ �env�
echo_stdin�
mix_stderrr c C s || _ |pi | _|| _|| _d S r )rJ rr rs rt )r rJ rr rs rt r r r r � s
zCliRunner.__init__�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 ru r r r �get_default_prog_name� s
z CliRunner.get_default_prog_name� overridesc C s t | j�}|r |�|� |S )z8Returns the environment overrides for invoking a script.)�dictrr �update)r rx r r r r �make_env� s
zCliRunner.make_envr
�colorc # s� � t || j�}d}tj}tj}tj}tj} dt_| �|�}t �
� }
| j r0t �
t jt||
�� }}t|| jddd� t_�| j rBd�_t|
| jddd�t_d} | jrVtjt_nt �
� } t| | jd dd
d �t_t|�dd t jt d
tf�fdd�
�} t|�dd t jt d
tf�fdd�
�}
t|�dtd
tfdd��}|� ddt jt j dt jt d
tf� fdd�
}tj}tj}tj}t j }| t_|
t_|t_|t _ i }zn|� � D ]%\}}t j �!|�||< |du r�zt j |= W q� t"y� Y q�w |t j |< q�|
| fV W |� � D ] \}}|du �rzt j |= W q� t"�y Y q�w |t j |< q�|t_|t_|t_|t_|t_|t_|t _ | t_dS |� � D ]"\}}|du �rYzt j |= W �q= t"�yX Y �q=w |t j |< �q=|t_|t_|t_|t_|t_|t_|t _ | t_w )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�| pd� � �� �d�}t j�|� d�� t j�� |S )N� ra rb )�sysr_ r r" �rstrip�flush)r� �val��
text_inputr r �
visible_input s
z*CliRunner.isolation.<locals>.visible_inputc s. t j�| pd� d�� t j�� � �� �d�S )Nr� rb ra )r� r_ r r� r" r� )r� r� r r � hidden_input s
z)CliRunner.isolation.<locals>.hidden_input�echoc S s* t j�d�}| rt j�|� t j�� |S )Nr )r� �stdinr r_ r r� )r� �charr r r �_getchar# s
z%CliRunner.isolation.<locals>._getcharr: r| c s |d u r� S | S r r )r: r| )�
default_colorr r �should_strip_ansi/ s z.CliRunner.isolation.<locals>.should_strip_ansir )NN)#rT rJ r� r� r_ re r � FORCED_WIDTHr{ rR rS rs r2 rM r3 r r< � _CHUNK_SIZErt r; rm r4 �boolrN r �visible_prompt_func�hidden_prompt_funcr� r r� �items�os�environ�get� Exception)r r
rr r| � bytes_input�
echo_input� old_stdin�
old_stdout�
old_stderr�old_forced_width� bytes_output� bytes_errorr� r� r� r� �old_visible_prompt_func�old_hidden_prompt_func�old__getchar_func�old_should_strip_ansi�old_env�key�valuer )r� r� r � isolation� s� �
�
�
�
� �
��� �
�
�
�zCliRunner.isolation�args�catch_exceptions�extrac K s� d}| j |||d���} d}
d} d} t|t�rt�|�}z|�d�}
W n ty0 | �|�}
Y nw z�z|jd|p9d|
d�|��}
W ne t y� } z=t
� � }t �
t jt jtt jf |j�}|du rcd}|dkri|} t|t�s~t
j�t|�� t
j�d� d}|} W Y d}~n!d}~w ty� } z|s�� |} d} t
� � }W Y d}~nd}~ww W t
j�� | d �� }| jr�d}n | d �� }nt
j�� | d �� }| jr�d}w | d �� }w W d � n1 s�w 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
rr r| r � prog_namer )r� r� rb r r^ )r� rP r4 �shlex�split�pop�KeyErrorrw �main�
SystemExitr� r] r2 rM rm �Unionr7 r5 �coder_ r r� r� �getvaluert rU )r ru r� r
rr r� r| r� r] �
outstreamsrZ r\ r[ r� �e�e_coder_ re r r r �invoke] sr ,
�
����
� � �.�zCliRunner.invoke�temp_dirc c s� � t �� }tj|d�}t �|� z"|V W t �|� |du r1zt�|� W dS ty0 Y dS w dS t �|� |du rMzt�|� W w tyL Y w w w )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� �cwdr2 r r r �isolated_filesystem� s, �
��
� ��z CliRunner.isolated_filesystem)rq NFTr )NNF)NNNTF) r/ r0 r1 rl r4 r2 rm �Mappingr� r rw r{ �
contextlib�contextmanagerr� r6 rN r9 ro rR rS r� �Sequencer5 rU r� r� �PathLiker� r r r r rW � s� �����
� ��
� ���� � ��������
�e���rW )#r� rR r� r� r� r� r� �typingr2 �typesr r� r r r �_compatr �
TYPE_CHECKING�corer r r� rm r9 r; �
TextIOWrapperr< r� r4 r6 rN r3 rT rU rW r r r r �<module> s8 " ��
�: