/usr/lib64/python3.9/unittest/__pycache__
Edit: /usr/lib64/python3.9/unittest/__pycache__/mock.cpython-39.opt-2.pyc (58581B)
a
i @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZmZ dd eeD Zd ZeZd
d Zd d
Zdd Zdd Zdd Zdd Z dddZ dd Z dd Z d d Z d d Z!dd!d"Z"d#d$ Z#d%d& Z$d'd( Z%G d)d* d*e&Z'G d+d, d,e&Z(e( Z)e)j*Z*e)j+Z,e)j-Z.h d-Z/d.d/ Z0G d0d1 d1e1Z2d2d3 Z3G d4d5 d5e&Z4G d6d7 d7e&Z5G d8d9 d9e5Z6e7e6j8Z9G d:d; d;e1Z:d
d? d?e5ZdDdE Z?G dFdG dGe&Z@dHdI ZAe*dddddfdJdKZBddLdMZCe*dddddfdNdOZDG dPdQ dQe&ZEdRdS ZFdTdU ZGeBeD_&eEeD_HeCeD_IeGeD_JdVeD_KdWZLdXZMdYNdZd[ eMO D ZPdYNd\d[ eMO D ZQh d]ZRd^d_ ZSd`d dYNeLeMePeQgO D ZTh daZUdbhZVeUeVB ZWeTeRB ZXeXeWB ZYh dcZZddde dfde dgde dhde diZ[e\e\e\e\djddddkdld djddm
Z]dndo Z^dpdq Z_drds Z`dtdu Zae^e_e`eadvZbdwdx ZcG dydz dze5ZdG d{d| d|ede6ZeG d}d~ d~edZfG dd dede=ZgG dd de5ZhG dd de5ZiG dd deiefe=ZjG dd de&Zkek Zldd ZmG dd denZoeoddZpdddZqdd ZrG dd de&ZseteqeteljufZvdawdd ZxdddZyG dd de=Zzdd Z{G dd dZ|dS ))Mock MagicMockpatchsentinelDEFAULTANYcallcreate_autospec AsyncMock
FILTER_DIRNonCallableMockNonCallableMagicMock mock_open PropertyMockseal N)iscoroutinefunction)CodeType
ModuleType
MethodType) safe_repr)wrapspartialc C s h | ]}| d s|qS _
startswith).0name r %/usr/lib64/python3.9/unittest/mock.py ' r Tc C s< t | rt| tsdS t| dr*t| d} t| p:t| S )NF__func__)_is_instance_mock
isinstancer hasattrgetattrr inspectZ isawaitableobjr r r
_is_async_obj/ s
r* c C s t | dd rt| S dS d S )N__code__F)r& r funcr r r _is_async_func7 s r. c C s t t| tS N)
issubclasstyper r( r r r r# > s r# c C s t | tp t | to t| tS r/ )r$
BaseExceptionr1 r0 r( r r r
_is_exceptionD s
r3 c C s" t | trt| dr| jS | S d S Nmock)r$
FunctionTypesr% r5 r( r r r
_extract_mockK s r7 c C s t | tr|s| j} d}n*t | tsDz
| j} W n tyB Y d S 0 |rTt| d }n| }z| t|fW S t y| Y d S 0 d S NT)
r$ r1 __init__r6 __call__AttributeErrorr r' signature
ValueError)r- Z as_instanceZeat_selfZsig_funcr r r _get_signature_objectT s
r> Fc sN t | || d u rd S \} fdd}t| | |t|_ t|_d S )Nc s j |i | d S r/ bindselfargskwargssigr r checksigv s z"_check_signature..checksig)r> _copy_func_detailsr1 _mock_check_sig
__signature__)r- r5 skipfirstinstancerG r rE r _check_signatureq s
rM c C s8 dD ].}zt ||t| | W q ty0 Y q0 qd S )N)__name____doc____text_signature__
__module__ __defaults____kwdefaults__)setattrr& r; )r- funcopy attributer r r rH } s
rH c C s@ t | trdS t | tttfr(t| jS t| dd d ur.checksigrU )Z
_checksig_r5 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs))r$ r1 r> rH rN isidentifierexec _setup_func) r5 originalrL rK resultr- rG r contextsrcrU r rE r _set_signature s$
rj c s _ fdd}fdd}fdd}fdd}fd d
}fd d }fd
d} fdd}
d _d _d _t _t _t _j _j _ j
_
| _ | _ | _
| _|
_| _| _| _| _ _d S )Nc s j | i |S r/ )assert_called_withrb r5 r r rk s z'_setup_func..assert_called_withc s j | i |S r/ )
assert_calledrb rl r r rm s z"_setup_func..assert_calledc s j | i |S r/ )assert_not_calledrb rl r r rn s z&_setup_func..assert_not_calledc s j | i |S r/ )assert_called_oncerb rl r r ro s z'_setup_func..assert_called_oncec s j | i |S r/ )assert_called_once_withrb rl r r rp s z,_setup_func..assert_called_once_withc s j | i |S r/ )assert_has_callsrb rl r r rq s z%_setup_func..assert_has_callsc s j | i |S r/ )assert_any_callrb rl r r rr s z$_setup_func..assert_any_callc s: t _t _ j} t| r6| ur6| d S r/ ) _CallList method_calls
mock_calls
reset_mock return_valuer# )retrU r5 r r rv s z _setup_func..reset_mockFr )r5 called
call_count call_argsrs call_args_listrt ru rw side_effect_mock_childrenrk rp rq rr rv rm rn ro rJ _mock_delegate) rU r5 rF rk rm rn ro rp rq rr rv r ry r re s8 re c sJ t jj _d _d _t _ fdd}dD ]}t |t|| q.d S )Nr c s t j| |i |S r/ )r& r5 )attrrC rD rl r r wrapper s z"_setup_async_mock..wrapper)assert_awaitedassert_awaited_onceassert_awaited_withassert_awaited_once_withassert_any_awaitassert_has_awaitsassert_not_awaited) asyncio
coroutines
_is_coroutine await_count
await_argsrs await_args_listrT r )r5 r rV r rl r _setup_async_mock s
r c C s d| dd | kS )N__%s__ r r r r r _is_magic s r c @ s$ e Zd Zdd Zdd Zdd ZdS )_SentinelObjectc C s
|| _ d S r/ r rB r r r r r9 s z_SentinelObject.__init__c C s
d| j S Nz sentinel.%sr rB r r r __repr__ s z_SentinelObject.__repr__c C s
d| j S r r r r r r
__reduce__ s z_SentinelObject.__reduce__N)rN rQ __qualname__r9 r r r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS ) _Sentinelc C s
i | _ d S r/ )
_sentinelsr r r r r9 " s z_Sentinel.__init__c C s |dkr t | j|t|S )N __bases__)r; r
setdefaultr r r r r __getattr__% s z_Sentinel.__getattr__c C s dS )Nr r r r r r r + s z_Sentinel.__reduce__N)rN rQ r r9 r r r r r r r s r > _mock_return_value_mock_new_parent_mock_new_namerw r~ _mock_side_effect _mock_parent
_mock_namec C s8 t | d| }| |fdd}| |fdd}t||S )NZ_mock_c S s" | j }|d u rt| |S t||S r/ )r r& )rB r _the_namerF r r r _get@ s
z"_delegating_property.._getc S s* | j }|d u r|| j|< n t||| d S r/ )r r^ rT )rB valuer r rF r r r _setE s z"_delegating_property.._set)_allowed_namesaddproperty)r r r r r r r _delegating_property= s
r c @ s e Zd Zdd Zdd ZdS )rs c C sl t |tst| |S t|}t| }||kr2dS td|| d D ]"}| ||| }||krD dS qDdS )NFr T)r$ rZ __contains__lenrange)rB r Z len_valueZlen_selfiZsub_listr r r r R s
z_CallList.__contains__c C s t t| S r/ )pprintZpformatrZ r r r r r ` s z_CallList.__repr__N)rN rQ r r r r r r r rs P s rs c C s| t |}t|sdS |js4|js4|jd us4|jd ur8dS | }|d urX||u rPdS |j}q<|rh| |_||_|rx| |_||_dS )NFT)r7 r# r r r r )parentr r new_name_parentr r r _check_and_set_parentd s* r c @ s e Zd Zdd Zdd ZdS ) _MockIterc C s t || _d S r/ )iterr) )rB r) r r r r9 s z_MockIter.__init__c C s
t | jS r/ )nextr) r r r r __next__ s z_MockIter.__next__N)rN rQ r r9 r r r r r r s r c @ s e Zd ZeZdZdd ZdS )BaseNc O s d S r/ r rA r r r r9 s z
Base.__init__)rN rQ r r r r r9 r r r r r s r c @ s` e Zd Zdd ZdKddZdd ZdLd
d ZdMd d
Zdd Zdd Z dZ
e ee e
Z e dd Z
edZedZedZedZedZdd Zd d Ze eeZdNddd d d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z d+d, Z d-d. Z d/d0 Z dOd2d3Z d4d5 Z!d6d7 Z"d8d9 Z#d:d; Z$dd? Z&d@dA Z'dPdBdCZ(dDdE Z)dFdG Z*dQdIdJZ+dS )Rr c O s| | f}t | tsTtj| g|R i |j}|d|d}|d urTt|rTt| f}t| j|d| j i}t
t | |}|S )Nspec_setspecrO )
r0 AsyncMockMixin _MOCK_SIGZ bind_partialZ argumentsr_ r* r1 rN rO _safe_superr __new__)clsrC kwbasesZ
bound_argsZspec_argnewrL r r r r s
zNonCallableMock.__new__N Fc K s |d u r |}| j }
||
d< ||
d< ||
d< ||
d< d|
d< |d urJ|}d}|
d u rZ|d u}
| ||| |
i |
d< ||
d < d |
d
< d|
d < d |
d < d
|
d< t |
d< t |
d< t |
d< | |
d< | r| jf i | tt| |||||| d S )Nr r r r F _mock_sealedTr _mock_wrapsr Z _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsrt _mock_unsafe)r^ _mock_add_specrs configure_mockr r r9 )rB r r r r r _spec_state _new_name _new_parent_spec_as_instance _eat_selfZunsaferD r^ r r r r9 s<
zNonCallableMock.__init__c C s0 t |}d |_d |_d|_d |_t| || d S )Nr )r7 r r r r rT )rB r5 rV Z
inner_mockr r r attach_mock s zNonCallableMock.attach_mockc C s | || d S r/ )r rB r r r r r
mock_add_spec s z NonCallableMock.mock_add_specc C s d }d }g }t |D ] }tt||d r|| q|d ur|t|s|t|trT|}nt|}t|||} | or| d }t |}| j}
||
d< ||
d< ||
d< ||
d< ||
d< d S )Nr _spec_class _spec_set_spec_signature
_mock_methods _spec_asyncs) dirr r& appendr\ r$ r1 r> r^ ) rB r r r r r r r r resr^ r r r r s*
z NonCallableMock._mock_add_specc C s8 | j }| jd ur| jj}|tu r4| j| dd}|| _|S )N()r r )r r rw r _get_child_mock)rB rx r r r Z__get_return_value s
z"NonCallableMock.__get_return_valuec C s, | j d ur|| j _n|| _t| |d d d S )Nr )r rw r r )rB r r r r Z__set_return_value s
z"NonCallableMock.__set_return_valuez1The value to be returned when the mock is called.c C s | j d u rt| S | j S r/ )r r1 r r r r __class__ s
zNonCallableMock.__class__rz r{ r| r} ru c C sN | j }|d u r| jS |j}|d urJt|sJt|tsJt|sJt|}||_|S r/ )r r r~ callabler$ r r3 )rB delegatedZsfr r r Z__get_side_effect' s z!NonCallableMock.__get_side_effectc C s( t |}| j}|d u r || _n||_d S r/ ) _try_iterr r r~ )rB r r r r r Z__set_side_effect2 s
z!NonCallableMock.__set_side_effectrw r~ c C s |d u r g }t | |v r d S |t | d| _d | _d| _t | _t | _t | _|r^t | _
|rhd | _ | j
D ](}t|tsr|tu rqr|j|||d qr| j
}t|r|| ur|| d S )NFr r )idr rz r| r{ rs ru r} rt r r r r valuesr$
_SpecState_deletedrv r# )rB Zvisitedrw r~ childrx r r r rv = s, zNonCallableMock.reset_mockc K sX t | dd dD ]>\}}|d}| }| }|D ]}t||}q6t||| qd S )Nc S s | d dS )Nr .)count)entryr r r h r! z0NonCallableMock.configure_mock..)keyr )sorteditemssplitpopr& rT )rB rD argvalrC finalr) r r r r r [ s
z NonCallableMock.configure_mockc C s |dv rt |n:| jd ur<|| jvs.|tv rLt d| nt|rLt || jsd|drdt d| j|}|tu rt |np|d u rd }| j d urt
| j |}| j | |||| d}|| j|< n.t |t
rt|j|j|j|j|j}|| j|< |S )N> r r z Mock object has no attribute %r)assertZassretz1Attributes cannot start with 'assert' or 'assret')r r r r r )r; r _all_magicsr r r r r_ r r r& r r$ r r r r rL r r )rB r rg r r r r r q s:
zNonCallableMock.__getattr__c C s | j g}| j}| }d}|dgkr$d}|d urZ|}||j | d}|j dkrRd}|j}q$tt|}|jpnd}t|dkr|d dvr|d7 }||d< d|S )Nr r r r5 r )r z().r )r r r rZ reversedr r join)rB Z
_name_listr lastdotZ_firstr r r _extract_mock_name s(
z"NonCallableMock._extract_mock_namec C s^ | }d}|dvr d| }d}| jd urDd}| jr8d}|| jj }dt| j||t| f S )Nr )r5 zmock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r r r rN r1 r )rB r Z name_stringZ spec_stringr r r r s
zNonCallableMock.__repr__c C sv t st| S | jpg }tt| }t| j}dd | j D }dd |D }dd |D }t
t || | | S )Nc S s g | ]\}}|t ur|qS r )r )r Zm_nameZm_valuer r r
s z+NonCallableMock.__dir__..c S s g | ]}| d s|qS r r r er r r r r! c S s" g | ]}| d rt|r|qS r )r r r r r r r s ) r
object__dir__r r r1 rZ r^ r r r set)rB extrasZ from_typeZ from_dictZfrom_child_mocksr r r r s
zNonCallableMock.__dir__c sT |t v rt||S jrHjd urH|jvrH|jvrHtd| n|tv rbd| }t|n|tv rjd ur|jvrtd| t |st
t |t || | fdd}n(t
|d | t
t || |j|< n,|dkr|_d S t
|||r|j|< jrFt|sF d| }td| t||S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c s g| R i |S r/ r rC r rf rB r r r r! z-NonCallableMock.__setattr__..r r z Cannot set )r r __setattr__r r r^ r; _unsupported_magicsr r# rT r1 _get_methodr r r r r% r )rB r r msg mock_namer r r r s<
zNonCallableMock.__setattr__c C s |t v r2|t| jv r2tt| | || jvr2d S | j|t}|| jv r\tt| | n|t
u rlt ||tur|| j|= t
| j|< d S r/ ) r r1 r^ delattrr r_ _missingr r __delattr__r r; )rB r r) r r r r s
zNonCallableMock.__delattr__c C s | j pd}t|||S r4 )r _format_call_signaturerB rC rD r r r r _format_mock_call_signature s
z+NonCallableMock._format_mock_call_signaturer c C s. d}| ||}| j}| j | }||||f S )Nz.expected %s not found.
Expected: %s
Actual: %s)r r| )rB rC rD actionmessageexpected_stringr| Z
actual_stringr r r _format_mock_failure_message s
z,NonCallableMock._format_mock_failure_messagec C sj |s
| j S d }|ddd}| j}|D ]:}||}|d u sJt|trP qfq*t|}|j}|j }q*|S )Nr r r )r replacer r r_ r$ r r7 )rB r rF namesZchildrenr r r r _get_call_signature_from_name s
z-NonCallableMock._get_call_signature_from_namec
C s t |tr&t|dkr&| |d }n| j}|d urt|dkrNd}|\}}n
|\}}}z"|j|i |}t||j|jW S t y } z|
d W Y d }~S d }~0 0 n|S d S )Nr r r ) r$ r[ r r r r@ r rC rD TypeErrorwith_traceback)rB _callrF r rC rD Z
bound_callr r r r
_call_matcherA s
"z NonCallableMock._call_matcherc C s0 | j dkr,d| jpd| j | f }t|d S )Nr z9Expected '%s' to not have been called. Called %s times.%sr5 r{ r _calls_reprAssertionErrorrB r r r r rn \ s
z!NonCallableMock.assert_not_calledc C s$ | j dkr d| jpd }t|d S )Nr z"Expected '%s' to have been called.r5 )r{ r r r! r r r rm f s
z NonCallableMock.assert_calledc C s0 | j dks,d| jpd| j | f }t|d S )Nr z:Expected '%s' to have been called once. Called %s times.%sr5 r r! r r r ro n s
z"NonCallableMock.assert_called_oncec s j d u r. }d}d||f }t| fdd}t fdd}j }||krt|trv|nd }t| |d S )Nz not called.z0expected call not found.
Expected: %s
Actual: %sc s } | S r/ r r rC rD rB r r _error_message s z:NonCallableMock.assert_called_with.._error_messageTtwo)r| r r r _Callr$ Exception)rB rC rD expectedactualZ
error_messager% causer r$ r rk x s
z"NonCallableMock.assert_called_withc O s< | j dks,d| jpd| j | f }t|| j|i |S )Nr z3Expected '%s' to be called once. Called %s times.%sr5 )r{ r r r rk rB rC rD r r r r rp s
z'NonCallableMock.assert_called_once_withc s fdd|D }t dd |D d }t fdd jD }|s||vr|d u rXd}nddd |D }t| d t| jd
d d |d S t|}g }|D ]0}z|| W q t y |
| Y q0 q|rtd
j pdt ||f |d S )Nc s g | ]} |qS r r r cr r r r r! z4NonCallableMock.assert_has_calls..c s s | ]}t |tr|V qd S r/ r$ r) r r r r r! z3NonCallableMock.assert_has_calls..c 3 s | ]} |V qd S r/ r. r/ r r r r2 r! zCalls not found.z+Error processing expected calls.
Errors: {}c S s g | ]}t |tr|nd qS r/ r1 r r r r r s
Expected: ZActual)prefixr z@%r does not contain all of %r in its call list, found %r insteadr5 )
r rs ru formatr r rstriprZ remover= r r r[ ) rB calls any_orderr* r, Z all_callsproblem not_foundkallr r r rq sN
z NonCallableMock.assert_has_callsc sj t||fdd}t|tr$|nd } fdd jD }|sL|t|vrf ||}td| |d S )NTr& c s g | ]} |qS r r. r/ r r r r r! z3NonCallableMock.assert_any_call..z%s call not found)r r( r$ r) r} _AnyComparerr r rB rC rD r* r, r+ r r r r rr s z NonCallableMock.assert_any_callc K s | d}|| jd v r&tf i |S | jrZd|v rBd|d nd}| | }t|t| }t|trz|t v rzt}nbt|t
r|t v s| j r|| j v rt}qt}n4t|t
st|trt}qt|trt}n
|jd }|f i |S )Nr r r r r r )r_ r^ r r r r; r1 r0 r _async_method_magicsr _all_sync_magicsr
CallableMixinr r r r] )rB r r rV r _typeklassr r r r s2
z NonCallableMock._get_child_mockCallsc C s" | j s
dS d| dt| j dS )Nr
z: r )ru r )rB r4 r r r r s zNonCallableMock._calls_repr) NNNNNNr NFNF)F)FF)N)r )F)rD ),rN rQ r r r9 r r r Z"_NonCallableMock__get_return_valueZ"_NonCallableMock__set_return_valueZ"_NonCallableMock__return_value_docr rw r r rz r{ r| r} ru Z!_NonCallableMock__get_side_effectZ!_NonCallableMock__set_side_effectr~ rv r r r r r r r r r r r rn rm ro rk rp rq rr r r r r r r r s^
-
''
!
-&r c @ s e Zd Zdd ZdS )r= c C s, | D ]"}t dd t||D r dS qdS )Nc S s g | ]\}}||kqS r r )r r* r+ r r r r s z-_AnyComparer.__contains__..TF)allzip)rB itemr r r r r s z_AnyComparer.__contains__N)rN rQ r r r r r r r=
s r= c C sJ | d u r | S t | r| S t| r$| S z
t| W S tyD | Y S 0 d S r/ )r3 rY r r r( r r r r s
r c
@ sR e Zd Zddedddddddf
ddZdd Zdd Zd d
Zd d Zd
d Z dS )rA Nr c K s: || j d< tt| j||||||| |
fi | || _d S )Nr )r^ r rA r9 r~ ) rB r r~ rw r r r r r r r rD r r r r9 . s
zCallableMixin.__init__c O s d S r/ r rA r r r rI : s z CallableMixin._mock_check_sigc O s0 | j |i | | j|i | | j|i |S r/ )rI _increment_mock_call
_mock_callrA r r r r: ? s zCallableMixin.__call__c O s | j |i |S r/ )_execute_mock_callrA r r r rJ G s zCallableMixin._mock_callc O s d| _ | jd7 _t||fdd}|| _| j| | jd u}| j}| j}|dk}| j td||f | j
}|d ur|r|j t|||f |jd u}|r|jd | }t|||f} |j | |jr|rd}
nd}
|jdk}|j|
| }|j
}qpd S )NTr r& r r r ) rz r{ r( r| r} r r r r ru r rt ) rB rC rD r Zdo_method_callsZmethod_call_nameZmock_call_nameZ is_a_callr Zthis_mock_callr r r r rI J s4
z"CallableMixin._increment_mock_callc O s | j }|d urTt|r |n,t|s:t|}t|rH|n||i |}|turT|S | jturd| jS | jd ur~| j|i |S | jS r/ )r~ r3 rY r r r rw r )rB rC rD effectrg r r r rK x s
z CallableMixin._execute_mock_call)
rN rQ r r r9 rI r: rJ rI rK r r r r rA , s
.rA c @ s e Zd ZdS )r NrN rQ r r r r r r s r c C s6 z t | |W S ty0 t| t | | Y S 0 d S r/ )r& r;
__import__)thingcomp import_pathr r r _dot_lookup s
rR c C sB | d}|d}t|}|D ] }|d| 7 }t|||}q |S )Nr r z.%s)r r rN rR )target
componentsrQ rO rP r r r _importer s
rU c @ sz e Zd ZdZg Zdd Zdd Zdd Zdd Ze j
d
d Z d d
Z dd Z
dd Zdd Zdd Zdd Zdd ZdS )_patchNc
C sn |d ur(|t urtd|d ur(td|| _|| _|| _|| _|| _|| _d| _|| _ || _
| | _ g | _ d S )Nz,Cannot use 'new' and 'new_callable' togetherz1Cannot use 'autospec' and 'new_callable' togetherF)
r r= getterrV r new_callabler createZ has_localr autospecrD additional_patchers)
rB rW rV r r rY r rZ rX rD r r r r9 s( z_patch.__init__c
C sH t | j| j| j| j| j| j| j| j| j }| j
|_
dd | j D |_ |S )Nc S s g | ] }| qS r )copy)r pr r r r s z _patch.copy..) rV rW rV r r rY r rZ rX rD attribute_namer[ )rB patcherr r r r\ s z _patch.copyc C s2 t |tr| |S t|r(| |S | |S r/ )r$ r1 decorate_classr' r decorate_async_callabledecorate_callablerB r- r r r r: s
z_patch.__call__c C sN t |D ]@}|tjsqt||}t|ds0q| }t|||| q|S Nr: )r r r TEST_PREFIXr& r% r\ rT )rB rC r
attr_valuer_ r r r r` s
z_patch.decorate_classc c s g }t f}|jD ]8}||}|jd ur8|| q|jtu r|| q|t |7 }||fV W d n1 sx0 Y d S r/ )
contextlib ExitStack patchings
enter_contextr^ updater r r r[ )rB patchedrC keywargs
extra_args
exit_stackpatchingr r r r decoration_helper s
z_patch.decoration_helperc s> t dr j S t fddg_S )Nri c sD | |"\}} |i |W d S 1 s60 Y d S r/ rq rC rm ZnewargsZ newkeywargsr- rl rB r r rl 3 s z)_patch.decorate_callable..patchedr% ri r r rc r rt r rb - s
z_patch.decorate_callablec s> t dr j S t fddg_S )Nri c sJ | |(\}} |i |I d H W d S 1 s<0 Y d S r/ rr rs rt r r rl D s z/_patch.decorate_async_callable..patchedru rc r rt r ra > s
z _patch.decorate_async_callablec C s | }| j}t}d}z|j| }W n" ttfyF t||t}Y n0 d}|tv rdt|t rdd| _
| j
s|tu rtd||f ||fS )NFTz!%s does not have the attribute %r) rW rV r r^ r; KeyErrorr& _builtinsr$ r rY )rB rS r rf localr r r get_originalO s
z_patch.get_originalc C s | j | j| j }}}| j| j }}| j}| | _|du r@d }|du rLd }|du rXd }|d urp|d urptd|d us|d ur|dvrtd| \}}|t
u r|d u rd} |du r|}|du r|}d }n&|d ur|du r|}d }n |du r|}|d us |d ur.|t
u r tdt |t r.d} |d u rHt
|rHt}
nt}
i } |d ur`|}
n^|d ust|d ur|} |d ur|} t| rd| v}
n
t| }
t
| rt}
n
|
rt}
|d ur|| d< |d ur|| d < t |
t r t|
tr | jr | j| d
< | | |
f i | }| rt|r|} |d urF|} t| s^t| s^t}
| d
|
f |d d | |_nl|d ur|t
urtd
|t
u rtdt|}|du r|}t |f|| jd|}n|rtd|}|| _ || _ t | _!zrt"| j| j| | j#d urti }| j t
u r>||| j#< | j$D ](}| j!%|}|j t
u rD|| qD|W S |W S | j&t'( s Y n0 d S )NFz Can't specify spec and autospec)TNz6Can't provide explicit spec_set *and* spec or autospecTz!Can't use 'spec' with create=Truer: r r r r r zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=True)r _namez.Can't pass kwargs to a mock we aren't creating))r r r rZ rD rX rW rS r ry r r$ r1 r* r r r\ r r r0 r rV rk r# ra r rw boolr
temp_originalis_localrg rh _exit_stackrT r^ r[ rj __exit__sysexc_info)rB r r r rZ rD rX rf rx ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrrn rp r r r r __enter__g s
z_patch.__enter__c G s | j r$| jtur$t| j| j| j n>t| j| j | jsbt| j| jrP| jdv rbt| j| j| j | `| ` | `| j }| ` |j
| S )N)rO rQ rR __annotations__rS ) r} r| r rT rS rV r
rY r% r~ r )rB r ro r r r r s z_patch.__exit__c C s | }| j| |S r/ )r _active_patchesr rB rg r r r start s z _patch.startc C s4 z| j | W n ty$ Y d S 0 | d d d S r/ )r r7 r= r r r r r stop s
z _patch.stop)rN rQ r r^ r r9 r\ r: r` rg contextmanagerrq rb ra ry r r r r r r r r rV s
rV c
sP z dd\ }W n& tttfy: td Y n0 fdd}||fS )Nr r z,Need a valid target to patch. You supplied: c s t S r/ rU r rS r r r r! z _get_target..)rsplitr r= r; )rS rV rW r r r _get_target s
r c
s> t tu rt d fdd} t| |||||||| S )Nz3 must be the actual object to be patched, not a strc s S r/ r r r r r r 5 r! z _patch_object..)r1 strr rV )
rS rV r r rY r rZ rX rD rW r r r
_patch_object s
r c
s t tu r fdd}n fdd}|s2tdt| }|d \} }
t|| |
|||||i } | | _|dd D ]2\} }
t|| |
|||||i } | | _| j| qt| S )Nc s t S r/ r r r r r r S r! z!_patch_multiple..c s S r/ r r r r r r U r! z=Must supply at least one keyword argument with patch.multipler r ) r1 r r= rZ r rV r^ r[ r )
rS r rY r rZ rX rD rW r rV r r_ Z this_patcherr r r _patch_multiple< s, r c
K s$ t | \}} t|| ||||||| S r/ )r rV )
rS r r rY r rZ rX rD rW rV r r r r m s
G
r c @ sV e Zd ZdddZdd Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dd Z dS ) _patch_dictr Fc K s, || _ t|| _| j| || _d | _d S r/ )in_dictdictr rk clear _original)rB r r r rD r r r r9 s
z_patch_dict.__init__c s. t tr S t fdd}|S )Nc s0 z | i |W S 0 d S r/ )r
_unpatch_dictr frB r r _inner s z$_patch_dict.__call__.._inner)r$ r1 r` r )rB r r r r r r: s
z_patch_dict.__call__c C sX t |D ]J}t||}|tjrt|drt| j| j| j }||}t
||| q|S rd ) r r& r r re r% r r r r rT )rB rC r rf decoratorZ decoratedr r r r` s
z_patch_dict.decorate_classc C s | | jS r/ )r r r r r r r s z_patch_dict.__enter__c C s | j }t| jtr t| j| _| j}| j}z | }W n, tyb i }|D ]}|| ||< qLY n0 || _|rvt | z|
| W n( ty |D ]}|| ||< qY n0 d S r/ ) r r$ r r rU r r\ r; r _clear_dictrk )rB r r r rf r r r r r s& z_patch_dict._patch_dictc C sP | j }| j}t| z|| W n( tyJ |D ]}|| ||< q4Y n0 d S r/ )r r r rk r; )rB r rf r r r r r s z_patch_dict._unpatch_dictc G s | j d ur| dS NF)r r )rB rC r r r r ) s
z_patch_dict.__exit__c C s | }tj| |S r/ )r rV r r r r r r r 0 s z_patch_dict.startc C s4 zt j| W n ty$ Y d S 0 | d d d S r/ )rV r r7 r= r r r r r r 7 s
z_patch_dict.stopN)r F) rN rQ r r9 r: r` r r r r r r r r r r r s
r c C s< z | W n* ty6 t| }|D ]
}| |= q&Y n0 d S r/ )r r; rZ )r keysr r r r r B s r c C s t tjD ] } | q
d S r/ )r rV r r )r r r r _patch_stopallK s 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 zHadd sub mul matmul div floordiv mod lshift rshift and xor or pow truediv c c s | ]}d | V qdS )zi%sNr r nr r r r2 j r! r2 c c s | ]}d | V qdS )zr%sNr r r r r r2 k r! > __set__ __getstate__
__reduce_ex____get__ __setstate__
__getformat__ __missing__
__format__r __getnewargs_ex__
__delete__
__setformat____getinitargs____getnewargs__r __subclasses__ __reversed__r c s fdd}| |_ |S )Nc s | g|R i |S r/ r rB rC r r, r r method| s z_get_method..method)rN )r r- r r r, r r
z s r
c C s h | ] }d | qS )r r )r r r r r r s >
__aenter__ __aexit__ __anext__ __aiter__> r9 __subclasscheck____instancecheck__r __prepare__r r __del__c C s
t | S r/ )r __hash__r r r r r r! r c C s
t | S r/ )r __str__r r r r r r! c C s
t | S r/ )r
__sizeof__r r r r r r! c C s" t | j d| dt| S )N/)r1 rN r r r r r r r r! )r r r
__fspath__r y ?g ?)
__lt____gt____le____ge____int__r __len__r __complex__ __float____bool__ __index__r c s fdd}|S )Nc s$ j j}|tur|S | u r dS tS r8 )__eq__r r NotImplemented)otherret_valr r r r s z_get_eq..__eq__r )rB r r r r _get_eq s r c s fdd}|S )Nc s j jturtS | u r dS tS r )__ne__r r r )r r r r r s
z_get_ne..__ne__r )rB r r r r _get_ne s r c s fdd}|S )Nc s j j} | tu rtg S t| S r/ )__iter__r r r r r r r r s z_get_iter..__iter__r )rB r r r r _get_iter s r c s fdd}|S )Nc s( j j} | tu r ttg S tt| S r/ )r r r _AsyncIteratorr r r r r r s z"_get_async_iter..__aiter__r )rB r r r r _get_async_iter s r )r r r r c C sb t |t}|tur ||_d S t|}|d urB|| }||_d S t|}|d ur^|| |_d S r/ )_return_valuesr_ r rw _calculate_return_value_side_effect_methodsr~ )r5 r r fixedZreturn_calculatorrw Z
side_effectorr r r _set_return_value s
r c @ s e Zd Zdd Zdd ZdS )
MagicMixinc O s* | tt| j|i | | d S r/ )_mock_set_magicsr r r9 r r r r r9 s zMagicMixin.__init__c C s t tB }|}t| dd d urX|| j}t }|| }|D ] }|t| jv r:t| | q:|tt| j }t| }|D ]}t ||t
|| qvd S )Nr ) _magicsr? r& intersectionr r r1 r^ r
rT
MagicProxy)rB Z orig_magicsZ these_magicsZ
remove_magicsr rB r r r r s zMagicMixin._mock_set_magicsN)rN rQ r r9 r r r r r r s r c @ s e Zd ZdddZdS )r Fc C s | || | d S r/ r r r r r r r s z"NonCallableMagicMock.mock_add_specN)FrN rQ r r r r r r r s r c @ s e Zd Zdd ZdS )AsyncMagicMixinc O s* | tt| j|i | | d S r/ )r r r r9 r r r r r9 s zAsyncMagicMixin.__init__NrN rQ r r9 r r r r r s r c @ s e Zd ZdddZdS )r Fc C s | || | d S r/ r r r r r r + s zMagicMock.mock_add_specN)Fr r r r r r s r c @ s& e Zd Zdd Zdd ZdddZdS ) r c C s || _ || _d S r/ r r )rB r r r r r r9 7 s zMagicProxy.__init__c C s8 | j }| j}|j|||d}t||| t||| |S )N)r r r )r r r rT r )rB r r mr r r create_mock; s zMagicProxy.create_mockNc C s | S r/ )r )rB r) rB r r r r D s zMagicProxy.__get__)N)rN rQ r r9 r r r r r r r 6 s r c s e Zd ZedZedZedZ fddZdd Zdd Z d
d Z
d d
Z dd Z dd Z
dddZdd Z fddZ ZS )r r r r c s` t j|i | tjj| jd< d| jd< d | jd< t | jd< ttd}t j
|_ || jd< d S )Nr r Z_mock_await_countZ_mock_await_argsZ_mock_await_args_listr r+ ) superr9 r r r r^ rs r r r' Z CO_COROUTINEco_flags)rB rC rD code_mockr r r r9 M s
zAsyncMockMixin.__init__c s t ||fdd}| jd7 _|| _| j| | j}|d urt|rL|nft|sz t|}W n t yv t
Y n0 t|r|n,t |r||i |I d H }n||i |}|t ur|S | j
t ur| jS | jd ur
t | jr| j|i |I d H S | j|i |S | jS )NTr& r )r( r r r r r~ r3 rY r
StopIterationStopAsyncIterationr r r rw r )rB rC rD r rL rg r r r rK ] s6
z!AsyncMockMixin._execute_mock_callc C s( | j dkr$d| jpd d}t|d S )Nr Expected r5 z to have been awaited.r r r r! r r r r s
z AsyncMockMixin.assert_awaitedc C s0 | j dks,d| jpd d| j d}t|d S Nr r r5 z$ to have been awaited once. Awaited times.r r! r r r r s
z"AsyncMockMixin.assert_awaited_oncec s j d u r& }td| d fdd}t fdd}j }||kr~t|trn|nd }t| |d S )NzExpected await: z
Not awaitedc s j dd} | S )Nawait)r r" r# r$ r r r% s z:AsyncMockMixin.assert_awaited_with.._error_messageTr& )r r r r r( r$ r) )rB rC rD r* r% r+ r, r r$ r r s
z"AsyncMockMixin.assert_awaited_withc O s< | j dks,d| jpd d| j d}t|| j|i |S r )r r r r r- r r r r s
z'AsyncMockMixin.assert_awaited_once_withc sj t||fdd}t|tr$|nd } fdd jD }|sL|t|vrf ||}td| |d S )NTr& c s g | ]} |qS r r. r/ r r r r r! z3AsyncMockMixin.assert_any_await..z%s await not found)r r( r$ r) r r= r r r> r r r r s z AsyncMockMixin.assert_any_awaitFc s fdd|D }t dd |D d }t fdd jD }|s||vr|d u rXd}nddd |D }t| d t| d
j |d S t|}g }|D ]0}z|| W q ty || Y q0 q|rtd t |f |d S ) Nc s g | ]} |qS r r. r/ r r r r r! z4AsyncMockMixin.assert_has_awaits..c s s | ]}t |tr|V qd S r/ r1 r r r r r2 r! z3AsyncMockMixin.assert_has_awaits..c 3 s | ]} |V qd S r/ r. r/ r r r r2 r! zAwaits not found.z,Error processing expected awaits.
Errors: {}c S s g | ]}t |tr|nd qS r/ r1 r r r r r s r3 z
Actual: z %r not all found in await list)
r rs r r5 r rZ r7 r= r r[ ) rB r8 r9 r* r, Z
all_awaitsr: r; r< r r r r sF z AsyncMockMixin.assert_has_awaitsc C s0 | j dkr,d| jpd d| j d}t|d S )Nr r r5 z# to not have been awaited. Awaited r r r! r r r r s
z!AsyncMockMixin.assert_not_awaitedc s* t j|i | d| _d | _t | _d S Nr )r rv r r rs r rA r r r rv s zAsyncMockMixin.reset_mock)F)rN rQ r r r r r r9 rK r r r r r r r rv
__classcell__r r r r r H s (
, r c @ s e Zd ZdS )r NrM r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )_ANYc C s dS r8 r rB r r r r r - s z _ANY.__eq__c C s dS r r r r r r r 0 s z _ANY.__ne__c C s dS )Nzr r r r r r 3 s z
_ANY.__repr__N)rN rQ r r r r r r r r r * s r c C s` d| }d}d dd |D }d dd | D }|r@|}|rX|rP|d7 }||7 }|| S )Nz%s(%%s)r z, c S s g | ] }t |qS r )repr)r r r r r r = r! z*_format_call_signature..c S s g | ]\}}d ||f qS )z%s=%rr )r r r r r r r > s )r r )r rC rD r Zformatted_args args_stringZ
kwargs_stringr r r r : s
r c @ sv e Zd Zd ddZd dd Zd
d ZejZd d
Zdd Z dd Z
dd Z e dd Z
e dd Zdd Zdd ZdS ) r( r r NFTc C s d}i }t |}|dkr$|\}}}nr|dkrd|\} }
t| trX| }t|
trR|
}qb|
}q| |
}}n2|dkr|\}t|tr|}nt|tr|}n|}|rt| ||fS t| |||fS )Nr r r )r r$ r r[ r ) r r r r r' from_kallrC rD _lenfirstsecondr r r r _ s.
z
_Call.__new__c C s || _ || _|| _d S r/ )r r _mock_from_kall)rB r r r r' r r r r r9 s z_Call.__init__c C s\ z t |}W n ty" t Y S 0 d}t | dkr>| \}}n
| \}}}t| dd rpt|dd rp| j|jkrpdS d}|dkrdi }}n|dkr|\}}}n|dkr|\} t| tr| }i }n"t| tr| }di }}nd}| }nV|dkr4|\}
} t|
tr(|
}t| tr | i }}n
d| }}n
|
| }}ndS |rL||krLdS ||f||fkS ) Nr r r Fr r r r )r r r r& r r$ r[ r ) rB r Z len_otherZ self_nameZ self_argsZ self_kwargsZ
other_nameZ
other_argsZ other_kwargsr r r r r r r sN
z _Call.__eq__c O s< | j d u r td||fddS | j d }t| j ||f|| dS )Nr r r r r r( r r r r r: s
z_Call.__call__c C s2 | j d u rt|ddS d| j |f }t|| ddS )NF)r r z%s.%s)r r r r )rB r r r r r r s
z_Call.__getattr__c C s |t jv rtt | |S r/ )r[ r^ r; __getattribute__)rB r r r r r s
z_Call.__getattribute__c C s( t | dkr| \}}n
| \}}}||fS )Nr )r r r r r _get_call_arguments s
z_Call._get_call_argumentsc C s | d S r r r r r r rC s z
_Call.argsc C s | d S )Nr r r r r r rD s z _Call.kwargsc C s| | j s&| jpd}|dr"d| }|S t| dkr@d}| \}}n0| \}}}|sTd}n |dshd| }nd| }t|||S )Nr r zcall%sr zcall.%s)r r r r r )rB r rC rD r r r r s
z_Call.__repr__c C s4 g }| }|d ur(|j r || |j}qtt|S r/ )r r r rs r )rB ZvalsrO r r r call_list s
z_Call.call_list)r r NFT)r NNFT)rN rQ r r r9 r r r r: r r r r rC rD r r r r r r r( L s$
5
r( )r c K sh t | rt| } t| t}t| }d| i}|r8d| i}n | d u rDi }|rT|rTd|d< || t} t| rri }n8|r|rtdt } n"t
| st } n|r|rt | st } |
d|}|}
|d u rd}
| f |||
|d|} t| trt| | } |rt| nt| | || |d ur,|s,| |j|< |rV|sVd |vrVt| |dd
| d | _t| D ]} t| rrq^zt| | }
W n ty Y q^Y n0 d|
i}|rd|
i}t|
tst|
|| | |}|| j| < nn| }t| tr| j}t| | |}||d < t |
rt }nt}|f || | |d
|}|| j| < t|
||d t|tr^t | | | q^| S )Nr r Tr zJInstance can not be True when create_autospec is mocking an async functionr r )r r r r rw r )rL rz r r )r r r r )rK ) r\ r1 r$ r. rk r r' Zisdatadescriptor RuntimeErrorr rY r ra r r6 rj r rM r r rw r r r& r; r r5
_must_skipr rT )r r rL r rz rD is_typeZ
is_async_funcr r r r5 r rf r r rK Z child_klassr r r r
s
r c C sx t | ts$|t| di v r dS | j} | jD ]H}|j|t}|tu rFq*t |tt frZ dS t |t
rl| S dS q*|S )Nr^ F) r$ r1 r& r r] r^ r_ r rW rX r6 )r r r rC rg r r r r
s
r c @ s e Zd ZdddZdS )r FNc C s( || _ || _|| _|| _|| _|| _d S r/ )r idsr r rL r )rB r r r r r rL r r r r9
s z_SpecState.__init__)FNNNFr r r r r r
s r c C s" t | trt| S t| S d S r/ )r$ bytesioBytesIOStringIO) read_datar r r
_to_stream
s
r r c s& t }|d gfdd}fdd} fddfdd fd d
}td u rd d l}ttt|jtt|ja| d u rt d t
d
} t tdj _ d j
_ d j_ d j_ d j_ |j_ d< d j_|j_ j_|j_fdd}|| _| _ | S )Nc s( j jd urj jS d j | i |S r ) readlinesrw rb _statehandler r _readlines_side_effect
s z)mock_open.._readlines_side_effectc s( j jd urj jS d j | i |S r )readrw rb r r r _read_side_effect
s z$mock_open.._read_side_effectc ? s( E d H d j | i |V q d S r )readlinerb )_iter_side_effectr r r _readline_side_effect
s z(mock_open.._readline_side_effectc 3 s0 j jd urj jV q d D ]
} | V q d S r )r rw )liner r r r
s z$mock_open.._iter_side_effectc s j jd urj jS t d S r )r rw r r r r r _next_side_effect
s z$mock_open.._next_side_effectr open)r r )r r c s6 t d< jjd kr2 d< d j_tS )Nr r )r r r~ r rb )r r r r r r
reset_data s
z mock_open..reset_data)r file_spec_iorZ r r
TextIOWrapperunionr r r r rw writer r r r~ r r )r5 r Z
_read_datar r r r r r )r r r r r r r
s8 "
r
c @ s& e Zd Zdd ZdddZdd ZdS ) r c K s t f i |S r/ )r )rB rD r r r r s z PropertyMock._get_child_mockNc C s | S r/ r )rB r) Zobj_typer r r r s zPropertyMock.__get__c C s | | d S r/ r )rB r) r r r r r ! s zPropertyMock.__set__)N)rN rQ r r r r r r r r r s
r c C sp d| _ t| D ]\}zt| |}W n ty6 Y qY n0 t|tsDqt|j|trXq|j | u rt
| qd S r8 ) r r r& r; r$ r r r_ r r r )r5 r r r r r r % s
r c @ s e Zd Zdd Zdd ZdS )r c C s& || _ ttd}tj|_|| jd< d S )Nr r+ )iteratorr r r' ZCO_ITERABLE_COROUTINEr r^ )rB r# r r r r r9 A s
z_AsyncIterator.__init__c s( z t | jW S ty Y n0 td S r/ )r r# r r r r r r r G s
z_AsyncIterator.__anext__N)rN rQ r r9 r r r r r r = s r )F)F)NFNNN)FFNN)Nr )}__all__r rg r r' r r builtinsr typesr r r Z
unittest.utilr functoolsr r r rw r
r r r* r. r# r3 r7 r> rM rH rY r\ ra rj re r r r r r r r MISSINGr ZDELETEDr r r rZ rs r r r r r< r9 r r= r rA r rR rU rV r r r r r r r r ZmultipleZstopallre Z
magic_methodsZnumericsr r ZinplacerightZ
_non_defaultsr
r r? Z_sync_async_magicsZ
_async_magicsr@ r r r r r r r r r r r r r r r r r r r r r r[ r( r r r r r1 r r6 r r r
r r r r r r r s0
1 h4 <
2
N
8+ :
N