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 ) iteritemsPY2.)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)selfinputoutput 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__Nr" ) __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
isinstancebytesencoder ioBytesIO)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 exceptionexc_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)zGiven 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 | <qW| VWd xY t | D]K \ } } | d k rvy t j | =Wqt k
rrYqXq8| 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 )sysstdoutr r rstripflush)promptval)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 )Zechocharr r r _getchar s
z%CliRunner.isolation.<locals>._getcharc s | d k r S| S)Nr )streamcolor)
default_colorr r should_strip_ansi s z.CliRunner.isolation.<locals>.should_strip_ansi) r/ r. rH rP rI stderrclickpkgZ
formattingZ FORCED_WIDTHrD r r r= r r, r-
TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcrR ZutilsrV r osenvironget 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_envkeyvaluer )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
SystemExitcoderH 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)zA context manager that creates a temporary folder and changes
the current working directory to it for isolated filesystem tests.
N) rZ getcwdtempfilemkdtempchdirshutilrmtreeOSErrorIOError)r cwdtr r r isolated_filesystem3 s
z CliRunner.isolated_filesystem) r# r$ r% r9 r rA rD
contextlibcontextmanagerr` 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