a xd62@sddlZddlZddlZddlZddlZddlZddlmZddlmZddlm 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@sPeZdZddZddZddZddd Zdd d Zd d ZddZ ddZ dS) EchoingStdincCs||_||_dSN)_input_output)selfinputoutputrE/var/www/html/Ranjet/env/lib/python3.9/site-packages/click/testing.py__init__szEchoingStdin.__init__cCs t|j|Sr )getattrr )rxrrr __getattr__szEchoingStdin.__getattr__cCs|j||Sr )rwrite)rrvrrr_echos zEchoingStdin._echocCs||j|Sr )rr readrnrrrr#szEchoingStdin.readcCs||j|Sr )rr readlinerrrrr&szEchoingStdin.readlinecsfddjDS)Ncsg|]}|qSrr.0rrrr *z*EchoingStdin.readlines..)r readlinesr#rr#rr&)szEchoingStdin.readlinescstfddjDS)Nc3s|]}|VqdSr r r!r#rr -r%z(EchoingStdin.__iter__..)iterr r#rr#r__iter__,szEchoingStdin.__iter__cCs t|jSr )reprr r#rrr__repr__/szEchoingStdin.__repr__N)r)r) __name__ __module__ __qualname__rrrrrr&r)r+rrrrr s  r cCsft|dr.tr|St|}|dur&|Std|durr?rrrrIs zResult.__init__cCs|jS)z(The (standard) output as unicode string.)stdoutr#rrrrYsz Result.outputcCs|j|jjdddS)z&The standard output as unicode string.replace  )r;decoder:r6rAr#rrrr@^sz Result.stdoutcCs,|jdurtd|j|jjdddS)z%The standard error as unicode string.Nzstderr not separately capturedrArBrC)r< ValueErrorrDr:r6rAr#rrrstderres  z Result.stderrcCs"dt|j|jrt|jndS)Nz<{} {}>okay)formattyper,r>r*r#rrrr+nszResult.__repr__)N) r,r-r.__doc__rpropertyrr@rFr+rrrrr8Fs    r8c@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,|dur d}||_|pi|_||_||_dS)Nzutf-8)r6env echo_stdin mix_stderr)rr6rMrNrOrrrrs  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)rclirrrget_default_prog_nameszCliRunner.get_default_prog_namecCst|j}|r|||S)z8Returns the environment overrides for invoking a script.)dictrMupdate)rZ overridesrrrrmake_envs  zCliRunner.make_envc#st|jtj}tj}tj}tj}dt_||}t rht }|j rNt ||t_|j st } | t_nZt}|j rt |tj|jdtj||jdt_|j st} tj| |jdt_|j rtjt_t_d fdd } d fdd } dd } |dfd d } tj}tj}tj}tj}| t_| t_| t_| t_i}zt|D]R\}}tj|||<|durz tj|=WntyYn0n |tj|<qJ||j o| fVWt|D]B\}}|durz tj|=WntyYn0n |tj|<q|t_|t_|t_|t_|t_|t_|t_|t_n~t|D]B\}}|durpz tj|=WntylYn0n |tj|<q:|t_|t_|t_|t_|t_|t_|t_|t_0dS)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}tjd|tj|S)NrB{} )sysr@rrrstriprHflush)promptvalrrr visible_inputs  z*CliRunner.isolation..visible_inputcs.tjd|pdtjdS)NrZrYrB)r[r@rrHr]rr\)r^r`rr hidden_inputs z)CliRunner.isolation..hidden_inputcSs*tjd}|r&tj|tj|S)Nr)r[stdinrr@rr])Zechocharrrr_getchars    z%CliRunner.isolation.._getcharcs|dur S| Sr r)streamcolor) default_colorrrshould_strip_ansisz.CliRunner.isolation..should_strip_ansi)N)N)NN)r7r6r[rcr@rFrZ FORCED_WIDTHrVrr rNr rOr4r5 TextIOWrapperrZvisible_prompt_funcZhidden_prompt_funcrerrirosenvironget Exception)rrrMrgZ old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputZ bytes_errorrarbreriZold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envkeyvaluer)rhrr isolations          zCliRunner.isolationc Ksd}|j|||d} d} d} t|tr4t|}z|d} Wnty^||} Yn0zz|jf|prd| d|Wnt y} z\t }| j } | durd} | dkr| } t| t st jt| t jdd} WYd} ~ nFd} ~ 0ty2} z$|s| } d} t }WYd} ~ n d} ~ 00Wt j| d}|jrZd}n | d}n2t j| d}|jrd}n | d}0Wdn1s0Yt|||| | |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)rrMrgr prog_namer)argsrrrCrr9)rqr1rshlexsplitpopKeyErrorrSmain SystemExitr[r?codeintr@rstrrnr]getvaluerOr8)rrRrsrrMZcatch_exceptionsrgextrar?Z outstreamsr>r=rrer@rFrrrinvokes^(          .zCliRunner.invokeccst}t}t|z:|VWt|zt|WqttfyRYq0n2t|zt|WnttfyYn00dS)zA context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rkgetcwdtempfilemkdtempchdirshutilrmtreeOSErrorIOError)rcwdtrrrisolated_filesystemos   zCliRunner.isolated_filesystem)NNFT)N)NNF)NNNTF) r,r-r.rJrrSrV contextlibcontextmanagerrqrrrrrrrLts   v \rL)rrkrtrr[rrYrrr_compatrrr cStringIOr r4r objectr r7r8rLrrrrs&       .