U
e5d @ s d Z dZddlZddlZddlZddlZddlZddlZddlZddl 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.d-d.d/d0d1d2d3d4hZ/d5d6 Z0G d7d8 d8e1Z2d9d: Z3G d;d< d d>e&Z5G d?d@ d@e5Z6dAdB Z7G dCdD dDe5Z8G dEdF dFe8e6Z9dGdH Z:dIdJ Z;G dKdL dLe&ZddQdRZ?e*dddddfdSdTZ@G dUdV dVe&ZAdWdX ZBdYdZ ZCe>e@_&eAe@_De?e@_EeCe@_Fd[e@_Gd\ZHd]ZId^Jd_d` eIK D ZLd^Jdad` eIK D ZMdbdcdddedfdgdhdidjdkdldmdndodpdqdrdshZNdtdu ZOdvd d^JeHeIeLeMgK D ZPdwdxdyhZQdzhZReQeRB ZSePeNB ZTeTeSB ZUd{d|d}d~ddddhZVdd dd dd dd dZWeXeXeXeXddddddd ddd
ZYdd ZZdd Z[dd Z\dd Z]eZe[e\e]dZ^dd Z_G dd de5Z`G dd de`e6ZaG dd de`ZbG dd de`e9ZcG dd de5ZdG dd de5ZeG dd deeebe9ZfG dd de&Zgeg Zhdd ZiG dd dejZkekddZldddZmdd ZnG dd de&ZoepemepehjqfZrdasdd ZtdddZuG dd de9Zvdd ZwG dd dZxdS ))Mock MagicMockpatchsentinelDEFAULTANYcallcreate_autospec AsyncMock
FILTER_DIRNonCallableMockNonCallableMagicMock mock_open PropertyMocksealz1.0 N)CodeType
ModuleType
MethodType) safe_repr)wrapspartialc C s h | ]}| d s|qS _
startswith).0name r %/usr/lib64/python3.8/unittest/mock.py ( s
r Tc C s> t | rt| tsdS t| dr*t| d} t| p Fc sN t | || d krd S \} fdd}t| | |t|_ t|_d S )Nc s j || d S r/ bindselfargskwargssigr r checksigw s z"_check_signature..checksig)r> _copy_func_detailsr1 _mock_check_sig
__signature__)r- r5 skipfirstinstancerG r rE r _check_signaturer s
rM c C s: dD ]0}zt ||t| | W q tk
r2 Y qX 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 k r.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 | |S r/ )assert_called_withrb r5 r r rk s z'_setup_func..assert_called_withc s j | |S r/ )
assert_calledrb rl r r rm s z"_setup_func..assert_calledc s j | |S r/ )assert_not_calledrb rl r r rn s z&_setup_func..assert_not_calledc s j | |S r/ )assert_called_oncerb rl r r ro s z'_setup_func..assert_called_oncec s j | |S r/ )assert_called_once_withrb rl r r rp s z,_setup_func..assert_called_once_withc s j | |S r/ )assert_has_callsrb rl r r rq s z%_setup_func..assert_has_callsc s j | |S r/ )assert_any_callrb rl r r rr s z$_setup_func..assert_any_callc s: t _t _ j} t| r6| k r6| 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| ||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) r%
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 rw _mock_return_valuer~ _mock_side_effect _mock_parent_mock_new_parent
_mock_name_mock_new_namec C s8 t | d| }| |fdd}| |fdd}t||S )NZ_mock_c S s" | j }|d krt| |S t||S r/ )r r$ )rB r _the_namerF r r r _getA s
z"_delegating_property.._getc S s* | j }|d kr|| j|< n t||| d S r/ )r r^ rT )rB valuer r rF r r r _setF 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 S s
z_CallList.__contains__c C s t t| S r/ )pprintZpformatrZ r r r r r a s z_CallList.__repr__N)rN rQ r r r r r r r rs Q s rs c C s| t |}t|sdS |js4|js4|jd k s4|jd k r8dS | }|d k rX||krPdS |j}q<|rh| |_||_|rx| |_||_dS )NFT)r7 r! r r r r )parentr r new_name_parentr r r _check_and_set_parente 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 | ts^ttj}|j| f||j}dd | D }|r^t ||d r^t
| f}t | j |d| j
i}tt| |}|S )Nc S s g | ]}| d r|qS specr r argr r r
s
z+NonCallableMock.__new__..r rO )r0 r r' r< r r9 Z bind_partialZ argumentskeysr* AsyncMockMixinr1 rN rO _safe_super__new__) clsrC kwbasesrF Z
bound_argsZspec_argnewrL r r r r s
zNonCallableMock.__new__N Fc K s |d kr |}| j }
||
d< ||
d< ||
d< ||
d< d|
d< |d k rJ|}d}|
d krZ|d k }
| ||| |
i |
d< ||
d < d |
d
< d|
d < d |
d < d
|
d< t |
d< t |
d< t |
d< | |
d< | r| jf | 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 spec_setr _spec_state _new_name _new_parent_spec_as_instance _eat_selfZunsaferD r^ r r r r9 sD
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 k r~t|s~t|trV|}nt|}t|||} | ot| d }t |}| j }
||
d< ||
d< ||
d< ||
d< ||
d< d S )Nr _spec_class _spec_set_spec_signature
_mock_methods _spec_asyncs)
dirr% r& 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 k r| jj}|tkr4| 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 k r|| 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 krt| S | j S r/ )r r1 r r r r __class__! s
zNonCallableMock.__class__rz r{ r| r} ru c C sN | j }|d kr| jS |j}|d k rJt|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 kr || _n||_d S r/ ) _try_iterr r r~ )rB r r r r r Z__set_side_effect9 s
z!NonCallableMock.__set_side_effect)rw r~ c C s |d kr g }t | |kr d S |t | d| _d | _d| _t | _t | _t | _|r^t | _
|rhd | _ | j
D ]"}t|tsr|tkrqr|| qr| j
}t|r|| k r|| d S )NFr )idr rz r| r{ rs ru r} rt r r r r valuesr"
_SpecState_deletedrv r! )rB Zvisitedrw r~ childrx r r r rv D 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 o z0NonCallableMock.configure_mock..)keyr )sorteditemssplitpopr$ rT )rB rD r valrC finalr) r r r r r b s
z NonCallableMock.configure_mockc C s |dkrt |n:| jd k r<|| jks.|tkrLt d| nt|rLt || jsd|drdt d| j|}|tkrt |np|d krd }| j d k rt
| 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 x sF
zNonCallableMock.__getattr__c C s | j g}| j}| }d}|dgkr$d}|d k rZ|}||j | d}|j dkrRd}|j}q$tt|}|jpnd}t|dkr|d dkr|d7 }||d< d|S )Nr r r r5 r )r z().r )r r r rZ reversedr r join)rB Z
_name_listr ZlastdotZ_firstr r r _extract_mock_name s(
z"NonCallableMock._extract_mock_namec C s^ | }d}|dkr d| }d}| jd k rDd}| 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 k r|qS r )r )r Zm_nameZm_valuer r r r s z+NonCallableMock.__dir__..c S s g | ]}| d s|qS r r r er r r r s
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 ZextrasZ from_typeZ from_dictZfrom_child_mocksr r r r s
zNonCallableMock.__dir__c sT |t krt||S jrHjd k rH|jkrH|jkrHtd| n|tkrbd| }t|n|tkrjd k r|jkrtd| 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 f| |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 kr2|t| jkr2tt| | || jkr2d S | j|t}|| jkr\tt| | n|t
krlt ||tk r|| 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 ksJt|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 k rt|dkrNd}|\}}n
|\}}}z||j||fW S tk
r } z|d W Y S d }~X Y qX n|S d S )Nr r r )r" r[ r r r r@ TypeErrorwith_traceback)rB _callrF r rC rD r r r r
_call_matcherH 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 b 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 l 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 t s
z"NonCallableMock.assert_called_oncec s j d kr. }d}d||f }t| fdd} f}j }||kr~t|trn|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_message)r| r r r r" Exception)rB rC rD expectedactualZ
error_messager$ causer r# r rk ~ s
z"NonCallableMock.assert_called_withc O s8 | j dks,d| jpd| j | f }t|| j||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||kr|d krXd}nddd |D }t| d t| jd
d d |d S t|}g }|D ]2}z|| W q t k
r |
| Y qX q|rtd
j pdt ||f |d S )Nc s g | ]} |qS r r r cr r r r s z4NonCallableMock.assert_has_calls..c s s | ]}t |tr|V qd S r/ r" r% r r r r s
z3NonCallableMock.assert_has_calls..c 3 s | ]} |V qd S r/ r* r+ r r r r. s zCalls not found.z+Error processing expected calls.
Errors: {}c S s g | ]}t |tr|nd qS r/ r- 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 sH
" z NonCallableMock.assert_has_callsc sZ ||f} fdd jD }||krVt|tr8|nd } ||}td| |d S )Nc s g | ]} |qS r r* r+ r r r r s z3NonCallableMock.assert_any_call..z%s call not found)r r} r" r% r r rB rC rD r& r' r( r r r r rr s z NonCallableMock.assert_any_callc K s | d}|| jd kr"tf |S t| }t|trB|tkrBt}nbt|trp|tksd| j rj|| j krjt}qt}n4t|t
st|t rt}qt|t rt
}n
|jd }| jrd|krd|d nd}| | }t||f |S )Nr r r r r r )r_ r^ r r1 r0 r _async_method_magicsr _all_sync_magicsr
CallableMixinr r r r] r r r; )rB r r _typeklassrV r r r r r s2
z NonCallableMock._get_child_mockCallsc C s" | j s
dS d| dt| j dS )Nr
z: r )ru r )rB r0 r r r r s zNonCallableMock._calls_repr) NNNNNNr NFNF)F)FF)N)r )F)r? ),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 sn
-
''
!
-'r c C sL | d kr | S t | r| S t| r$| S z
t| W S tk
rF | Y S X 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 )r< Nr c K s6 || j d< tt| j||||||| |
f| || _d S )Nr )r^ r r< 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 s$ | j || | j|| | j||S r/ )rI _increment_mock_call
_mock_callrA r r r r: 4 s zCallableMixin.__call__c O s | j ||S r/ )_execute_mock_callrA r r r rB < s zCallableMixin._mock_callc O s d| _ | jd7 _t||fdd}|| _| j| | jd k }| j}| j}|dk}| j td||f | j
}|d k r|r|j t|||f |jd k }|r|jd | }t|||f} |j | |jr|rd}
nd}
|jdk}|j|
| }|j
}qpd S )NTr twor r r ) rz r{ _Callr| 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 rA ? s4
z"CallableMixin._increment_mock_callc O s| | j }|d k rPt|r |n(t|s:t|}t|rD|n
|||}|tk rP|S | jtk r`| jS | jd k rv| j||S | jS r/ )r~ r3 rY r r r rw r )rB rC rD effectrg r r r rC m s
z CallableMixin._execute_mock_call)
rN rQ r r r9 rI r: rB rA rC r r r r r< ! s
.r< c @ s e Zd ZdS )r NrN rQ r r r r r r s r c C s8 z t | |W S tk
r2 t| t | | Y S X d S r/ )r$ r;
__import__)thingcomp import_pathr r r _dot_lookup s
rM c C sB | d}|d}t|}|D ] }|d| 7 }t|||}q |S )Nr r z.%s)r r rI rM )targetZ
componentsrL rJ rK r r r _importer s
rO 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 k r(|t k rtd|d k r(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 rQ rV r r rS r rT rR 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..) rP rQ rV r r rS r rT rR rD attribute_namerU )rB patcherr r r rV 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# rV rT )rB r> r
attr_valuerY r r r rZ s
z_patch.decorate_classc c sr g }t \}|jD ]8}||}|jd k r8|| q|jtkr|| q|t |7 }||fV W 5 Q R X d S r/ )
contextlib ExitStack patchings
enter_contextrX 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 )Nrc c
s4 | | \}} ||W 5 Q R S Q R X d S r/ rk rC rg ZnewargsZ newkeywargsr- rf rB r r rf ( s z)_patch.decorate_callable..patchedr# rc r r r] r rn r r\ " s
z_patch.decorate_callablec s> t dr j S t fddg_S )Nrc c
s: | |"\}} ||I d H W 5 Q R S Q R X d S r/ rl rm rn r r rf 9 s z/_patch.decorate_async_callable..patchedro r] r rn r r[ 3 s
z _patch.decorate_async_callablec C s | }| j}t}d}z|j| }W n$ ttfk
rH t||t}Y nX d}|tkrft|t rfd| _
| j
s|tkrtd||f ||fS )NFTz!%s does not have the attribute %r) rQ rV r r^ r; KeyErrorr$ _builtinsr" r rS )rB rN r rf localr r r get_originalD s
z_patch.get_originalc C s | j | j| j }}}| j| j }}| j}| | _|dkr@d }|dkrLd }|dkrXd }|d k rp|d k rptd|d k s|d k r|dkrtd| \}}|t
kr||d kr|d} |dkr|}|dkr|}d }n&|d k r|dkr|}d }n |dkr|}|d k s |d k r.|t
kr tdt |t r.d} |d krHt
|rHt}
nt}
i } |d k r`|}
n^|d k st|d k r|} |d k r|} t| rd| k}
n
t| }
t
| rt}
n
|
rt}
|d k r|| d< |d k r|| d < t |
t r t|
tr | jr | j| d
< | | |
f | }| rt|r|} |d k rB|} t| sZt| sZt}
| d
|
f |d d | |_nl|d k r|t
k rtd
|t
krtdt|}|dkr|}t |f|| jd|}n|rtd|}|| _ || _ t | _!zrt"| j| j| | j#d k rpi }| j t
kr:||| j#< | j$D ](}| j!%|}|j t
kr@|| q@|W S |W S | j&t'( s Y nX 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 rT rD rR rQ rN r rs r r" r1 r* r r r\ r r r0 r rV re r! ra r rw boolr
temp_originalis_localra rb _exit_stackrT rX rU rd __exit__sysexc_info)rB r r r rT rD rR rf rr ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrrh rj r r r r __enter__\ s
z_patch.__enter__c G s | j r$| jtk r$t| j| j| j n>t| j| j | jsbt| j| jrP| jdkrbt| j| j| j | `| ` | `| j }| ` |j
| S )N)rO rQ rR __annotations__rS ) rw rv r rT rN rV r rS r# rx ry )rB r{ ri r r r ry s z_patch.__exit__c C s | }| j| |S r/ )r~ _active_patchesr )rB rg r r r start s z _patch.startc C s6 z| j | W n tk
r& Y d S X | d d d S r/ )r r3 r= ry r r r r stop s
z _patch.stop)rN rQ r rX r r9 rV r: rZ ra contextmanagerrk r\ r[ rs r~ ry r r r r r r rP s
rP c sP z dd\ }W n& ttfk
r: td f Y nX fdd}||fS )Nr r z.Need a valid target to patch. You supplied: %rc s t S r/ rO r rN r r r r z _get_target..)rsplitr r= )rN rV rQ r r r _get_target
s r c
s> t tkrt 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 * r z _patch_object..)r1 strr rP )
rN rV r r rS r rT rR rD rQ r r r
_patch_object s r c
s t tkr 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 H r z!_patch_multiple..c s S r/ r r r r r r J r z=Must supply at least one keyword argument with patch.multipler r ) r1 r r= rZ r rP rX rU r )
rN r rS r rT rR rD rQ r rV r rY Z this_patcherr r r _patch_multiple1 sH r c
K s$ t | \}} t|| ||||||| S r/ )r rP )
rN r r rS r rT rR rD rQ rV r r r r b s F r c @ sN e Zd ZdddZdd Zdd Zd d
Zd d Zd
d Zdd Z eZ
e Z dS ) _patch_dictr Fc K s, || _ t|| _| j| || _d | _d S r/ )in_dictdictr re 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 s& z | |W S X d S r/ )r
_unpatch_dictr frB r r _inner s z$_patch_dict.__call__.._inner)r" r1 rZ 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 r^ ) r r$ r r r_ r# r r r r rT )rB r> r r` Z decoratorZ decoratedr r r rZ 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. tk
rd i }|D ]}|| ||< qNY nX || _|rxt | z|
| W n* tk
r |D ]}|| ||< qY nX d S r/ ) r r" r r rO r rV r; r _clear_dictre )rB r r r rf r r r r r s& z_patch_dict._patch_dictc C sR | j }| j}t| z|| W n* tk
rL |D ]}|| ||< q6Y nX d S r/ )r r r re r; )rB r rf r r r r r s z_patch_dict._unpatch_dictc G s | dS NF)r )rB rC r r r ry s z_patch_dict.__exit__N)r F) rN rQ r r9 r: rZ r~ r r ry r r r r r r r s
r c C s> z | W n, tk
r8 t| }|D ]
}| |= q(Y nX d S r/ )r r; rZ )r r r r r r r & s r c C s t tjD ] } | q
d S r/ )r rP r r )r r r r _patch_stopall/ s r Ztestzlt 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 r. N s r. c c s | ]}d | V qdS )zr%sNr r r r r r. O s __get____set__
__delete__ __reversed__ __missing__r
__reduce_ex__Z__getinitargs____getnewargs__ __getstate__ __setstate__
__getformat__Z
__setformat__r r __subclasses__
__format____getnewargs_ex__c s fdd}| |_ |S )Nc s | f||S r/ r rB rC r r, r r method` s z_get_method..method)rN )r r- r r r, r r ^ s r c C s h | ] }d | qS )r r )r r r r r r f s
__aenter__ __aexit__ __anext__ __aiter__r r r9 r __prepare____instancecheck____subclasscheck____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__ry __complex__ __float____bool__ __index__r c s fdd}|S )Nc s$ j j}|tk r|S | kr 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 jtk rtS | kr 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} | tkrtg 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} | tkr 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}|tk r ||_d S t|}|d k rB|| }||_d S t|}|d k r^|| |_d S r/ )_return_valuesr_ r rw _calculate_return_value_side_effect_methodsr~ )r5 r r ZfixedZreturn_calculatorrw Z
side_effectorr r r _set_return_value s
r c @ s e Zd Zdd Zdd ZdS )
MagicMixinc O s& | tt| j|| | d S r/ )_mock_set_magicsr r r9 r r r r r9 s zMagicMixin.__init__c C s t tB }|}t| dd d k rX|| j}t }|| }|D ] }|t| jkr: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 r= 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|| | 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 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) r= r r r r ( s zMagicProxy.__get__)N)rN rQ r r9 r r r r r r r 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|| 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 1 s
zAsyncMockMixin.__init__c s t ||fdd}| jd7 _|| _| j| | j}|d k rt|rL|nbt|sz t|}W n t k
rx t
Y nX t|r|n&t |r|||I d H }n
|||}|t
k r|S | jt
k r| jS | jd k rt | jr| j||I d H S | j||S | jS )NTrD r )rF r r r r r~ r3 rY r
StopIterationStopAsyncIterationr% r& r r rw r )rB rC rD r rG rg r r r rC A 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 i 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 q s
z"AsyncMockMixin.assert_awaited_oncec sz j d kr& }td| d fdd} f}j }||krvt|trf|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_message)r r r r r" r% )rB rC rD r& r$ r' r( r r# r r z s
z"AsyncMockMixin.assert_awaited_withc O s8 | j dks,d| jpd d| j d}t|| j||S r )r r r r r) r r r r s
z'AsyncMockMixin.assert_awaited_once_withc sZ ||f} fdd jD }||krVt|tr8|nd } ||}td| |d S )Nc s g | ]} |qS r r* r+ r r r r s z3AsyncMockMixin.assert_any_await..z%s await not found)r r r" r% r r r9 r r r r s z AsyncMockMixin.assert_any_awaitFc s fdd|D }t dd |D d }t fdd jD }|s||kr|d krXd}nddd |D }t| d t| d
j |d S t|}g }|D ]2}z|| W q tk
r || Y qX q|rtd t |f |d S ) Nc s g | ]} |qS r r* r+ r r r r s z4AsyncMockMixin.assert_has_awaits..c s s | ]}t |tr|V qd S r/ r- r r r r r. s
z3AsyncMockMixin.assert_has_awaits..c 3 s | ]} |V qd S r/ r* r+ r r r r. s zAwaits not found.z,Error processing expected awaits.
Errors: {}c S s g | ]}t |tr|nd qS r/ r- r r r r r s r/ z
Actual: z %r not all found in await list)
r rs r r1 r rZ r3 r= r r[ ) rB r4 r5 r& r( Z
all_awaitsr6 r7 r8 r r r r s> 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|| 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 rC r r r r r r r rv
__classcell__r r r r r , s (
, r c @ s e Zd ZdS )r NrH 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 s z _ANY.__ne__c C s dS )Nzr r r r r r 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 )reprr r r r r ! s 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_argsZ args_stringZ
kwargs_stringr r r r s
r c @ s e Zd Zd ddZd!dd Zd
d ZejZd d
Zdd Z dd Z
dd Z dd Z dd Z
edd Zedd Zd d Zd d ZdS )"rF 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 rE from_kallrC rD _lenfirstsecondr r r r C s.
z
_Call.__new__c C s || _ || _|| _d S r/ )r r _mock_from_kall)rB r r r rE r r r r r9 c s z_Call.__init__c C sh |t kr dS z t|}W n tk
r. Y dS X d}t| dkrJ| \}}n
| \}}}t| dd r|t|dd r|| j|jkr|dS d}|dkrdi }}n|dkr|\}}}n|d kr|\} t| tr| }i }n"t| tr| }di }}nd}| }nV|dkr@|\}
} t|
tr4|
}t| tr(| i }}n
d| }}n
|
| }}ndS |rX||krXdS ||f||fkS )
NTFr r r r 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 j sR
z _Call.__eq__c O s< | j d kr td||fddS | j d }t| j ||f|| dS )Nr r r r r rF r r r r r: s
z_Call.__call__c C s2 | j d krt|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 jkrtt | |S r/ )r[ r^ r; __getattribute__)rB r r r r r s
z_Call.__getattribute__c O s | d||S )Nr r rA r r r r s z _Call.countc O s | d||S )Nindexr rA r r r r s z _Call.indexc 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 k r(|j r || |j}qtt|S r/ )r r r rs r )rB ZvalsrJ 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 r r rC rD r r r r r r rF 0 s(
7
rF )r c K sl t | rt| } t| t}t| }d| i}|r8d| i}n | d krDi }|rT|rTd|d< || t} t| rri }n8|r|rtdt } n"t
| st } n|r|rt | st } |
d|}|}
|d krd}
| f |||
|d|} t| trt| | } |rt| nt| | || |d k r,|s,| |j|< |rV|sVd |krVt| |dd
| d | _t| D ]} t| rrq^zt| | }
W n tk
r Y q^Y nX d|
i}|rd|
i}t|
tst|
|| | |}|| j| < np| }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 rt r r )r r r r )rK ) r\ r1 r" r. re r r' Zisdatadescriptor RuntimeErrorr rY r ra r r6 rj r rM r r rw r r r$ r; r r5
_must_skipr% r& rT )r r rL r rt 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 kr dS | j} | jD ]H}|j|t}|tkrFq*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 r> rg r r r r w
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 krd d l}ttt|jtt|ja| d krt 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 k rj jS d j | |S r ) readlinesrw rb _statehandler r _readlines_side_effect
s z)mock_open.._readlines_side_effectc s$ j jd k rj jS d j | |S r )readrw rb r r r _read_side_effect
s z$mock_open.._read_side_effectc ? s$ E d H d j | |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 k rj 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 k rj 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 |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 s^ d| _ t| D ]J}zt| |}W n tk
r8 Y qY nX t|tsFq|j| krt| qd S r8 )r r r$ r; r" r r r )r5 r r r r r r s
r c @ s$ e Zd Zdd 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 + s
z_AsyncIterator.__init__c C s | S r/ r r r r r r 1 s z_AsyncIterator.__aiter__c s* z t | jW S tk
r Y nX td S r/ )r r r r r r r r r 4 s
z_AsyncIterator.__anext__N)rN rQ r r9 r r r r r r r ' s r )F)F)NFNNN)FFNN)Nr )y__all__ __version__r% ra r r' r rz builtinstypesr r r Z
unittest.utilr functoolsr r r rq 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 r< r rM rO rP r r r r r r r r ZmultipleZstopallr_ 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[ rF r r r r r1 r r6 r r r
r r r r r r r s~
1 h4 <
2
Mw
8+ B
N