SˠS v d dl Z d dlmZmZmZ d dlmZ d dlmZm Z d dl
m Z m Z d dl
mZ d dlmZ d dlmZmZmZmZmZmZmZ d d
lmZmZm Z ee e f Z ee ee f Z!d e!d e fd
Z" G d de# Z$d efdZ% G d de# Z& G d de# Z'dZ( ee(e'j) e'j* g dS ) N)wrapspartialreduce)
TextIOWrapper)chain zip_longest)STDOUTPIPE)
TemporaryFile)extend_docstring)UnionCallableListOptionalSequenceTupleDict )Mockcall_Callcommandreturnc n t | t s d d | D } | S )N c 3 > K | ]}t j | V d S N)pipesquote).0parts d/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/testfixtures/popen.py z shell_join.. s, AA5;t,,AAAAAA )
isinstancestrjoin)r s r"
shell_joinr( s: gs # # B((AAAAAAA Nr$ c : e Zd ZdZ d
dededed ed
ef
d Zd S )PopenBehaviourzp
An object representing the behaviour of a :class:`MockPopen` when
simulating a particular command.
r$ r stdoutstderr
returncodepid
poll_countc L || _ || _ || _ || _ || _ d S r )r- r. r/ r0 r1 )selfr- r. r/ r0 r1 s r" __init__zPopenBehaviour.__init__ s+ $$r$ N)r$ r$ r r+ r, )__name__
__module__ __qualname____doc__bytesintr4 r$ r" r* r* sx
% % % % %
%
% % % % % %r$ r* c < t fd }|S )Nc J | j j fg|R i | | g|R i |S r )_recordr5 )r3 argskwfuncs r" recorderzrecord..recorder, sJ dm%3333333tD&4&&&2&&&r$ )r )rA rB s ` r" recordrC + s3
4[[' ' ' ' [' Or$ c
6 e Zd ZU dZdZeed< dZeed< dZ eed< dd
Z
d Z d Z d
Z
eddedefd Zeddededeeef fd Zedee fd Zededdfd Zedd Zedd ZdS ) MockPopenInstancez;
A mock process as returned by :class:`MockPopen`.
Nstdinr- r. r FTr; c h t | _ |j j | _ || _ g | _ |j | _ t | }|j ||j
}|t d|z t | r
||| }|| _
|j }|j }|t k rit# j t' | d | d }d d |D }d }|j | _ d||fd||ffD ]\ } } } d } | t. u rdt1 } | | | | d |s|s|rt9 | || } t; | | | |t. k rIt | _ d
D ]3} t? | j d | f }!|!tC | j | _" 4|j# | _# d | _$ || _% d S ) Nz Nothing specified for command %r)r rF Tr$ c 3 K | ]}||V d S r r; )r ls r" r# z-MockPopenInstance.__init__..i s' #B#B!#BA#B#B#B#B#B#Br$ r- r. r )encodingerrors)writecloserF )&r mockPopen_instanceclass_instance_mock root_callcalls all_callsr( commandsgetdefault_behaviourKeyErrorcallable behaviourr- r. r r
from_iterabler
splitlinesr' r1 r
r rL flushseekr setattrrF r r> getattr side_effectr0 r/ r? )"r3
mock_classrQ r? bufsize
executablerF r- r.
preexec_fn close_fdsshellcwdenvuniversal_newlines startupinfo
creationflagsrestore_signalsstart_new_sessionpass_fdsrJ rK textcmdrY stdout_value stderr_value
line_iteratornameoption
mock_valuevaluemethod
record_writess" r" r4 zMockPopenInstance.__init__D sS && )3 )G ) "$
%/%9 '++C1MNN =CDD D I <! #U;;;I)2 ' ' V ! / ''-- ''--1 1 M 88#B#B}#B#B#BBBL L ( 3
v| ,
v| ,)
' ' $D&* E~~% J'''
1
% S S S )%(6 R R RE D$ & & & & D== DJ*
H
H ' w6G H H
:G
F++77!
)- ! r$ c | j | j fD ] } t t || |i | t | j f| j | j ffD ]4\ }}| t t || |i | 5d S r ) rP rN r r_ r rR rQ rS append)r3 namesr? r@ rN base_callstores r" r> zMockPopenInstance._record s ,di7 6 6D (F7E4 ( ($ 5" 5 5 5 5
4:
^T^ ,!
I I Iu
LL: ::DGBGG H H H H I Ir$ c | S r r; r3 s r" __enter__zMockPopenInstance.__enter__ s r$ c | | | j | j fD ]}|r| d S r )waitr- r. rM )r3 exc_typeexc_valexc_tbstreams r" __exit__zMockPopenInstance.__exit__ sG k4;. F
r$ timeoutr c 2 | j j | _ | j S )z/Simulate calls to :meth:`subprocess.Popen.wait`)rY r/ )r3 r s r" r zMockPopenInstance.wait s .3 r$ inputc | j j | _ | j o| j | j o| j fS )z6Simulate calls to :meth:`subprocess.Popen.communicate`)rY r/ r- readr. )r3 r r s r" communicatez MockPopenInstance.communicate sJ .3 2 0 0 2 2 2 0 0 2 24 4r$ c r | j r| j | xj dz c_ dS | j j | _ | j S )z/Simulate calls to :meth:`subprocess.Popen.poll`Nr )r1 r/ rY r s r" pollzMockPopenInstance.poll s@ o $/"9 OOq OO4
.3 r$ signalc dS )z6Simulate calls to :meth:`subprocess.Popen.send_signal`Nr; )r3 r s r" send_signalz MockPopenInstance.send_signal
r$ c dS )z4Simulate calls to :meth:`subprocess.Popen.terminate`Nr; r s r" terminatezMockPopenInstance.terminate r r$ c dS )z/Simulate calls to :meth:`subprocess.Popen.kill`Nr; r s r" killzMockPopenInstance.kill r r$ )r NNNNNFFNNFNr TFr; NNNr )NN)r N)r5 r6 r7 r8 rF r __annotations__r- r r. r4 r> r r rC floatr: r AnyStrr r r r r r r r; r$ r" rE rE 3 s E4 !FM !FM .215DH.3DH3526@" @" @" @"DI I I E S V
4 4 4 4%PVX^P^J_ 4 4 4 V4
hsm
V
#
$
V
V
V
r$ rE c e Zd ZU dZdZeed< d Zd Z dd
e d e
d e
d
e de de de ee
f fdZ ddZd ZdS ) MockPopena
A specialised mock for testing use of :class:`subprocess.Popen`.
An instance of this class can be used in place of the
:class:`subprocess.Popen` and is often inserted where it's needed using
:func:`unittest.mock.patch` or a :class:`~testfixtures.Replacer`.
NrV c H i | _ t | _ g | _ d S r )rT r rN rS r s r" r4 zMockPopen.__init__ s! 35
&& &(r$ c 0 |t ||||| S |S r )r* r3 r- r. r/ r0 r1 rY s r" _resolve_behaviourz MockPopen._resolve_behaviour s- !
C
r$ r$ r r+ r, r r- r. r/ r0 r1 rY c d | |||||| | j t | < dS )z
Set the behaviour of this mock when it is used to simulate the
specified command.
:param command: A :class:`str` representing the command to be simulated.
N)r rT r( )r3 r r- r. r/ r0 r1 rY s r" set_commandzMockPopen.set_command s; .2-D-D FJZ.
.
j))***r$ c D | |||||| | _ dS )z
Set the behaviour of this mock when it is used to simulate commands
that have no explicit behavior specified using
:meth:`~MockPopen.set_command`.
N)r rV r s r" set_defaultzMockPopen.set_default s/ "&!8!8 FJZ"
"
r$ c | j j |i | t j |i |}| j | t | |g|R i |S r )rN Popenr rS r{ rE )r3 r? r@ rQ s r" __call__zMockPopen.__call__ se $$$$J+++ i((( y>4>>>2>>>r$ )r$ r$ r r+ r, N)r5 r6 r7 r8 rV r* r r4 r r& r9 r: r
r r r r r; r$ r" r r s )-~,,,( ( ( 9=
^X 56
( >?6:
? ? ? ? ?r$ r a
:param stdout:
:class:`bytes` representing the simulated content written by the process
to the stdout pipe.
:param stderr:
:class:`bytes` representing the simulated content written by the process
to the stderr pipe.
:param returncode:
An integer representing the return code of the simulated process.
:param pid:
An integer representing the process identifier of the simulated
process. This is useful if you have code the prints out the pids
of running processes.
:param poll_count:
Specifies the number of times :meth:`~MockPopenInstance.poll` can be
called before :attr:`~MockPopenInstance.returncode` is set and returned
by :meth:`~MockPopenInstance.poll`.
If supplied, ``behaviour`` must be either a :class:`PopenBehaviour`
instance or a callable that takes the ``command`` string representing
the command to be simulated and the ``stdin`` supplied when instantiating
the :class:`subprocess.Popen` with that command and should
return a :class:`PopenBehaviour` instance.
)+r functoolsr r r ior itertoolsr r
subprocessr r
tempfiler testfixtures.utilsr typingr
r r r r r r rN r r r Callr& r9 r Commandr( objectr* rC rE r set_command_paramsr r r; r$ r" r s
, , , , , , , , , , ( ( ( ( ( ( ( ( # # # # # # # # " " " " " " / / / / / / I I I I I I I I I I I I I I I I I I + + + + + + + + + +
sEz
Xc]"
# C % % % % %V % % %*H L
L
L
L
L
L
L
L
^=? =? =? =? =? =? =? =?@ 6 #')>?A A A A Ar$