← Back
Editing: testing.cpython-38.pyc
U L�[3 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZm Z ej e � dd�d Z erld dlmZ nd dlZddlmZ G dd� de�Zd d � ZG d d � d e�ZG d d� de�ZdS )� N� )� iteritems�PY2� string_types�.)�StringIO)�_find_binary_readerc @ sP e Zd Zdd� Zdd� Zdd� Zddd �Zdd d �Zd d � Zdd� Z dd� Z dS )� EchoingStdinc C s || _ || _d S �N)�_input�_output)�self�input�output� r �//usr/lib/python3/dist-packages/click/testing.py�__init__ s zEchoingStdin.__init__c C s t | j|�S r )�getattrr )r �xr r r � __getattr__ s zEchoingStdin.__getattr__c C s | j �|� |S r )r �write)r �rvr r r �_echo" s zEchoingStdin._echo���c C s | � | j�|��S r )r r �read�r �nr r r r & s zEchoingStdin.readc C s | � | j�|��S r )r r �readliner r r r r ) s zEchoingStdin.readlinec s � fdd�� j �� D �S )Nc s g | ]}� � |��qS r �r ��.0r �r r r � <listcomp>- s z*EchoingStdin.readlines.<locals>.<listcomp>)r � readlinesr! r r! r r# , s zEchoingStdin.readlinesc s t � fdd�� jD ��S )Nc 3 s | ]}� � |�V qd S r r r r! r r � <genexpr>0 s z(EchoingStdin.__iter__.<locals>.<genexpr>)�iterr r! r r! r �__iter__/ s zEchoingStdin.__iter__c C s t | j�S r )�reprr r! r r r �__repr__2 s zEchoingStdin.__repr__N)r )r ) �__name__� __module__� __qualname__r r r r r r# r& r( r r r r r s r c C sf t | d�r.tr| S t| �}|d k r&|S td��| d kr<d} nt| t�sP| �|�} tr\t| �S t� | �S )Nr z.Could not find binary reader for input stream.� ) �hasattrr r � TypeError� isinstance�bytes�encoder �io�BytesIO)r �charsetr r r r �make_input_stream6 s r5 c @ sF e Zd ZdZd dd�Zedd� �Zedd� �Zed d � �Zd d � Z dS )�Resultz3Holds the captured result of an invoked CLI script.Nc C s( || _ || _|| _|| _|| _|| _d S r ��runner� stdout_bytes� stderr_bytes� exit_code� exception�exc_info)r r8 r9 r: r; r< r= r r r r L s zResult.__init__c C s | j S )z(The (standard) output as unicode string.)�stdoutr! r r r r [ s z Result.outputc C s | j �| jjd��dd�S )z&The standard output as unicode string.�replace� � )r9 �decoder8 r4 r? r! r r r r> ` s �z Result.stdoutc C s( | j std��| j �| jjd��dd�S )z%The standard error as unicode string.z stderr not separately capturedr? r@ rA )r: � ValueErrorrB r8 r4 r? r! r r r �stderrf s �z Result.stderrc C s" dt | �j| jrt| j�p df S )Nz<%s %s>Zokay)�typer) r< r' r! r r r r( o s �zResult.__repr__)N) r) r* r+ �__doc__r �propertyr r> rD r( r r r r r6 I s � r6 c @ sT e Zd ZdZddd�Zdd� Zdd d �Zejdd d ��Z dd d�Z ejdd� �Z dS )� CliRunnera� The 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 NFTc C s, |d kr d}|| _ |pi | _|| _|| _d S )Nzutf-8)r4 �env� echo_stdin� mix_stderr)r r4 rI rJ rK r r r r � s zCliRunner.__init__c C s |j pdS )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 �clir r r �get_default_prog_name� s z CliRunner.get_default_prog_namec C s t | j�}|r|�|� |S )z8Returns the environment overrides for invoking a script.)�dictrI �update)r Z overridesr r r r �make_env� s zCliRunner.make_envc # sZ t �| j��tj}tj}tj}tjj}dtj_| � |�}t rlt � }| j rRt �|��|t_| js�t � } | t_nZt�� }| j r�t �|��tj�| jd��tj|| jd�t_| js�t�� } tj| | jd�t_| jr�tjt_�t_d �fdd� } d �fdd� } dd � } |� d� fd d � } tjj}tjj}tjj}tjj}| tj_| tj_| tj_| tj_i }zvt|�D ]T\}}tj� |�||< |dk�r�z tj|= W n tk �r� Y nX n |tj|< �q^|| j �o�| fV W 5 t|�D ]D\}}|dk�r z tj|= W n tk �r Y nX n |tj|< �q�|t_|t_|t_|tj_|tj_|tj_|tj_|tj_X dS )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 )�encodingNc s< t j�| p d� � �� �d�}t j�|d � t j�� |S )N� r@ rA )�sysr> r r �rstrip�flush)�prompt�val�r r r � visible_input� s z*CliRunner.isolation.<locals>.visible_inputc s, t j�| p dd � t j�� � �� �d�S )NrU rA r@ )rV r> r rX r rW )rY r[ r r � hidden_input� s z)CliRunner.isolation.<locals>.hidden_inputc S s* t j�d�}| r&t j�|� t j�� |S )Nr )rV �stdinr r> r rX )Zecho�charr r r �_getchar� s z%CliRunner.isolation.<locals>._getcharc s |d kr� S | S r r )�stream�color)� default_colorr r �should_strip_ansi� s z.CliRunner.isolation.<locals>.should_strip_ansi)N)N)NN) r5 r4 rV r^ r> rD �clickpkgZ formattingZ FORCED_WIDTHrR r r rJ r rK r2 r3 � TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcr` Zutilsrd r �os�environ� Exception�get)r r rI rb Z old_stdinZ old_stdoutZ old_stderrZold_forced_widthZ bytes_outputZ bytes_errorr\ r] r` rd Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_env�key�valuer )rc r r � isolation� s� � � zCliRunner.isolationc K s� d} | j |||d���P} d} d} t|t�r4t�|�}z|�d�} W n tk r` | �|�} Y nX z�z |j f |prd| d�|�� W n� t k r� } zTt�� } |j} | dkr�d} | dkr�|} t| t�s�tj �t| �� tj �d� d} W 5 d}~X Y n< tk �r. } z |�s� |} d} t�� } W 5 d}~X Y nX W 5 tj � � | d � � }| d �o^| d � � }X W 5 Q R X t| ||| | | 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)r rI rb r � prog_namer r )�argsrn rA r7 )rm r/ r �shlex�split�pop�KeyErrorrO rV r> rX �getvalue�main� SystemExitr= �code�intr �strri r6 )r rN ro r rI Zcatch_exceptionsrb rK Zextrar= Z outstreamsr< r; rn r> rD �er r r �invoke sN "�zCliRunner.invokec c s^ t �� }t�� }t �|� z |V W 5 t �|� zt�|� W n ttfk rV Y nX X dS )z�A context manager that creates a temporary folder and changes the current working directory to it for isolated filesystem tests. N) rg �getcwd�tempfileZmkdtemp�chdir�shutilZrmtree�OSError�IOError)r �cwd�tr r r �isolated_filesystemg s z CliRunner.isolated_filesystem)NNFT)N)NNF)NNNTFF) r) r* r+ rF r rO rR � contextlib�contextmanagerrm r{ r� r r r r rH v s � u � OrH )rg rV r r} r� rp Z_compatr r r �modulesr) �rsplitre � cStringIOr r2 r �objectr r5 r6 rH r r r r �<module> s -
Save File
Cancel