e([V�* � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z e j e j
d d � d Z e r� d d l m
Z
n d d l Z d d l m Z Gd d � d e � Z d d
� Z Gd d � d e � Z Gd
d � d e � Z d S)� N� )� iteritems�PY2�.)�StringIO)�_find_binary_readerc @ sv e Z d Z d d � Z d d � Z d d � Z d d d � Z d d
d � Z d d
� Z d d � Z d d � Z
d S)� EchoingStdinc C s | | _ | | _ d S)N)�_input�_output)�self�input�output� r � /testing.py�__init__ s zEchoingStdin.__init__c C s t | j | � S)N)�getattrr )r �xr r r � __getattr__ s zEchoingStdin.__getattr__c C s | j j | � | S)N)r
�write)r �rvr r r �_echo! s zEchoingStdin._echor c C s | j | j j | � � S)N)r r �read)r �nr r r r % s zEchoingStdin.readc C s | j | j j | � � S)N)r r �readline)r r r r r r ( s zEchoingStdin.readlinec s � f d d � � j j � D� S)Nc s g | ] } � j | � � q Sr )r )�.0r )r r r �
<listcomp>, s z*EchoingStdin.readlines.<locals>.<listcomp>)r � readlines)r r )r r r + s zEchoingStdin.readlinesc s t � f d d � � j D� � S)Nc 3 s | ] } � j | � Vq d S)N)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)N)�reprr )r r r r �__repr__1 s zEchoingStdin.__repr__N���r" ) �__name__�
__module__� __qualname__r r r r r r r r! r r r r r s r c C s� t | d � rA t r | St | � } | d k r5 | St d � � | d k rV d } n t | t � st | j | � } t r� t | � St j | � 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
r/ c @ sC e Z d Z d Z d d d � Z e d d � � Z d d � Z d S) �Resultz3Holds the captured result of an invoked CLI script.Nc C s1 | | _ | | _ | | _ | | _ | | _ d S)N)�runner� output_bytes� exit_code� exception�exc_info)r r1 r2 r3 r4 r5 r r r r K s
zResult.__init__c C s% | j j | j j d � j d d � S)z The output as unicode string.�replacez
�
)r2 �decoder1 r. r6 )r r r r r
X s z
Result.outputc C s# d | j r t | j � p d f S)Nz <Result %s>Zokay)r4 r )r r r r r! ^ s zResult.__repr__)r# r$ r% �__doc__r �propertyr
r! r r r r r0 H s r0 c @ s� e Z d Z d Z d d d d d � Z d d � Z d d d � Z e j d d d d
d � � Z d d d d d d
d � Z
e j d d � � Z d S)� 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 s7 | d k r d } | | _ | p$ i | _ | | _ d S)Nzutf-8)r. �env�
echo_stdin)r r. r< r= r r r r t s
zCliRunner.__init__c C s
| j p d S)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" | j | � | S)z8Returns the environment overrides for invoking a script.)�dictr< �update)r Z overridesr r r r �make_env� s
zCliRunner.make_envc # s� t � | j � � t j } t j } t j } t j j } d t j _ | j | � } t
r� t � t _ t _ } | j r� t
� | � � n^ t j � } | j r� t
� | � � t j � d | j �� t j | d | j �t _ t _ � t _ d � f d d � } d � f d d � }
d d � } | � d d � f d
d � } t j j }
t j j } t j j } t j j } | t j _ |
t j _ | t j _ | t j _ i } z{ xo t | � D]a \ } } t j j | � | | <| d k ry t j | =Wq t k
r
Yq Xq�| t j | <q�W| VWd xY t | � D]K \ } } | d k rvy t j | =Wq�t k
rrYq�Xq8| t j | <q8W| t _ | t _ | t _ |
t j _ | t j _ | t j _ | t j _ | t j _ Xd S) 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 sP t j j | p d � � j � j d � } t j j | d � t j j � | S)N� z
r7 )�sys�stdoutr r �rstrip�flush)�prompt�val)r r r �
visible_input� s
z*CliRunner.isolation.<locals>.visible_inputc s: t j j | p d d � t j j � � j � j d � S)NrG r7 z
)rH rI r rK r rJ )rL )r r r � hidden_input� s
z)CliRunner.isolation.<locals>.hidden_inputc S s9 t j j d � } | r5 t j j | � t j j � | S)Nr )rH �stdinr rI r rK )Zecho�charr r r �_getchar� s
z%CliRunner.isolation.<locals>._getcharc s | d k r � S| S)Nr )�stream�color)�
default_colorr r �should_strip_ansi� s z.CliRunner.isolation.<locals>.should_strip_ansi) r/ r. rH rP rI �stderr�clickpkgZ
formattingZ FORCED_WIDTHrD r r r= r r, r- �
TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcrR ZutilsrV r �os�environ�get� Exception)r r r< rT Z old_stdinZ
old_stdoutZ
old_stderrZold_forced_widthZ bytes_outputrN rO rR rV Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_env�key�valuer )rU r r � isolation� sr
zCliRunner.isolationTc & K s~ d } | j d | d | d | � �2} d }
d } zy, | j d | pK f d | j | � | � Wn� t k
r� } zj | j d k r� | }
t j � } | j } t | t � s� t j j
t | � � t j j
d � d } WYd d } ~ XnF t k
r3} z& | s � | }
d } t j � } WYd d } ~ Xn XWd t j j
� | j � }
XWd QRXt d
| d |
d | d
|
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.
Nr r< rT r �argsZ prog_namer7 r r1 r2 r3 r4 r5 r" )r` �mainrA �
SystemExit�coderH r5 r) �intrI r �strr] rK �getvaluer0 )r r@ ra r r< Zcatch_exceptionsrT �extrar5 �outr4 r3 �er
r r r �invoke� s: ! #
zCliRunner.invokec c so t j � } t j � } t j | � z | VWd t j | � y t j | � Wn t t f k
ri Yn XXd S)z�A context manager that creates a temporary folder and changes
the current working directory to it for isolated filesystem tests.
N) rZ �getcwd�tempfile�mkdtemp�chdir�shutil�rmtree�OSError�IOError)r �cwd�tr r r �isolated_filesystem3 s
z CliRunner.isolated_filesystem) r# r$ r% r9 r rA rD �
contextlib�contextmanagerr` rk rv r r r r r; d s i ?r; )rZ rH rp rm rw Z_compatr r �modulesr# �rsplitrX � cStringIOr r, r �objectr r/ r0 r; r r r r �<module> s