U
e([V�* � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZmZ eje �
dd�d Z er`d dl m
Z
nd dlZddlmZ G dd� de�Zd d
� ZG d d � d e�ZG d
d� de�ZdS )� N� )� iteritems�PY2�.)�StringIO)�_find_binary_readerc @ sP e Zd Zdd� Zdd� Zdd� Zddd �Zdd
d �Zd d
� Zdd� Z dd� Z
dS )� EchoingStdinc C s || _ || _d S �N)�_input�_output)�self�input�output� r � /testing.py�__init__ s zEchoingStdin.__init__c C s t | j|�S r )�getattrr
)r �xr r r � __getattr__ s zEchoingStdin.__getattr__c C s | j �|� |S r )r �write)r �rvr r r �_echo! s zEchoingStdin._echo���c C s | � | j�|��S r )r r
�read�r �nr r r r % s zEchoingStdin.readc C s | � | j�|��S r )r r
�readliner r r r r ( s 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" + s zEchoingStdin.readlinesc s t � fdd�� jD ��S )Nc 3 s | ]}� � |�V qd S r r r r r r � <genexpr>/ s z(EchoingStdin.__iter__.<locals>.<genexpr>)�iterr
r r r r �__iter__. s zEchoingStdin.__iter__c C s
t | j�S r )�reprr
r r r r �__repr__1 s zEchoingStdin.__repr__N)r )r ) �__name__�
__module__� __qualname__r r r r r r" r% r' r r r r r s
r c C sf t | d�r.tr| S t| �}|d k r&|S td��| d kr<d} nt| t�sP| �|�} tr\t| �S t� | �S )Nr z.Could not find binary reader for input stream.� )
�hasattrr r � TypeError�
isinstance�bytes�encoder �io�BytesIO)r
�charsetr r r r �make_input_stream5 s
r4 c @ s. e Zd ZdZd dd�Zedd� �Zdd� ZdS )
�Resultz3Holds the captured result of an invoked CLI script.Nc C s" || _ || _|| _|| _|| _d S r ��runner� output_bytes� exit_code� exception�exc_info)r r7 r8 r9 r: r; r r r r K s
zResult.__init__c C s | j �| jjd��dd�S )z The output as unicode string.�replace�
�
)r8 �decoder7 r3 r< r r r r r X s �z
Result.outputc C s d| j rt| j �pdf S )Nz <Result %s>Zokay)r: r& r r r r r' ^ s �zResult.__repr__)N)r( r) r* �__doc__r �propertyr r' r r r r r5 H s �
r5 c @ sT e Zd ZdZddd�Zdd� Zddd �Zejdd
d ��Z dd
d�Z
ejdd� �Z dS )� CliRunneraI 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. This is
UTF-8 by default and should not be changed currently as
the reporting to Click only works in Python 2 properly.
: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.
NFc C s& |d kr d}|| _ |pi | _|| _d S )Nzutf-8)r3 �env�
echo_stdin)r r3 rC rD r r r r t s
zCliRunner.__init__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)�name)r �clir r r �get_default_prog_name{ s z CliRunner.get_default_prog_namec C s t | j�}|r|�|� |S )z8Returns the environment overrides for invoking a script.)�dictrC �update)r Z overridesr r r r �make_env� s
zCliRunner.make_envc # s t �| j��tj}tj}tj}tjj}dtj_| � |�}t
r`t � t_ t_}| j r�t
�|��n@t�� }| j rxt
�|��tj�| jd��tj|| jd� t_t_�t_d �fdd� } d
�fdd� }
dd � } |� d� fd
d � } tjj}
tjj}tjj}tjj}| tj_|
tj_| tj_| tj_i }zht|�D ]T\}}tj�|�||< |dk�rrz tj|= W n tk
�rn Y nX n
|tj|< �q*|V W 5 t|�D ]D\}}|dk�r�z tj|= W n tk
�r� Y nX n
|tj|< �q�|t_|t_|t_|
tj_|tj_|tj_|tj_|tj_X dS )a� 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.
.. versionadded:: 4.0
The ``color`` parameter was added.
: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.
�P )�encodingNc s< t j�| p d� � �� �d�}t j�|d � t j�� |S )N� r= r> )�sys�stdoutr r �rstrip�flush)�prompt�val�r
r r �
visible_input� s
z*CliRunner.isolation.<locals>.visible_inputc s, t j�| p dd � t j�� � �� �d�S )NrN r> r= )rO rP r rR r rQ )rS rU r r � hidden_input� s
z)CliRunner.isolation.<locals>.hidden_inputc S s* t j�d�}| r&t j�|� t j�� |S )Nr )rO �stdinr rP r rR )Zecho�charr r r �_getchar� s
z%CliRunner.isolation.<locals>._getcharc s |d kr� S | S r r )�stream�color)�
default_colorr r �should_strip_ansi� s z.CliRunner.isolation.<locals>.should_strip_ansi)N)N)NN) r4 r3 rO rX rP �stderr�clickpkgZ
formattingZ FORCED_WIDTHrK r r rD r r1 r2 �
TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcrZ Zutilsr^ r �os�environ� Exception�get)r r
rC r\ Z old_stdinZ
old_stdoutZ
old_stderrZold_forced_widthZ bytes_outputrV rW rZ r^ Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_env�key�valuer )r] r
r � isolation� sv
�
zCliRunner.isolationTc K s d}| j |||d���} d}
d} z�z$|jf |p.d| �|�d�|�� W n� tk
r� }
zJ|
jdkrd|
}
t� � }|
j} t
| t �s�tj� t
| �� tj� d� d} W 5 d}
~
X Y n8 tk
r� }
z|s |
}
d} t� � }W 5 d}
~
X Y nX W 5 tj�� | �� } X W 5 Q R X t| | | |
|d �S )
ae 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.
.. versionadded:: 3.0
The ``catch_exceptions`` parameter was added.
.. versionchanged:: 3.0
The result object now has an `exc_info` attribute with the
traceback if available.
.. versionadded:: 4.0
The ``color`` parameter was added.
:param cli: the command to invoke
:param args: the arguments to invoke
: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.
N)r
rC r\ r r )�argsZ prog_namer> r r r6 )rh rO rP rR �getvalue�mainrH �
SystemExit�coder; r. �intr �strrd r5 )r rG ri r
rC Zcatch_exceptionsr\ Zextrar; �outr: r9 r �er r r �invoke� sB ��
�zCliRunner.invokec c s^ t �� }t�� }t �|� z
|V W 5 t �|� zt�|� W n ttfk
rV Y nX X dS )z�A context manager that creates a temporary folder and changes
the current working directory to it for isolated filesystem tests.
N) rb �getcwd�tempfileZmkdtemp�chdir�shutilZrmtree�OSError�IOError)r �cwd�tr r r �isolated_filesystem3 s
z CliRunner.isolated_filesystem)NNF)N)NNF)NNNTF) r( r) r* r@ r rH rK �
contextlib�contextmanagerrh rr r{ r r r r rB d s
i �
@rB )rb rO rv rt r| Z_compatr r �modulesr( �rsplitr` Z cStringIOr r1 r �objectr r4 r5 rB r r r r �<module> s