U L[3@sddlZddlZddlZddlZddlZddlZddlmZmZm Z ej e dddZ erlddlmZnddlZddlmZGdddeZd d ZGd d d eZGd ddeZdS)N) iteritemsPY2 string_types.)StringIO)_find_binary_readerc@sPeZdZddZddZddZddd Zdd d Zd d ZddZ ddZ dS) EchoingStdincCs||_||_dSN)_input_output)selfinputoutputr//usr/lib/python3/dist-packages/click/testing.py__init__szEchoingStdin.__init__cCs t|j|Sr )getattrr )r xrrr __getattr__szEchoingStdin.__getattr__cCs|j||Sr )r write)r rvrrr_echo"s zEchoingStdin._echocCs||j|Sr )rr readr nrrrr&szEchoingStdin.readcCs||j|Sr )rr readlinerrrrr)szEchoingStdin.readlinecsfddjDS)Ncsg|]}|qSrr.0rr rr -sz*EchoingStdin.readlines..)r readlinesr!rr!rr#,szEchoingStdin.readlinescstfddjDS)Nc3s|]}|VqdSr rrr!rr 0sz(EchoingStdin.__iter__..)iterr r!rr!r__iter__/szEchoingStdin.__iter__cCs t|jSr )reprr r!rrr__repr__2szEchoingStdin.__repr__N)r)r) __name__ __module__ __qualname__rrrrrr#r&r(rrrrr s  r cCsft|dr.tr|St|}|dk r&|Std|dkr`sz Result.stdoutcCs(|jstd|j|jjdddS)z%The standard error as unicode string.zstderr not separately capturedr?r@rA)r: ValueErrorrBr8r4r?r!rrrstderrfs z Result.stderrcCs"dt|j|jrt|jpdfS)Nz<%s %s>Zokay)typer)r<r'r!rrrr(oszResult.__repr__)N) r)r*r+__doc__rpropertyrr>rDr(rrrrr6Is    r6c@sTeZdZdZdddZddZdd d Zejdd d Z dd dZ ejddZ dS) CliRunneraThe 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. :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 NFTcCs,|dkr d}||_|pi|_||_||_dS)Nzutf-8)r4env echo_stdin mix_stderr)r r4rIrJrKrrrrs  zCliRunner.__init__cCs |jpdS)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 clirrrget_default_prog_nameszCliRunner.get_default_prog_namecCst|j}|r|||S)z8Returns the environment overrides for invoking a script.)dictrIupdate)r Z overridesrrrrmake_envs  zCliRunner.make_envc#sZt|jtj}tj}tj}tjj}dtj_| |}t rlt }|j rRt ||t_|jst } | t_nZt}|j rt |tj|jdtj||jdt_|jst} tj| |jdt_|jrtjt_t_d fdd } d fdd } dd } |dfd d } tjj}tjj}tjj}tjj}| tj_| tj_| tj_| tj_i}zvt|D]T\}}tj|||<|dkrz tj|=Wntk rYnXn |tj|<q^||j o| fVW5t|D]D\}}|dkr z tj|=Wntk rYnXn |tj|<q|t_|t_|t_|tj_|tj_|tj_|tj_|tj_XdS)aA 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)encodingNcs<tj|p dd}tj|dtj|S)Nr@rA)sysr>rrrstripflush)promptvalrrr visible_inputs  z*CliRunner.isolation..visible_inputcs,tj|p ddtjdS)NrUrAr@)rVr>rrXrrW)rYr[rr hidden_inputs z)CliRunner.isolation..hidden_inputcSs*tjd}|r&tj|tj|S)Nr)rVstdinrr>rrX)Zechocharrrr_getchars    z%CliRunner.isolation.._getcharcs|dkr S| Sr r)streamcolor) default_colorrrshould_strip_ansisz.CliRunner.isolation..should_strip_ansi)N)N)NN)r5r4rVr^r>rDclickpkgZ formattingZ FORCED_WIDTHrRrrrJr rKr2r3 TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcr`Zutilsrdrosenviron Exceptionget)r rrIrbZ old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputZ bytes_errorr\r]r`rdZold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envkeyvaluer)rcrr isolations        zCliRunner.isolationc Ksd} |j|||dP} d} d} t|tr4t|}z|d} Wntk r`||} YnXzz|j f|prd| d|Wnt k r}zTt} |j} | dkrd} | dkr|} t| tstj t| tj dd} W5d}~XYn<tk r.}z|s|} d} t} W5d}~XYnXW5tj | d }| do^| d }XW5QRXt|||| | | 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. .. 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. 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. N)rrIrbr prog_namerr)argsrnrAr7)rmr/rshlexsplitpopKeyErrorrOrVr>rXgetvaluemain SystemExitr=codeintrstrrir6)r rNrorrIZcatch_exceptionsrbrKZextrar=Z outstreamsr<r;rnr>rDerrrinvokesN       "zCliRunner.invokeccs^t}t}t|z |VW5t|zt|Wnttfk rVYnXXdS)zA context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rggetcwdtempfileZmkdtempchdirshutilZrmtreeOSErrorIOError)r cwdtrrrisolated_filesystemgs   zCliRunner.isolated_filesystem)NNFT)N)NNF)NNNTFF) r)r*r+rFrrOrR contextlibcontextmanagerrmr{rrrrrrHvs   u OrH)rgrVrr}rrpZ_compatrrrmodulesr)rsplitre cStringIOrr2robjectr r5r6rHrrrrs -