j f, @ s d d d d d d d d d d d
d d f
Z d
Z d d l Z d d l Z d d l Z d d l m Z m Z e f Z d e j
k r d d l Z e e j j
f Z n d Z e Z d d Z d d Z Gd d d e Z e e j e f Z d d Z d d d Z d d Z d d! Z d" d# Z d$ d% Z d d& d' Z d( d) Z d* d+ Z! Gd, d- d- e Z" Gd. d/ d/ e Z# e# Z$ e$ j% Z% e$ j& Z' e$ j( Z) d0 d1 Z* e+ d2 d3 d4 d5 d6 d7 d8 d9 g Z, d: d; Z- Gd< d= d= e. Z/ d> d? Z0 Gd@ dA dA e Z1 GdB dC dC e Z2 GdD d d e2 Z3 dE dF Z4 GdG dH dH e2 Z5 GdI d d e5 e3 Z6 dJ dK Z7 dL dM Z8 dN dO Z9 GdP dQ dQ e Z: dR dS Z; e% d d d d d dT dU Z< d d d d d dV dW Z= e% d d d d d dX d Z> GdY dZ dZ e Z? d[ d\ Z@ d] d^ ZA e< e> _ e? e> _B e= e> _C eA e> _D d_ e> _E d` ZF da ZG db jH dc dd eG jI D ZJ db jH de dd eG jI D ZK e+ df dd dg dh di dj dk dl dm dn do dp dq dr ds dt du dv dw g D ZL dx dy ZM e+ dz dd db jH eF eG eJ eK g jI D ZN eN eL BZO e+ d{ d| d} d~ d d d g ZP i d d d 6d d d 6d d d 6ZQ i eR d 6eR d 6eR d 6eR d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6ZS d d ZT d d ZU d d ZV i eT d 6eU d 6eV d 6ZW d d ZX Gd d d e ZY Gd d
d
eY e3 ZZ Gd d d eY e6 Z[ Gd d d e Z\ Gd d d e Z] e] Z^ d d Z_ Gd d d e` Za ea d d Zb d d d d d d Zc d d Zd d d Ze Gd d d e Zf e ec e e^ jg f Zh e e^ jg ji f Zj d ak d d Zl d d d d Zm Gd d d e6 Zn d S)Mock MagicMockpatchsentinelDEFAULTANYcallcreate_autospec
FILTER_DIRNonCallableMockNonCallableMagicMock mock_open PropertyMockz1.0 N)wrapspartialjavaTc C s t t | t S)N)
issubclasstyper
)obj r 2/opt/alt/python34/lib64/python3.4/unittest/mock.py_is_instance_mock. s r c C s+ t | t p* t | t o* t | t S)N)
isinstanceBaseExceptionsr r )r r r r
_is_exception4 s r c @ s e Z d Z d g Z d S)_slottedaN)__name__
__module__ __qualname__ __slots__r r r r r ; s r c C s t | t rE | rE y
| j } Wn t k
r; d SYn Xd } n8 t | t s} y
| j } Wq} t k
ry d SYq} Xn | r t | d } n | } y | t j | f SWn t k
r d SYn Xd S)z
Given an arbitrary, possibly callable object, try to create a suitable
signature object.
Return a (reduced func, signature) tuple, or None.
NT)
r r __init__AttributeError
FunctionTypes__call__r inspectZ signature
ValueError)funcZ as_instanceZeat_selfZsig_funcr r r _get_signature_objectE s$
r( Fc s` t | | | d k r" d S \ } f d d } t | | | t | _ d S)Nc s j | | d S)N)bind)
_mock_selfargskwargs)sigr r checksigj s z"_check_signature..checksig)r( _copy_func_detailsr _mock_check_sig)r' mock skipfirstinstancer. r )r- r _check_signaturee s
r4 c # C s | j | _ | j | _ y | j | _ Wn t k
r< Yn Xy | j | _ Wn t k
ra Yn Xy | j | _ Wn t k
r Yn Xy | j | _ Wn t k
r Yn Xd S)N)r __doc____text_signature__r" r __defaults____kwdefaults__)r' funcopyr r r r/ p s$
r/ c C s3 t | t r d St | d d d k r/ d Sd S)NTr$ F)r r getattr)r r r r _callable s
r; c C s t | t t f k S)N)r listtuple)r r r r _is_list s r> c C s` t | t s% t | d d d k Sx4 | f | j D]" } | j j d d k r6 d Sq6 Wd S)ztGiven an object, return True if the object is callable.
For classes, return True if instances would be callable.r$ NTF)r r r: __mro____dict__get)r baser r r _instance_callable s rC c s t | s d St | t } t | | | } | d k rA d S| \ } f d d } t | | | j } | j s d } n i | d 6| d 6} d | } t | | | | }
t |
| |
S)Nc s j | | d S)N)r) )r+ r, )r- r r r. s z _set_signature..checksigr9 Z
_checksig_r1 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs)) r; r r r( r/ r isidentifierexec _setup_func) r1 originalr3 r2 resultr' r. namecontextsrcr9 r )r- r _set_signature s$
rL c s _ t s d S f d d } f d d } f d d } f d d } f d d
} d _ d _ d _ t _ t _ t _ j _ j
_
j _ | _ | _
| _ | _ | _ _ d S)
Nc s j | | S)N)assert_called_with)r+ r, )r1 r r rM s z'_setup_func..assert_called_withc s j | | S)N)assert_called_once_with)r+ r, )r1 r r rN s z,_setup_func..assert_called_once_withc s j | | S)N)assert_has_calls)r+ r, )r1 r r rO s z%_setup_func..assert_has_callsc s j | | S)N)assert_any_call)r+ r, )r1 r r rP s z$_setup_func..assert_any_callc sT t _ t _ j j } t | rP | k rP | j n d S)N) _CallList method_calls
mock_calls
reset_mock return_valuer )ret)r9 r1 r r rT s
z _setup_func..reset_mockFr )r1 r called
call_count call_argsrQ call_args_listrR rS rU side_effect_mock_childrenrM rN rO rP rT _mock_delegate)r9 r1 rM rN rO rP rT r )r9 r1 r rF s. rF c C s d | d d | k S)Nz__%s__ r )rI r r r _is_magic s r` c @ s. e Z d Z d Z d d Z d d Z d S)_SentinelObjectz!A unique, named, sentinel object.c C s
| | _ d S)N)rI )selfrI r r r r! s z_SentinelObject.__init__c C s d | j S)Nz sentinel.%s)rI )rb r r r __repr__ s z_SentinelObject.__repr__N)r r r r5 r! rc r r r r ra s ra c @ s. e Z d Z d Z d d Z d d Z d S) _SentinelzAAccess attributes to return a named object, usable as a sentinel.c C s
i | _ d S)N)
_sentinels)rb r r r r! s z_Sentinel.__init__c C s. | d k r t n | j j | t | S)N __bases__)r" re
setdefaultra )rb rI r r r __getattr__ s z_Sentinel.__getattr__N)r r r r5 r! rh r r r r rd s rd c C s2 t | t t t t f k r. t | | S| S)N)r dictr< r= set)valuer r r _copy s rl rU _mock_return_valuer[ _mock_side_effect _mock_parent_mock_new_parent
_mock_name_mock_new_namec C sH t j | d | } | | d d } | | d d } t | | S)NZ_mock_c S s/ | j } | d k r" t | | St | | S)N)r] r: )rb rI _the_namer- r r r _get s
z"_delegating_property.._getc S s9 | j } | d k r% | | j | ._set)_allowed_namesaddproperty)rI rs rt rv r r r _delegating_property s
rz c @ s( e Z d Z d d Z d d Z d S)rQ c C s t | t s t j | | St | } t | } | | k rG d SxC t d | | d D]* } | | | | } | | k r_ d Sq_ Wd S)NFr T)r r< __contains__lenrange)rb rk Z len_valueZlen_selfiZsub_listr r r r| 0 s z_CallList.__contains__c C s t j t | S)N)pprintZpformatr< )rb r r r rc > s z_CallList.__repr__N)r r r r| rc r r r r rQ . s rQ c C s t | s d S| j s@ | j s@ | j d k s@ | j d k rD d S| } x) | d k ru | | k ri d S| j } qM W| r | | _ | | _ n | r | | _ | | _ n d S)NFT)r rq rr ro rp )parentrk rI new_name_parentr r r _check_and_set_parentB s$
r c @ s4 e Z d Z d d Z d d Z d d Z d S) _MockIterc C s t | | _ d S)N)iterr )rb r r r r r! \ s z_MockIter.__init__c C s | S)Nr )rb r r r __iter__^ s z_MockIter.__iter__c C s
t | j S)N)nextr )rb r r r __next__` s z_MockIter.__next__N)r r r r! r r r r r r r [ s r c @ s( e Z d Z e Z d Z d d Z d S)BaseNc O s d S)Nr )rb r+ r, r r r r! f s z
Base.__init__)r r r r rm rn r! r r r r r c s r c @ s e Z d Z d Z d d Z d d d d d d d d d d d d
Z d d
Z d d d Z d d d
d Z d d Z d d Z
d Z e e e
e Z
e d d Z e d Z e d Z e d Z e d Z e d Z d d Z d d Z e e e Z d d d Z d! d" Z d# d$ Z d% d& Z d' d( Z d) d* Z d+ d, Z d- d. Z d/ d0 Z d1 d2 Z! d3 d4 Z" d5 d6 Z# d d7 d8 Z$ d9 d: Z% d; d< Z& d S)=r
z A non-callable version of `Mock`c O s5 t | j | f i | j d 6 } t j | } | S)Nr5 )r r r5 object__new__)clsr+ kwnewr3 r r r r n s "zNonCallableMock.__new__N Fc
K s4 | d k r | } n | j } | | d <| | d <| | d <| | d <| d k ra | } d } n |
d k r| | d k }
n | j | | | |
i | d <| | d r r
_get_classr( dirr@ ) rb r r r r r r resr@ r r r r s
z NonCallableMock._mock_add_specc C s[ | j } | j d k r' | j j } n | t k rW | j d | d d } | | _ n | S)Nr r z())rm r] rU r _get_child_mock)rb rV 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)Nz())r] rU rm r )rb rk 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 k r t | S| j S)N)r r )rb r r r __class__ s
zNonCallableMock.__class__rW rX rY rZ rS c C sj | j } | d k r | j S| j } | d k rf t | rf t | t rf t | } | | _ n | S)N)r] rn r[ callabler r )rb delegatedZsfr r r Z__get_side_effect s ) z!NonCallableMock.__get_side_effectc C s: t | } | j } | d k r- | | _ n | | _ d S)N) _try_iterr] rn r[ )rb rk r r r r Z__set_side_effect s
z!NonCallableMock.__set_side_effectc C s | d k r g } n t | | k r+ d S| j t | d | _ d | _ d | _ t | _ t | _ t | _ x9 | j j
D]( } t | t r q n | j
| q W| j } t | r | | k r | j
| n d S)z-Restore the mock object to its initial state.NFr )idappendrW rY rX rQ rS rZ rR r\ valuesr
_SpecStaterT rm r )rb ZvisitedZchildrV r r r rT
s$ zNonCallableMock.reset_mockc K s x} t | j d d d D]] \ } } | j d } | j } | } x | D] } t | | } qS Wt | | | q Wd S)aZ Set attributes on the mock through keyword arguments.
Attributes plus return values and side effects can be set on child
mocks using standard dot notation and unpacking a dictionary in the
method call:
>>> attrs = {'method.return_value': 3, 'other.side_effect': KeyError}
>>> mock.configure_mock(**attrs)keyc S s | d j d S)Nr .)count)entryr r r 0 s z0NonCallableMock.configure_mock..r N)sorteditemssplitpopr: ru )rb r, argvalr+ finalr r r r r r # s
z NonCallableMock.configure_mockc C s] | d k r t | n[ | j d k r[ | | j k sE | t k rv t d | qv n t | rv t | n | j j | } | t k r t | n | d k rd } | j d k r t | j | } n | j d | d | d | d | d | } | | j | )r r )zmockr )rr rp rj r r rx r< reversedrq r} joinr r r r )
rb Z
_name_listr lastdotseenZ_firstrI Z name_stringZ spec_stringr r r rc \ sL
zNonCallableMock.__repr__c C s t s t j | S| j p g } t t | } t | j } d d | D } d d | D } t t | | | t | j
S)z8Filter the output of `dir(mock)` to only useful members.c S s% g | ] } | j d s | q S)_)
startswith).0er r r
s z+NonCallableMock.__dir__..c S s2 g | ]( } | j d s( t | r | q S)r )r r` )r r r r r r s ) r r __dir__r r r r< r@ r rj r\ )rb ZextrasZ from_typeZ from_dictr r r r s
zNonCallableMock.__dir__c s | t k r t j | | S j rh j d k rh | j k rh | j k rh t d | n"| t k r d | } t | n | t k rL j d k r | j k r t d | n t | st
t | t | | | f d d } qt
| d | t
t | | | j | | d k re| _ d St
| | | r| j | .r )rw r __setattr__r r r@ r" _unsupported_magicsr r ru r _get_methodr r\ r )rb rI rk msgr )rG rb r r s2
zNonCallableMock.__setattr__c C s | t k rJ | t | j k rJ t t | | | | j k rJ d Sn | | j k rl t j | | n | j j | t } | t k r t
| n | t k r | j | =n t | j | ._error_message)rY r AssertionErrorr r Exception)r* r+ r, expectedr actualcauser )r+ r, rb r rM s z"NonCallableMock.assert_called_withc O sP | } | j d k s@ d | j p$ d | j f } t | n | j | | S)zVassert that the mock was called exactly once and with the specified
arguments.r{ z1Expected '%s' to be called once. Called %s times.r1 )rX rq r rM )r* r+ r, rb r r r r rN s z'NonCallableMock.assert_called_once_withc s
f d d | D } t | t r. | n d } t f d d j D } | s | | k r t d | j f | n d St | } g } xA | D]9 } y | j | Wq t k
r | j | Yq Xq W| rt d t | f | n d S)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 s g | ] } j | q Sr )r )r c)rb r r r ! s z4NonCallableMock.assert_has_calls..Nc 3 s | ] } j | Vq d S)N)r )r r )rb r r # s z3NonCallableMock.assert_has_calls..z(Calls not found.
Expected: %r
Actual: %rz %r not all found in call list)
r r rQ rS r r< remover& r r= )rb ZcallsZ any_orderr r Z all_callsZ not_foundZkallr )rb r rO s*
"
z NonCallableMock.assert_has_callsc s j | | f } f d d j D } | | k r t | t rR | n d } j | | } t d | | n d S)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.c s g | ] } j | q Sr )r )r r )rb r r r A s z3NonCallableMock.assert_any_call..Nz%s call not found)r rZ r r r r )rb r+ r, r r r r r )rb r rP : s
z NonCallableMock.assert_any_callc K se t | } t | t sN t | t r3 t } q[ t | t r[ t } q[ n
| j d } | | 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{ )r r
CallableMixinr r r
r r? )rb r _typeklassr r r r J s
z NonCallableMock._get_child_mock)'r r r r5 r r! r r r Z"_NonCallableMock__get_return_valueZ"_NonCallableMock__set_return_valueZ"_NonCallableMock__return_value_docry rU r rz rW rX rY rZ rS Z!_NonCallableMock__get_side_effectZ!_NonCallableMock__set_side_effectr[ rT r rh rc r r r r r r rM rN rO rP r r r r r r
k sL
(
# 2 "
# c C s[ | d k r | St | r | St | r0 | Sy t | SWn t k
rV | SYn Xd S)N)r r; r r )r r r r r ^ s
r c @ s^ e Z d Z d d e d d d d d d d d d
Z d d Z d d Z d d
Z d S) r Nr c
K sH | | j d .) r r r r r r r r r r, attribute_namer )rb r r r r r @ s z _patch.copyc C s) t | t r | j | S| j | S)N)r r decorate_classdecorate_callable)rb r' r r r r$ M s
z_patch.__call__c C sy xr t | D]d } | j t j s+ q
n t | | } t | d sO q
n | j } t | | | | q
W| S)Nr$ )r r r TEST_PREFIXr: r r ru )rb r attr
attr_valuer r r r r S s z_patch.decorate_classc sT t d r# j j | St f d d | g _ S)N patchingsc s g } g } t } z y xh j D]] } | j } | j | | j d k rc | j | q% | j t k r% | j | q% q% W| t | 7} | | SWn? | | k r t | r | j | n t j
} Yn XWd x! t | D] } | j | q WXd S)N)
r= r __enter__r r updater r r sysexc_infor __exit__)r+ Zkeywargs
extra_argsZentered_patchersr patchingr )r' patchedr r r
f s*
z)_patch.decorate_callable..patched)r r r r )rb r' r )r' r
r r a s ! z_patch.decorate_callablec C s | j } | j } t } d } y | j | } Wn* t t f k
r^ t | | t } Yn Xd } | j r | t k r t d | | f n | | f S)NFTz!%s does not have the attribute %r)r r r r@ r" KeyErrorr: r )rb r rI rG localr r r get_original s z_patch.get_originalc C s | j | j | j } } } | j | j } } | j } | j | _ | d k r] d } n | d k rr d } n | d k r d } n | d k r | d k r t d n | d k s | d k r | d k r t d n | j \ } } | t
k r\| d k r\d } | d k rA| } | d k r| } d } qn? | d k rk| d k r| } d } qn | d k r| } n | d k s| d k r| t
k rt d n t | t rd } qn t
}
i } | d k r| }
nm | d k s| d k r\| } | d k r"| } n t | r=d | k }
n
t | }
|
r\t }
q\n | d k ru| | d r r r r
r r r rC r rU boolr
temp_originalr ru r r r )rb r r r r r, r rG r ZinheritKlass_kwargsZ this_specZ not_callableZnew_attrr r r r r r r s
z_patch.__enter__c G s t | s t d n | j rO | j t k rO t | j | j | j nO t | j | j | j r t
| j | j r t | j | j | j n | ` | ` | ` x3 t | j D]" } t | r | j
| q q Wd S)zUndo the patch.z stop called on unstarted patcherN)r RuntimeErrorr r r ru r r r r r r r r
)rb r r r r r r
s z_patch.__exit__c C s | j } | j j | | S)z-Activate a patch, returning any created mock.)r _active_patchesr )rb rH r r r start1 s z _patch.startc C s3 y | j j | Wn t k
r( Yn X| j S)zStop an active patch.)r r r& r
)rb r r r stop8 s
z _patch.stop)r r r r r r! r r$ r r r r r
r r r r r r r s
( ~ r c sf y j d d \ } Wn+ t t f k
rI t d f Yn X f d d } | | f S)Nr r{ z.Need a valid target to patch. You supplied: %rc s
t S)N)r r )r r r r J s z _get_target..)rsplitr r& )r r r r )r r _get_targetD s r c
s4 f d d } 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.
c s S)Nr r )r r r r ` s z _patch_object..)r )
r r r r r r r r r, r r )r r
_patch_objectN s r c
s t t k r' f d d } n f d d } | sN t d n t | j } | d \ } }
t | | |
| | | | | i } | | _ x[ | d d D]I \ } }
t | | |
| | | | | i } | | _ | j j | q W| 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 s
t S)N)r r )r r r r ~ s z!_patch_multiple..c s S)Nr r )r r r r s z=Must supply at least one keyword argument with patch.multipler r{ N) r strr& r< r r r r r )
r r r r r r r, r r r r r Z this_patcherr )r r _patch_multipleg s& r c
K s4 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 a
`MagicMock`. 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 `MagicMock` is
used.
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` takes arbitrary keyword arguments. These will be passed to
the `Mock` (or `new_callable`) on construction.
`patch.dict(...)`, `patch.multiple(...)` and `patch.object(...)` are
available for alternate use-cases.
)r r )
r r r r r r r r, r r r r r r s Ec @ s| e Z d Z d Z f d d d Z d d Z d d Z d d
Z d d Z d
d Z d d Z
e Z e
Z d S) _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.
Fc K s\ t | t r t | } n | | _ t | | _ | j j | | | _ d | _ d S)N) r r r in_dictri r r clear _original)rb r r r! r, r r r r! s z_patch_dict.__init__c sA t t r j St f d d } | S)Nc
s- j z | | SWd j Xd S)N)r
_unpatch_dict)r+ r )frb r r _inner s
z$_patch_dict.__call__.._inner)r r r r )rb r$ r% r )r$ rb r r$ s
!z_patch_dict.__call__c C s x~ t | D]p } t | | } | j t j r
t | d r
t | j | j | j } | | } t
| | | q
q
W| S)Nr$ ) r r: r r r r r r r r! ru )rb r r r Z decoratorZ decoratedr r r r s z_patch_dict.decorate_classc C s | j d S)zPatch the dict.N)r )rb r r r r % s z_patch_dict.__enter__c C s | j } | j } | j } y | j } Wn7 t k
rd i } x | D] } | | | | .method)r )rI r' r+ r )r' r r s r c c s | ] } d | Vq d S)z__%s__Nr )r r+ r r r r s rh r r! r Z __prepare____instancecheck____subclasscheck____del__c C s
t j | S)N)r __hash__)rb r r r r s r r3 c C s
t j | S)N)r __str__)rb r r r r s r4 c C s
t j | S)N)r
__sizeof__)rb r r r r s r5 __lt____gt____le____ge__r{ __int__r| __len__r
y ? __complex__g ? __float____bool__ __index__c s f d d } | S)Nc s& j j } | t k r | S | k S)N)__eq__rm r )otherr )rb r r r@ s z_get_eq..__eq__r )rb r@ r )rb r _get_eq s rB c s f d d } | S)Nc s j j t k r t S | k S)N)__ne__rm r )rA )rb r r rC s z_get_ne..__ne__r )rb rC r )rb r _get_ne s rD c s f d d } | S)Nc s, j j } | t k r" t g St | S)N)r rm r r )r )rb r r r s
z_get_iter..__iter__r )rb r r )rb r _get_iter s rE r@ rC r c C s t j | t } | t k r+ | | _ d St j | } | d k r y | | } Wn t k
rv t | } Yn X| | _ d St j | } | d k r | | | _ n d S)N)_return_valuesrA r rU _calculate_return_valuer" _side_effect_methodsr[ )r1 r+ rI ZfixedZreturn_calulatorrU Z
side_effectorr r r _set_return_value s
rI c @ s( e Z d Z d d Z d d Z d S)
MagicMixinc O s1 | j t t | j | | | j d S)N)_mock_set_magicsr rJ r! )rb r+ r r r r r! s
zMagicMixin.__init__c C s t } t | d d d k r| t j | j } t } t | } x6 | D]+ } | t | j k rJ t | | qJ qJ Wn | t t | j } t | } x' | D] } t | | t | | q Wd S)Nr )
_magicsr: intersectionr rj r r@ r ru
MagicProxy)rb Z these_magicsZ
remove_magicsr r r r r rK s
zMagicMixin._mock_set_magicsN)r r r r! rK r r r r rJ s rJ c @ s% e Z d Z d Z d d d Z d S)r z-A version of `MagicMock` that isn't callable.Fc C s | j | | | j d S)zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r rK )rb r r r r r r s z"NonCallableMagicMock.mock_add_specN)r r r r5 r r r r r r s c @ s% e Z d Z d Z d d d Z d S)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 C s | j | | | j d S)zAdd a spec to a mock. `spec` can either be an object or a
list of strings. Only attributes on the `spec` can be fetched as
attributes from the mock.
If `spec_set` is True then only attributes on the spec can be set.N)r rK )rb r r r r r r ( s zMagicMock.mock_add_specN)r r r r5 r r r r r r s
c @ sC e Z d Z d d Z d d Z d d Z d d d Z d S)
rN c C s | | _ | | _ d S)N)rI r )rb rI r r r r r! 4 s zMagicProxy.__init__c O s | j } | | | S)N) create_mock)rb r+ r, mr r r r$ 8 s zMagicProxy.__call__c C sT | j } | j } | j d | d | d | } t | | | t | | | | S)NrI r r )rI r r ru rI )rb r r rP r r r rO < s zMagicProxy.create_mockNc C s
| j S)N)rO )rb r r r r r __get__E s zMagicProxy.__get__)r r r r! r$ rO rQ r r r r rN 3 s rN c @ s: e Z d Z d Z d d Z d d Z d d Z d S) _ANYz2A helper object that compares equal to everything.c C s d S)NTr )rb rA r r r r@ M s z _ANY.__eq__c C s d S)NFr )rb rA r r r rC P s z _ANY.__ne__c C s d S)Nzr )rb r r r rc S s z
_ANY.__repr__N)r r r r5 r@ rC rc r r r r rR J s rR c C s d | } d } d j d d | D } d j d d | j D } | r] | } n | r | rv | d 7} n | | 7} n | | S)Nz%s(%%s)r z, c S s g | ] } t | q Sr )r/ )r r r r r r ] s z*_format_call_signature..c S s& g | ] \ } } d | | f q S)z%s=%rr )r r rk r r r r _ s )r r )rI r+ r, r Zformatted_argsZ args_stringZ
kwargs_stringr r r r Z s
r c @ s e Z d Z d Z f d d d d d d Z f d d d d d d Z d d
Z d d Z d
d Z d d Z d d Z
d d Z d S)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.
NFTc C s d } f } i } t | } | d k r< | \ } } } n | d k r | \ } }
t | t r | } t |
t r |
} q |
} q | |
} } nN | d k r | \ } t | t r | } q t | t r | } q | } n | rt j | | | f St j | | | | f S)Nr r r^ r{ )r} r r r= r ) r rk rI r r from_kallr+ r, _lenfirstsecondr r r r s0 z
_Call.__new__c C s | | _ | | _ | | _ d S)N)rI r rS )rb rk rI r r rS r r r r! s z_Call.__init__c C s | t k r d Sy t | } Wn t k
r8 d SYn Xd } t | d k r` | \ } } n | \ } } } d } | d k r f i } } n | d k r | \ } } } n | d k r| \ } t | t r | } i } qt | t r| } f i } } qf } | } np | d k r| \ }
} t |
t rr|
} t | t rb| i } } qf | } } q|
| } } n d S| r| | k rd S| | f | | f k S)NTFr r^ r r r{ )r r} r r r= r ) rb rA Z len_otherZ self_nameZ self_argsZ self_kwargsZ
other_nameZ
other_argsZ other_kwargsrk rU rV r r r r@ sJ
z _Call.__eq__c C s | j | S)N)r@ )rb rA r r r rC s z _Call.__ne__c O sW | j d k r( t d | | f d d S| j d } t | j | | f d | d | S)Nr rI z()r )rI r )rb r+ r, rI r r r r$ s
z_Call.__call__c C sN | j d k r" t d | d d Sd | j | f } t d | d | d d S)NrI rS Fz%s.%sr )rI r )rb r rI r r r rh s z_Call.__getattr__c C s | j s8 | j p d } | j d r4 d | } n | St | d k r_ d } | \ } } nD | \ } } } | s} d } n& | j d s d | } n
d | } t | | | S)Nr z()zcall%sr^ zcall.%s)rS rI r r} r )rb rI r+ r, r r r rc s
z_Call.__repr__c C sQ g } | } x2 | d k r@ | j r4 | j | n | j } q Wt 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.N)rS r r rQ r )rb valsr r r r call_list s
z_Call.call_list) r r r r5 r r! r@ rC r$ rh rc rX r r r r r l s 3 r rS c K s t | r t | } n t | t } i | d 6} | rM i | d 6} n | d k rb i } n | r{ | r{ d | d r r r r DescriptorTypesr; r rC r r# rL r4 r\ r rU r r` r: r" r r1
_must_skipru )r r r3 r r r, is_typer r r r1 r rG r r r2 r r r r st
c C s t | t s7 | t | d i k r+ d S| j } n xt | j D]i } | j j | t } | t k rn qA n t | t t f r d St t | d d t
r | Sd SqA W| S)z[
Return whether we should skip the first argument on spec's `entry`
attribute.
r@ FrQ N) r r r: r r? r@ rA r staticmethod classmethodMethodWrapperTypes)r r r[ r rH r r r rZ s rZ c C s. y | j SWn t k
r) t | SYn Xd S)N)r r" r )r r r r r s
r c @ s+ e Z d Z d d d d d d d Z d S)r FNc C s: | | _ | | _ | | _ | | _ | | _ | | _ d S)N)r idsr r r3 rI )rb r r r rI r_ r3 r r r r! s z_SpecState.__init__)r r r r! r r r r r s r c # s t | t r d n d f d d | j D } | d k r` | d d } n | d d d | d
.r{ rc rc rc rc )r bytesr ) read_dataZ data_as_listliner )rb r _iterate_read_data s "
rg r c sw f d d } f d d } f d d t d k r d d l } t t t | j j t t | j a n | d k r t d d
d t } n t d t j
_ t d g d j
_ d j _ d j _ d j _ | j _ d < d j _ | j _ f d
d } | | _ | _ | 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` methoddline`, and `readlines` of the
file handle to return. This is an empty string by default.
c s* j j d k r j j St d S)Nr ) readlinesrU r< )r+ r, )_statehandler r _readlines_side_effect s
z)mock_open.._readlines_side_effectc s6 j j d k r j j St j d S)Nr )readrU r r )r+ r, )ri rj re r r _read_side_effect s
z$mock_open.._read_side_effectc 3 sE j j d k r' x j j Vq Wn x d D] } | Vq2 Wd S)Nr )readlinerU )rf )ri rj r r _readline_side_effect s
z(mock_open.._readline_side_effectNr rI openr r{ c sJ t d < j j d k rF d < d j _ n t S)Nr r{ )rg rn r[ r )r+ r, )ro ri rj re r r
reset_data s
z mock_open..reset_data) file_spec_ior< rj r
TextIOWrapperunionBytesIOr rp r rU rg writerl rn rh r[ )r1 re rk rm rs rq r )ro ri rj re r r s. 6
c @ s: e Z d Z d Z d d Z d d Z d d Z d S) 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 K s
t | S)N)r )rb r, r r r r * s z PropertyMock._get_child_mockc C s | S)Nr )rb r Zobj_typer r r rQ - s zPropertyMock.__get__c C s | | d S)Nr )rb r r r r r __set__/ s zPropertyMock.__set__N)r r r r5 r rQ rx r r r r r
! s )o__all__ __version__r% r r functoolsr r
BaseExceptionr platformr ZlangZ Throwabler superr r r r r r r ry rY r( r4 r/ r; r> rC rL rF r` ra rd r r ZMISSINGr ZDELETEDr rl rj rw rz r< rQ r r r r
r r r r r r r r r r r r r& r( ri ZmultipleZstopallr Z
magic_methodsZnumericsr r ZinplacerightZ
_non_defaultsr rL r r rG NotImplementedrF rB rD rE rH rI rJ r r rN rR r r r= r r r rZ r r r@ r# rQ r^ rr rg r r
r r r r s + e5 % 1 Jv "" (
w B