HJ> d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
ddl m Z ddl m
Z
ddl mZ ddlmZ ej rddlmZ G d d
Ze j d ej e d ej d fd
Z G d dej Zdej ej e e ej ej f de d ej fdZ! G d d Z" G d d Z#dS ) N)
TracebackType )
formatting)termui)utils)_find_binary_reader) BaseCommandc e Zd Zdej dej ddfdZdedej fdZde de fd Z
dd e de fd Z dd e de fd
Z
dd e de fdZdej e fdZdej e fdZdefdZdS ) EchoingStdininputoutputreturnNc 0 || _ || _ d| _ d S )NF)_input_output_paused)selfr r
s m/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/click/testing.py__init__zEchoingStdin.__init__ s xc , t | j | S N)getattrr )r r s r __getattr__zEchoingStdin.__getattr__ s t{A&&&r rvc H | j s| j | |S r )r r write)r r s r _echozEchoingStdin._echo s' | # L r " " " r nc \ | | j | S r )r r readr r! s r r# zEchoingStdin.read# s$ zz$+**1--...r c \ | | j | S r )r r read1r$ s r r& zEchoingStdin.read1& s$ zz$+++A..///r c \ | | j | S r )r r readliner$ s r r( zEchoingStdin.readline) s$ zz$+..q11222r c N fd j D S )Nc : g | ]} | S r .0r r s r
z*EchoingStdin.readlines..- s# ???!
1
???r )r readlinesr s `r r0 zEchoingStdin.readlines, s+ ????t{'<'<'>'>????r c D t fd j D S )Nc 3 B K | ]} | V d S r r, r- s r z(EchoingStdin.__iter__..0 s- 77aDJJqMM777777r )iterr r1 s `r __iter__zEchoingStdin.__iter__/ s( 77774;777777r c * t | j S r )reprr r1 s r __repr__zEchoingStdin.__repr__2 s DK r )r )__name__
__module__ __qualname__tBinaryIOr strAnyr bytesr intr# r& r( Listr0 Iteratorr6 r9 r+ r r r r s_ aj !*
'S 'QU ' ' ' ' % / /c /5 / / / /0 0s 0E 0 0 0 03 3# 3u 3 3 3 3@16%= @ @ @ @8!*U+ 8 8 8 8!# ! ! ! ! ! !r r streamr c # > K | d V d S d| _ d V d| _ d S )NTF)r )rE s r _pause_echorG 6 s3
~
r c
e Zd Zdej dededej ddf
fd Zedefd Z edefd Z
xZ S )
_NamedTextIOWrapperbuffernamemodekwargsr Nc X t j |fi | || _ || _ d S r )superr _name_mode)r rJ rK rL rM __class__s r r z _NamedTextIOWrapper.__init__A s6 **6***
r c | j S r )rP r1 s r rK z_NamedTextIOWrapper.nameH
zr c | j S r )rQ r1 s r rL z_NamedTextIOWrapper.modeL rT r ) r: r; r< r= r> r? r@ r propertyrK rL
__classcell__)rR s @r rI rI @ s j(+36BC%
c X c X r rI r charsetc F t | d rOt t j t j t j | }||S t
d | d} n*t | t r| | } t j | S )Nr# z.Could not find binary reader for input stream.r ) hasattrr r= castIOr@ TypeError
isinstancer? encodeioBytesIO)r rX r s r make_input_streamrb Q s uf J
QU U!;!;
<
<
>IHIII } E3 & W%%
:e r c e Zd ZdZ ddddedej e dej ded ej e d
ej ej
ej e e e f fd Z
ed efd
Zed efd Zed efd Zd efdZdS )Resultz3Holds the captured result of an invoked CLI script.Nrunner CliRunner stdout_bytes stderr_bytes return_value exit_code exceptionexc_infoc h || _ || _ || _ || _ || _ || _ || _ d S r re rg rh ri rj rk rl )r re rg rh ri rj rk rl s r r zResult.__init__h s> ( ( )""
r r c | j S )z(The (standard) output as unicode string.)stdoutr1 s r r
z
Result.output s {r c t | j | j j d dd S )z&The standard output as unicode string.replace
)rg decodere rX rr r1 s r rp z
Result.stdout s9 '' (;YGGOO D
r c | j t d | j | j j d dd S )z%The standard error as unicode string.Nz stderr not separately capturedrr rs rt )rh
ValueErrorru re rX rr r1 s r stderrz
Result.stderr sQ $ =>> > '' (;YGGOO D
r c r | j rt | j nd}dt | j d| dS )Nokay< >)rk r8 typer: )r exc_strs r r9 zResult.__repr__ s? *..D$t~&&&f34::&333333r r )r: r; r< __doc__rA r= Optionalr@ rB
BaseExceptionTupleTyper r rV r? r
rp rx r9 r+ r r rd rd e s; ==
! !! ! j' !
e !
! :m,! *
GAF=)=-G H
! ! ! !: X
X
X
4# 4 4 4 4 4 4r rd c e Zd ZdZ ddedej ej eej e f ded ed
df
d Z d d
d
efdZ
d dej ej eej e f d
ej eej e f fdZ e j
d dej ej eeej ej f dej ej eej e f ded
ej ej ej ej ej f fd Z d d d
dej ej eej e f dej ej eeej ej f dej ej eej e f dededej d
efdZe j
d dej ej edf d
ej e fd ZdS ) rf a 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.
: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
utf-8NFTrX env
echo_stdin
mix_stderrr c B || _ |pi | _ || _ || _ d S r )rX r r r )r rX r r r s r r zCliRunner.__init__ s( 47I2$$r clir c |j pdS )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)rK )r r s r get_default_prog_namez CliRunner.get_default_prog_name s
x!6!r overridesc \ t | j }|r| | |S )z8Returns the environment overrides for invoking a script.)dictr update)r r r s r make_envzCliRunner.make_env s0 $(^^ ! IIi r r colorc # K t || j }d}t j }t j }t j }t j } dt _ | | }t j
}
| j r/t j
t j t ||
x}}t! || j dd xt _ | j rd_ t! |
| j dd t _ d} | j rt j t _ n6t j
} t! | | j d dd
t _ t' | dd t j t* d
t* ffd
} t' | dd t j t* d
t* ffd
}
t' | dt, d
t* fd }| ddt j t j t j dt j t, d
t, ffd
}t2 j }t2 j }t2 j }t: j }| t2 _ |
t2 _ |t2 _ |t: _ i } | D ]W\ }}t@ j! " | ||< | t@ j! |= 8# tF $ r Y Dw xY w|t@ j! |< X|
| fV | D ]5\ }}| t@ j! |= # tF $ r Y "w xY w|t@ j! |< 6|t _ |t _ |t _ |t2 _ |t2 _ |t2 _ |t: _ | t _ dS # | D ]5\ }}| t@ j! |= # tF $ r Y "w xY w|t@ j! |< 6|t _ |t _ |t _ |t2 _ |t2 _ |t2 _ |t: _ | t _ w xY w)an 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.
: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.
.. versionchanged:: 8.0
``stderr`` is opened with ``errors="backslashreplace"``
instead of the default ``"strict"``.
.. versionchanged:: 4.0
Added the ``color`` parameter.
NP zr)encodingrK rL r zwzbackslashreplace)r rK rL errorspromptr c t j | pd d }t j | d t j |S )N rs rt )sysrp r r( rstripflush)r val
text_inputs r
visible_inputz*CliRunner.isolation..visible_input sq J V\r * * * %%''..v66C J ZZZ ( ( ( J Jr c t j | pd d t j d S )Nr rt rs )r rp r r r( r )r r s r hidden_inputz)CliRunner.isolation..hidden_input sZ J " 0 0 0 1 1 1 J &&((//77 7r echoc t j d }| r t j | t j |S )Nr )r stdinr# rp r r )r chars r _getcharz%CliRunner.isolation.._getchar# sL 9>>!$$D
'
&&& J Kr rE r c | S | S r r+ )rE r
default_colors r should_strip_ansiz.CliRunner.isolation..should_strip_ansi/ s }((( 9 r r )NN)$rb rX r r rp rx r FORCED_WIDTHr r` ra r r= r[ r> r rI _CHUNK_SIZEr rG r r? boolr\ r@ r visible_prompt_funchidden_prompt_funcr r r itemsosenvironget Exception)r r r r bytes_input
echo_input old_stdin
old_stdout
old_stderrold_forced_width bytes_output bytes_errorr r r r old_visible_prompt_funcold_hidden_prompt_funcold__getchar_funcold_should_strip_ansiold_envkeyvaluer r s @@r isolationzCliRunner.isolation s 6 (t|<<
I Z
Z
%2"$
mmC z|| ? '(v
Ll C C( (
K* "5 $,YS"
"
"
J ? ' &'J "( 4F % /FO&;E #&6J # # # &mmoo
,
,
U= JsOO$ ',BJsOO#CJ#CJ!CI)@F &(>F % /FO&;E #&6J # 6 6 6 6sg 2=O 0
K>=O >
L O
L O
M
MM Q<-
O;:Q<;
P Q<P A4Q<argscatch_exceptionsextrac 0 d}| ||| 5 } d}
d} d} t |t rt j | } | d }
n%# t $ r | | }
Y nw xY w |j d|pd|
d|}
n# t $ r}t j }t j
t j t j t t j f |j }|d}|dk r|} t |t sMt j t | t j d d}|} Y d}~n4d}~wt* $ r$}|s |} d} t j }Y d}~nd}~ww xY wt j | d }| j rd}n|| d }na# t j | d }| j rd}n| d }w xY wddd n # 1 swxY w Y t3 | |||
| | | 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.
: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.
.. versionchanged:: 8.0
The result object has the ``return_value`` attribute with
the value returned from the invoked command.
.. versionchanged:: 4.0
Added the ``color`` parameter.
.. versionchanged:: 3.0
Added the ``catch_exceptions`` parameter.
.. versionchanged:: 3.0
The result object has the ``exc_info`` attribute with the
traceback if available.
N)r r r r prog_namer+ )r r rt r rn )r r^ r? shlexsplitpopKeyErrorr main
SystemExitr rl r= r[ r UnionrB r@ coderp r r r getvaluer rd )r r r r r r r r rl
outstreamsri rk rj r ee_coderp rx s r invokezCliRunner.invoke] s X
^^%S^
>
> , 6* L37II$$$
) {4((
< !IIk 2 2
<
<
< 6 6s ; ;
<
6 'sx VTZR9 V VPU V V
#
#
# <>>
173:+> ?HH> FQ;; !I!&#.. J$$S[[111J$$T*** F "
*
*
*' <>>
*
"""#A //11? 6 !FF '] 3 3 5 5FF
"""#A //11? 6 !FF '] 3 3 5 5F5555Y, 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6 , 6\ %
s 0I7 A"!I7" BI7BI7B H
F*&CE94H
9
F*F% H
%F**H
-A I7
A I((I77I;>I;temp_dirzos.PathLike[str]c # K t j }t j | }t j | |V t j | |( t j | dS # t $ r Y dS w xY wdS # t j | |& t j | w # t $ r Y w w xY ww xY w)a A context manager that creates a temporary directory and
changes the current working directory to it. This isolates tests
that affect the contents of the CWD to prevent them from
interfering with each other.
:param temp_dir: Create the temporary directory under this
directory. If given, the created directory is not removed
when exiting.
.. versionchanged:: 8.0
Added the ``temp_dir`` parameter.
)dirN)r getcwdtempfilemkdtempchdirshutilrmtreeOSError)r r cwddts r isolated_filesystemz CliRunner.isolated_filesystem s ikk
(
+
+
+
HHH HSMMM M"%%%%% DD
HSMMM M"%%%% D sB B A1 1
A?>A?C B10C1
B>;C=B>>C)r NFTr )NNF)NNNTF)r: r; r< r r? r= r Mappingr r r r
contextlibcontextmanagerr rA r\ r@ rD r r` ra r Sequencerd r r r+ r r rf rf s ( ;?
%
%
% Z #qz#"67
8
%
%
%
%
%
%
%" "3 " " " " HL AIc1:c?.B$CD
3
3' ( ?C;? L7 L7z!'#uad15k"9:;L7 Z #qz#"67
8L7 L7
AGBJ
2:(> >? @ L7 L7 L7 L7b ;?>B;?!%c
c
c
jajo!567c
z!'#uad15k"9:; c
Z #qz#"67
8 c
c
c
c
c
c
c
c
J GK
1730B+B#CD
C r rf )$r r` r r r r r typingr= typesr r r r r _compatr
TYPE_CHECKINGcorer r r r rD rG
TextIOWrapperrI r r? rA r\ r@ r> rb rd rf r+ r r r s
( ( ( ( ( (? "!!!!!! ! ! ! ! ! ! ! !D
<0 QZ5E "* " :agc5!$qu+ 56 7 BE Z (74 74 74 74 74 74 74 74t@ @ @ @ @ @ @ @ @ @r