YE0E d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZ ddlmZmZ ddlmZ ddlmZ dd lmZ G d
d e Zd ee D Zd
Ze Z d Z d Z d Z d Z!d Z"d Z#ddZ$d Z%d Z&d Z'd Z(ddZ)ddZ*d Z+d Z,d Z- G d d e. Z/ G d! d"e. Z0 e0 Z1e1j2 Z2e1j3 Z4e1j5 Z6h d#Z7d$ Z8 G d% d&e9 Z:d' Z; G d( d)e. Z< G d* d+e. Z= G d, d-e= Z> e?j@ Z@n# eA$ r d. Z@Y nw xY w eBd/ ee> D ZC G d0 d1e9 ZDd2 ZE G d3 d4e= ZF G d5 d6eFe> ZGd7 ZHd8 ZId9 ZJ G d: d;e. ZKd< ZLe2dddddfdd=d>ZM dd?ZNe2dddddfdd=d@ZO G dA dBe. ZPdC ZQdD ZReMeO_. ePeO_S eNeO_T eReO_U dEeO_V dFZWereWX dGdH ZWdIZYdJZ dK eY[ D Z\dJZ dL eY[ D Z]h dMZ^dN Z_dO dJZ eWeYe\e]g [ D Z`h dPZadQhZbeaebz Zce`e^z Zdedecz Zeh dRZfdS dT dU dV dWZgehehehehdddddXdYd
dddZ
Zid[ Zjd\ Zkd] Zld^ Zmejekelemd_Znd` Zo G da dbe= Zp G dc ddepe> Zq G de dfep Zr G dg dhepeG Zs G di dje= Zt ee
Zu ejv ee
jw d Zx G dk dle= Zy G dm dneyereG Zz G do dpe. Z{ e{ Z|dq Z} G dr dse~ Z edt Z ddd=duZdv Z G dw dxe. Z ee ee|j fZdadady ZddzZ G d{ d|eG Ze1j Z G d} d~e= Z G d deepeG Zd Z G d d ZdS ))Mock MagicMockpatchsentinelDEFAULTANYcallcreate_autospec AsyncMock
ThreadingMock
FILTER_DIRNonCallableMockNonCallableMagicMock mock_open PropertyMockseal N)CodeType
ModuleType
MethodType) safe_repr)wrapspartial)RLock)IS_PYPY )iscoroutinefunctionc e Zd ZdZdS )InvalidSpecErrorz8Indicates that an invalid value was used as a mock spec.N__name__
__module__ __qualname____doc__ [/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/mock/mock.pyr r - s BBBBr% r c < h | ]}| d |S _
startswith.0names r& r/ 1 s) H H Hd4??33G3G HT H H Hr% Tc t | rt | t sdS t | d rt | d } t | pt
j | S )NF__func__)_is_instance_mock
isinstancer
hasattrgetattrr inspect isawaitableobjs r&
_is_async_objr: 9 sg ji&@&@ usJ 'c:&& s # # ?w':3'?'??r% c F t | dd rt | S dS )N__code__F)r5 r )funcs r& _is_async_funcr> A s) tZ&& "4(((ur% c F t t | t S N)
issubclasstyper
r8 s r& r2 r2 H s d3ii 1 11r% c t | t p)t | t ot | t S r@ )r3
BaseExceptionrB rA r8 s r&
_is_exceptionrE N s6 3
&& A3 @*S-"@"@r% c ^ t | t rt | d r| j S | S Nmock)r3
FunctionTypesr4 rH r8 s r&
_extract_mockrJ U s3 #}%% '#v*>*> x
r% c t | t r |s
| j } d}njt | t t f r t | t rd}| j } n/t | t s | j } n# t $ r Y dS w xY w|rt | d }n| } | t j | fS # t $ r Y dS w xY w)z
Given an arbitrary, possibly callable object, try to create a suitable
signature object.
Return a (reduced func, signature) tuple, or None.
TN)
r3 rB __init__ classmethod staticmethodr1 rI __call__AttributeErrorr r6 signature
ValueError)r= as_instanceeat_selfsig_funcs r& _get_signature_objectrV ^ s $ k } D; 5 6 6 dK ( ( H}
m
,
, =DD 44 4&&W&x0000 tts$ 3A; ;
B B "B8 8
CCFc t | || d S \ } fd}t | | |t | _ t | _ d S )Nc " j |i | d S r@ bind)
_mock_selfargskwargssigs r& checksigz"_check_signature..checksig $!&!!!!!r% )rV _copy_func_detailsrB _mock_check_sig
__signature__)r= rH skipfirstinstancer_ r^ s @r& _check_signaturerf sr
h
:
:C
{ID#" " " " "tX&&&!)DJJ "DJJ r% c p dD ]2} t ||t | | ## t $ r Y /w xY wd S )N)r r# __text_signature__r! __defaults____kwdefaults__)setattrr5 rP )r= funcopy attributes r& ra ra sa GYi(@(@ A A A A D
s &
33c t | t rdS t | t t t f rt | j S t | dd dS dS )NTrO F)r3 rB rN rM r _callabler1 r5 r8 s r& ro ro s^ #t t# k:>?? ' &&&sJ%%1t 5r% c < t | t t fv S r@ )rB listtupler8 s r& _is_listrs s 99u
%%r% c t | t st | dd duS | f| j z D ] }|j d dS dS )ztGiven an object, return True if the object is callable.
For classes, return True if instances would be callable.rO NTF)r3 rB r5 __mro____dict__get)r9 bases r& _instance_callablery sn c4 :sJ--T99 $ = Z ( ( 444 5 5r% c 0 t |t }t ||| }|| S |\ } fd}t || |j }| sd}|| d}d|z } t
| | || }
t |
| |
S )Nc " j | i | d S r@ rY r\ r] r^ s r& r_ z _set_signature..checksig r` r% rl
_checksig_rH zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs))r3 rB rV ra r isidentifierexec _setup_func) rH originalre rd resultr= r_ r. contextsrcrl r^ s @r& _set_signaturer s
8T**I
"8Xy
A
AF
~ ID#" " " " "tX&&& D %t44G $&* +C #wdmGs### Nr% c t |t }t ||| \ } fd}t || |j }|| d}d|z } t | | || }
t
|
| t |
|
S )Nc " j | i | d S r@ rY r| s r& r_ z&_set_async_signature..checksig r` r% r} zeasync def %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return await mock(*args, **kwargs))r3 rB rV ra r r r _setup_async_mock) rH r re
is_async_mockrd r= r_ r. r r rl r^ s @r& _set_async_signaturer s
8T**I%h)DDID#" " " " "tX&&& D %t44G *,0 1C #wdmGs###g Nr% c _ fd}fd}fd}fd}fd}fd}fd} fd}
d _ d
_ d _ t _ t _ t _ j _ j _ j
_
| _ | _ | _
| _ |
_ | _ | _ | _ | _ _ d S ) Nc j | i |S r@ )assert_called_withr\ r] rH s r& r z'_setup_func..assert_called_with &t&7777r% c j | i |S r@ )
assert_calledr s r& r z"_setup_func..assert_called s !t!426222r% c j | i |S r@ )assert_not_calledr s r& r z&_setup_func..assert_not_called s %t%t6v666r% c j | i |S r@ )assert_called_oncer s r& r z'_setup_func..assert_called_once r r% c j | i |S r@ )assert_called_once_withr s r& r z,_setup_func..assert_called_once_with s +t+T.assert_has_calls s $t$d5f555r% c j | i |S r@ )assert_any_callr s r& r z$_setup_func..assert_any_call s #t#T4V444r% c t _ t _ j } t | r| ur| d S d S d S r@ ) _CallList method_calls
mock_calls
reset_mock return_valuer2 )retrl rH s r& r z _setup_func..reset_mock so ({{ &[[ " S ! ! #++ NN ++r% Fr )rH called
call_count call_argsr call_args_listr r r side_effect_mock_childrenr r r r r r r r rc _mock_delegate) rl rH r^ r r r r r r r r s `` r& r r s GL8 8 8 8 83 3 3 3 37 7 7 7 78 8 8 8 8= = = = =6 6 6 6 65 5 5 5 5 GNG G &[[G$;;G"G ,G *G ! 0G!3G &=G# /G -G#G )G 1G !3G G!Dr% c t j j _ d _ d _ t _ fd}dD ]!}t |t || "d S )Nr c : t j | |i |S r@ )r5 rH )attrr\ r] rH s r& wrapperz"_setup_async_mock..wrapper$ s$ 'wty$''8888r% )assert_awaitedassert_awaited_onceassert_awaited_withassert_awaited_once_withassert_any_awaitassert_has_awaitsassert_not_awaited) asyncio
coroutines
_is_coroutine await_count
await_argsr await_args_listrk r )rH r rm s ` r& r r s +9DDDO$;;D
9 9 9 9 9, > > i)! >r% c $ d| dd z | k S )N__%s__ r$ r. s r& _is_magicr 6 s d1R4j D ((r% c $ e Zd ZdZd Zd Zd ZdS )_SentinelObjectz!A unique, named, sentinel object.c || _ d S r@ r selfr. s r& rL z_SentinelObject.__init__< s
r% c d| j z S Nz sentinel.%sr r s r& __repr__z_SentinelObject.__repr__? ty((r% c d| j z S r r r s r&
__reduce__z_SentinelObject.__reduce__B r r% N)r r! r" r# rL r r r$ r% r& r r : sG '' ) ) )) ) ) ) )r% r c $ e Zd ZdZd Zd Zd ZdS ) _SentinelzAAccess attributes to return a named object, usable as a sentinel.c i | _ d S r@ )
_sentinelsr s r& rL z_Sentinel.__init__H s
r% c l |dk rt | j |t | S )N __bases__)rP r
setdefaultr r s r& __getattr__z_Sentinel.__getattr__K s3 ; ))$0E0EFFFr% c dS )Nr r$ r s r& r z_Sentinel.__reduce__Q s zr% N)r r! r" r# rL r r r$ r% r& r r F sJ KK G G G r% r >
_mock_namer _mock_parentr _mock_new_name_mock_new_parent_mock_side_effect_mock_return_valuec x t | d| z }| |fd}| |fd}t || S )N_mock_c T | j }|t | | S t || S r@ )r r5 )r r. _the_namer^ s r& _getz"_delegating_property.._getf s/ ! ;4++ +sD!!!r% c R | j }| || j |< d S t ||| d S r@ )r rv rk )r valuer. r r^ s r& _setz"_delegating_property.._setk s9 ! ;',DM) $ $ $ Cu % % % % %r% )_allowed_namesaddproperty)r. r r r s r& _delegating_propertyr c se t 4I " " " "
$y & & & & D$ r% c e Zd Zd Zd ZdS )r c t |t st | | S t | }t | }||k rdS t d||z
dz D ]}| |||z }||k r dS dS )NFr r T)r3 rq __contains__lenrange)r r len_valuelen_selfisub_lists r& r z_CallList.__contains__x s %&& 2$$T511 1JJ t99 x 5q(Y.233 AAa kM*H5 tt !ur% c D t j t | S r@ )pprintpformatrq r s r& r z_CallList.__repr__ s ~d4jj)))r% N)r r! r" r r r$ r% r& r r v s2 * * * * *r% r c t | }t | sdS |j s|j s|j |j dS | }|||u rdS |j }||r| |_ ||_ |r| |_ ||_ dS )NFT)rJ r2 r r r r )parentr r. new_name_parents r& _check_and_set_parentr s % E U # # u U 1 ' +uG
e 5*
(!' ' # 4r% c e Zd Zd Zd ZdS ) _MockIterc . t | | _ d S r@ )iterr9 )r r9 s r& rL z_MockIter.__init__ s 99r% c * t | j S r@ )nextr9 r s r& __next__z_MockIter.__next__ s DH~~ r% N)r r! r" rL r r$ r% r& r r s2 r% r c e Zd ZeZdZd ZdS )BaseNc d S r@ r$ r r\ r] s r& rL z
Base.__init__ s r% )r r! r" r r r rL r$ r% r& r r s/
r% r c e Zd ZdZ e Z d-dZ d-dZd Zd.dZ d/d Z
d
Z d Z d Z
ee e e
Zed
Z ed Z ed Z ed Z ed Z ed Zd Zd Z eee Zd0ddddZd Zd Z d Z d Z d Z d Z d Z!d Z"d1d Z#d! Z$d" Z%d# Z&d$ Z'd% Z(d& Z)d' Z*d.d(Z+d) Z,d* Z-d2d,Z.dS )3r
z A non-callable version of `Mock`N Fc | f}
t | t s |p|}|t | r t | f}
t | j |
d| j i }t
t | | }|S )Nr# ) rA AsyncMockMixinr: rB r r# _safe_superr
__new__)clsspecr r. spec_setr _spec_state _new_name _new_parent_spec_as_instance _eat_selfunsafer] basesspec_argnewre s r& r
zNonCallableMock.__new__ s #~.. . '4H#
h(?(?#'-3<CK(@AA 44<>&& D ' C !$D
r% c b | j || j _ d S || _ t | |d d d S )Nr9 )r r r r )r r s r& __set_return_valuez"NonCallableMock.__set_return_valueJ s> */4D , , ,&+D # !$tT : : : : :r% z1The value to be returned when the mock is called.c < | j t | S | j S r@ )r+ rB r s r& __class__zNonCallableMock.__class__V s #:: r% r r r r r c | j }|| j S |j }|It | s:t |t
s%t
| st | }||_ |S r@ )r r r callabler3 r rE )r delegatedsfs r& __get_side_effectz!NonCallableMock.__get_side_effectc sj ' ) )
" N8B<>> attrs = {'method.return_value': 3, 'other.side_effect': KeyError}
>>> mock.configure_mock(**attrs)c 8 | d d S )Nr .)count)entrys r& z0NonCallableMock.configure_mock.. s q 1D1D r% )keyrS N)sorteditemssplitpopr5 rk )r r] argvalr\ finalr9 rU s r& r! z NonCallableMock.configure_mock s v||~~ $E#D F F F
%
%HC
99S>>DHHJJEC
*
* c5)) C $ $ $ $
%
%r% c
|dv rt | | j %|| j vs |t v rt d|z n t | rt | | j sC| j r || j vr3| d s |t v rt |d|d t j 5 | j
| }|t u rt | |Cd }| j t | j | }| | |||| }|| j |< nt |t rv t# |j |j |j |j |j }n># t. $ r1 | j d p| }t/ d|d |d
| d |j d w xY w|| j |< d d d n # 1 swxY w Y |S )
N> r r. z Mock object has no attribute %r)assertassretasertaseertassrtz6 is not a valid assertion. Use a spec for the mock if z is meant to be an attribute.)r r. r r r r Cannot autospec attr
from target , as it has already been mocked out. [target=, attr=r* )rP r. _all_magicsr r r+ _ATTRIB_DENY_LISTr
_lockr rw rN r r5 r; r3 rM r r r
re r r. r rv )r r. r r target_names r& r zNonCallableMock.__getattr__ s 4 4 4 && &
+4---1D1D$%F%MNNN 2E
t__ ' && & N$*< NDL^@^@^ OPP
NTX\mTmTm$ M M'+M M MN N N
" 4 4(,,T22F!!$T*** #/ $D$4d ; ;E -- d%4 $ . .4#D))FJ//
4
D , V_fo
v{ FF ( D D D"&- "="EK*C C C&C C#'C C28+C C CD D DD .4#D); 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4>
s+ 4B
G,E/.G/;F**
GG
Gc n | j g}| j }| }d}|dgk rd}|7|}| |j |z d}|j dk rd}|j }|7t t | }|j pd}t
| dk r|d dvr|dz
}||d< d | S )NrS r9 r rH r )r9 z().r )r r r3 rq reversedr r join)r
_name_listr lastdot_firsts r& _extract_mock_namez"NonCallableMock._extract_mock_name s )*
' $ C!D g 4s : ; ; ;C%-- .G ! (:..//
*F z??Q !}M11#
1
wwz"""r% c | }d}|dvrd|z }d}| j d}| j rd}|| j j z }dt | j ||dt | dS ) Nr )rH zmock.z name=%rz spec=%rz spec_set=%r)rt r+ r, r rB rK )r r. name_string spec_strings r& r zNonCallableMock.__repr__ s &&(( ( ( ($t+K '$K~
- , %(8(AAK JJ K KK tHHHH
r% c v t st | S | j pg }t t | }t
| j }d | j D }d |D }d |D }t t ||z |z |z S )z8Filter the output of `dir(mock)` to only useful members.c * g | ]\ }}|t u|S r$ )rN )r- m_namem_values r&
z+NonCallableMock.__dir__.. s1 ( ( (&vwh&&
&&&r% c < g | ]}| d |S r( r* r- es r& r} z+NonCallableMock.__dir__.. s) CCC1 c1B1BCQCCCr% c Z g | ](}| d rt | &|)S r( )r+ r r s r& r} z+NonCallableMock.__dir__.. sE # # #1 c1B1B # q\\#Q # # #r% ) r object__dir__r. r0 rB rq rv r rY rX set)r extras from_type from_dictfrom_child_mockss r& r zNonCallableMock.__dir__
s (>>$'' '#)rT
OO '' ( (*.*=*C*C*E*E ( ( ( DC CCC # # # # # c&9,y8;KKLLMMMr% c T |t v r t || S j r+ j $| j vr| j vrt
d|z |t v rd|z }t
| |t v r j | j vrt
d|z t | s5t t |t || | fd}nft |d | t t || | j |< n+|dk r | _ d S t ||| r
| j |< j r;t# | s+ d| }t
d| t || S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c g| R i |S r@ r$ )r\ kwr r s r& rV z-NonCallableMock.__setattr__..1 s! HHT,GD,G,G,GB,G,G r% r@ rS z Cannot set )r r __setattr__r, r. rv rP _unsupported_magicsri r2 rk rB _get_methodr r r+ r r4 rt )r r. r msg mock_namer s ` @r& r zNonCallableMock.__setattr__ s > ! !%%dD%88 8n 2!3!? * * *
% % !Dt!KLL L
(
(
(BTIC %% %
[
!-$d>P2P2P$%H4%OPPP$U++
2T
D+dE*B*BCCC GGGGG &dE4>>>T
D%000,1#D))
[
$D F$T5$==
2,1#D) |t | j v r(t t | | || j vrd S | j |t }|| j v r)t t | | n|t u rt | |t ur| j |= t | j |< d S r@ ) ri rB rv delattrr rw _missingr r
__delattr__rN rP )r r. r9 s r& r zNonCallableMock.__delattr__F s ; 44::+>#>#> DJJ % % %4=(( !%%dH55 4= . . : :4 @ @ @ @
H__ && & h #D)$,D!!!r% c 6 | j pd}t ||| S rG )r _format_call_signaturer r\ r] r. s r& _format_mock_call_signaturez+NonCallableMock._format_mock_call_signatureX s (&%dD&999r% r c d d}| || }| j } | j | }||||fz S )Nz.expected %s not found.
Expected: %s
Actual: %s)r r )r r\ r] actionmessageexpected_stringr
actual_strings r& _format_mock_failure_messagez,NonCallableMock._format_mock_failure_message] sD D ::4HHN 88)D
&/=AAAr% c |s| j S d}| dd d }| j }|D ]M}| | }|t |t r n t | }|j }|j }N|S )aH
* If call objects are asserted against a method/function like obj.meth1
then there could be no name for the call object to lookup. Hence just
return the spec_signature of the method/function being asserted against.
* If the name is not empty then remove () and split by '.' to get
list of names to iterate through the children until a potential
match is found. A child mock is created only during attribute access
so if we get a _SpecState then no attributes of the spec were accessed
and can be safely exited.
Nr9 r rS )r- replacerZ r rw r3 rM rJ )r r. r^ nameschildrenrP s r& _get_call_signature_from_namez-NonCallableMock._get_call_signature_from_namee s (' ' T2&&,,S11&
,
,D LL&&E}
5* = =}
&e,, /+
r% c t |t r/t | dk r | |d }n| j }|vt | dk rd}|\ }}n|\ }}} |j |i |}t
||j |j S # t $ r }|
d cY d}~S d}~ww xY w|S )a
Given a call (or simply an (args, kwargs) tuple), return a
comparison key suitable for matching with other calls.
This is a best effort method which relies on the spec's signature,
if available, or falls back on the arguments themselves.
r r Nr ) r3 rr r r r- rZ r r\ r] TypeErrorwith_traceback)r _callr^ r. r\ r]
bound_callr s r&
_call_matcherz NonCallableMock._call_matcher s eU # # 'E
Q44U1X>>CC&C ?5zzQ $ ff%*"dF
. %SXt 6v 6 6
D*/:3DEEE
.
.
.''--------
. Ls 0'B
C"B<6C<Cc | }|j dk r8d|j pdd|j d| }t | dS )z/assert that the mock was never called.
r
Expected 'rH z"' to not have been called. Called times.Nr r _calls_reprAssertionErrorr[ r r s r& r z!NonCallableMock.assert_not_called sc ?a o///ooo&&(((*C !%% % r% c V | }|j dk rd|j pdz }t | dS )z6assert that the mock was called at least once
r z"Expected '%s' to have been called.rH N)r r r r s r& r z NonCallableMock.assert_called s@ ?a 7?,f.C %% % r% c | }|j dk s8d|j pdd|j d| }t | dS )z3assert that the mock was called only once.
r r rH z#' to have been called once. Called r Nr r s r& r z"NonCallableMock.assert_called_once sc !###o///ooo&&(((*C !%% % $#r% c x | j / }d}d|d|}t | fd} t fd } j }||k r1t |t r|nd}t | |dS )zassert that the last call was made with the specified arguments.
Raises an AssertionError if the args and keyword args passed in are
different to the last call to the mock.Nz not called.z#expected call not found.
Expected:
Actual: c 4 } | S r@ r r r\ r] r s r& _error_messagez:NonCallableMock.assert_called_with.._error_message s 33D&AACJr% Ttwo)r r r r _Callr3 Exception) r[ r\ r] expectedactual
error_messager causer s `` @r& r z"NonCallableMock.assert_called_with s
> !77fEEH"FF xx )M // / %%eT6N&E&E&EFF##DN33 X *8Y ? ?IHHTE !1!122 = r% c | }|j dk s8d|j pdd|j d| }t | |j |i |S )ziassert that the mock was called exactly once and that that call was
with the specified arguments.r r rH z ' to be called once. Called r )r r r r r r[ r\ r] r r s r& r z'NonCallableMock.assert_called_once_with ss !###o///ooo&&(((*C !%% %&t&7777r% c fd|D }t d |D d }t fd j D }|su||vro|d}n d d |D }t | dt | d
d |dS t | }g }|D ]=} | | # t $ r |
| Y :w xY w|r-t j pd d
t | d|d |dS )a assert the mock has been called with the specified calls.
The `mock_calls` list is checked for the calls.
If `any_order` is False (the default) then the calls must be
sequential. There can be extra calls before or after the
specified calls.
If `any_order` is True then the calls can be in any order, but
they must all appear in `mock_calls`.c : g | ]} | S r$ r r- cr s r& r} z4NonCallableMock.assert_has_calls.. ' 999aD&&q))999r% c 3 D K | ]}t |t |V d S r@ r3 r r s r& z3NonCallableMock.assert_has_calls.. s1 FFAZ9-E-EFaFFFFFFr% Nc 3 B K | ]} | V d S r@ r r s r& r z3NonCallableMock.assert_has_calls.. s1 M Md 0 0 3 3 M M M M M Mr% zCalls not found.z+Error processing expected calls.
Errors: {}c @ g | ]}t |t r|nd S r@ r r s r& r} z4NonCallableMock.assert_has_calls.. ; $7 $7 $7() *4Ay)A)A$KAAt $7 $7 $7r%
Expected: Actual)prefixrS rH z does not contain all of z in its call list, found z instead)
r r r formatr r rstriprq removerR r3 r rr ) r calls any_orderr r all_callsproblem not_foundkalls ` r& r z NonCallableMock.assert_has_calls s :9995999FFFFFMM M M M MT_ M M MMM y((= 0GG ,-3V$7 $7-5$7 $7 $7.8 .8 % G G!*5!1!1G''x'88??DDG G
FOO ' 'D
' &&&&
'
'
' &&&&&
' &*o&?&?&?&+I&6&6&6&6 C
s C-- DDc $ t ||fd }t |t r|nd} fd j D }|s|t | vr) || }t d|z |dS )zassert the mock has been called with the specified arguments.
The assert passes if the mock has *ever* been called, unlike
`assert_called_with` and `assert_called_once_with` that only pass if
the call is the most recent one.Tr Nc : g | ]} | S r$ r r s r& r} z3NonCallableMock.assert_any_call.. s' EEEA$$$Q''EEEr% z%s call not found)r r r3 r r _AnyComparerr r )r r\ r] r r r r s ` r& r z NonCallableMock.assert_any_call s %%eT6N&E&E&EFF &x ; ;EEEEE1DEEE HL$8$888 " > >tV L LO #o5
98r% c | j r7d|v r d|d nd}| |z }t | | d }|| j d v r t di |S t
| }t |t r|t v rt
}nt |t r)|t v s| j r|| j v rt }ndt
}n\t |t s:t |t rt }n*t |t rt }n
|j d } |di |S )aP Create the child mocks for attributes and return value.
By default child mocks will be the same type as the parent.
Subclasses of Mock may want to override this to customize the way
child mocks are made.
For non-callable mocks the callable variant will be used (rather than
any custom subclass).r. rS r9 r r/ r r$ )r rt rP rw rv r
rB rA r _async_method_magicsr _all_sync_magicsr.
CallableMixinr r
r ru )r r rm r r _typeklasss r& r; z NonCallableMock._get_child_mock! s[ ,,2bLL(BvJ(((dI //11I=I ++ +FF;''
n5 5 5 ??r?? "T
eY ' ' %I9M,M,M EE
~
.
.
% ---& .+48J+J+J!!E=11 %%!566
!E?33
M!$Eu{{r{{r% Callsc J | j sdS d| dt | j dS )zRenders self.mock_calls as a string.
Example: "
Calls: [call(1), call(2)]."
If self.mock_calls is empty, an empty string is returned. The
output will be truncated if very long.
r
z: rS )r r )r r s r& r zNonCallableMock._calls_reprG s6 2;F;;i 8 8;;;;r% ) NNNNNNr NFNFFFFr@ )r )r )/r r! r" r# r rk r
rL r$ r( r "_NonCallableMock__get_return_value"_NonCallableMock__set_return_value"_NonCallableMock__return_value_docr r r@ r r r r r r !_NonCallableMock__get_side_effect!_NonCallableMock__set_side_effectr r r! r rt r r r r r r r r r r r r r r r r; r r$ r% r& r
r
s **
EGGE =AEIBEI > >.
8
8
8* * * *Z
# # #L
<
<
<
<
<
.k s1 $Hf F" r% TF)r allzip)r itemr s r& r z_AnyComparer.__contains__h s{ Et99E
**** (+D%(8(8
tt
ur% N)r r! r" r# r r$ r% r& r r c s- r% r c | | S t | r| S t | r| S t | S # t $ r | cY S w xY wr@ )rE ro r r r8 s r& rG rG s sk
{
S
~~
Cyy
s 7 AAc
H e Zd Zddedddddddf
dZd Zd Zd Zd Zd Z dS ) r Nr c
x || j d< t t | j ||||||| |
fi | || _ d S )Nr )rv r r rL r ) r r r r r r. r
r r r r r] s r& rL zCallableMixin.__init__ sa /;
*+1 M4((1 %x K
39
'r% c d S r@ r$ r s r& rb z CallableMixin._mock_check_sig s r% c P | j |i | | j |i | | j |i |S r@ )rb _increment_mock_call
_mock_callr[ r\ r] s r& rO zCallableMixin.__call__ sL #
"D3F333'
'8888$z$d5f555r% c | j |i |S r@ )_execute_mock_callr s r& r zCallableMixin._mock_call s ,z,d=f===r% c | }d|_ |xj dz
c_ t ||fd }||_ |j | |j d u}|j }|j }|dk }|j t d||f |j
} | |rB| j t |||f | j d u}|r
| j dz |z }t |||f }
| j |
| j r |rd} nd} | j dk }| j | z |z }| j
} | d S d S )NTr r r9 r rS ) r r r r r r3 r r r r r r ) r[ r\ r] r r do_method_callsmethod_call_namemock_call_name is_a_callr this_mock_callrr s r& r z"CallableMixin._increment_mock_call s 1
tVn$/// ""5))) +47 ? ,"d* ub$%7 8 8999 + %
W(//7Gv6V0W0WXXX"-":$"F" W'2'='CFV'V$ #ND&#A B BN " ) ). 9 9 9)
S CC C ' 6$ > !,!;c!AN!R &6K- %%%%%r% c | }|j }|Tt | r|t | s!t | }t | r|n ||i |}|t ur|S |j t ur|j S |j
|j |i |S |j S r@ )r rE ro r r r r r )r[ r\ r] r effectr s r& r z CallableMixin._execute_mock_call s ! V$$
1 v&&
1 f (( ! L! 000W$$
"' 1 1$ $ '#4#T4V44 4 r% )
r r! r" r rL rb rO r r r r$ r% r& r r s d$d !RT ' ' ' '
6 6 6> > >-7 -7 -7^! ! ! ! !r% r c e Zd ZdZdS )r a
Create a new `Mock` object. `Mock` takes several optional arguments
that specify the behaviour of the Mock object:
* `spec`: This can be either a list of strings or an existing object (a
class or instance) that acts as the specification for the mock object. If
you pass in an object then a list of strings is formed by calling dir on
the object (excluding unsupported magic attributes and methods). Accessing
any attribute not in this list will raise an `AttributeError`.
If `spec` is an object (rather than a list of strings) then
`mock.__class__` returns the class of the spec object. This allows mocks
to pass `isinstance` tests.
* `spec_set`: A stricter variant of `spec`. If used, attempting to *set*
or get an attribute on the mock that isn't on the object passed as
`spec_set` will raise an `AttributeError`.
* `side_effect`: A function to be called whenever the Mock is called. See
the `side_effect` attribute. Useful for raising exceptions or
dynamically changing return values. The function is called with the same
arguments as the mock, and unless it returns `DEFAULT`, the return
value of this function is used as the return value.
If `side_effect` is an iterable then each call to the mock will return
the next value from the iterable. If any of the members of the iterable
are exceptions they will be raised instead of returned.
* `return_value`: The value returned when the mock is called. By default
this is a new Mock (created on first access). See the
`return_value` attribute.
* `unsafe`: By default, accessing any attribute whose name starts with
*assert*, *assret*, *asert*, *aseert*, or *assrt* raises an AttributeError.
Additionally, an AttributeError is raised when accessing
attributes that match the name of an assertion method without the prefix
`assert_`, e.g. accessing `called_once` instead of `assert_called_once`.
Passing `unsafe=True` will allow access to these attributes.
* `wraps`: Item for the mock object to wrap. If `wraps` is not None then
calling the Mock will pass the call through to the wrapped object
(returning the real result). Attribute access on the mock will return a
Mock object that wraps the corresponding attribute of the wrapped object
(so attempting to access an attribute that doesn't exist will raise an
`AttributeError`).
If the mock has an explicit `return_value` set then calls are not passed
to the wrapped object and the `return_value` is returned instead.
* `name`: If the mock has a name then it will be used in the repr of the
mock. This can be useful for debugging. The name is propagated to child
mocks.
Mocks can also be called with arbitrary keyword arguments. These will be
used to set attributes on the mock after it is created.
Nr r$ r% r& r r s 7 7 7 7r% r c t | | S # t $ r" t | t | | cY S w xY wr@ )r5 rP
__import__)thingcomp import_paths r& _dot_lookupr ' sV $ud### $ $ $; ud#####$s )>>c | d }| d }t | }|D ]}|d|z z
}t ||| } |S )NrS r z.%s)rZ r[ r
r )target
componentsr r r s r& _importerr / sf c""J ..##K { # #E 6 6ut|# E455 Lr% c @ d}|D ]}|| v rt |d d S )N) autospect auto_specset_specz5 might be a typo; use unsafe=True if this is intended) RuntimeError)kwargs_to_checktypostypos r& _check_spec_arg_typosr < sN 2E ? " " PPP
# r% c ~ e Zd ZdZg ZdddZd Zd Zd Ze j
d Z d Z d
Z
d Zd Zd
Zd Zd ZdS )_patchNFr c
|)|t urt d |t d |
st | t | rt d|d|d t | rt d|d|d || _ || _ || _ || _ || _ || _
d| _ || _ || _
| | _ g | _ d S )Nz,Cannot use 'new' and 'new_callable' togetherz1Cannot use 'autospec' and 'new_callable' togetherzCannot spec attr z0 as the spec has already been mocked out. [spec=r* z? as the spec_set target has already been mocked out. [spec_set=F)r rR r r2 r getterrm r new_callabler create has_localr
autospecr] additional_patchers) r r# rm r r r% r
r' r$ r] r s r& rL z_patch.__init__J sV #'!! B # G * !& ) ) ) T " " A"@I @ @6:@ @ @A A
A X & & P"OI O OAIO O OP P
P " (
#% r% c t | j | j | j | j | j | j | j | j | j }| j
|_
d | j D |_ |S )Nc 6 g | ]}| S r$ )copy)r- ps r& r} z _patch.copy..v s- '
'
'
AFFHH'
'
'
r% ) r r# rm r r r% r
r' r$ r] attribute_namer( )r patchers r& r+ z _patch.copyo sq K 49 K M4,dk
"&!4 '
'
" 6'
'
'
# r% c t |t r| | S t j | r| | S | | S r@ r3 rB decorate_classr6 r decorate_async_callabledecorate_callable)r r= s r& rO z_patch.__call__| sc dD ! ! -&&t,, , &t , , 6//55 5%%d+++r% c t | D ]q}| t j s"t || }t |d sC| }t || || r|S NrO )r0 r+ r TEST_PREFIXr5 r4 r+ rk )r r r
attr_valuer. s r& r1 z_patch.decorate_class s JJ 6 6D??5#455
--J:z22
iikkG E4!4!4 5 5 5 5 r% c # T K g }t j 5 }|j D ]W}| | }|j | | 4|j t u r| | X|t | z
}||fV d d d d S # 1 swxY w Y d S r@ )
contextlib ExitStack patchings
enter_contextr- updater r r3 rr )r patchedr\ keywargs
extra_args
exit_stackpatchingr\ s r& decoration_helperz_patch.decoration_helper s
!
#
# #z # -
+
+ ..x88*6 OOC(((( \W,, %%c*** E*%% %D" " " " # # # # # # # # # # # # # # # # # #s A8B B!$B!c t d r j S t fd g_ S )Nr; c | | | 5 \ }} |i |cd d d S # 1 swxY w Y d S r@ rC r\ r? newargs newkeywargsr= r> r s r& r> z)_patch.decorate_callable..patched s ''(,(02 2
55Kg{tW4 44
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5s 155r4 r; r3 r r r= r> s ``@r& r3 z_patch.decorate_callable sv 4 % % N ! !$ ' ' 'K t 5 5 5 5 5 5
5 "Fr% c t d r j S t fd g_ S )Nr; c K | | 5 \ }} |i | d {V cd d d S # 1 swxY w Y d S r@ rF rG s r& r> z/_patch.decorate_async_callable..patched s ''(,(02 2
;5Kg{ !T7 :k : :::::::
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;s
9= =rJ rK s ``@r& r2 z _patch.decorate_async_callable sv 4 % % N ! !$ ' ' 'K t ; ; ; ; ; ;
; "Fr% c ` | }| j }t }d} |j | }d}n-# t t
f$ r t
||t }Y nw xY w|t v r t |t rd| _
| j
s |t u rt |d| ||fS )NFTz does not have the attribute ) r# rm r rv rP KeyErrorr5 _builtinsr3 r r% )r r r. r locals r& get_originalz_patch.get_original s ~ t,H EE ) 6 6 6 vtW55HHH 6
9 FJ!?!? DK{ x7 2 2 7=vvttD
s
6 'A A c | j | j | j }}}| j | j }}| j }| | _ |du rd}|du rd}|du rd}||t d |||dvrt d | \ }}|t u r|d} |du r |}|du r|}d}n| |du r|}d}n|du r|}||/|t u rt d t |t rd} |t | rt }
nt }
i } ||}
nN||J|} ||} t! | rd| v}
nt# | }
t | rt }
n |
rt$ }
||| d < ||| d
< t |
t r&t' |
t( r| j r
| j | d < | | |
di | }| r_t/ | rP|} ||} t! | st1 | st$ }
| d |
d|d d
| |_ n||t urt d |t u rt d t7 | }|du r|}t/ | j r#t9 d| j d| j d|d t/ | rAt; | j d| j }t9 d| j d|d| j d|d t= |f|| j d|}n|rt d |}|| _ || _ tC j" | _# tI | j | j | | j% ci }| j t u r
||| j% < | j& D ]?}| j# ' | }|j t u r| | @|S |S # | j( tS j* s Y dS xY w)zPerform the patch.FNz Can't specify spec and autospec)TNz6Can't provide explicit spec_set *and* spec or autospecTz!Can't use 'spec' with create=TruerO r r
r. r9 r: zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=Truere z: as the patch target has already been mocked out. [target=rh r* r rf rg )r
_namez.Can't pass kwargs to a mock we aren't creatingr$ )+r r r
r' r] r$ r# r r rR r r3 rB r: r
r rs rB r rA r
rm r= r2 ry r[ r boolr r5 r
temp_originalis_localr9 r: _exit_stackrk r- r( r< __exit__sysexc_info)r r r r
r' r] r$ r rQ inheritKlass_kwargs this_spec not_callablerl new_attrr@ rB r\ s r& __enter__z_patch.__enter__ s "h 4=8T =$+&( kkmm 5==D u H u H 4 =>> >
!5 L ( ( TUU U ++--% '>>h .Gt|| t## 'H D!t## #H DT!!# 8#7w&&#$GHHH h-- # "G|
h 7 7|!!G'$!X%9 ' (II&& ;#-Y#>LL'/ ':':#:L ++ 1 %EE! 1 0E "&#&.
# 5$''
15/22
17;~
1"&. NN6 " " "%""'""C
4,S11
4 ! ' (I ++ 1&y111 0E F####(5 $4SD $4 $4+2$4 $4
! '!! ( 7"" GHHHH~~H4 # --
D&CDN C C #{C C5=C C CD D D !**
D %dk:t{ K K &CDN C C"C C #{C C5=C C CD D D
"( BX(,B B:@B BCC
N LMM M %
%/11 DK : : :".
8w&&7:Jt 23 $ 8 / /H *88BBC |w.."))#...!!J 4=#,..1
s BO O O<c h | j r/| j t ur!t | j | j | j ndt
| j | j | j sCt | j | j r | j dv r t | j | j | j | `| ` | `| j }| ` |j
| S )zUndo the patch.)r# r! ri __annotations__rj ) rW rV r rk r rm r r% r4 rX rY )r r[ rA s r& rY z_patch.__exit__m s = IT /w > > DK 1C D D D D DK 0 0 0;
I T^(L(L
I += = = T^T5GHHH M K%
"z"H--r% c b | }| j | |S z-Activate a patch, returning any created mock.)rb _active_patchesr3 r r s r& startz _patch.start s- !! ##D)))
r% c | j | n# t $ r Y dS w xY w| ddd S zStop an active patch.N)rg r rR rY r s r& stopz _patch.stop s[ ' ' - - - - 44 }}T4...s
++)r r! r" r- rg rL r+ rO r1 r9 contextmanagerrC r3 r2 rR rb rY ri rl r$ r% r& r r E s NO AF"& "& "& "& "&J
, , , # # # " " 0P P Pd. . .* / / / / /r% r c dd \ }n-# t t t f$ r t d w xY w fd}||fS )NrS r z,Need a valid target to patch. You supplied: c " t S r@ r r s r& rV z _get_target.. s Yv&& r% )rsplitr rR rP )r rm r# s ` r& _get_targetrs s G "MM#q 1 1 z> 2 G G G E6 E EG G GG '
&
&
&F 9 s *Ar! c t t u rt d fd}
t |
|||||||| |
S )a
patch the named member (`attribute`) on an object (`target`) with a mock
object.
`patch.object` can be used as a decorator, class decorator or a context
manager. Arguments `new`, `spec`, `create`, `spec_set`,
`autospec` and `new_callable` have the same meaning as for `patch`. Like
`patch`, `patch.object` takes arbitrary keyword arguments for configuring
the mock object it creates.
When used as a class decorator `patch.object` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
z3 must be the actual object to be patched, not a strc S r@ r$ rq s r& rV z _patch_object.. s V r% r! )rB strr r ) r rm r r r% r
r' r$ r r] r# s ` r&
_patch_objectrw sn $ F||s L L L
^^^F 3f(L& r% c t t u r fd}n fd}|st d t | }|d \ } }
t || |
|||||i } | | _ |dd D ]=\ } }
t || |
|||||i } | | _ | j | >| S )a Perform multiple patches in a single call. It takes the object to be
patched (either as an object or a string to fetch the object by importing)
and keyword arguments for the patches::
with patch.multiple(settings, FIRST_PATCH='one', SECOND_PATCH='two'):
...
Use `DEFAULT` as the value if you want `patch.multiple` to create
mocks for you. In this case the created mocks are passed into a decorated
function by keyword, and a dictionary is returned when `patch.multiple` is
used as a context manager.
`patch.multiple` can be used as a decorator, class decorator or a context
manager. The arguments `spec`, `spec_set`, `create`,
`autospec` and `new_callable` have the same meaning as for `patch`. These
arguments will be applied to *all* patches done by `patch.multiple`.
When used as a class decorator `patch.multiple` honours `patch.TEST_PREFIX`
for choosing which methods to wrap.
c " t S r@ rp rq s r& rV z!_patch_multiple.. s 6** r% c S r@ r$ rq s r& rV z!_patch_multiple.. s r% z=Must supply at least one keyword argument with patch.multipler r N) rB rv rR rq rY r r- r( r3 )
r r r% r
r' r$ r] r# rY rm r r. this_patchers
` r& _patch_multipler| s , F||s****
K
E1XNIs 3fh, G 'G ) 9 9 3 IsD&( lB
'0 ##**<8888 Nr% c X t | \ } }
t | |
||||||||
S )a:
`patch` acts as a function decorator, class decorator or a context
manager. Inside the body of the function or with statement, the `target`
is patched with a `new` object. When the function/with statement exits
the patch is undone.
If `new` is omitted, then the target is replaced with an
`AsyncMock if the patched object is an async function or a
`MagicMock` otherwise. If `patch` is used as a decorator and `new` is
omitted, the created mock is passed in as an extra argument to the
decorated function. If `patch` is used as a context manager the created
mock is returned by the context manager.
`target` should be a string in the form `'package.module.ClassName'`. The
`target` is imported and the specified object replaced with the `new`
object, so the `target` must be importable from the environment you are
calling `patch` from. The target is imported when the decorated function
is executed, not at decoration time.
The `spec` and `spec_set` keyword arguments are passed to the `MagicMock`
if patch is creating one for you.
In addition you can pass `spec=True` or `spec_set=True`, which causes
patch to pass in the object being mocked as the spec/spec_set object.
`new_callable` allows you to specify a different class, or callable object,
that will be called to create the `new` object. By default `AsyncMock` is
used for async functions and `MagicMock` for the rest.
A more powerful form of `spec` is `autospec`. If you set `autospec=True`
then the mock will be created with a spec from the object being replaced.
All attributes of the mock will also have the spec of the corresponding
attribute of the object being replaced. Methods and functions being
mocked will have their arguments checked and will raise a `TypeError` if
they are called with the wrong signature. For mocks replacing a class,
their return value (the 'instance') will have the same spec as the class.
Instead of `autospec=True` you can pass `autospec=some_object` to use an
arbitrary object as the spec instead of the one being replaced.
By default `patch` will fail to replace attributes that don't exist. If
you pass in `create=True`, and the attribute doesn't exist, patch will
create the attribute for you when the patched function is called, and
delete it again afterwards. This is useful for writing tests against
attributes that your production code creates at runtime. It is off by
default because it can be dangerous. With it switched on you can write
passing tests against APIs that don't actually exist!
Patch can be used as a `TestCase` class decorator. It works by
decorating each test method in the class. This reduces the boilerplate
code when your test methods share a common patchings set. `patch` finds
tests by looking for method names that start with `patch.TEST_PREFIX`.
By default this is `test`, which matches the way `unittest` finds tests.
You can specify an alternative prefix by setting `patch.TEST_PREFIX`.
Patch can be used as a context manager, with the with statement. Here the
patching applies to the indented block after the with statement. If you
use "as" then the patched object will be bound to the name after the
"as"; very useful if `patch` is creating a mock object for you.
Patch will raise a `RuntimeError` if passed some common misspellings of
the arguments autospec and spec_set. Pass the argument `unsafe` with the
value True to disable that check.
`patch` takes arbitrary keyword arguments. These will be passed to
`AsyncMock` if the patched object is asynchronous, to `MagicMock`
otherwise or to `new_callable` if specified.
`patch.dict(...)`, `patch.multiple(...)` and `patch.object(...)` are
available for alternate use-cases.
r! )rs r ) r r r r% r
r' r$ r r] r# rm s r& r r sD V $F++FI 3f(L& r% c V e Zd ZdZddZd Zd Zd Zd Zd Z d
Z
d Z d Z d
Z
d ZdS ) _patch_dicta#
Patch a dictionary, or dictionary like object, and restore the dictionary
to its original state after the test.
`in_dict` can be a dictionary or a mapping like container. If it is a
mapping then it must at least support getting, setting and deleting items
plus iterating over keys.
`in_dict` can also be a string specifying the name of the dictionary, which
will then be fetched by importing it.
`values` can be a dictionary of values to set in the dictionary. `values`
can also be an iterable of `(key, value)` pairs.
If `clear` is True then the dictionary will be cleared before the new
values are set.
`patch.dict` can also be called with arbitrary keyword arguments to set
values in the dictionary::
with patch.dict('sys.modules', mymodule=Mock(), other_module=Mock()):
...
`patch.dict` can be used as a context manager, decorator or class
decorator. When used as a class decorator `patch.dict` honours
`patch.TEST_PREFIX` for choosing which methods to wrap.
r$ Fc || _ t | | _ | j | || _ d | _ d S r@ )in_dictdictrL r= clear _original)r r rL r r] s r& rL z_patch_dict.__init__\ s> 6ll 6"""
r% c t |t r| | S t j | r| | S | | S r@ r0 )r fs r& rO z_patch_dict.__call__e sc a *&&q)) ) &q ) ) 3//22 2%%a(((r% c @ t fd }|S )Nc | i | S # w xY wr@ r
_unpatch_dictr\ r r r s r& _innerz-_patch_dict.decorate_callable.._innern sV
%q$~"~~""$$$$""$$$$s 3 A r r r r s `` r& r3 z _patch_dict.decorate_callablem 9 q % % % % %
%
r% c @ t fd }|S )Nc K | i | d {V S # w xY wr@ r r s r& r z3_patch_dict.decorate_async_callable.._innerz so
% Q^^^+++++++""$$$$""$$$$s
< Ar r s `` r& r2 z#_patch_dict.decorate_async_callabley r r% c t | D ]}}t || }| t j rLt |d r=A>B1 1CCc | j }| j }t | | | d S # t $ r |D ]
}|| ||< Y d S w xY wr@ )r r r r= rP )r r r rW s r& r z_patch_dict._unpatch_dict s ,>G - NN8 $ $ $ $ $ - - -
-
- '}
-
-
- -s 6 AAc < | j | dS )zUnpatch the dict.NF)r r )r r\ s r& rY z_patch_dict.__exit__ s! > % ur% c l | }t j | |S rf )rb r rg r3 rh s r& ri z_patch_dict.start s- !! %%d+++
r% c t j | n# t $ r Y dS w xY w| ddd S rk )r rg r rR rY r s r& rl z_patch_dict.stop s[ " ) )$ / / / / 44 }}T4...s "
00N)r$ F)r r! r" r# rL rO r3 r2 r1 rb r r rY ri rl r$ r% r& r r ? s 8 ) ) ) + + +8
-
-
- / / / / /r% r c | d S # t $ r t | }|D ]}| |= Y d S w xY wr@ )r rP rq )r keysrW s r& r r se
G}} C s !==c f t t j D ]} | dS )z7Stop all active patches. LIFO to unroll nested patches.N)rn r rg rl )r s r& _patch_stopallr s5 &011
r% testzlt le gt ge eq ne getitem setitem delitem len contains iter hash str sizeof enter exit divmod rdivmod neg pos abs invert complex int float index round trunc floor ceil bool next fspath aiter zsizeof r zDadd sub mul matmul truediv floordiv mod lshift rshift and xor or pow c # K | ] }d |z V
dS )zi%sNr$ r- ns r& r r s& 77519777777r% c # K | ] }d |z V
dS )zr%sNr$ r s r& r r s& 55q555555r% > r __get____set__r
__delete__
__format__r __missing__ __getstate__ __reversed__ __setstate__
__getformat__
__reduce_ex____getnewargs____subclasses____getinitargs____getnewargs_ex__c fd}| |_ |S )z:Turns a callable object (like a mock) into a real functionc | g|R i |S r@ r$ )r r\ r r= s r& methodz_get_method..method s# tD&4&&&2&&&r% )r )r. r= r s ` r& r r s( ' ' ' ' 'FO Mr% c h | ]}d |z S )r r$ )r- r s r& r/ r/ s* Hv r% > __aexit__ __anext__
__aenter__ __aiter__> __del__r
rL r __prepare__r __instancecheck____subclasscheck__c 6 t | S r@ )r __hash__r s r& rV rV 2 s V__T 2 2 r% c 6 t | S r@ )r __str__r s r& rV rV 3 s FNN4 0 0 r% c 6 t | S r@ )r
__sizeof__r s r& rV rV 4 s v 0 0 6 6 r% c x t | j d| dt | S )N/)rB r rt rK r s r& rV rV 5 s; $t**"5 ^ ^8O8O8Q8Q ^ ^TVW[T\T\ ^ ^ r% )r r r
__fspath__y ?g ?)
__lt____gt____le____ge____int__r __len__rY __complex__ __float____bool__ __index__r c fd}|S )Nc L j j }|t ur|S | u rdS t S NT)__eq__r r NotImplemented)otherret_valr s r& r z_get_eq..__eq__J s1 +0 ' ! !N 5==4 r% r$ )r r s ` r& _get_eqr I s# Mr% c fd}|S )Nc R j j t urt S | u rdS t S NF)__ne__r r r )r r s r& r z_get_ne..__ne__T s, ; ) 8 8N 5==5 r% r$ )r r s ` r& _get_ner S s# Mr% c fd}|S )Nc j j j } | t u rt g S t | S r@ )__iter__r r r r r s r& r z_get_iter..__iter__] s1 -2 g 88O G}} r% r$ )r r s ` r& _get_iterr \ s# Or% c fd}|S )Nc j j } | t u r t t g S t t | S r@ )r r r _AsyncIteratorr r s r& r z"_get_async_iter..__aiter__g s@ .3 g !$r((++ + d7mm,,,r% r$ )r r s ` r& _get_async_iterr f s$ - - - - -
r% )r r r r c & t |t }|t ur ||_ d S t | }| || }||_ d S t
| }| || |_ d S d S r@ )_return_valuesrw r r _calculate_return_value_side_effect_methodsr )rH r r. fixedreturn_calculatorr
side_effectors r& _set_return_valuer w s tW - -E G #/33D99$((.. *(,,T22M *]4 0 0 ! r% c e Zd Zd Zd ZdS )
MagicMixinc | t t | j |i | | d S r@ )_mock_set_magicsr r rL )r r\ r s r& rL zMagicMixin.__init__ sN . J%%.;;;; r% c t t z }|}t | dd X| | j }t }||z
}|D ](}|t
| j v rt | | )|t t
| j z
}t
| }|D ]!}t ||t || "d S )Nr. ) _magicsr r5 intersectionr. r rB rv r rk
MagicProxy)r orig_magics these_magics
remove_magicsrU r s r& r zMagicMixin._mock_set_magics s 44 " 4 $ / / ;&33D4FGGL EEM ', 6M&
)
)DJJ ///D%((( $c$t***=&>&>> T
! ; ;E E5*UD"9"9 : : : : ; ;r% N)r r! r" rL r r$ r% r& r r s2 ; ; ; ; ;r% r c e Zd ZdZddZdS )r z-A version of `MagicMock` that isn't callable.Fc Z | || | dS r& r r r' s r& r( z"NonCallableMagicMock.mock_add_spec 2
D(+++ r% Nr r r! r" r# r( r$ r% r& r r s. 77 r% r c e Zd ZdS )AsyncMagicMixinN)r r! r" r$ r% r& r r s Dr% r c e Zd ZdZddZdS )r a
MagicMock is a subclass of Mock with default implementations
of most of the magic methods. You can use MagicMock without having to
configure the magic methods yourself.
If you use the `spec` or `spec_set` arguments then *only* magic
methods that exist in the spec will be created.
Attributes and the return value of a `MagicMock` will also be `MagicMocks`.
Fc Z | || | dS r& r r' s r& r( zMagicMock.mock_add_spec r r% Nr r r$ r% r& r r s2 r% r c " e Zd Zd Zd ZddZdS )r c " || _ || _ d S r@ r. r )r r. r s r& rL zMagicProxy.__init__ s r% c | j }| j }| ||| }t ||| t ||| |S )N)r. r r )r. r r; rk r )r rU r ms r& create_mockzMagicProxy.create_mock sZ " "/5
#
7
7q!!!&!U+++r% Nc * | S r@ )r
)r r9 r s r& r zMagicProxy.__get__ s !!!r% r@ )r r! r" rL r
r r$ r% r& r r sF " " " " " "r% r c e Zd Z ed Z ed Z ed Z fdZd Zd Z d Z
d Z d Z d
Z
dd Zd
Z fdZ xZS )r r r r c T t j |i | t j j | j d< d| j d< d | j d< t
| j d< t t }t |j d< t |j d< t j t j
z t j z |_ d|_ d |_ d|_ n# t& $ r Y nw xY wd|_ || j d
< d | j d < t+ | j d
< i | j d< d | j d< d S )Nr r _mock_await_count_mock_await_args_mock_await_args_listr
r+ r- )r\ r] r< r
r ri rj rd )superrL r r r rv r r
_CODE_ATTRSr _CODE_SIGr6 CO_COROUTINE
CO_VARARGSCO_VARKEYWORDSco_flags co_argcount co_varnamesco_posonlyargcountrP co_kwonlyargcountrr )r r\ r] code_mockr@ s r& rL zAsyncMockMixin.__init__ sA $)&))) *1);)I
o&-.
)*,0
()1:
-.#[999 ,4 =)09 ,-
!$
%
!" 2 +,I ( ( D '( #$-
j!$/
j!(-
n%*,
&'+/
'(((s C
C C c d K | }t ||fd }|xj dz
c_ ||_ |j | |j }|t
| r|t | s8 t | }n# t $ r t w xY wt
| r|n&t | r ||i | d {V }n ||i |}|t ur|S |j
t ur|j S |j 4t |j r |j |i | d {V S |j |i |S |j S )NTr r )r r r r r3 r rE ro r
StopIterationStopAsyncIterationr r r r r )r[ r\ r] r r r
r s r& r z!AsyncMockMixin._execute_mock_call s tVn$/// A ##E***! V$$
1 v&&
1- !&\\FF$ - - - -,- !(( ! L!$V,,
1 %vt 6v 6 6666666 000W$$
"' 1 1$ $ '"4#344
? -T -t >v > >>>>>>>>#4#T4V44 4 s 3B Bc X | }|j dk r d|j pd d}t | dS )zA
Assert that the mock was awaited at least once.
r Expected rH z to have been awaited.Nr r r r s r& r z AsyncMockMixin.assert_awaited4 sD q Odo 7OOOC %% % ! r% c h | }|j dk s$d|j pd d|j d}t | dS )z@
Assert that the mock was awaited exactly once.
r r" rH $ to have been awaited once. Awaited r Nr# r s r& r z"AsyncMockMixin.assert_awaited_once= s\ 1$$9t 8& 9 9 # /9 9 9C %% % %$r% c l | j ) }t d| d fd} t fd } j }||k r1t |t r|nd}t | |dS )zN
Assert that the last await was with the specified arguments.
NzExpected await: z
Not awaitedc 8 d } | S )Nawait)r r r s r& r z:AsyncMockMixin.assert_awaited_with.._error_messageP s" 33D&3QQCJr% Tr )r r r r r r3 r )r[ r\ r] r r r r r s `` @r& r z"AsyncMockMixin.assert_awaited_withG s ? "77fEEH !KH!K!K!KLL L %%eT6N&E&E&EFF##DO44 X *8Y ? ?IHHTE !1!122 = r% c ~ | }|j dk s$d|j pd d|j d}t | |j |i |S )zi
Assert that the mock was awaited exactly once and with the specified
arguments.
r r" rH r% r )r r r r r s r& r z'AsyncMockMixin.assert_awaited_once_withZ sl
1$$9t 8& 9 9 # /9 9 9C %% %'t'8888r% c ( | t ||fd }t |t r|nd}fdj D }|s|t | vr) || }t d|z |dS )zU
Assert the mock has ever been awaited with the specified arguments.
Tr Nc : g | ]} | S r$ r r s r& r} z3AsyncMockMixin.assert_any_await..m s' FFFA$$$Q''FFFr% z%s await not found)r r r3 r r r r r )r[ r\ r] r r r r r s @r& r z AsyncMockMixin.assert_any_awaitf s %%eT6N&E&E&EFF &x ; ;EFFFF1EFFF HL$8$888 " > >tV L LO $6
98r% Fc 2 | fd|D }t d |D d x}}t fd j D }|sT||vrN|d}n d d |D }t | dt | d j |dS t | }g }|D ]=} | | # t $ r | | Y :w xY w|r t t | d
|dS ) a
Assert the mock has been awaited with the specified calls.
The :attr:`await_args_list` list is checked for the awaits.
If `any_order` is False (the default) then the awaits must be
sequential. There can be extra calls before or after the
specified awaits.
If `any_order` is True then the awaits can be in any order, but
they must all appear in :attr:`await_args_list`.
c : g | ]} | S r$ r r s r& r} z4AsyncMockMixin.assert_has_awaits.. r r% c 3 D K | ]}t |t |V d S r@ r r s r& r z3AsyncMockMixin.assert_has_awaits.. s1 N NAZ95M5M Na N N N N N Nr% Nc 3 B K | ]} | V d S r@ r r s r& r z3AsyncMockMixin.assert_has_awaits.. s1 S St 1 1! 4 4 S S S S S Sr% zAwaits not found.z,Error processing expected awaits.
Errors: {}c @ g | ]}t |t r|nd S r@ r r s r& r} z4AsyncMockMixin.assert_has_awaits.. r r% r r z not all found in await list)
r r r r r rq r rR r3 rr )
r[ r r r r
all_awaitsr r r r s
@r& r z AsyncMockMixin.assert_has_awaitst s 99995999 N N N N NPTUUU S S S Sd>R S S SSS
z))= 1GG ,-3V$7 $7-5$7 $7 $7.8 .8 % 6 6!*5!1!16 6 # 36 6
F*%%
' 'D
'!!$''''
'
'
' &&&&&
' 49)4D4D4D4DF
s :C C21C2c h | }|j dk r$d|j pd d|j d}t | dS )z9
Assert that the mock was never awaited.
r r" rH z# to not have been awaited. Awaited r Nr# r s r& r z!AsyncMockMixin.assert_not_awaited s\ q 9t 8& 9 9 # /9 9 9C %% % ! r% c | t j |i | d| _ d| _ t | _ dS )0
See :func:`.Mock.reset_mock()`
r N)r r r r r r r r\ r] r@ s r& r zAsyncMockMixin.reset_mock sB D+F+++ ({{ r% r )r r! r" r r r r rL r r r r r r r r r
__classcell__r@ s @r& r r s &&}55K%%l33J**+<==O 0 0 0 0 0D'! '! '!R& & && & &> > >&
9
9
9 + + + +Z& & &+ + + + + + + + +r% r c e Zd ZdZdS )r
aY
Enhance :class:`Mock` with features allowing to mock
an async function.
The :class:`AsyncMock` object will behave so the object is
recognized as an async function, and the result of a call is an awaitable:
>>> mock = AsyncMock()
>>> iscoroutinefunction(mock)
True
>>> inspect.isawaitable(mock())
True
The result of ``mock()`` is an async function which will have the outcome
of ``side_effect`` or ``return_value``:
- if ``side_effect`` is a function, the async function will return the
result of that function,
- if ``side_effect`` is an exception, the async function will raise the
exception,
- if ``side_effect`` is an iterable, the async function will return the
next value of the iterable, however, if the sequence of result is
exhausted, ``StopIteration`` is raised immediately,
- if ``side_effect`` is not defined, the async function will return the
value defined by ``return_value``, hence, by default, the async function
returns a new :class:`AsyncMock` object.
If the outcome of ``side_effect`` or ``return_value`` is an async function,
the mock async function obtained when the mock object is called will be this
async function itself (and not an async function returning an async
function).
The test author can also specify a wrapped object with ``wraps``. In this
case, the :class:`Mock` object behavior is the same as with an
:class:`.Mock` object: the wrapped object may have methods
defined as async function functions.
Based on Martin Richard's asynctest project.
Nr r$ r% r& r
r
s ' ' ' 'r% r
c $ e Zd ZdZd Zd Zd ZdS )_ANYz2A helper object that compares equal to everything.c dS r r$ r r s r& r z _ANY.__eq__ s tr% c dS r r$ r< s r& r z _ANY.__ne__ s ur% c dS )Nzr$ r s r& r z
_ANY.__repr__ s wr% N)r r! r" r# r r r r$ r% r& r: r: sG 88 r% r: c d| z }d}d d |D }d d | D }|r|}|r |r|dz
}||z
}||z S )Nz%s(%%s)r z, c , g | ]}t | S r$ )repr)r- r\ s r& r} z*_format_call_signature.. s 7 7 73T#YY 7 7 7r% c " g | ] \ }}|d |
S )=r$ )r- rW r s r& r} z*_format_call_signature.. s4 #-3333 r% )ro rY )r. r\ r] r formatted_args args_string
kwargs_strings r& r r s $ GN)) 7 7$ 7 7 788KII 17 M %$ ( # d "N-' ^ ##r% c e Zd ZdZ ddZ ddZd Zej Zd
Z d Z
d Z d
Z e
d Ze
d Zd Zd ZdS )r a
A tuple for holding the results of a call to a mock, either in the form
`(args, kwargs)` or `(name, args, kwargs)`.
If args or kwargs are empty then a call tuple will compare equal to
a tuple without those values. This makes comparisons less verbose::
_Call(('name', (), {})) == ('name',)
_Call(('name', (1,), {})) == ('name', (1,))
_Call(((), {'a': 'b'})) == ({'a': 'b'},)
The `_Call` object provides a useful shortcut for comparing with call::
_Call(((1, 2), {'a': 3})) == call(1, 2, a=3)
_Call(('foo', (1, 2), {'a': 3})) == call.foo(1, 2, a=3)
If the _Call has no name then it will match any name.
r$ r NFTc d}i }t | }|dk r|\ }}}n~|dk r<|\ } }
t | t r | }t |
t r|
}nD|
}nA| |
}}n<|dk r6|\ }t |t r|}nt |t r|}n|}|r t | ||f S t | |||f S )Nr$ r r )r r3 rv rr r
) r r r. r r from_kallr\ r] _lenfirstseconds r& r
z
_Call.__new__
s 5zz 199!& D$
QYY !ME6%%%
- fe,, $!DD #FF $ff
QYYFE%%%
E5))
6==tVn55 5}}S4v"6777r% c 0 || _ || _ || _ d S r@ )r r _mock_from_kall)r r r. r r rJ s r& rL z_Call.__init__5
s " ( r% c r t | }n# t $ r
t cY S w xY wd}t | dk r| \ }}n| \ }}}t | dd r#t |dd r| j |j k rdS d}|dk rdi }}n|dk r|\ }}}n|dk r?|\ } t | t r| }i }nit | t r| }di }}nMd}| }nH|dk r@|\ }
} t |
t r!|
}t | t r| i }}n d| }}n|
| }}ndS |r||k rdS ||f||fk S ) Nr r r Fr r$ rI r )r r r r5 r r3 rr rv ) r r len_other self_name self_args self_kwargs
other_name
other_args other_kwargsr rL rM s r& r z _Call.__eq__<
s "E
II " " "! ! ! ! " t99>>%) "I{{04 -Iy+ D.$ / / GE>SW4X4X %);;;5
>>')2 JJ
!^^38 0J
LL
!^^FE%''
% "
! E3''
% "
+-rL
$
!^^ !ME6%%%
9 "
fe,, :/5r JJ/16 JJ+0&L
5 y005 L)i-EEEs &&c | j t d||fd S | j dz }t | j ||f|| S )Nr r9 r r r r r s r& rO z_Call.__call__t
sN ? ""dF+$777 7%dotV44MMMMr% c n | j t |d S | j d|}t || d S )NF)r. rJ rS )r. r rJ rY )r r r. s r& r z_Call.__getattr__|
sD ? "de444 4 ///440$tu====r% c b |t j v rt t | | S r@ )rr rv rP __getattribute__)r r s r& r\ z_Call.__getattribute__
s+ 5> ! ! %%dD111r% c H t | dk r| \ }}n| \ }}}||fS )Nr r r s r& _get_call_argumentsz_Call._get_call_arguments
s2 t99>> LD&&!% D$V|r% c 6 | d S Nr r^ r s r& r\ z
_Call.args
''))!,,r% c 6 | d S )Nr ra r s r& r] z _Call.kwargs
rb r% c | j s%| j pd}| d rd|z }|S t | dk rd}| \ }}n+| \ }}}|sd}n | d sd|z }nd|z }t ||| S )Nr r9 zcall%sr zcall.%s)rO r r+ r r )r r. r\ r] s r& r z_Call.__repr__
s # ?,fDt$$
' $K t99>>D LD&&!% D$
' __T**
' 4' $%dD&999r% c g }| }|%|j r| | |j }|%t t | S )zFor a call object that represents multiple calls, `call_list`
returns a list of all the intermediate calls as well as the
final call.)rO r3 r r rn )r valsr s r& call_listz_Call.call_list
sW $
# E"""&E $ (((r% )r$ r NFT)r$ NNFT)r r! r" r# r
rL r r r rO r r\ r^ r r\ r] r rg r$ r% r& r r
s $ :? 8 8 8 8@ >C ) ) ) )2F 2F 2Fj ]FN N N> > >2 2 2 - - X- - - X-: : :*
)
)
)
)
)r% r )rJ c t | rt | } t | t }t | rt d| d t | }d| i} |rd| i} n| i } | r|rd| d< |st
| | | t }
t j
| ri } nL|r|rt d t }
n1t | st }
n|r|rt | st }
| d | }|} |d
} |
d||| |d | } t | t" r$|rt% | | } n#t' | | } nt) | | || | |s
| |j |< |r |s d |vrt- | |dd
| | _ t1 | D ]"}
t3 |
r t5 | |
}n# t6 $ r Y 1w xY wd|i}|rd|i}t |t" s t9 ||| |
| }|| j |
< n| }t | t" r| j }t= | |
| }||d< t? | rt }nt } |d||
|
|d|}|| j |
< | |_ t) ||| t |t" rtA | |
| $| S )a Create a mock object using another object as a spec. Attributes on the
mock will use the corresponding attribute on the `spec` object as their
spec.
Functions or methods being mocked will have their arguments checked
to check that they are called with the correct signature.
If `spec_set` is True then attempting to set attributes that don't exist
on the spec object will raise an `AttributeError`.
If a class is used as a spec then the return value of the mock (the
instance of the class) will have the same spec. You can use a class as the
spec for an instance object by passing `instance=True`. The returned mock
will only be callable if instances of the mock are callable.
`create_autospec` will raise a `RuntimeError` if passed some common
misspellings of the arguments autospec and spec_set. Pass the argument
`unsafe` with the value True to disable that check.
`create_autospec` also takes arbitrary keyword arguments that are passed to
the constructor of the created mock.z'Cannot autospec a Mock object. [object=r* r r
NTr zJInstance can not be True when create_autospec is mocking an async functionr. r )r r r r. r r9 )re rT r r )r r. r r )rd r$ )!rs rB r3 r2 r r> r r= r r6 isdatadescriptorr r
ro r ry r[ rI r r rf r r r r0 r r5 rP rM rH
_must_skipr rk )r r
re r rT r r] is_type
is_async_funcr^ r] r rH rU r r r rd child_klasss r& r r
s . ~~ Dzzt$$G 5 4*. 4 4 4 5 5 5"4((MtnG t$
,8 ,'+#$ & f%%% NN6 E %% % % ? > ? ?
?
t__ %$ %X %&8&>&> %$ KK & &EI 5 (W ( ( & ( (D $
&& 8 .'d33DD!$--DDtWh7778(, u% Fx FN&$@$@ +D(T26 F F F T 1& 1& U tU++HH H (# , (+F(M22 A XxuhGGC),D & &F$
..
# "488I"+F; "8,,
( ' ' + (V%5*0( ( &( (C *-D & *{}}C Xsi @ @ @ @ c= ) ) & D% % % % Ks G$$
G10G1c D t | t s |t | di v rdS | j } | j D ]f}|j |t }|t u r,t |t t f r dS t |t r|c S dS |S )z[
Return whether we should skip the first argument on spec's `entry`
attribute.
rv F) r3 rB r5 r@ ru rv rw r rN rM rI )r rU rk r r s r& rj rj L s
dD ! ! GD*b11 1 15~ ##E733 W f|[ 9 : : 55
.
. NNN55 Nr% c e Zd Z ddZdS )rM FNc Z || _ || _ || _ || _ || _ || _ d S r@ )r idsr
r re r. )r r r
r r. rq re s r& rL z_SpecState.__init__j s0
r% )FNNNF)r r! r" rL r$ r% r& rM rM h s. 48/4 r% rM c | t | t rt j | S t j | S r@ )r3 bytesioBytesIOStringIO) read_datas r&
_to_streamrx s4 )U## &z)$$${9%%%r% c x
t }|dg fd} fd} fd
fd fd} fd}t dddl}t t t |j t t |j at 2ddl}t t t |j
a | t d t
} t t j _
d j _
d j _
d j _
d j _
| j _
d < d j _ | j _ j _ | j _ | j _
fd
}|| _ | _
| S )a
A helper function to create a mock to replace the use of `open`. It works
for `open` called directly or used as a context manager.
The `mock` argument is the mock object to configure. If `None` (the
default) then a `MagicMock` will be created for you, with the API limited
to methods or attributes available on standard file handles.
`read_data` is a string for the `read`, `readline` and `readlines` of the
file handle to return. This is an empty string by default.
Nc Z j j j j S d j | i |S r` ) readlinesr r\ r] _statehandles r& _readlines_side_effectz)mock_open.._readlines_side_effect s7 ( 4#0 0"vay"D3F333r% c Z j j j j S d j | i |S r` )readr r| s r& _read_side_effectz$mock_open.._read_side_effect s4 ; # /;+ + vay~t.v...r% c ? V K E d {V d j | i |V NTr )readline)r\ r] _iter_side_effectr} s r& _readline_side_effectz(mock_open.._readline_side_effect sT $$&&&&&&&&& 6$&)$d5f55 5 5 5 6r% c 3 b K j j j j V d D ]} | V d S r )r r )liner} r~ s r& r z$mock_open.._iter_side_effect sU ? ' 3
3 o2222
31I DJJJJ r% c ^ j j j j S t d S r` )r r r )r} r~ s r& _next_side_effectz$mock_open.._next_side_effect s) ? ' 3?/ /F1I r% c 0 d S r@ )close)exctypeexcinstexctbr~ s r& _exit_side_effectz$mock_open.._exit_side_effect s r% r open)r. r )r r c t d< j j d k r d< d j _ t S )Nr r )rx r r r )r\ r] r r} r~ rw s r&
reset_dataz mock_open..reset_data sM y))q ? &&) 3 3--//F1I*0)FO 'r% )rx file_spec_iorq r r0
TextIOWrapperunionru open_specr r rb r writer r r{ r r r rY )
rH rw
_read_datar r r r r r r r r} r~ s
` @@@@r& r r s; I&&J$
F4 4 4 4 4 4
/ / / / / /
6 6 6 6 6 6
S!2 3 344::3s3;?O?O;P;PQQRR
S]]++,, |f9555
I
&
&
&F$*F! $FL #FK #'FO $(F! /FK%%''F1I"()FO #9F "3FO "3FO "3FO "D D Kr% c & e Zd ZdZd ZddZd ZdS )r aW
A mock intended to be used as a property, or other descriptor, on a class.
`PropertyMock` provides `__get__` and `__set__` methods so you can specify
a return value when it is fetched.
Fetching a `PropertyMock` instance from an object calls the mock, with
no args. Setting it calls the mock with the value being set.
c t di |S )Nr$ )r )r r] s r& r; z PropertyMock._get_child_mock s ""6"""r% Nc | S r@ r$ )r r9 obj_types r& r zPropertyMock.__get__ s
tvv
r% c | | d S r@ r$ )r r9 r] s r& r zPropertyMock.__set__ s S r% r@ )r r! r" r# r; r r r$ r% r& r r sP # # # r% r c ^ e Zd ZdZ fdZed fd
Z fdZd Z fdZ eddZ
d Z xZ S )
ThreadingMixinNc t | d t r|d j |d< n8t | d t r|d j |d< t j di |S )Nr timeoutr r$ )r3 rw r _mock_wait_timeoutr r; )r r r@ s r& r; z ThreadingMixin._get_child_mock s bffX&& 7 7 A xL ;ByMM
}--~
>
> A } - @ByM&uww&,,,,,r% r c t j |i | |t u r| j }t j | j d< g | j d< t j | j d< || j d< d S )N _mock_event_mock_calls_events_mock_calls_events_lockr )r rL _timeout_unsetDEFAULT_TIMEOUT threadingEventrv Lock)r r r\ r] r@ s r& rL zThreadingMixin.__init__ s{ $)&))) n $ $*G'0'8'8
m$.0
*+3<>3C3C
/0.5
*+++r% c t j |i | t j | j d< g | j d< dS )r4 r r N)r r r r rv r5 s r& r zThreadingMixin.reset_mock sH D+F+++'0'8'8
m$.0
*+++r% c | j 5 | j D ] \ }}}||f||fk r|c cd d d S !t j }| j |||f d d d n # 1 swxY w Y |S r@ )r r r r r3 )r
expected_argsexpected_kwargsr\ r] event new_events r& __get_eventzThreadingMixin.__get_event
s
) X X'+'>
!
!#fe&>m_%EEE LL X X X X X X X XE!))I # * *M?I+V W W W X X X X X X X X X X X X X X X s A-1A--A14A1c t j |i |}| || }| | j |S r@ )r r _ThreadingMixin__get_eventr r )r r\ r] ret_value
call_eventr@ s r& r zThreadingMixin._mock_call s^ &EGG&777 %%dF33
r% c |t u r| j }| j | s | j pd d| d}t | dS )zWait until the mock object is called.
`timeout` - time to wait for in seconds, waits forever otherwise.
Defaults to the constructor provided timeout.
Use None to block undefinetively.
r rH z was not called before timeout(z).N)r r r waitr r )r r r s r& wait_until_calledz ThreadingMixin.wait_until_called ss n $ $-G $$W$55 &o/ + + &+ + +C %% % & &r% c | || }| | j s(| || }t | d dS )zWait until the mock object is called with given args.
Waits for the timeout in seconds provided in the constructor.
r z call not foundN)r r r r r )r r\ r] r r s r& wait_until_any_call_withz'ThreadingMixin.wait_until_any_call_with. sk
v..zz$"9z:: F " > >tV L LO O!D!D!DEE E F Fr% )
r r! r" r r; r rL r r r r r r6 r7 s @r& r r s O- - - - - '5 6 6 6 6 6 6 61 1 1 1 1 ,: & & & & & F F F F F F Fr% r c e Zd ZdZdS )r a
A mock that can be used to wait until on calls happening
in a different thread.
The constructor can take a `timeout` argument which
controls the timeout in seconds for all `wait` calls of the mock.
You can change the default timeout of all instances via the
`ThreadingMock.DEFAULT_TIMEOUT` attribute.
If no timeout is set, it will block undefinetively.
Nr r$ r% r& r r 9 s Dr% r c 4 d| _ t | D ]} t | | }n# t $ r Y w xY wt |t
s:t |j | t rh|j | u rt | dS )a Disable the automatic generation of child mocks.
Given an input Mock, seals it to ensure no further mocks will be generated
when accessing an attribute that was not already defined.
The operation recursively seals the mock passed in, meaning that
the mock itself, any mocks generated by accessing one of its attributes,
and all assigned mocks without a name or spec will be sealed.
TN) r r0 r5 rP r3 r
r rw rM r r )rH r r s r& r r I s DD
d##AA H !_-- a&**400* = = % % GGG
s +
88c e Zd ZdZd Zd ZdS )r z8
Wraps an iterator in an asynchronous iterator.
c t || _ t t }t j |_ || j d< d S )Nr r< )iteratorr
r r6 CO_ITERABLE_COROUTINEr rv )r r r s r& rL z_AsyncIterator.__init__e s6
#X666 $ : $-
j!!!r% c ^ K t | j S # t $ r Y nw xY wt r@ )r r r r r s r& r z_AsyncIterator.__anext__k sA
&& & D s
%%N)r r! r" r# rL r r$ r% r& r r a s< . . . ! ! ! ! !r% r r r )NFNNN)FFNN)Nr )__all__r r9 rt r6 r rZ r builtinstypesr r r
unittest.utilr functoolsr r r rH r backportsr r r r0 rP r r r r: r> r2 rE rJ rV rf ra ro rs ry r r r r r r r r r r MISSINGr DELETEDrN r r rq r r r r r
rv r rP frozensetrj r rG r r r r r r rs rw r| r r r r r multiplestopallr6
magic_methodsr numericsro rZ inplaceright
_non_defaultsr r r _sync_async_magics
_async_magicsr ri r r r r r r r r r r r r r r r r rQ rL r r r
r: r r rr r r r rj rM rB r rI r r rx r r
TIMEOUT_UNSETr r r r r r$ r% r& r s (
2 2 2 2 2 2 2 2 2 2 # # # # # # $ $ $ $ $ $ $ $ * * * * * *C C C C Cy C C C
I Hcc(mm H H H
@ @ @ 2 2 2 F # # # #
& & & 4 .." ." ."b> > >6) ) ) ) ) ) ) )f ) ) ) 9;;
&* * * * * * * *( 6
6
^
< ^
< ^
< ^
< ^
: j
D ," " " " " " " "$ c(mm G ggh&7 > > ? ? N+ N+ N+ N+ N+T N+ N+ N+b( ( ( ( ( ( ( (V
6
dff$ $ $$v) v) v) v) v)E v) v) v)r
uu CG K*/K K K K K\ 8 D D
& & &S S S Sl 4 $ 'EF EF EF EF EFT EF EF EFP
NJ
0! ! ! ! ! ! ! ! ! !s %D- -D87D8