3 pXú*ã@s°ddlZddlZddlZddlZddlZddlmZmZeje j ddƒdZ er`ddl m Z nddlZddlmZGdd„deƒZd d „ZGd d „d eƒZGd d„deƒZdS)éNé)Ú iteritemsÚPY2Ú.)ÚStringIO)Ú_find_binary_readerc@sPeZdZdd„Zdd„Zdd„Zddd „Zdd d „Zd d „Zdd„Z dd„Z dS)Ú EchoingStdincCs||_||_dS)N)Ú_inputÚ_output)ÚselfÚinputÚoutput©rú//usr/lib/python3/dist-packages/click/testing.pyÚ__init__szEchoingStdin.__init__cCs t|j|ƒS)N)Úgetattrr )r ÚxrrrÚ __getattr__szEchoingStdin.__getattr__cCs|jj|ƒ|S)N)r Úwrite)r ÚrvrrrÚ_echo!s zEchoingStdin._echorcCs|j|jj|ƒƒS)N)rr Úread)r Únrrrr%szEchoingStdin.readcCs|j|jj|ƒƒS)N)rr Úreadline)r rrrrr(szEchoingStdin.readlinecs‡fdd„ˆjjƒDƒS)Ncsg|]}ˆj|ƒ‘qSr)r)Ú.0r)r rrú ,sz*EchoingStdin.readlines..)r Ú readlines)r r)r rr+szEchoingStdin.readlinescst‡fdd„ˆjDƒƒS)Nc3s|]}ˆj|ƒVqdS)N)r)rr)r rrú /sz(EchoingStdin.__iter__..)Úiterr )r r)r rÚ__iter__.szEchoingStdin.__iter__cCs t|jƒS)N)Úreprr )r rrrÚ__repr__1szEchoingStdin.__repr__Néÿÿÿÿ)r"r")r") Ú__name__Ú __module__Ú __qualname__rrrrrrrr!rrrrrs  rcCsft|dƒr.tr|St|ƒ}|dk r&|Stdƒ‚|dkrZokay)r4r )r rrrr!^szResult.__repr__)N)r#r$r%Ú__doc__rÚpropertyr r!rrrrr0Hs r0c@sTeZdZdZddd„Zdd„Zddd „Zejdd d „ƒZ dd d„Z ejdd„ƒZ dS)Ú CliRunneraIThe 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. NFcCs&|dkr d}||_|pi|_||_dS)Nzutf-8)r.ÚenvÚ echo_stdin)r r.r<r=rrrrts  zCliRunner.__init__cCs |jpdS)zžGiven 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 ÚclirrrÚget_default_prog_name{szCliRunner.get_default_prog_namecCst|jƒ}|r|j|ƒ|S)z8Returns the environment overrides for invoking a script.)Údictr<Úupdate)r Z overridesrrrrÚmake_env‚s  zCliRunner.make_envc#s tˆ|jƒ‰tj}tj}tj}tjj}dtj_|j |ƒ}t r`t ƒt_t_}|j r t ˆ|ƒ‰n@tjƒ}|j rxt ˆ|ƒ‰tjˆ|jd‰tj||jdt_t_ˆt_d ‡fdd„ } d ‡fdd„ } dd „} |‰d‡fd d „ } tjj} tjj}tjj}tjj}| tj_| tj_| tj_| tj_i}zlx`t|ƒD]T\}}tjj|ƒ||<|dkrty tj|=Wntk rpYnXn |tj|<q,W|VWdxPt|ƒD]D\}}|dkrÐy tj|=Wntk rÌYnXn |tj|<q˜W|t_|t_|t_| tj_|tj_|tj_|tj_|tj_XdS)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)ÚencodingNcs<tjj|p dƒˆjƒjdƒ}tjj|dƒtjjƒ|S)NÚz r7)ÚsysÚstdoutrrÚrstripÚflush)ÚpromptÚval)r rrÚ visible_input³s  z*CliRunner.isolation..visible_inputcs,tjj|p ddƒtjjƒˆjƒjdƒS)NrGr7z )rHrIrrKrrJ)rL)r rrÚ hidden_inputºs z)CliRunner.isolation..hidden_inputcSs*tjjdƒ}|r&tjj|ƒtjjƒ|S)Nr)rHÚstdinrrIrrK)ZechoÚcharrrrÚ_getchar¿s    z%CliRunner.isolation.._getcharcs|dkrˆ S| S)Nr)ÚstreamÚcolor)Ú default_colorrrÚshould_strip_ansiÇsz.CliRunner.isolation..should_strip_ansi)N)N)NN)r/r.rHrPrIÚstderrÚclickpkgZ formattingZ FORCED_WIDTHrDrrr=rr,r-Ú TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcrRZutilsrVrÚosÚenvironÚgetÚ Exception)r r r<rTZ old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputrNrOrRrVZold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envÚkeyÚvaluer)rUr rÚ isolation‰sr         zCliRunner.isolationTc"Ksd}|j|||dì} d} d} zÈy$|jf|p.f|j|ƒdœ|—ŽWnžtk r¬} zL| jdkrd| } tjƒ}| j} t| tƒsœtj j t | ƒƒtj j dƒd} WYdd} ~ Xn8t k râ} z|s‚| } d} tjƒ}WYdd} ~ XnXWdtj j ƒ| jƒ} XWdQRXt|| | | |dS) aeInvokes 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. N)r r<rTr)ÚargsZ prog_namer7r)r1r2r3r4r5r")r`ÚmainrAÚ SystemExitÚcoderHr5r)ÚintrIrÚstrr]rKÚgetvaluer0)r r@rar r<Zcatch_exceptionsrTZextrar5Úoutr4r3Úer rrrÚinvokeós:     zCliRunner.invokeccs^tjƒ}tjƒ}tj|ƒz |VWdtj|ƒytj|ƒWnttfk rVYnXXdS)z‘A context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rZÚgetcwdÚtempfileZmkdtempÚchdirÚshutilZrmtreeÚOSErrorÚIOError)r ÚcwdÚtrrrÚisolated_filesystem3s   zCliRunner.isolated_filesystem)NNF)N)NNF)NNNTF) r#r$r%r9rrArDÚ contextlibÚcontextmanagerr`rjrsrrrrr;ds   i ?r;)rZrHrnrlrtZ_compatrrÚmodulesr#ÚrsplitrXÚ cStringIOrr,rÚobjectrr/r0r;rrrrÚs