<ݻ d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddlm Z ddlm
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddl m Z ddl m!Z! ej" rddl#m$Z$ dgZ% G d d Z& G d d ee
Z'dS ) z#ProcessDispatcher class declaration N)deque)deepcopy)partial)Any)AnyStr) Awaitable)Callable)Deque)Dict)Generic)List)Optional)Tuple)Type)Union )asyncio_subprocess)
exceptions)AsyncFakePopen) FakePopen)COMMAND)Command) FakeProcessProcessDispatcherc
V e Zd ZU dZg Zed ed< dZee ed< dZ
ee ed< dZ e
ed< e Zedede f f ed < dZe
ed
< d Zeed < ed d Zed d Zededee deeej f fd Z edee e f dedee e j j! f fd Z"edee e f dee e f dedee e j j! f fd Z#edede dedee e j j! f fd Z$edee%e e%e f deded
edeee f f
d Z&ededee fd Z'edede(ee) ee*eee
f ed f fd Z+dS ) r z"Main class for handling processes.r process_listNbuilt_in_popenbuilt_in_async_subprocessF_allow_unregistered_cache_keep_last_processr _pidprocessreturnc F | j sft j | _ t t _ t
j | _ | j t
_ | j t
_
t t
_ d | j D | j |< | j
| d S )Nc 8 i | ]}|t |j S )r definitions).0procs v/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/pytest_subprocess/process_dispatcher.py
z.ProcessDispatcher.register..; s2
15D(4+,,
)r
subprocessPopenr FakePopenWrapperasyncior async_shellcreate_subprocess_shell
async_execcreate_subprocess_execr r append)clsr# s r+ registerzProcessDispatcher.register0 s 4!+!1C /J ,3,>C ).1oG +-0^G *!3G
9<9I
7 (((((r- c | j | | j | }| D ] \ }}||_
| j sq| j t _ d | _ | j t j | j t _ | j j
t _
| j j t _ d | _ d S d S N)r remover popitemsr( r r. r/ r r PluginInternalErrorr1 r3 r5 )r7 r# cacher* processess r+
deregisterz ProcessDispatcher.deregister@ s (((
w'' ${{}} ) )OD) (D 1 " 1J !%C ,4 44!$!>G -E
+ -D
* -1C ) ) ) 1 1r- commandkwargsc | | }|#| j t j | j |fi |}|S | t
||| }t
|t
s t j | |S )z:This method will be used instead of the subprocess.Popen()) _ProcessDispatcher__dispatchr r r> _prepare_instancer
isinstance
run_thread)r7 rB rC r# popenresults r+ dispatchzProcessDispatcher.dispatchV s
..)) ?!) 44&8c&8&K&KF&K&KEL&&y'67KK&),, 1 0 0
r- cmdc K t |t t f st d t | j j |fi |}t |t r| }| ||| d{V S )z1Replacement of asyncio.create_subprocess_shell()zcmd must be a stringN) rG strbytes
ValueErrorr r r3 decode _call_async)r7 rL rC methods r+ r2 z ProcessDispatcher.async_shellj s
#U|,, 5 344 4 ) A
c5 ! ! **,,C__S&&999999999r- programargsc < K t |t t f st d t | j j |g|R i |}t |t r| }|gd |D }| ||| d{V S )z/Replacement of asyncio.create_subprocess_exec()zprogram must be a stringc d g | ]-}t |t r| n|.S r' )rG rO rQ )r) args r+
z0ProcessDispatcher.async_exec.. s3
N
N
N#je 4 4=cjjlll#
N
N
Nr- N) rG rN rO rP r r r5 rQ rR )r7 rT rU rC rS rB s r+ r4 z ProcessDispatcher.async_execz s
'C<00 9 788 8 ) @
gu % % ' nn&&G
N
N
N
N
N
__Wff=========r- async_methodc K | | }|%| j t j | }| d {V S t j dk rLt
t j t j
rt d | t ||| }t
|t s t j | |S )Nwin32z0The SelectorEventLoop doesn't support subprocess)rE r r r> sysplatformrG r1 get_event_loop_policyget_event_loopSelectorEventLoopNotImplementedErrorrF r evaluate)r7 rB rZ rC r# r2 rJ s r+ rR z ProcessDispatcher._call_async s ..)) ?,4 44AM K$$$$$$$ $ <7 " "z ) + + : : < W(
(
" &B
&&~wPP&.11 1 0 0
r- klassc | }||d< | d } |di |}|j | | j |_ |j di | |S )NrB recorderr' )copyr< callsr6 r" pid configure)r7 rd rB rC r# fake_popen_kwargsrf rJ s r+ rF z#ProcessDispatcher._prepare_instance s $LLNN'.)$$((44++*++ f%%%X
""6"""
r- c | | \ }}}|r|j | |s`| j d j sLt j dt |t r|n d d |D d d S |
}|s4|2| j d j r| | n
|r|j |= | xj
dz
c_
t |t rd S |S )Nz
The process ' c 3 4 K | ]}t | V d S r: )rN )r) items r+ z/ProcessDispatcher.__dispatch.. s( )H)H#d)))H)H)H)H)H)Hr- z' was not registered.r ) _get_processrh r6 r r r ProcessNotRegisteredErrorrG rN joinpopleftr! r( r" bool)r7 rB command_instancer@ process_instancer# s r+
__dispatchz ProcessDispatcher.__dispatch sJ 8;8H8H8Q8Q5)%5 3 " ) )' 2 2 2
#B';
::
*'3 7 7 IGG!$)H)H)H)H)H!H!H I I t##%% C - 9 #6
C ))))!
C$01AB A
gt $ $ 4r- c t | j D ]X}t fd|j D d \ }}|}|r t |t r|||fc S YdS )Nc 3 2 K | ]\ }}|k ||fV d S r: r' )r) keyvaluerB s r+ rq z1ProcessDispatcher._get_process.. s= "Ug~~ %L%~~~ r- )NN)NNN)reversedr nextr( r= rG r )r7 rB r* rw r@ rx s ` r+ rr z ProcessDispatcher._get_process s S -.. E ED*. &*&6&<&<&>&>
+ + ' i $
EZ 5 9 9
E'4DDDDD r- )r# r r$ N),__name__
__module__ __qualname____doc__r r
__annotations__r r r r AnyTyper rv dictr r r! r" int classmethodr8 rA r r r r. r/ rK rN rO r r1 Processr2 r4 rR r rF rE r r r
rr r' r- r+ r r % s@ ,,(*L$}%***)-NHX&---37 x 0777 %%%%@DFD ]G%; << =FFF $$$$D#MMM
)
)
) [
) 1 1 1 [1* )1$ y*** + [&
:U
#
:/3
: ~w 1 99 :
:
:
: [
: >CJ'>05c5j0A>MQ> ~w 1 99 :> > > [>*
~w 1 99 : [8 T)_d>&::;
y.( )
[, Xd^ [ < 8E%d
*;$<=x
?VV
[ r- c . e Zd Zdedee defdZdS )r0 rB rC r$ c & t j |fi |S r: )r rK )r7 rB rC s r+ __new__zFakePopenWrapper.__new__ s !)'<r s< ) )
! ! ! ! ! ! & & & & & & ! ! ! ! ! ! *))))))
P P P P P P P P f= = = = =wv = = = = =r-