_62@s/ddlZddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddl m Z e rdd lmZnddlZdd l mZGd d d eZd dZGdddeZGdddeZdS)N) formatting)termui)utils) iteritems)PY2) string_types)StringIO)_find_binary_readerc@sveZdZddZddZddZddd Zdd d Zd d ZddZ ddZ dS) EchoingStdincCs||_||_dS)N)_input_output)selfinputoutputr./tmp/pip-build-5gj8f0j9/click/click/testing.py__init__s zEchoingStdin.__init__cCst|j|S)N)getattrr )rxrrr __getattr__szEchoingStdin.__getattr__cCs|jj||S)N)r write)rrvrrr_echoszEchoingStdin._echorcCs|j|jj|S)N)rr read)rnrrrr#szEchoingStdin.readcCs|j|jj|S)N)rr readline)rrrrrr&szEchoingStdin.readlinecs fddjjDS)Ncsg|]}j|qSr)r).0r)rrr *s z*EchoingStdin.readlines..)r readlines)rr)rrr)szEchoingStdin.readlinescs tfddjDS)Nc3s|]}j|VqdS)N)r)rr)rrr -sz(EchoingStdin.__iter__..)iterr )rr)rr__iter__,szEchoingStdin.__iter__cCs t|jS)N)reprr )rrrr__repr__/szEchoingStdin.__repr__Nr%) __name__ __module__ __qualname__rrrrrrr"r$rrrrr s      r cCst|drAtr|St|}|dk r5|Std|dkrVd}nt|tst|j|}trt|Stj |S)Nrz.Could not find binary reader for input stream.) hasattrrr TypeError isinstancebytesencoder ioBytesIO)rcharsetrrrrmake_input_stream3s      r2c@sgeZdZdZdddZeddZeddZed d Zd d Z dS) Resultz3Holds the captured result of an invoked CLI script.NcCs:||_||_||_||_||_||_dS)N)runner stdout_bytes stderr_bytes exit_code exceptionexc_info)rr4r5r6r7r8r9rrrrIs      zResult.__init__cCs|jS)z(The (standard) output as unicode string.)stdout)rrrrrYsz Result.outputcCs%|jj|jjdjddS)z&The standard output as unicode string.replacez  )r5decoder4r1r;)rrrrr:^sz Result.stdoutcCs@|jdkrtd|jj|jjdjddS)z%The standard error as unicode string.Nzstderr not separately capturedr;z r<)r6 ValueErrorr=r4r1r;)rrrrstderres z Result.stderrcCs1djt|j|jr*t|jndS)Nz<{} {}>okay)formattyper&r8r#)rrrrr$nszResult.__repr__) r&r'r(__doc__rpropertyrr:r?r$rrrrr3Fs  r3c@seZdZdZddddddZddZdd d Zejdddd d Z dddddd 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@|dkrd}||_|p$i|_||_||_dS)Nzutf-8)r1env echo_stdin mix_stderr)rr1rFrGrHrrrrs    zCliRunner.__init__cCs |jp dS)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)rclirrrget_default_prog_nameszCliRunner.get_default_prog_namecCs&t|j}|r"|j||S)z8Returns the environment overrides for invoking a script.)dictrFupdate)rZ overridesrrrrmake_envs zCliRunner.make_envc#st|jtj}tj}tj}tj}dt_|j|}t rt }|j rut ||t_|j s#t } | t_ntj}|j rt |tjd|jtj|d|jt_|j s#tj} tj| d|jt_|j r8tjt_t_dfdd} dfdd} dd } |ddfd d } tj}tj}tj}tj}| t_| t_| t_| t_i}zxot|D]a\}}tjj|||<|dkrGytj|=WqTtk rCYqTXq|tj|.visible_inputcs?tjjdj|pdtjjjjdS)Nz{} rRz )rSr:rrArUrrT)rV)rrr hidden_inputs z)CliRunner.isolation..hidden_inputcSs9tjjd}|r5tjj|tjj|S)Nr)rSstdinrr:rrU)Zechocharrrr_getchars  z%CliRunner.isolation.._getcharcs|dkr S| S)Nr)streamcolor) default_colorrrshould_strip_ansis z.CliRunner.isolation..should_strip_ansi)r2r1rSrZr:r?rZ FORCED_WIDTHrOrr rGr rHr/r0 TextIOWrapperrZvisible_prompt_funcZhidden_prompt_funcr\rr`rosenvironget Exception)rrrFr^Z old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputZ bytes_errorrXrYr\r`Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envkeyvaluer)r_rr isolations                                      zCliRunner.isolationc0Ksd}|jd|d|d|} d} d} t|trQtj|}y|jd} Wn!tk r|j|} YnXzy#|jd|pfd| |Wnt k rI} zyt j }| j } | dkrd} | dkr| } t| t s7t jjt| t jjdd } WYdd} ~ XnFtk r} z&|sd| } d } t j }WYdd} ~ XnXWdt jj| dj}|jrd}n| d j}XWdQRXtd |d |d |d | d| 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. NrrFr^r prog_nameargsr<rr4r5r6r7r8r9)rhr,rshlexsplitpopKeyErrorrLmain SystemExitrSr9codeintr:rstrrerUgetvaluerHr3)rrKrjrrFZcatch_exceptionsr^extrar9Z outstreamsr8r7rier:r?rrrinvokesR(! #    #   zCliRunner.invokeccsotj}tj}tj|z |VWdtj|ytj|Wnttfk riYnXXdS)zA context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rbgetcwdtempfilemkdtempchdirshutilrmtreeOSErrorIOError)rcwdtrrrisolated_filesystemos     zCliRunner.isolated_filesystem) r&r'r(rCrrLrO contextlibcontextmanagerrhrwrrrrrrEts  vUrE)rrbrkr|rSryrRrrr_compatrrr cStringIOr r/r objectr r2r3rErrrrs&        .