Your IP : 216.73.216.213
�
j f�� � @ s[ d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m Z m
Z
m Z m Z m
Z
m Z m Z m Z m Z m Z m Z m Z m Z d Z e j e Z d e Z d d d � Z e j a e j Z Gd
d � d e � Z d e Z e d
d � Z! Gd d � d e � Z" Gd d � d e j# � Z$ e d k rWe j% � n d S)� N)�support)� SomeClass� is_instance)
�NonCallableMock�
CallableMixin�patch�sentinel� MagicMock�Mock�NonCallableMagicMockr �_patch�DEFAULT�call� _get_targetr �builtinsz %s.PTModuleTc sb G� f d d � d t � } | s[ � f d d � } � f d d � } | | _ | | _ n | � S)Nc s" e Z d Z � f d d � Z d S)z_get_proxy.<locals>.Proxyc s
t � | � S)N)�getattr)�self�name)�obj� �E/opt/alt/python34/lib64/python3.4/unittest/test/testmock/testpatch.py� __getattr__ s z%_get_proxy.<locals>.Proxy.__getattr__N)�__name__�
__module__� __qualname__r r )r r r �Proxy s r c s t � | | � d S)N)�setattr)r r �value)r r r � __setattr__ s z _get_proxy.<locals>.__setattr__c s t � | � d S)N)�delattr)r r )r r r � __delattr__ s z _get_proxy.<locals>.__delattr__)�objectr r )r �get_onlyr r r r )r r �
_get_proxy s r# c @ sP e Z d Z d d � Z d d � Z d d � Z d Z Gd d � d e � Z d
S) �Fooc C s d S)Nr )r �ar r r �__init__- s z Foo.__init__c C s d S)Nr )r r% r r r �f/ s zFoo.fc C s d S)Nr )r r r r �g1 s zFoo.g�barc @ s e Z d Z d d � Z d S)zFoo.Barc C s d S)Nr )r r r r r% 6 s z Foo.Bar.aN)r r r r% r r r r �Bar5 s r* N) r r r r& r' r( �foor! r* r r r r r$ , s
r$ z%s.Fooc C s d S)Nr )r% �br r r �function< s r- c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) � Containerc C s
i | _ d S)N)�values)r r r r r& A s zContainer.__init__c C s | j | S)N)r/ )r r r r r � __getitem__D s zContainer.__getitem__c C s | | j | <d S)N)r/ )r r r r r r � __setitem__G s zContainer.__setitem__c C s | j | =d S)N)r/ )r r r r r � __delitem__J s zContainer.__delitem__c C s
t | j � S)N)�iterr/ )r r r r �__iter__M s zContainer.__iter__N)r r r r& r0 r1 r2 r4 r r r r r. @ s
r. c @ s� e Z d 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 d � Z 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$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z d> d? � Z! d@ dA � Z" dB dC � Z# dD dE � Z$ dF dG � Z% dH dI � Z& dJ dK � Z' dL dM � Z( dN dO � Z) dP dQ � Z* dR dS � Z+ dT dU � Z, dV dW � Z- dX dY � Z. dZ d[ � Z/ d\ d] � Z0 d^ d_ � Z1 d` da � Z2 db dc � Z3 dd de � Z4 df dg � Z5 dh di � Z6 dj dk � Z7 dl dm � Z8 dn do � Z9 dp dq � Z: dr ds � Z; dt du � Z< dv dw � Z= dx dy � Z> dz d{ � Z? d| d} � Z@ d~ d � ZA d� d� � ZB d� d� � ZC d� d� � ZD d� d� � ZE d� d� � ZF d� d� � ZG d� d� � ZH d� d� � ZI d� d� � ZJ d� d� � ZK d� d� � ZL d� d� � ZM d� d� � ZN d� d� � ZO d� d� � ZP d� d� � ZQ eR d� d� � d� d� � � ZS eR d� d� � d� d� � � ZT d� d� � ZU d� d� � ZV d� d� � ZW d� d� � ZX d� d� � ZY d� d� � ZZ d� d� � Z[ d� d� � Z\ d� d� � Z] d� d� � Z^ d� d� � Z_ d� d� � Z` d� d� � Za d� d� � Zb d� S)�� PatchTestTc C sU t } | s t } n | j t | � | j t | | � � | j t | t � � d S)N)r r � assertRaises� TypeError�
assertTruer � assertFalser )r r �magic� MockClassr r r �assertNotCallableT s zPatchTest.assertNotCallablec sg Gd d � d t � � t j � d t j � � � f d d � � } | � � j � j t j d � d S)Nc @ s e Z d Z e j Z d S)z4PatchTest.test_single_patchobject.<locals>.SomethingN)r r r r �Original� attributer r r r � Something_ s r? r> c s � j � j t j d � d S)N� unpatched)� assertEqualr> r �Patchedr )r? r r r �testb s z/PatchTest.test_single_patchobject.<locals>.testzpatch not restored)r! r r rB rA r> r= )r rC r )r? r r �test_single_patchobject^ s
-z!PatchTest.test_single_patchobjectc sd Gd d � d t � � t j � d d � � � f d d � � } | � � j � j t j d � d S)Nc @ s e Z d Z e j Z d S)z7PatchTest.test_patchobject_with_none.<locals>.SomethingN)r r r r r= r> r r r r r? l s r? r> c s � j � j d � d S)Nr@ )� assertIsNoner> r )r? r r r rC o s z2PatchTest.test_patchobject_with_none.<locals>.testzpatch not restored)r! r rA r> r r= )r rC r )r? r r �test_patchobject_with_nonek s
*z$PatchTest.test_patchobject_with_nonec s� Gd d � d t � � t j � d t j � t j � d t j � � � f d d � � � } | � � j � j t j d � � j � j t j d � d S)Nc @ s" e Z d Z e j Z e j Z d S)z6PatchTest.test_multiple_patchobject.<locals>.SomethingN)r r r r r= r> � Original2�next_attributer r r r r? y s r? r> rH c s6 � j � j t j d � � j � j t j d � d S)Nr@ )rA r> r rB rH �Patched2r )r? r r r rC } s z1PatchTest.test_multiple_patchobject.<locals>.testzpatch not restored)
r! r r rB rI rA r> r= rH rG )r rC r )r? r r �test_multiple_patchobjectx s 0z#PatchTest.test_multiple_patchobjectc
C s] t } t d t t j � d d � � } z' t j a | � | j t t j � Wd | a Xd S)Nz %s.somethingc S s d S)Nr r r r r rC � s z8PatchTest.test_object_lookup_is_quite_lazy.<locals>.test)� somethingr r r �
Something2Zreplacement_valuerA )r �originalrC r r r � test_object_lookup_is_quite_lazy� s " z*PatchTest.test_object_lookup_is_quite_lazyc sP t d t t j � � f d d � � } | � � j t j t j d � t d t t j � t d t t j � � f d d � � � } � j t j t j d � � j t j t j d � | � � j t j t j d � � j t j t j d � t
� } t j | _ t d t
| � � f d d � � } | � | � � j t | d � d S) Nz %s.somethingc s � j t j t j d � d S)Nr@ )rA �PTModulerK r rL r )r r r rC � s z"PatchTest.test_patch.<locals>.testzpatch not restoredz%s.something_elsec s6 � j t j t j d � � j t j t j d � d S)Nr@ )rA rO rK r rL �something_else�
SomethingElser )r r r rC � s z%s.openc s# � j t d d � t j d � d S)N�filename�rzopen not patched)rA �openr �Handler )r r r rC � s )r r r rL rA rO rK r? rQ rP r
rU � return_value�builtin_string�assertNotEqualrT )r rC �mockr )r r �
test_patch� s* (+ %zPatchTest.test_patchc sI t d t t j � � f d d � � } | � � j t j j d � d S)Nz %s.SomeClass.class_attributec s � j t j j t j d � d S)Nr@ )rA rO r �class_attributer �ClassAttributer )r r r rC � s z2PatchTest.test_patch_class_attribute.<locals>.testzpatch not restored)r r r r\ rE rO r r[ )r rC r )r r �test_patch_class_attribute� s
z$PatchTest.test_patch_class_attributec s� Gd d � d t � � t j � d � � � f d d � � } | � t j � d � t j � d � � � � � f d d � � � } d � � | t j t j � \ � � | t j t j � d S)Nc @ s" e Z d Z e j Z e j Z d S)z:PatchTest.test_patchobject_with_default_mock.<locals>.TestN)r r r r r= rK rG �
something2r r r r �Test� s r_ rK c s- � j | � j d � � j | t d � d S)Nz"Mock not passed into test functionz.patch with two arguments did not create a mock)rA rK �assertIsInstancer )rY )r_ r r r rC � s z:PatchTest.test_patchobject_with_default_mock.<locals>.testr^ c s� � j | t j d � � j | t j d � � j | � j d � � j | � j d � � j | t d � � j | t d � � j � | d � � j � | d � | | f S)Nz0Patched function didn't receive initial argumentz/Patched function didn't receive second argumentz"Mock not passed into test functionz)Second Mock not passed into test functionz.patch with two arguments did not create a mockzunexpected value for mock1) rA r �this1�this2r^ rK r` r rX )ra rb Zmock1Zmock2)r_ �
outerMock1�
outerMock2r r r rC � s )r! r r ra rb )r rC r )r_ rc rd r r �"test_patchobject_with_default_mock� s '0
z,PatchTest.test_patchobject_with_default_mockc s3 t d t d t �� f d d � � } | � d S)Nz %s.SomeClass�specc sC � j t | � � j t t j t � � � j t d d � � d S)Nc S s t j S)N)r �
not_wibbler r r r �<lambda> s z>PatchTest.test_patch_with_spec.<locals>.test.<locals>.<lambda>)rA r r8 r �wibbler r6 �AttributeError)�
MockSomeClass)r r r rC � s z,PatchTest.test_patch_with_spec.<locals>.test)r r r )r rC r )r r �test_patch_with_spec� s (z PatchTest.test_patch_with_specc s5 t j t d d t �� f d d � � } | � d S)Nr[ rf c sI � j t j | � � j t t j j t � � � j t d d � � d S)Nc S s
t j j S)N)r r[ rg r r r r rh
s zDPatchTest.test_patchobject_with_spec.<locals>.test.<locals>.<lambda>) rA r r[ r8 r ri r r6 rj )�
MockAttribute)r r r rC s
z2PatchTest.test_patchobject_with_spec.<locals>.test)r r! r )r rC r )r r �test_patchobject_with_spec s *z$PatchTest.test_patchobject_with_specc s6 t d t d d g �� f d d � � } | � d S)Nz %s.SomeClassrf ri c sC � j t | � � j t t j t � � � j t d d � � d S)Nc S s t j S)N)r rg r r r r rh s zFPatchTest.test_patch_with_spec_as_list.<locals>.test.<locals>.<lambda>)rA r r8 r ri r r6 rj )rk )r r r rC s z4PatchTest.test_patch_with_spec_as_list.<locals>.test)r r )r rC r )r r � test_patch_with_spec_as_list s +z&PatchTest.test_patch_with_spec_as_listc s8 t j t d d d g �� f d d � � } | � d S)Nr[ rf ri c sI � j t j | � � j t t j j t � � � j t d d � � d S)Nc S s
t j j S)N)r r[ rg r r r r rh # s zLPatchTest.test_patchobject_with_spec_as_list.<locals>.test.<locals>.<lambda>) rA r r[ r8 r ri r r6 rj )rm )r r r rC s
z:PatchTest.test_patchobject_with_spec_as_list.<locals>.test)r r! r )r rC r )r r �"test_patchobject_with_spec_as_list s -z,PatchTest.test_patchobject_with_spec_as_listc sF t d t � t d t d d g �� f d d � � � } | � d S)Nz%s.openz %s.SomeClassrf ri c sC � j t | � � j t t j t � � � j t d d � � d S)Nc S s t j S)N)r rg r r r r rh / s zMPatchTest.test_nested_patch_with_spec_as_list.<locals>.test.<locals>.<lambda>)rA r r8 r ri r r6 rj )rk ZMockOpen)r r r rC * s z;PatchTest.test_nested_patch_with_spec_as_list.<locals>.test)r rW r )r rC r )r r �#test_nested_patch_with_spec_as_list( s
.z-PatchTest.test_nested_patch_with_spec_as_listc s3 t d t d d �� f d d � � } | � d S)Nz %s.SomeClassrf Tc s7 � j t � � � j � j t � f d d � � d S)Nc s � j S)N)rg r )rk r r rh : s zIPatchTest.test_patch_with_spec_as_boolean.<locals>.test.<locals>.<lambda>)rA r ri r6 rj )rk )r )rk r rC 4 s z7PatchTest.test_patch_with_spec_as_boolean.<locals>.test)r r )r rC r )r r � test_patch_with_spec_as_boolean3 s (z)PatchTest.test_patch_with_spec_as_booleanc s5 t j t d d d �� f d d � � } | � d S)Nr rf Tc s7 � j t � � � j � j t � f d d � � d S)Nc s � j S)N)rg r )rk r r rh F s zPPatchTest.test_patch_object_with_spec_as_boolean.<locals>.test.<locals>.<lambda>)rA r ri r6 rj )rk )r )rk r rC @ s z>PatchTest.test_patch_object_with_spec_as_boolean.<locals>.test)r r! rO )r rC r )r r �&test_patch_object_with_spec_as_boolean? s *z0PatchTest.test_patch_object_with_spec_as_booleanc s3 t d t d d �� f d d � � } | � d S)Nz %s.SomeClassrf Tc sS � j t | t � � | � � � j � � � j � j t � f d d � � d S)Nc s � j S)N)rg r )�instancer r rh T s zVPatchTest.test_patch_class_acts_with_spec_is_inherited.<locals>.test.<locals>.<lambda>)r8 r r r< ri r6 rj )rk )r )rt r rC L s
zDPatchTest.test_patch_class_acts_with_spec_is_inherited.<locals>.test)r r )r rC r )r r �,test_patch_class_acts_with_spec_is_inheritedK s (
z6PatchTest.test_patch_class_acts_with_spec_is_inheritedc sO t d t t j d d �� f d d � � } | � � j t d d � � d S)Nz
%s.frooble�createTc s � j t t j � d S)N)rA �froobler �Froobler )r r r rC Z s zLPatchTest.test_patch_with_create_mocks_non_existent_attributes.<locals>.testc S s t S)N)rw r r r r rh _ s zPPatchTest.test_patch_with_create_mocks_non_existent_attributes.<locals>.<lambda>)r rW r rx r6 � NameError)r rC r )r r �4test_patch_with_create_mocks_non_existent_attributesY s .z>PatchTest.test_patch_with_create_mocks_non_existent_attributesc sQ t j t d t j d d �� f d d � � } | � � j t t d � � d S)Nrw rv Tc s � j t j t j � d S)N)rA r rw r rx r )r r r rC c s zRPatchTest.test_patchobject_with_create_mocks_non_existent_attributes.<locals>.test)r r! r r rx r9 �hasattr)r rC r )r r �:test_patchobject_with_create_mocks_non_existent_attributesb s 0zDPatchTest.test_patchobject_with_create_mocks_non_existent_attributesc so y3 t d t t j � � f d d � � } | � Wn t k
rG Yn X� j d � � j t d d � � d S)Nz
%s.frooblec s � j t t j � d S)N)rA rw r rx r )r r r rC m s z9PatchTest.test_patch_wont_create_by_default.<locals>.testz,Patching non existent attributes should failc S s t S)N)rw r r r r rh w s z=PatchTest.test_patch_wont_create_by_default.<locals>.<lambda>)r rW r rx rj �failr6 ry )r rC r )r r �!test_patch_wont_create_by_defaultk s (
z+PatchTest.test_patch_wont_create_by_defaultc sq y5 t j t d t j � � f d d � � } | � Wn t k
rI Yn X� j d � � j t t d � � d S)Nrw c s � j d � d S)Nz,Patching non existent attributes should fail)r} r )r r r rC | s z?PatchTest.test_patchobject_wont_create_by_default.<locals>.testz,Patching non existent attributes should fail) r r! r r rx rj r} r9 r{ )r rC r )r r �'test_patchobject_wont_create_by_defaultz s *
z1PatchTest.test_patchobject_wont_create_by_defaultc sp Gd d � d t � � t j � d t d d � � � � � f d d � � } | � � j � j � t j � d S)Nc @ s" e Z d Z e d d � � Z d S)z5PatchTest.test_patch_with_static_methods.<locals>.Fooc S s t j S)N)r �Staticr r r r �woot� s z:PatchTest.test_patch_with_static_methods.<locals>.Foo.wootN)r r r � staticmethodr� r r r r r$ � s r$ r� c S s t j S)N)r rB r r r r rh � s z:PatchTest.test_patch_with_static_methods.<locals>.<lambda>c s � j � j � t j � d S)N)rA r� r rB r )r$ r r r � anonymous� s z;PatchTest.test_patch_with_static_methods.<locals>.anonymous)r! r r� rA r� r r� )r r� r )r$ r r � test_patch_with_static_methods� s 6z(PatchTest.test_patch_with_static_methodsc sN t j } t j t d d � � f d d � � } | � � j t j | � d S)Nr$ c s � j t j d � d S)Nr$ )rA r r$ r )r r r r� � s z-PatchTest.test_patch_local.<locals>.anonymous)r r$ r r! rA )r r+ r� r )r r �test_patch_local� s 'zPatchTest.test_patch_localc sv Gd d � d t � } | � � t j � _ t j � d d � � � f d d � � } | � � j � j t j � d S)Nc @ s e Z d Z d Z d S)z'PatchTest.test_patch_slots.<locals>.Foor$ N)zFoo)r r r � __slots__r r r r r$ � s r$ c s � j � j d � d S)Nr$ )rA r$ r )r+ r r r r� � s z-PatchTest.test_patch_slots.<locals>.anonymous)r! r r$ r rA )r r$ r� r )r+ r r �test_patch_slots� s *zPatchTest.test_patch_slotsc s� Gd d � d t � � G� � f d d � d t � } t j � d t j � | � } | � } | j � | j � � j � j t j d � d S)Nc @ s e Z d Z e j Z d S)z=PatchTest.test_patchobject_class_decorator.<locals>.SomethingN)r r r r r= r> r r r r r? � s r? c s: e Z d Z � � f d d � Z � � f d d � Z d S)z7PatchTest.test_patchobject_class_decorator.<locals>.Fooc s � j � j t j d � d S)Nr@ )rA r> r rB )�
other_self)r? r r r � test_method� s zCPatchTest.test_patchobject_class_decorator.<locals>.Foo.test_methodc s � j � j t j d � d S)Nznon-test method patched)rA r> r r= )r� )r? r r r �not_test_method� s zGPatchTest.test_patchobject_class_decorator.<locals>.Foo.not_test_methodN)r r r r� r� r )r? r r r r$ � s r$ r> zpatch not restored) r! r r rB r� r� rA r> r= )r r$ r' r )r? r r � test_patchobject_class_decorator� s
z*PatchTest.test_patchobject_class_decoratorc s� Gd d � d t � } G� f d d � d t � } t d t � | � } | � } | j � | j � � j | j t j d � � j t j
t j d � d S)Nc @ s e Z d Z e j Z d S)z7PatchTest.test_patch_class_decorator.<locals>.SomethingN)r r r r r= r> r r r r r? � s r? c s4 e Z d Z � f d d � Z � f d d � Z d S)z1PatchTest.test_patch_class_decorator.<locals>.Fooc s � j t j | d � d S)Nr@ )rA rO rK )r� Zmock_something)r r r r� � s z=PatchTest.test_patch_class_decorator.<locals>.Foo.test_methodc s � j t j t j d � d S)Nznon-test method patched)rA rO rK r r? )r� )r r r r� � s zAPatchTest.test_patch_class_decorator.<locals>.Foo.not_test_methodN)r r r r� r� r )r r r r$ � s r$ z %s.somethingzpatch not restored) r! r r r� r� rA r> r r= rO rK r? )r r? r$ r' r )r r �test_patch_class_decorator� s
z$PatchTest.test_patch_class_decoratorc s Gd d � d t � � t j � d t j � t j � d t j � � � f d d � � � } | � � j � j t j d � d S)Nc @ s" e Z d Z e j Z e j Z d S)z3PatchTest.test_patchobject_twice.<locals>.SomethingN)r r r r r= r> rG rH r r r r r? � s r? r> c s � j � j t j d � d S)Nr@ )rA r> r rB r )r? r r r rC � s z.PatchTest.test_patchobject_twice.<locals>.testzpatch not restored)r! r r rB rA r> r= )r rC r )r? r r �test_patchobject_twice� s 0z PatchTest.test_patchobject_twicec s� i t � d 6d d 6� � j � } t j � � � f d d � � } | � � j � | � t j � i d d 6� � � f d d � � } | � � j � | � t j � d
g � � � f d d � � } | � � j � | � d S) N�initialrK �otherc s d � d <� d =d � d <d S)N� r% r� zsomething elser� r r )r+ r r rC � s
z'PatchTest.test_patch_dict.<locals>.testr, r% c s. � j t � � d � � j � d d � d S)Nr� r% r, )rA �lenr )r+ r r r rC � s c s. � j t � � d � � j � d d � d S)Nr� r% r, )rA r� r )r+ r r r rC s )r% r, )r! �copyr �dictrA )r rM rC r )r+ r r �test_patch_dict� s !.*zPatchTest.test_patch_dictc s� t � � t � � d <d � d <� j j � } t j � � � f d d � � } | � � j � j | � t j � i d d 6� � � f d d � � } | � � j � j | � d S) Nr� rK r� c s d � d <� d =d � d <d S)Nr� r% r� zsomething elser� r r )r+ r r rC s
z=PatchTest.test_patch_dict_with_container_object.<locals>.testr, r% c s1 � j t � j � d � � j � d d � d S)Nr� r% r, )rA r� r/ r )r+ r r r rC s )r. r! r/ r� r r� rA )r rM rC r )r+ r r �%test_patch_dict_with_container_object
s
!.z/PatchTest.test_patch_dict_with_container_objectc s� i t � d 6d d 6� � j � } t j � d d �� � f d d � � } | � � j � | � t j � i d d 6d d �� � f d
d � � } | � � j � | � t j � d g d d �� � f d d � � } | � � j � | � d S)
Nr� rK r� �clearTc s( � j � i � d � d <d � d <d S)Nr� r% zsomething elser� )rA r )r+ r r r rC , s
z2PatchTest.test_patch_dict_with_clear.<locals>.testr, r% c s � j � i d d 6� d S)Nr, r% )rA r )r+ r r r rC 6 s c s � j � i d d 6� d S)Nr, r% )rA r )r+ r r r rC > s )r% r, )r! r� r r� rA )r rM rC r )r+ r r �test_patch_dict_with_clear( s *40z$PatchTest.test_patch_dict_with_clearc s� t � � t � � d <d � d <� j j � } t j � d d �� � f d d � � } | � � j � j | � t j � i d d 6d d �� � f d
d � � } | � � j � j | � d S) Nr� rK r� r� Tc s+ � j � j i � d � d <d � d <d S)Nr� r% zsomething elser� )rA r/ r )r+ r r r rC N s
zGPatchTest.test_patch_dict_with_container_object_and_clear.<locals>.testr, r% c s � j � j i d d 6� d S)Nr, r% )rA r/ r )r+ r r r rC X s )r. r! r/ r� r r� rA )r rM rC r )r+ r r �/test_patch_dict_with_container_object_and_clearG s
*4z9PatchTest.test_patch_dict_with_container_object_and_clearc C s i } t d t t � � t d t t � d d �t j t t � � t j | � d d � � � � � } | j | j d � d S)Nz %s.SomeClass�autospecTc S s d S)Nr r r r r � some_named s z0PatchTest.test_name_preserved.<locals>.some_namer� )r r r! r r� rA )r r+ r� r r r �test_name_preserveda s $z PatchTest.test_name_preservedc C sl i } t j | i d d 6� d d � � } y | � Wn t k
rJ Yn X| j d � | j | i � d S)Nr, r% c S s t d � � d S)NZKonrad)ry r r r r rC q s z1PatchTest.test_patch_with_exception.<locals>.testz NameError not raised by test)r r� ry r} rA )r r+ rC r r r �test_patch_with_exceptionn s %
z#PatchTest.test_patch_with_exceptionc s6 t j d i d d 6� � f d d � � } | � d S)Nz
os.environz
some value�
konrad_delongc s � j d t j � d S)Nr� )�assertIn�os�environr )r r r rC s z3PatchTest.test_patch_dict_with_string.<locals>.test)r r� )r rC r )r r �test_patch_dict_with_string~ s +z%PatchTest.test_patch_dict_with_stringc s� d SGd d � d t � � G� � � f d d � d t � � � j � j d � d � � j � � j � j d � d � � j � � � } � j | j d � d � | j � � j | j d � d � | j � )Nc @ s e Z d Z d Z d S)z0PatchTest.test_patch_descriptor.<locals>.NothingN)r r r r+ r r r r �Nothing� s r� c s� e Z d Z i Z e j � d d � e � � f d d � � � Z e j � d d � e d d � � � Z e j
e � e � � f d d � � � Z e j
e � e d d
� � � Z d S) z2PatchTest.test_patch_descriptor.<locals>.Somethingr+ � c s � j | � � d S)N)�assertIs)�cls)r? r r r �klass� s z8PatchTest.test_patch_descriptor.<locals>.Something.klassc S s | S)Nr )�argr r r �static� s z9PatchTest.test_patch_descriptor.<locals>.Something.staticc s � j | � � d S)N)r� )r� )r? r r r �
klass_dict� s z=PatchTest.test_patch_descriptor.<locals>.Something.klass_dictc S s | S)Nr )r� r r r � static_dict� s z>PatchTest.test_patch_descriptor.<locals>.Something.static_dictN)
r r r r+ r r! � classmethodr� r� r� r� r� r� r )r� r? r r r r? � s r? Zf00)r! rA r� r� r� r� )r rK r )r� r? r r �test_patch_descriptor� s "
z PatchTest.test_patch_descriptorc C s� t d t d t d d �d d � � } | j t | � t j t d d t d d �d d � � } | j t | � t d t d d �d d � � } | j t | � t j t d d d �d
d � � } | j t | � d S) Nz %s.SomeClassrf �spec_setTc S s
d | _ d S)Nr+ )�z)r; r r r rC � s z+PatchTest.test_patch_spec_set.<locals>.testr c S s
d | _ d S)Nr+ )r� )r; r r r rC � s c S s
d | _ d S)Nr+ )r� )r; r r r rC � s c S s
d | _ d S)Nr+ )r� )r; r r r rC � s )r r r r6 rj r! r )r rC r r r �test_patch_spec_set� s (*"$z PatchTest.test_patch_spec_setc C s6 t d t d d �d d � � } | j t | � d S)Nz %s.SomeClassr� Tc S s | � } d | _ d S)Nr+ )r� )r; rt r r r rC � s z-PatchTest.test_spec_set_inherit.<locals>.test)r r r6 rj )r rC r r r �test_spec_set_inherit� s "z PatchTest.test_spec_set_inheritc C sx t } t d t � } | j t | � | j � } z$ | j | | � | j t | � Wd | j � X| j t | � d S)Nz %s.something)rK r r r� �start� assertIsNot�stop)r rM �patcherrY r r r �test_patch_start_stop� s z PatchTest.test_patch_start_stopc C s) t t d d � } | j t | j � d S)Nr) r� )r �foo_namer6 � RuntimeErrorr� )r r� r r r �test_stop_without_start� s z!PatchTest.test_stop_without_startc C s} t } t j t d d � } | j t | � | j � } z$ | j | d � | j t | � Wd | j � X| j t | � d S)NrK r+ )rK r r! rO r� r� rA r� )r rM r� Zreplacedr r r �test_patchobject_start_stop� s z%PatchTest.test_patchobject_start_stopc C s� i d d 6} | j � } t j | d g d d �} | j | | � | j � z | j | i d d 6� Wd | j � X| j | | � d S)Nr) r+ �spam�eggsr� T)r� r� )r� r r� rA r� r� )r �drM r� r r r �test_patch_dict_start_stop� s
z$PatchTest.test_patch_dict_start_stopc s� | � i d d 6� � j � } G� � f d d � d t � } t j � i d d 6d d �| � } | j � | � | � } | j � | j � | � | j � | j � | � | � } | j � | j � | � | j � | j � | � d S) Nr� r� c s: e Z d Z � � f d d � Z � � f d d � Z d S)z7PatchTest.test_patch_dict_class_decorator.<locals>.Testc s � j � i d d 6� d S)Nr) r+ )rA )r )r� �thisr r �
test_first s zBPatchTest.test_patch_dict_class_decorator.<locals>.Test.test_firstc s � j � i d d 6� d S)Nr) r+ )rA )r )r� r� r r � test_second
s zCPatchTest.test_patch_dict_class_decorator.<locals>.Test.test_secondN)r r r r� r� r )r� r� r r r_ s r_ r) r+ r� T)r� r! r r� rA r� r� )r rM r_ rC r )r� r� r � test_patch_dict_class_decorator s
%
z)PatchTest.test_patch_dict_class_decoratorc s� Gd d � d t � } Gd d � d � } x� | | | � | f D]| } t | � � t j � d d � � � f d d � � } | � � j � j d � � j | j d � � j d � j � q? Wd S) Nc @ s e Z d Z d Z d S)z0PatchTest.test_get_only_proxy.<locals>.Somethingr+ N)r r r r+ r r r r r? " s r? c @ s e Z d Z d Z d S)z4PatchTest.test_get_only_proxy.<locals>.SomethingElser+ N)r r r r+ r r r r rQ $ s rQ r+ r) c s � j � j d � d S)Nr) )rA r+ r )�proxyr r r rC * s z+PatchTest.test_get_only_proxy.<locals>.test)r! r# r rA r+ � assertNotIn�__dict__)r r? rQ �thingrC r )r� r r �test_get_only_proxy! s *z PatchTest.test_get_only_proxyc s� Gd d � d t � } Gd d � d � } x� | | | � | f D]� } t | d d �� t j � d d � � � f d d
� � } | � � j � j d � � j | j d � � j d � j � q? Wd S) Nc @ s e Z d Z d Z d S)z6PatchTest.test_get_set_delete_proxy.<locals>.Somethingr+ N)r r r r+ r r r r r? 4 s r? c @ s e Z d Z d Z d S)z:PatchTest.test_get_set_delete_proxy.<locals>.SomethingElser+ N)r r r r+ r r r r rQ 6 s rQ r" Fr+ r) c s � j � j d � d S)Nr) )rA r+ r )r� r r r rC < s z1PatchTest.test_get_set_delete_proxy.<locals>.test)r! r# r rA r+ r� r� )r r? rQ r� rC r )r� r r �test_get_set_delete_proxy3 s *z#PatchTest.test_get_set_delete_proxyc C s� i t d 6d d 6t � d 6} t t | � } | j � } | j � | j t | � | j | j j � d � | j
| j t � d S)N� side_effect�! zfoo.bar.return_valuer+ ) �KeyErrorr r r� r� r� r6 rA r+ r) r` )r �kwargsr� rY r r r �test_patch_keyword_argsE s
z!PatchTest.test_patch_keyword_argsc C s� i t d 6d d 6t � d 6} t j t d | � } | j � } | j � | j t | � | j | j j
� d � | j | j t � d S)Nr� r� zfoo.bar.return_valuer+ r' ) r� r r r! r$ r� r� r6 rA r+ r) r` )r r� r� rY r r r � test_patch_object_keyword_argsR s
z(PatchTest.test_patch_object_keyword_argsc C s� i d d 6} | j � } t j | d d d d d d �} | j � z) | j | t d d d d d d � � Wd | j � X| j | | � d S)Nr) r+ r� � Zbaz� )r� r r� r� rA r� )r rM r� r� r r r � test_patch_dict_keyword_args_ s
!
) z&PatchTest.test_patch_dict_keyword_argsc s� Gd d � d t � } � f d d � � � f d d � � � � � f d d � } t t d d
�| � } | � } � j t | � | � t j t } t j | d d d
�| � } | � } � j t | � | � d S) Nc @ sP e Z d Z d d � Z d d � Z d d � Z d Z Gd d � d e � Z d
S) z$PatchTest.test_autospec.<locals>.Booc S s d S)Nr )r r% r r r r& p s z-PatchTest.test_autospec.<locals>.Boo.__init__c S s d S)Nr )r r% r r r r' r s z&PatchTest.test_autospec.<locals>.Boo.fc S s d S)Nr )r r r r r( t s z&PatchTest.test_autospec.<locals>.Boo.gr) c @ s e Z d Z d d � Z d S)z(PatchTest.test_autospec.<locals>.Boo.Barc S s d S)Nr )r r r r r% y s z*PatchTest.test_autospec.<locals>.Boo.Bar.aN)r r r r% r r r r r* x s r* N) r r r r& r' r( r+ r! r* r r r r �Booo s
r� c s+ | d � | j d � � j t | � d S)N� )�assert_called_withr6 r7 )rY )r r r �_test| s
z&PatchTest.test_autospec.<locals>._testc sn | j d � | j j d � � j t | j � | j � | j j � � j t | j d � � j t t | d � | j j � | j j j � � j t t | j d � | j � | j j � | j j
� | j j
j � � j t | j j
d � | j � j
� | j � j
j � � j t | j � j
d � � j t t | j d � � j t t | j � d � d S)Nr� �hr) r, ) r' r� r6 r7 r( rj r r+ �lowerr* r% )rY )r r r �_test2� s(
z'PatchTest.test_autospec.<locals>._test2c s8 � | � � | � � | d � � � j | t � | S)Nr� )r� r$ )rY )r� r� r r r r- � s
z)PatchTest.test_autospec.<locals>.functionr� Tr$ )r! r r� r� r$ �sys�modulesr )r r� r- rC rY �moduler )r� r� r r �
test_autospecn s
zPatchTest.test_autospecc s3 t d t d d �� f d d � � } | � d S)Nz %s.functionr� Tc s^ t d � t j d � t d d � t j d d � � j t t � � j t t t d � d S)Nr� r� r� r+ )r- r� r6 r7 rj r )rY )r r r rC � s
z.PatchTest.test_autospec_function.<locals>.test)r r )r rC r )r r �test_autospec_function� s (
z PatchTest.test_autospec_functionc C sE t d t d d d d �d d � � } | � } | j | d � d S)Nz %s.functionr� TrV r� c S s
t d d � S)Nr� r� )r- )Z
mock_functionr r r rC � s z.PatchTest.test_autospec_keywords.<locals>.test)r r rA )r rC �resultr r r �test_autospec_keywords� s z PatchTest.test_autospec_keywordsc C sq t d t d d d d �} | j t | j � t j t } t j | d d d d d �} | j t | j � d S)Nz %s.function�newr� r� Tr- )r r r6 r7 r� r� r� r! )r r� r� r r r �test_autospec_with_new� s
z PatchTest.test_autospec_with_newc
C sm Gd d � d t � } t t d | �} | j � } z' | j | | � | j | j t � Wd | j � Xd S)Nc @ s e Z d Z g Z d S)z0PatchTest.test_autospec_with_object.<locals>.BarN)r r r �extrar r r r r* � s r* r� )r$ r r� r� r` r� �listr� )r r* r� rY r r r �test_autospec_with_object� s z#PatchTest.test_autospec_with_objectc
C s` t } t t d d �} | j � } z* | j | | � | j | d � | � Wd | j � Xd S)Nr� Tr� )r$ r r� r� r` r� )r ZFooClassr� rY r r r �test_autospec_inherits� s z PatchTest.test_autospec_inheritsc C s� t t d d �} | j � } zn | j d t | � � | j d t | j � � | j d t | d � � � | j d t | d � j � � Wd | j � Xd S)Nr� Tz name='Foo'z
name='Foo.f'z
name='Foo()'z name='Foo().f')r r� r� r� �reprr' r� )r r� rY r r r �test_autospec_name� s #z PatchTest.test_autospec_namec
C s� t j t d t � � d d � � } y | � Wn t j � } Yn Xt j d d d � } | j | | � } | j d | � d S)Nr' c S s
t � d S)N)�AssertionErrorr r r r rC � s z'PatchTest.test_tracebacks.<locals>.testr zraise AssertionError) r r! r$ r� �exc_info�unittestZTextTestResultZ_exc_info_to_stringr� )r rC �errr� � tracebackr r r �test_tracebacks� s $ zPatchTest.test_tracebacksc C sv t t d t �} | j � } | j � | j � } | j � | j | | � x! | | f D] } | j | � q[ Wd S)N� new_callable)r r� r r� r� r� r< )r r� �m1�m2rY r r r �test_new_callable_patch
s
z!PatchTest.test_new_callable_patchc C s| t j t d d t �} | j � } | j � | j � } | j � | j | | � x! | | f D] } | j | � qa Wd S)Nr' r� )r r! r$ r r� r� r� r< )r r� r� r� rY r r r � test_new_callable_patch_object s
z(PatchTest.test_new_callable_patch_objectc
s� G� f d d � d t � � t t d � d d d d �} | j � } z< | j t | � � � | j � j t d d d d � � Wd | j � Xd S)Nc s( e Z d Z d Z � f d d � Z d S)z:PatchTest.test_new_callable_keyword_arguments.<locals>.BarNc s
| � _ d S)N)r� )r r� )r* r r r& ' s zCPatchTest.test_new_callable_keyword_arguments.<locals>.Bar.__init__)r r r r� r& r )r* r r r* % s r* r� Zarg1r� Zarg2r� )
r! r r� r� r� �typerA r� r� r� )r r� �mr )r* r �#test_new_callable_keyword_arguments$ s &z-PatchTest.test_new_callable_keyword_argumentsc s� G� f d d � d t � � t t d � d � �} | j � z | j � j t d � � � Wd | j � Xt t d � d � �} | j � z | j � j t d � � � Wd | j � Xd S)Nc s( e Z d Z d Z � f d d � Z d S)z-PatchTest.test_new_callable_spec.<locals>.BarNc s
| � _ d S)N)r� )r r� )r* r r r& 6 s z6PatchTest.test_new_callable_spec.<locals>.Bar.__init__)r r r r� r& r )r* r r r* 4 s r* r� rf r� )r! r r� r� rA r� r� r� )r r� r )r* r �test_new_callable_spec3 s
z PatchTest.test_new_callable_specc
C s| d t } t | d t �} | j t | j � t | d t d d �} | j � } z | j | d d �Wd | j � Xd S)Nz %s.weeeeer� rv Tr: F)r� r r r6 rj r� r< r� )r Znon_existent_attr�pr� r r r �test_new_callable_createH s
z"PatchTest.test_new_callable_createc C sN | j t t t d t � d t �| j t t j t d d t � d t �d S)Nr� r� r' )r6 �
ValueErrorr r� r! r r$ )r r r r �'test_new_callable_incompatible_with_newV s
z1PatchTest.test_new_callable_incompatible_with_newc C sH | j t t t d t d d �| j t t j t d d t d d �d S)Nr� r� Tr' )r6 r� r r� r r! r$ )r r r r �,test_new_callable_incompatible_with_autospec` s z6PatchTest.test_new_callable_incompatible_with_autospecc
C s� Gd d � d t � } t t t t | f } x~ | D]v } xm d D]e } i d | 6} t t d | | �} | j � } z# | j } | j t t
| d � Wd | j � Xq? Wq2 Wd S) Nc @ s e Z d Z d S)z>PatchTest.test_new_callable_inherit_for_mocks.<locals>.MockSubN)r r r r r r r �MockSubl s r� rf r� Tr� �x)zspeczspec_set) r
r r r r r� r� rV r6 rj r r� ) r r� Z MockClassesZKlassr� r� r� r� rt r r r �#test_new_callable_inherit_for_mocksk s
z-PatchTest.test_new_callable_inherit_for_mocksc C s� Gd d � d t � } t t d | d d �} | j � } z0 | j t | | � � | j t t | d � Wd | j � X| j
| j t � d S)Nc @ s e Z d Z d d � Z d S)z>PatchTest.test_new_callable_inherit_non_mock.<locals>.NotAMockc S s
| | _ d S)N)rf )r rf r r r r& � s zGPatchTest.test_new_callable_inherit_non_mock.<locals>.NotAMock.__init__N)r r r r& r r r r �NotAMock s r� r� rf TrV )
r! r r� r� r8 r r6 rj r r� rA rf r$ )r r� r� r� r r r �"test_new_callable_inherit_non_mock~ s z,PatchTest.test_new_callable_inherit_non_mockc sq | � t � G� � f d d � d t � } t t d t �| � } | � j � | � j � | j t � � d S)Nc s= e Z d Z � � f d d � Z d d � Z d d � Z d S)z>PatchTest.test_new_callable_class_decorating.<locals>.SomeTestc s4 � j t � � � j t | � � j t t � d S)N)r� r$ r� r` r )r �mock_foo)rM rC r r r� � s zDPatchTest.test_new_callable_class_decorating.<locals>.SomeTest._testc S s | j | � d S)N)r� )r r r r r �test_two� s zGPatchTest.test_new_callable_class_decorating.<locals>.SomeTest.test_twoc S s | j | � d S)N)r� )r r r r r �test_one� s zGPatchTest.test_new_callable_class_decorating.<locals>.SomeTest.test_oneN)r r r r� r r r )rM rC r r �SomeTest� s r r� )r$ r! r r� r r r r� )r r r )rM rC r �"test_new_callable_class_decorating� s
z,PatchTest.test_new_callable_class_decoratingc s( t � t j } t j } t j t d d d d �} t j t d d d d �} x� | | f D]� } | j � z: � j t � � � j t j d � � j t j d � Wd | j � X� j t � � � j t j | � � j t j | � q[ Wt j t d d d d �� � f d d � � } | � d S) Nr' r� r( r� r� r� c s: � j t � � � j t j d � � j t j d � d S)Nr� r� )r� r$ rA r' r( r )� original_foor r r rC � s z+PatchTest.test_patch_multiple.<locals>.test)
r$ r' r( r �multipler� r� r� rA r� )r �
original_f�
original_gZpatcher1Zpatcher2r� rC r )r r r �test_patch_multiple� s"
0z PatchTest.test_patch_multiplec C s0 | j t t j t � | j t t j t � d S)N)r6 r� r r r� r$ )r r r r � test_patch_multiple_no_kwargs� s z'PatchTest.test_patch_multiple_no_kwargsc s t � t j } t j } t j t d t d d d t �� � f d d � � } | � � j t j | � � j t j | � d S)Nr' r( r� r+ c sy � j t � � � j t j | � � j t j d � � j t j | � � j t | t � � � j t | t � � d S)Nr� ) r� r$ r' rA r( r+ r8 r r )r' r+ )r r r r rC � s z8PatchTest.test_patch_multiple_create_mocks.<locals>.test)r$ r' r( r r r� r
rA )r r r rC r )r r r � test_patch_multiple_create_mocks� s 6 z*PatchTest.test_patch_multiple_create_mocksc s� t j } t j } t j t d d � } d | _ t j t d t � } d | _ | g | _ | � f d d � � } | � � j t j | � � j t j | � d S)Nr' r� r( c s* � j t j | � � j t j d � d S)Nr� )r� r$ r( rA r' )r( )r r r rC � s zHPatchTest.test_patch_multiple_create_mocks_different_order.<locals>.test) r$ r' r( r r! �attribute_namer
�additional_patchersrA )r r r r� r� rC r )r r �0test_patch_multiple_create_mocks_different_order� s z:PatchTest.test_patch_multiple_create_mocks_different_orderc sP t � t j } t j } t j t d t �t j t d t �t t d � � f d d � � � � } t j t d t �t t d � t j t d t �� f d d � � � � } t t d � t j t d t �t j t d t �� f d d � � � � } � � f d
d � � | � | � | � � j t j | � � j t j | � d S) Nr' r+ z.gc s � | | � d S)Nr )r( r� )r� r r �test1� s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test1c s � | | � d S)Nr )r( r� )r� r r �test2� s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test2c s � | | � d S)Nr )r( r� )r� r r �test3 s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test3c s� | j d � } | j d � } � j | � � j t � � � j t j | � � j t j | � � j t j | � � j t | t � � � j t | t � � � j t | t � � d S)Nr' r+ )
�popr9 r� r$ r' r( r+ r8 r r )r( r� r' r+ )r r r r r� s
z?PatchTest.test_patch_multiple_stacked_decorators.<locals>._test)r$ r' r( r r r� r
rA )r r r r r r r )r� r r r �&test_patch_multiple_stacked_decorators� s$ (
-
-
z0PatchTest.test_patch_multiple_stacked_decoratorsc C s t } t j } t j } t j t d t d d d t �} | j � } z� | d } | d } | j t | � t d d g � � | j
t | � | j
t j | � | j
t j | � | j t
| t � � | j t
| t � � Wd | j � X| j t j | � | j t j | � d S)Nr' r( r� r+ )r$ r' r( r r r� r
r� rA �setr� r+ r8 r r r� )r r r r r� r� r' r+ r r r �(test_patch_multiple_create_mocks_patcher s" !
" z2PatchTest.test_patch_multiple_create_mocks_patcherc s� | � t � t j } t j } G� � f d d � d t � } t j t d t d d d t �| � } | � } | j � | j � | j
t j | � | j
t j | � d S)Nc s= e Z d Z � � f d d � Z d d � Z d d � Z d S)z@PatchTest.test_patch_multiple_decorating_class.<locals>.SomeTestc sy � j t � � � j t j | � � j t j d � � j t j | � � j t | t � � � j t | t � � d S)Nr� ) r� r$ r' rA r( r+ r8 r r )r r' r+ )r rC r r r� < s zFPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest._testc S s | j | | � d S)N)r� )r r' r+ r r r r D s zIPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest.test_twoc S s | j | | � d S)N)r� )r r' r+ r r r r F s zIPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest.test_oneN)r r r r� r r r )r rC r r r : s r r' r( r� r+ ) r$ r' r( r! r r r� r
r r rA )r r r r r� r )r rC r �$test_patch_multiple_decorating_class4 s
z.PatchTest.test_patch_multiple_decorating_classc C s� t j t d d �} | j t | j � t j t d d d d �} | j � z | j t j d � Wd | j � X| j t
t d � � d S)N�blamrv T) r r r$ r6 rj r� rA r r� r9 r{ )r r� r r r �test_patch_multiple_createU s
z$PatchTest.test_patch_multiple_createc C s� t j t d t d d d g �} | j � } z} | j t j | d � t j j d � t j j d � t j j j d � t j j j d � | j
t t t j d d � Wd | j
� Xd S)Nr+ r� r% r, r� r� �c)r r r$ r
r� rA r+ r% r, r� r6 rj r r� )r r� r� r r r � test_patch_multiple_spec_setc s ! z&PatchTest.test_patch_multiple_spec_setc
C s� Gd d � d t � } t j t d t d t d | �} | j � } zn | j t j | d � | j t j | d � | j t j | � | j t j | � | j
t j t j � Wd | j � Xd S)Nc @ s e Z d Z d S)z9PatchTest.test_patch_multiple_new_callable.<locals>.ThingN)r r r r r r r �Thingt s r r' r( r� ) r! r r r$ r
r� r� r' r( r` r� r� )r r r� r� r r r � test_patch_multiple_new_callables s z*PatchTest.test_patch_multiple_new_callablec C sG t j } t j } t j t d d � t j t d d � t j t d d � d d � � � � } t j t d d � t j t d d � t j t d d � d d � � � � } t j t d d � t j t d d � t j t d d � d d
� � � � } xM | | | f D]< } | j t | � | j t j | � | j t j | � qWd S) Nr( r� �missingr' c S s d S)Nr r r r r �thing1� s z3PatchTest.test_nested_patch_failure.<locals>.thing1c S s d S)Nr r r r r �thing2� s z3PatchTest.test_nested_patch_failure.<locals>.thing2c S s d S)Nr r r r r �thing3� s z3PatchTest.test_nested_patch_failure.<locals>.thing3)r$ r' r( r r! r6 rj rA )r r r r r r �funcr r r �test_nested_patch_failure� s '''z#PatchTest.test_nested_patch_failurec C sx t j } t j } t j } d d � } t j t d d � t j t d d | �t j t d d � d d � � � � } t j t d d | �t j t d d � t j t d d � d
d � � � � } t j t d d � t j t d d � t j t d d | �d d
� � � � } x` | | | f D]O } | j t | � | j t j | � | j t j | � | j t j | � q!Wd S)Nc S s t d � � d S)N�crasher)ry r r r r r# � s z4PatchTest.test_new_callable_failure.<locals>.crasherr( r� r+ r� r' c S s d S)Nr r r r r r � s z3PatchTest.test_new_callable_failure.<locals>.thing1c S s d S)Nr r r r r r � s z3PatchTest.test_new_callable_failure.<locals>.thing2c S s d S)Nr r r r r r � s z3PatchTest.test_new_callable_failure.<locals>.thing3) r$ r' r( r+ r r! r6 ry rA ) r r r r r# r r r r! r r r �test_new_callable_failure� s$ ''*z#PatchTest.test_new_callable_failurec C s� t j } t j } t j t d d � } d | _ t j t d d � } d | _ t j t d d � } d | _ xq | | g | | g f D]W } | | _ | d d � � } | j t | � | j t j | � | j t j | � q� Wd S)Nr' r� r( r c S s d S)Nr r r r r r! � s z3PatchTest.test_patch_multiple_failure.<locals>.func)
r$ r' r( r r! r r
r6 rj rA )r r r r� �good�bad� additionalsr! r r r �test_patch_multiple_failure� s z%PatchTest.test_patch_multiple_failurec
C s t j } t j } t j } d d � } t j t d d � } d | _ t j t d d � } d | _ t j t d d | �} d | _ x� | | g | | g f D]j } | | _ | d d � � } | j t | � | j
t j | � | j
t j | � | j
t j | � q� Wd S)
Nc S s t d � � d S)Nr# )ry r r r r r# � s zCPatchTest.test_patch_multiple_new_callable_failure.<locals>.crasherr' r� r( r+ r� c S s d S)Nr r r r r r! � s z@PatchTest.test_patch_multiple_new_callable_failure.<locals>.func) r$ r' r( r+ r r! r r
r6 ry rA )
r r r r r# r� r% r& r' r! r r r �(test_patch_multiple_new_callable_failure� s" z2PatchTest.test_patch_multiple_new_callable_failurec sm t d t f i d d 6� } | � � t j � d d �� � f d d � � } | � � j � j d � d S)Nr$ Ztasty�fishz nearly gonec s � j � j d � d S)Nz nearly gone)rA r* r )r+ r r r rC � s z=PatchTest.test_patch_multiple_string_subclasses.<locals>.test)r� �strr r rA r* )r r$ rC r )r+ r r �%test_patch_multiple_string_subclasses� s
*z/PatchTest.test_patch_multiple_string_subclassesz unittest.mock.patch.TEST_PREFIXr+ c C s� Gd d � d t � } t j | d d � | � } | � } | j | j � d � | j | j � d � | j | j � d � | j | j � d � d S)Nc @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) z-PatchTest.test_patch_test_prefix.<locals>.FoorM c S s | j S)N)r� )r r r r �foo_one s z5PatchTest.test_patch_test_prefix.<locals>.Foo.foo_onec S s | j S)N)r� )r r r r �foo_two s z5PatchTest.test_patch_test_prefix.<locals>.Foo.foo_twoc S s | j S)N)r� )r r r r r
s z6PatchTest.test_patch_test_prefix.<locals>.Foo.test_onec S s | j S)N)r� )r r r r r s z6PatchTest.test_patch_test_prefix.<locals>.Foo.test_twoN)r r r r� r- r. r r r r r r r$ s
r$ r� �changedrM )r! r rA r- r. r r )r r$ r+ r r r �test_patch_test_prefix s z PatchTest.test_patch_test_prefixr) c s� G� f d d � d t � } i d d 6� t j � d d �| � } | � } | j | j � i d d 6� | j | j � i d d 6� | j | j � i d d 6� | j | j � i d d 6� d S)Nc sX e Z d Z � f d d � Z � f d d � Z � f d d � Z � f d d � Z d S)
z2PatchTest.test_patch_dict_test_prefix.<locals>.Fooc s
t � � S)N)r� )r )�the_dictr r �bar_one s z:PatchTest.test_patch_dict_test_prefix.<locals>.Foo.bar_onec s
t � � S)N)r� )r )r1 r r �bar_two s z:PatchTest.test_patch_dict_test_prefix.<locals>.Foo.bar_twoc s
t � � S)N)r� )r )r1 r r r " s z;PatchTest.test_patch_dict_test_prefix.<locals>.Foo.test_onec s
t � � S)N)r� )r )r1 r r r $ s z;PatchTest.test_patch_dict_test_prefix.<locals>.Foo.test_twoN)r r r r2 r3 r r r )r1 r r r$ s r$ rM �keyr/ )r! r r� rA r2 r3 r r )r r$ r+ r )r1 r �test_patch_dict_test_prefix s
z%PatchTest.test_patch_dict_test_prefixc C s� x� d
D]� } t d t i d | 6� } | j � } zh | j d t | � � | j d t | j � � | j d t | � � � | j d t | � j � � Wd | j � Xq Wd S) Nrf r� r� z %s.SomeClassTz name='SomeClass'z! name='SomeClass.class_attribute'z name='SomeClass()'z# name='SomeClass().class_attribute')zspeczautospeczspec_set)r r r� r� r� r[ r� )r r� r� r� r r r � test_patch_with_spec_mock_repr1 s
z(PatchTest.test_patch_with_spec_mock_reprc C s` t d d d ��H } | j d t | j j � � � | j d t | j � j � � � Wd QXd S)Nz unittest.test.testmock.supportr� Tz" name='support.SomeClass.wibble()'z$ name='support.SomeClass().wibble()')r r� r� r ri )r r� r r r � test_patch_nested_autospec_repr@ s
z)PatchTest.test_patch_nested_autospec_reprc C s[ xTd D]L} t d t i d | 6� } | j � } z| j � t j � g } | j | j | � | j | j | � | j | j j t � g � | � } | j t � � | j | j | � | j � | j t � j � � | j | j | � | j | j t j � g � | j | j j t � g � | j | j t j � g � Wd | j � Xq Wd S)Nrf r� r� z %s.SomeClassT)zspeczautospeczspec_set)
r r r� ri r rA Z
mock_callsZ method_calls�appendr� )r r� r� r� Zkallsr� r r r �test_mock_calls_with_patchI s&
z$PatchTest.test_mock_calls_with_patchc C s� t j j d d � t d � } | j t | j � t � } d | _ | t j d <t d � } d | _ | j � | j � | j
| j d � d S)N�squizzz
squizz.squozz� r� ) r� r� r r r6 � ImportErrorr� r
Zsquozzr� rA )r Zp1r: r r r �test_patch_imports_lazilyd s
z#PatchTest.test_patch_imports_lazilyc s� Gd d � d � � G� f d d � d t � � � f d d � } | d � d d � � } | j t | � | j � j d
t � | j � j d d � | j � j d
d � d S)Nc @ s e Z d Z d Z d S)z<PatchTest.test_patch_propogrates_exc_on_exit.<locals>.holderN)NNN)r r r r� r r r r �holderu s r> c s1 e Z d Z d d d � f d d � Z e Z d S)zBPatchTest.test_patch_propogrates_exc_on_exit.<locals>.custom_patchNc s, t j | | | | � | | | f � _ d S)N)r �__exit__r� )r �etype�val�tb)r> r r r? y s zKPatchTest.test_patch_propogrates_exc_on_exit.<locals>.custom_patch.__exit__)r r r r? r� r )r> r r � custom_patchx s rC c
s4 t | � \ } } � | | t d d d d d i � S)NF)r r
)�target�getterr> )rC r r �with_custom_patch~ s zGPatchTest.test_patch_propogrates_exc_on_exit.<locals>.with_custom_patchz
squizz.squozzc S s
t � d S)N)r� )rY r r r rC � s z:PatchTest.test_patch_propogrates_exc_on_exit.<locals>.testr r� z exception value not propgatedr� z!exception traceback not propgated)r r6 r� r� r� ZassertIsNotNone)r rF rC r )rC r> r �"test_patch_propogrates_exc_on_exitt s z,PatchTest.test_patch_propogrates_exc_on_exitc C s� x� d D] } t d t d d i d | 6�} | j t | j � | j t d d � � t t d d i d | 6�} | j � | j � q Wd S)
Nrf r� r� z%s.doesnotexistrv Tc S s t S)N)� doesnotexistr r r r rh � s z1PatchTest.test_create_and_specs.<locals>.<lambda>)zspeczspec_setzautospec)r r r6 r7 r� ry �MODNAMEr� )r �kwargr� r r r �test_create_and_specs� s
z PatchTest.test_create_and_specsc C s� t } xM d D]E } t t d d i d | 6�} | j t | j � | j t | � q
WxM d D]E } t t d d i d | 6�} | j t | j � | j t | � q] WxM d D]E } t t d d i d | 6�} | j t | j � | j t | � q� Wd S)Nrf r� r� r )�spec�spec_set)rL r� )rM r� )rO r rI r6 r7 r� r� )r rM rJ r� r r r �test_multiple_specs� s
z PatchTest.test_multiple_specsc
C sP t t d d d d d d �} | j � } z | j d | _ Wd | j � Xd S)Nrf Fr� r� r� )r rI r� Zdoes_not_existr� )r r� rY r r r � test_specs_false_instead_of_none� s
z*PatchTest.test_specs_false_instead_of_nonec C s_ xX d D]P } t t i d | 6� } | j � } z | j t t | d � Wd | j � Xq Wd S)Nrf r� r� r Z doesnotexit)zspeczautospeczspec_set)r rI r� r6 rj r r� )r rJ r� r� r r r �test_falsey_spec� s
zPatchTest.test_falsey_specc
C s~ xw d D]o } t t d d i d | 6�} | j � } z3 | j t t | d d � | j t t | d � Wd | j � Xq Wd S)Nrf r� r� TrH rK )zspeczautospec)r rI r� r6 rj r r r� )r rJ r� r� r r r �test_spec_set_true� s
z PatchTest.test_spec_set_truec
C sM d } t t d | �} | j � } z | j t | � � Wd | j � Xd S)N�__call__rf )z__call__)r rI r� r8 �callabler� )r rf r� r� r r r �test_callable_spec_as_list� s z$PatchTest.test_callable_spec_as_listc
C sM d } t t d | �} | j � } z | j t | � � Wd | j � Xd S)Nr+ r) rf )zfoozbar)r rI r� r9 rS r� )r rf r� r� r r r � test_not_callable_spec_as_list� s z(PatchTest.test_not_callable_spec_as_listc s� t j � t j � t j } t d t � j � t d t � j � t d � � � � f d d � � } | � � j t j | � d S)Nz os.unlinkzos.chdirzos.pathc sG t j � � j t j | � � j t j � � � j t j � � d S)N)r �stopallr� r� �path�unlink�chdir)Z mock_path)rY r rX r r �patched� s
z-PatchTest.test_patch_stopall.<locals>.patched) r� rX rY rW r rK r� rP r� )r rW rZ r )rY r rX r �test_patch_stopall� s $z PatchTest.test_patch_stopallc so g � Gd d � d t � � � � f d d � � � f d d � d
D� t j � | j � d d d g � d S) Nc @ s e Z d Z d Z Z Z d S)z*PatchTest.test_stopall_lifo.<locals>.thingN)r r r �one�two�threer r r r r� � s r� c
sP G� � � f d d � d t � � � � f d d � � d d d d d d i � S)Nc s+ e Z d Z � � � � f d d � Z � S)z?PatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.mypatchc s � j � � t � | � j � S)N)r8 �superr� )r )� __class__r> �mypatch�stoppedr r r� � s
zDPatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.mypatch.stop)r r r r� r )r> ra rb )r` r ra � s ra c s � S)Nr r )r� r r rh s z@PatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.<lambda>F)r )r> )rb r� )r> ra r � get_patch� s "z.PatchTest.test_stopall_lifo.<locals>.get_patchc s" g | ] } � | � j � � q Sr )r� )�.0rA )rc r r �
<listcomp> s z/PatchTest.test_stopall_lifo.<locals>.<listcomp>r\ r] r^ )zoneztwozthree)r! r rV rA )r r )rc rb r� r �test_stopall_lifo� s
zPatchTest.test_stopall_lifoN)cr r r r< rD rF rJ rN rZ r] re rl rn ro rp rq rr rs ru rz 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� 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 r r r r" r$ r( r) r, r r0 r5 r6 r7 r9 r= rG rK rN rO rP rQ rT rU r[ rf r r r r r5 R s�
+ *
!
+
G
+ ! !
r5 �__main__)&r� r� r� Zunittest.test.testmockr Z unittest.test.testmock.supportr r Z
unittest.mockr r r r r r
r r r
r r rW r� r rO rI r# r? rK rQ rP r! r$ r� r- r. ZTestCaser5 �mainr r r r �<module> s0 X
� � � � � � �