o [h62@sddlZddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddl m Z e rEdd lmZn ddlZdd 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)selfinputoutputrt/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/click/testing.py__init__s zEchoingStdin.__init__cCs t|j|Sr )getattrr )rxrrr __getattr__s zEchoingStdin.__getattr__cCs|j||Sr )rwrite)rrvrrr_echos zEchoingStdin._echocC||j|Sr )rr readrnrrrr#zEchoingStdin.readcCrr )rr readlinerrrrr!&r zEchoingStdin.readlinecsfddjDS)Ncsg|]}|qSrr.0rrrr *sz*EchoingStdin.readlines..)r readlinesr%rr%rr')zEchoingStdin.readlinescstfddjDS)Nc3s|]}|VqdSr r"r#r%rr -sz(EchoingStdin.__iter__..)iterr r%rr%r__iter__,r(zEchoingStdin.__iter__cCs t|jSr )reprr r%rrr__repr__/s zEchoingStdin.__repr__N)r) __name__ __module__ __qualname__rrrrr!r'r+r-rrrrr s   r cCsft|drtr |St|}|dur|Std|durd}n t|ts(||}tr.t|St |S)Nrz.Could not find binary reader for input stream.) hasattrrr TypeError isinstancebytesencoder ioBytesIO)rcharsetrrrrmake_input_stream3s    r:c@sHeZdZdZ d ddZeddZeddZed d Zd d Z dS)Resultz3Holds the captured result of an invoked CLI script.NcCs(||_||_||_||_||_||_dSr runner stdout_bytes stderr_bytes exit_code exceptionexc_info)rr=r>r?r@rArBrrrrIs  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=r9rDr%rrrrC^sz Result.stdoutcCs,|jdur td|j|jjdddS)z%The standard error as unicode string.Nzstderr not separately capturedrDrErF)r? ValueErrorrGr=r9rDr%rrrstderres z Result.stderrcCs$dt|j|jrt|jSdS)Nz<{} {}>okay)formattyper.rAr,r%rrrr-ns zResult.__repr__r ) r.r/r0__doc__rpropertyrrCrIr-rrrrr;Fs     r;c@s^eZdZdZdddZddZdd 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,|durd}||_|p i|_||_||_dS)Nzutf-8)r9env echo_stdin mix_stderr)rr9rPrQrRrrrrs   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_names zCliRunner.get_default_prog_namecCst|j}|r |||S)z8Returns the environment overrides for invoking a script.)dictrPupdate)rZ overridesrrrrmake_envs  zCliRunner.make_envc#st|jtj}tj}tj}tj}dt_||}t r5t }|j r(t ||t_|j s4t } | t_n-t}|j rAt |tj|jdtj||jdt_|j sbt} tj| |jdt_|j ritjt_t_d fdd } d fdd } dd } |d fd d } tj}tj}tj}tj}| t_| t_| t_| t_i}zpt|D]%\}}tj|||<|durztj|=WqtyYqw|tj|<q||j o| fVWt|D]\}}|durztj|=WqtyYqw|tj|<q|t_|t_|t_|t_|t_|t_|t_|t_dSt|D]"\}}|dur2ztj|=Wqty1Yqw|tj|<q|t_|t_|t_|t_|t_|t_|t_|t_w)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|pdd}tjd|tj|S)NrE{} )sysrCrr!rstriprKflush)promptvalrrr visible_inputs  z*CliRunner.isolation..visible_inputcs.tjd|pdtjdS)Nr]r\rE)r^rCrrKr`r!r_)rarcrr hidden_inputs z)CliRunner.isolation..hidden_inputcSs*tjd}|rtj|tj|S)Nr)r^stdinrrCrr`)Zechocharrrr_getchars   z%CliRunner.isolation.._getcharcs|dur S| Sr r)streamcolor) default_colorrrshould_strip_ansisz.CliRunner.isolation..should_strip_ansir )NN)r:r9r^rfrCrIrZ FORCED_WIDTHrYrr rQr rRr7r8 TextIOWrapperrZvisible_prompt_funcZhidden_prompt_funcrhrrlrosenvironget Exception)rrrPrjZ old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputZ bytes_errorrdrerhrlZold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envkeyvaluer)rkrr isolations            zCliRunner.isolationc Ksd}|j|||d} d} d} t|trt|}z|d} Wnty.||} Ynwzz|jd|p7d| d|WnVt yy} z.t }| j } | durTd} | dkrZ| } t| t sot jt| t jdd} WYd} ~ n!d} ~ wty} z|s| } d} t }WYd} ~ nd} ~ wwWt j| d}|jrd}n| d}nt j| d}|jrd}w| d}wWdn1swYt|||| | |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)rrPrjr prog_namer)argsrurFrr<)rtr4rshlexsplitpopKeyErrorrVmain SystemExitr^rBcodeintrCrstrrqr`getvaluerRr;)rrUrvrrPZcatch_exceptionsrjextrarBZ outstreamsrAr@ruerCrIrrrinvokesl(        *zCliRunner.invokeccst}t}t|z|VWt|zt|WdSttfy,YdSwt|zt|WwttfyDYww)zA context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rngetcwdtempfilemkdtempchdirshutilrmtreeOSErrorIOError)rcwdtrrrisolated_filesystemos$   zCliRunner.isolated_filesystem)NNFTr )NNF)NNNTF) r.r/r0rMrrVrY contextlibcontextmanagerrtrrrrrrrOts   v \rO)rrnrwrr^rr\rrrZ_compatrrr cStringIOr r7r objectr r:r;rOrrrrs(       .