File: //usr/local/lib/python3.7/unittest/test/testmock/__pycache__/testpatch.cpython-37.pyc
B
"��g�� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl
m Z m Z m
Z
mZmZmZmZmZmZmZmZ dZeje Zde Zdd d
�Ze
j a e
j Z G d d � d e �Z!d
e Z"e!fdd�Z#G dd� de �Z$G dd� dej%�Z&edkr�e�'� dS )� N)�support)� SomeClass� is_instance)�uncache) �NonCallableMock�
CallableMixin�sentinel� MagicMock�Mock�NonCallableMagicMock�patch�_patch�DEFAULT�call� _get_target�builtinsz %s.PTModuleTc sB G � fdd�dt �}|s<� fdd�}� fdd�}||_||_|� S )Nc s e Zd Z� fdd�ZdS )z_get_proxy.<locals>.Proxyc s
t � |�S )N)�getattr)�self�name)�obj� �</usr/local/lib/python3.7/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 Zd Zdd� Zdd� Zdd� ZdZedd � �Ze d
d � �Z
G d d
� d
e �Z dS )�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 �f0 s zFoo.fc C s d S )Nr )r r r r �g2 s zFoo.g�barc C s dS )N� r r r r r �
static_method6 s zFoo.static_methodc C s dS )N�* r )�clsr r r � class_method: s zFoo.class_methodc @ s e Zd Zdd� ZdS )zFoo.Barc C s d S )Nr )r r r r r& ? s z Foo.Bar.aN)r r r r& r r r r �Bar> s r0 N)
r r r r' r( r) �foo� staticmethodr, � classmethodr/ r" r0 r r r r r% - s r% z%s.Fooc C s d S )Nr )r& �br r r �functionE s r5 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � Containerc C s
i | _ d S )N)�values)r r r r r' J s zContainer.__init__c C s
| j | S )N)r7 )r r r r r � __getitem__M s zContainer.__getitem__c C s || j |< d S )N)r7 )r r r r r r � __setitem__P s zContainer.__setitem__c C s | j |= d S )N)r7 )r r r r r � __delitem__S s zContainer.__delitem__c C s
t | j�S )N)�iterr7 )r r r r �__iter__V s zContainer.__iter__N)r r r r' r8 r9 r: r< r r r r r6 I s
r6 c @ sJ e Zd Zd�dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLdM� Z(dNdO� Z)dPdQ� Z*dRdS� Z+dTdU� Z,dVdW� Z-dXdY� Z.dZd[� Z/d\d]� Z0d^d_� Z1d`da� Z2dbdc� Z3ddde� Z4dfdg� Z5dhdi� Z6djdk� Z7dldm� Z8dndo� Z9dpdq� Z:drds� Z;dtdu� Z<dvdw� Z=dxdy� Z>dzd{� Z?d|d}� Z@d~d� ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVeWd�d��d�d�� �ZXeWd�d��d�d�� �ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d Zbd�dĄ Zcd�dƄ Zdd�dȄ Zed�dʄ Zfd�d̄ Zgd�d΄ Zhd�S )�� PatchTestTc C s< t }|s t}| �t|� | �t||�� | �t|t�� d S )N)r r � assertRaises� TypeError�
assertTruer � assertFalser )r r �magic� MockClassr r r �assertNotCallable] s zPatchTest.assertNotCallablec sJ G dd� dt �� t� � dtj�� �fdd��}|� ��� jtjd� d S )Nc @ s e Zd ZejZdS )z4PatchTest.test_single_patchobject.<locals>.SomethingN)r r r r �Original� attributer r r r � Somethingh s rG rF c s �� � jtjd� d S )N� unpatched)� assertEqualrF r �Patchedr )rG r r r �testk s z/PatchTest.test_single_patchobject.<locals>.testzpatch not restored)r" r r rJ rI rF rE )r rK r )rG r r �test_single_patchobjectg s
z!PatchTest.test_single_patchobjectc C s, d}| � t|�� t�dd� W d Q R X d S )Nz>'Something' must be the actual object to be patched, not a strrG Z do_something)ZassertRaisesRegexr? r r" )r �msgr r r �&test_patchobject_with_string_as_targets s z0PatchTest.test_patchobject_with_string_as_targetc sH G dd� dt �� t� � dd �� �fdd��}|� ��� jtjd� d S )Nc @ s e Zd ZejZdS )z7PatchTest.test_patchobject_with_none.<locals>.SomethingN)r r r r rE rF r r r r rG y s rG rF c s �� � jd� d S )NrH )� assertIsNonerF r )rG r r r rK | s z2PatchTest.test_patchobject_with_none.<locals>.testzpatch not restored)r" r rI rF r rE )r rK r )rG r r �test_patchobject_with_nonex s
z$PatchTest.test_patchobject_with_nonec sl G dd� dt �� t� � dtj�t� � dtj�� �fdd���}|� ��� jtjd� ��� jtj d� d S )Nc @ s e Zd ZejZejZdS )z6PatchTest.test_multiple_patchobject.<locals>.SomethingN)r r r r rE rF � Original2�next_attributer r r r rG � s rG rF rR c s( �� � jtjd� �� � jtjd� d S )NrH )rI rF r rJ rR �Patched2r )rG r r r rK � s z1PatchTest.test_multiple_patchobject.<locals>.testzpatch not restored)
r" r r rJ rS rI rF rE rR rQ )r rK r )rG r r �test_multiple_patchobject� s z#PatchTest.test_multiple_patchobjectc C sF t }tdt tj�dd� �}z tja |� | �t tj� W d |a X d S )Nz %s.somethingc S s d S )Nr r r r r rK � s z8PatchTest.test_object_lookup_is_quite_lazy.<locals>.test)� somethingr r r �
Something2Zreplacement_valuerI )r �originalrK r r r � test_object_lookup_is_quite_lazy� s z*PatchTest.test_object_lookup_is_quite_lazyc s� t dt tj�� fdd��}|� � �tjtjd� t dt tj�t dt tj�� fdd���}� �tjtjd� � �tj tjd� |� � �tjtjd� � �tj tjd� t
� }tj |_ t dt
|�� fdd��}|� |� � �t|d� d S ) Nz %s.somethingc s � � tjtjd� d S )NrH )rI �PTModulerU r rV r )r r r rK � s z"PatchTest.test_patch.<locals>.testzpatch not restoredz%s.something_elsec s( � � tjtjd� � � tjtjd� d S )NrH )rI rY rU r rV �something_else�
SomethingElser )r r r rK � s z%s.openc s � � tdd�tjd� d S )N�filename�rzopen not patched)rI �openr �Handler )r r r rK � s )r r r rV rI rY rU rG r[ rZ r
r_ � return_value�builtin_string�assertNotEqualr^ )r rK �mockr )r r �
test_patch� s* zPatchTest.test_patchc s6 t dt tj�� fdd��}|� � �tjjd� d S )Nz %s.SomeClass.class_attributec s � � tjjtjd� d S )NrH )rI rY r �class_attributer �ClassAttributer )r r r rK � s
z2PatchTest.test_patch_class_attribute.<locals>.testzpatch not restored)r r r rf rO rY r re )r rK r )r r �test_patch_class_attribute� s
z$PatchTest.test_patch_class_attributec s� G dd� dt �� t� � d�� �fdd��}|� t� � d�t� � d�� ���fdd���}d ��|tjtj�\��|tjtj� d S )Nc @ s e Zd ZejZejZdS )z:PatchTest.test_patchobject_with_default_mock.<locals>.TestN)r r r r rE rU rQ �
something2r r r r �Test� s ri rU c s" �� | � jd� ��| td� d S )Nz"Mock not passed into test functionz.patch with two arguments did not create a mock)rI rU �assertIsInstancer )rc )ri r r r rK � s
z:PatchTest.test_patchobject_with_default_mock.<locals>.testrh c s� �� | tjd� �� |tjd� �� |� jd� �� |� jd� ��|td� ��|td� ���|d� ���|d� ||fS )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) rI r �this1�this2rh rU rj r rb )rk rl Zmock1Zmock2)ri �
outerMock1�
outerMock2r r r rK � s
)r" r r rk rl )r rK r )ri rm rn r r �"test_patchobject_with_default_mock� s
z,PatchTest.test_patchobject_with_default_mockc s& t dt td�� fdd��}|� d S )Nz %s.SomeClass)�specc s2 � � t| � � �ttjt�� � �tdd� � d S )Nc S s t jS )N)r �
not_wibbler r r r �<lambda> � z>PatchTest.test_patch_with_spec.<locals>.test.<locals>.<lambda>)rI r r@ r �wibbler r> �AttributeError)�
MockSomeClass)r r r rK
s z,PatchTest.test_patch_with_spec.<locals>.test)r r r )r rK r )r r �test_patch_with_spec s z PatchTest.test_patch_with_specc s& t jtdtd�� fdd��}|� d S )Nre )rp c s6 � � tj| � � �ttjjt�� � �tdd� � d S )Nc S s t jjS )N)r re rq r r r r rr rs zDPatchTest.test_patchobject_with_spec.<locals>.test.<locals>.<lambda>) rI r re r@ r rt r r> ru )�
MockAttribute)r r r rK s
z2PatchTest.test_patchobject_with_spec.<locals>.test)r r" r )r rK r )r r �test_patchobject_with_spec s z$PatchTest.test_patchobject_with_specc s( t dt dgd�� fdd��}|� d S )Nz %s.SomeClassrt )rp c s2 � � t| � � �ttjt�� � �tdd� � d S )Nc S s t jS )N)r rq r r r r rr $ rs zFPatchTest.test_patch_with_spec_as_list.<locals>.test.<locals>.<lambda>)rI r r@ r rt r r> ru )rv )r r r rK s z4PatchTest.test_patch_with_spec_as_list.<locals>.test)r r )r rK r )r r � test_patch_with_spec_as_list s z&PatchTest.test_patch_with_spec_as_listc s( t jtddgd�� fdd��}|� d S )Nre rt )rp c s6 � � tj| � � �ttjjt�� � �tdd� � d S )Nc S s t jjS )N)r re rq r r r r rr 0 rs zLPatchTest.test_patchobject_with_spec_as_list.<locals>.test.<locals>.<lambda>) rI r re r@ r rt r r> ru )rx )r r r rK * s
z:PatchTest.test_patchobject_with_spec_as_list.<locals>.test)r r" r )r rK r )r r �"test_patchobject_with_spec_as_list) s z,PatchTest.test_patchobject_with_spec_as_listc s4 t dt �t dt dgd�� fdd���}|� d S )Nz%s.openz %s.SomeClassrt )rp c s2 � � t| � � �ttjt�� � �tdd� � d S )Nc S s t jS )N)r rq r r r r rr < rs zMPatchTest.test_nested_patch_with_spec_as_list.<locals>.test.<locals>.<lambda>)rI r r@ r rt r r> ru )rv ZMockOpen)r r r rK 7 s z;PatchTest.test_nested_patch_with_spec_as_list.<locals>.test)r ra r )r rK r )r r �#test_nested_patch_with_spec_as_list5 s
z-PatchTest.test_nested_patch_with_spec_as_listc s& t dt dd�� fdd��}|� d S )Nz %s.SomeClassT)rp c s* �� t� � � j ��t� fdd�� d S )Nc s � j S )N)rq r )rv r r rr G rs zIPatchTest.test_patch_with_spec_as_boolean.<locals>.test.<locals>.<lambda>)rI r rt r> ru )rv )r )rv r rK A s z7PatchTest.test_patch_with_spec_as_boolean.<locals>.test)r r )r rK r )r r � test_patch_with_spec_as_boolean@ s z)PatchTest.test_patch_with_spec_as_booleanc s& t jtddd�� fdd��}|� d S )Nr T)rp c s* �� t� � � j ��t� fdd�� d S )Nc s � j S )N)rq r )rv r r rr S rs zPPatchTest.test_patch_object_with_spec_as_boolean.<locals>.test.<locals>.<lambda>)rI r rt r> ru )rv )r )rv r rK M s z>PatchTest.test_patch_object_with_spec_as_boolean.<locals>.test)r r" rY )r rK r )r r �&test_patch_object_with_spec_as_booleanL s z0PatchTest.test_patch_object_with_spec_as_booleanc s& t dt dd�� fdd��}|� d S )Nz %s.SomeClassT)rp c s> �� t| t�� | � � ��� � � j ��t� fdd�� d S )Nc s � j S )N)rq r )�instancer r rr a rs zVPatchTest.test_patch_class_acts_with_spec_is_inherited.<locals>.test.<locals>.<lambda>)r@ r r rD rt r> ru )rv )r )r r rK Y s
zDPatchTest.test_patch_class_acts_with_spec_is_inherited.<locals>.test)r r )r rK r )r r �,test_patch_class_acts_with_spec_is_inheritedX s
z6PatchTest.test_patch_class_acts_with_spec_is_inheritedc s: t dt tjdd�� fdd��}|� � �tdd� � d S )Nz
%s.froobleT)�createc s � � ttj� d S )N)rI �froobler �Froobler )r r r rK g s zLPatchTest.test_patch_with_create_mocks_non_existent_attributes.<locals>.testc S s t S )N)r� r r r r rr l rs zPPatchTest.test_patch_with_create_mocks_non_existent_attributes.<locals>.<lambda>)r ra r r� r> � NameError)r rK r )r r �4test_patch_with_create_mocks_non_existent_attributesf s z>PatchTest.test_patch_with_create_mocks_non_existent_attributesc s: t jtdtjdd�� fdd��}|� � �ttd�� d S )Nr� T)r� c s � � tjtj� d S )N)rI r r� r r� r )r r r rK p s zRPatchTest.test_patchobject_with_create_mocks_non_existent_attributes.<locals>.test)r r" r r r� rA �hasattr)r rK r )r r �:test_patchobject_with_create_mocks_non_existent_attributeso s zDPatchTest.test_patchobject_with_create_mocks_non_existent_attributesc sZ y&t dt tj�� fdd��}|� W n tk
r: Y n X � �d� � �tdd� � d S )Nz
%s.frooblec s � � ttj� d S )N)rI r� r r� r )r r r rK z s z9PatchTest.test_patch_wont_create_by_default.<locals>.testz,Patching non existent attributes should failc S s t S )N)r� r r r r rr � rs z=PatchTest.test_patch_wont_create_by_default.<locals>.<lambda>)r ra r r� ru �failr> r� )r rK r )r r �!test_patch_wont_create_by_defaultx s
z+PatchTest.test_patch_wont_create_by_defaultc sZ y&t �tdtj�� fdd��}|� W n tk
r: Y n X � �d� � �ttd�� d S )N�ordc s � � d� d S )Nz,Patching non existent attributes should fail)r� r )r r r rK � s z?PatchTest.test_patchobject_wont_create_by_default.<locals>.testz,Patching non existent attributes should fail) r r" r r r� ru r� rA r� )r rK r )r r �'test_patchobject_wont_create_by_default� s
z1PatchTest.test_patchobject_wont_create_by_defaultc C sH t td �dd� �}t td �dd� �}| �|� d� | �|� d� d S ) Nz.ordc S s d| _ td�S )N�e �c)r` r� )Zmock_ordr r r �test_ord� s z>PatchTest.test_patch_builtins_without_create.<locals>.test_ordz.openc S s* | j }d|j_ td�}|�� }|�� |S )N�abcdzdoesnotexists.txt)r` �readr^ �close)Z mock_open�mZfobj�datar r r � test_open� s z?PatchTest.test_patch_builtins_without_create.<locals>.test_openr� r� )r r rI )r r� r� r r r �"test_patch_builtins_without_create� s
z,PatchTest.test_patch_builtins_without_createc sP G dd� dt �� t� � dtdd� ��� �fdd��}|� ��� �� tj� d S )Nc @ s e Zd Zedd� �ZdS )z5PatchTest.test_patch_with_static_methods.<locals>.Fooc S s t jS )N)r �Staticr r r r �woot� s z:PatchTest.test_patch_with_static_methods.<locals>.Foo.wootN)r r r r2 r� r r r r r% � s r% r� c S s t jS )N)r rJ r r r r rr � rs z:PatchTest.test_patch_with_static_methods.<locals>.<lambda>c s �� � �� tj� d S )N)rI r� r rJ r )r% r r r � anonymous� s z;PatchTest.test_patch_with_static_methods.<locals>.anonymous)r" r r2 rI r� r r� )r r� r )r% r r � test_patch_with_static_methods� s $z(PatchTest.test_patch_with_static_methodsc s8 t j}t�t dd�� fdd��}|� � �t j|� d S )Nr% c s � � tjd� d S )Nr% )rI r r% r )r r r r� � s z-PatchTest.test_patch_local.<locals>.anonymous)r r% r r" rI )r r1 r� r )r r �test_patch_local� s zPatchTest.test_patch_localc sT G dd� dt �}|� � tj� _t� � dd�� �fdd��}|� ��� jtj� d S )Nc @ s e Zd ZdZdS )z'PatchTest.test_patch_slots.<locals>.Foo)r% N)r r r � __slots__r r r r r% � s r% c s �� � jd� d S )Nr% )rI r% r )r1 r r r r� � s z-PatchTest.test_patch_slots.<locals>.anonymous)r" r r% r rI )r r% r� r )r1 r r �test_patch_slots� s zPatchTest.test_patch_slotsc sf G dd� dt �� G � �fdd�dt �}t� � dtj�|�}|� }|�� |�� ��� jtjd� d S )Nc @ s e Zd ZejZdS )z=PatchTest.test_patchobject_class_decorator.<locals>.SomethingN)r r r r rE rF r r r r rG � s rG c s( e Zd Z� �fdd�Z� �fdd�ZdS )z7PatchTest.test_patchobject_class_decorator.<locals>.Fooc s �� � jtjd� d S )NrH )rI rF r rJ )�
other_self)rG r r r � test_method� s zCPatchTest.test_patchobject_class_decorator.<locals>.Foo.test_methodc s �� � jtjd� d S )Nznon-test method patched)rI rF r rE )r� )rG r r r �not_test_method� s zGPatchTest.test_patchobject_class_decorator.<locals>.Foo.not_test_methodN)r r r r� r� r )rG r r r r% � s r% rF zpatch not restored) r" r r rJ r� r� rI rF rE )r r% r( r )rG r r � test_patchobject_class_decorator� s z*PatchTest.test_patchobject_class_decoratorc sr G dd� dt �}G � fdd�dt �}tdt �|�}|� }|�� |�� � �|jtjd� � �t j
tj d� d S )Nc @ s e Zd ZejZdS )z7PatchTest.test_patch_class_decorator.<locals>.SomethingN)r r r r rE rF r r r r rG � s rG c s$ e Zd Z� fdd�Z� fdd�ZdS )z1PatchTest.test_patch_class_decorator.<locals>.Fooc s � � tj|d� d S )NrH )rI rY rU )r� Zmock_something)r r r r� � s
z=PatchTest.test_patch_class_decorator.<locals>.Foo.test_methodc s � � tjtjd� d S )Nznon-test method patched)rI rY rU r rG )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� rI rF r rE rY rU rG )r rG r% r( r )r r �test_patch_class_decorator� s z$PatchTest.test_patch_class_decoratorc sZ G dd� dt �� t� � dtj�t� � dtj�� �fdd���}|� ��� jtjd� d S )Nc @ s e Zd ZejZejZdS )z3PatchTest.test_patchobject_twice.<locals>.SomethingN)r r r r rE rF rQ rR r r r r rG � s rG rF c s �� � jtjd� d S )NrH )rI rF r rJ r )rG r r r rK s z.PatchTest.test_patchobject_twice.<locals>.testzpatch not restored)r" r r rJ rI rF rE )r rK r )rG r r �test_patchobject_twice� s z PatchTest.test_patchobject_twicec s� t � dd�� � �� }t�� �� fdd��}|� ��� |� t�� ddi�� �fdd��}|� ��� |� t�� dg�� �fd d��}|� ��� |� d S )
NrU )�initial�otherc s d� d< � d= d� d< d S )N� r& r� zsomething elser� r r )r1 r r rK s z'PatchTest.test_patch_dict.<locals>.testr& r4 c s$ �� t� �d� �� � d d� d S )Nr� r& r4 )rI �lenr )r1 r r r rK s )r& r4 c s$ �� t� �d� �� � d d� d S )Nr� r& r4 )rI r� r )r1 r r r rK $ s )r" �copyr �dictrI )r rW rK r )r1 r r �test_patch_dict
s zPatchTest.test_patch_dictc s� t � � t� � d< d� d< � j�� }t�� �� fdd��}|� ��� j|� t�� ddi�� �fdd��}|� ��� j|� d S ) Nr� rU r� c s d� d< � d= d� d< d S )Nr� r& r� zsomething elser� r r )r1 r r rK 5 s z=PatchTest.test_patch_dict_with_container_object.<locals>.testr& r4 c s& �� t� j�d� �� � d d� d S )Nr� r& r4 )rI r� r7 r )r1 r r r rK ? s )r6 r" r7 r� r r� rI )r rW rK r )r1 r r �%test_patch_dict_with_container_object. s
z/PatchTest.test_patch_dict_with_container_objectc s� t � dd�� � �� }tj� dd�� �fdd��}|� ��� |� tj� ddidd�� �fd d��}|� ��� |� tj� d
gdd�� �fd d��}|� ��� |� d S ) NrU )r� r� T)�clearc s �� � i � d� d< d� d< d S )Nr� r& zsomething elser� )rI r )r1 r r r rK M s z2PatchTest.test_patch_dict_with_clear.<locals>.testr& r4 c s �� � ddi� d S )Nr& r4 )rI r )r1 r r r rK W s )r& r4 c s �� � ddi� d S )Nr& r4 )rI r )r1 r r r rK _ s )r" r� r r� rI )r rW rK r )r1 r r �test_patch_dict_with_clearI s " z$PatchTest.test_patch_dict_with_clearc s� t � � t� � d< d� d< � j�� }tj� dd�� �fdd��}|� ��� j|� tj� dd idd�� �fd
d��}|� ��� j|� d S ) Nr� rU r� T)r� c s" �� � ji � d� d< d� d< d S )Nr� r& zsomething elser� )rI r7 r )r1 r r r rK o s zGPatchTest.test_patch_dict_with_container_object_and_clear.<locals>.testr& r4 c s �� � jddi� d S )Nr& r4 )rI r7 r )r1 r r r rK y s )r6 r" r7 r� r r� rI )r rW rK r )r1 r r �/test_patch_dict_with_container_object_and_clearh s
"z9PatchTest.test_patch_dict_with_container_object_and_clearc C sZ i }t dt t� �t dt t� dd�t �tt� �t �|�dd� ����}| �|jd� d S )Nz %s.SomeClassT)�autospecc S s d S )Nr r r r r � some_name� s z0PatchTest.test_name_preserved.<locals>.some_namer� )r r r" r r� rI )r r1 r� r r r �test_name_preserved� s z PatchTest.test_name_preservedc C sV i }t �|ddi�dd� �}y
|� W n tk
r: Y n X | �d� | �|i � d S )Nr& r4 c S s t d��d S )NZKonrad)r� r r r r rK � s z1PatchTest.test_patch_with_exception.<locals>.testz NameError not raised by test)r r� r� r� rI )r r1 rK r r r �test_patch_with_exception� s
z#PatchTest.test_patch_with_exceptionc s& t �dddi�� fdd��}|� d S )Nz
os.environ�
konrad_delongz
some valuec s � � dtj� d S )Nr� )�assertIn�os�environr )r r r rK � s z3PatchTest.test_patch_dict_with_string.<locals>.test)r r� )r rK r )r r �test_patch_dict_with_string� s z%PatchTest.test_patch_dict_with_stringc sZ t j�� }t�dddi�� fdd��}z&ddit _|� � �t jddi� W d |t _X d S )Nz%unittest.test.testmock.support.targetr* �BARc s � � tjddd�� d S )N�BAZr� )r1 r* )rI r �targetr )r r r rK � s z<PatchTest.test_patch_dict_decorator_resolution.<locals>.testr1 r� )r r� r� r r� rI )r rW rK r )r r �$test_patch_dict_decorator_resolution� s
z.PatchTest.test_patch_dict_decorator_resolutionc s d S )Nc @ s e Zd ZdZdS )z0PatchTest.test_patch_descriptor.<locals>.NothingN)r r r r1 r r r r �Nothing� s r� c s| e Zd Zi Ze�� dd�e��fdd���Ze�� dd�edd� ��Z e�
e�e��fdd���Z e�
e�ed d
� ��Z d S ) z2PatchTest.test_patch_descriptor.<locals>.Somethingr1 � c s �� | � � d S )N)�assertIs)r. )rG 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 �� | � � d S )N)r� )r. )rG 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 r1 r r" r3 r� r2 r� r� r� r� r )r� rG r r r rG � s rG Zf00)r" rI r� r� r� r� )r rU r )r� rG r r �test_patch_descriptor� s z PatchTest.test_patch_descriptorc C s� t dt tdd�dd� �}| �t|� t jtdtdd�dd� �}| �t|� t dt dd�d d� �}| �t|� t jtddd�d
d� �}| �t|� d S ) Nz %s.SomeClassT)rp �spec_setc S s
d| _ d S )Nr1 )�z)rC r r r rK � s z+PatchTest.test_patch_spec_set.<locals>.testr c S s
d| _ d S )Nr1 )r� )rC r r r rK � s )r� c S s
d| _ d S )Nr1 )r� )rC r r r rK � s c S s
d| _ d S )Nr1 )r� )rC r r r rK � s )r r r r> ru r" r )r rK r r r �test_patch_spec_set� s z PatchTest.test_patch_spec_setc C s( t dt dd�dd� �}| �t|� d S )Nz %s.SomeClassT)r� c S s | � }d|_ d S )Nr1 )r� )rC r r r r rK � s z-PatchTest.test_spec_set_inherit.<locals>.test)r r r> ru )r rK r r r �test_spec_set_inherit� s z PatchTest.test_spec_set_inheritc C s\ t }tdt �}| �t |� |�� }z | �||� | �t |� W d |�� X | �t |� d S )Nz %s.something)rU r r r� �start� assertIsNot�stop)r rW �patcherrc r r r �test_patch_start_stop s
z PatchTest.test_patch_start_stopc C s t tdd�}| �t|j� d S )Nr* r� )r �foo_namer> � RuntimeErrorr� )r r� r r r �test_stop_without_start s z!PatchTest.test_stop_without_startc C s^ t }t�tdd�}| �t |� |�� }z | �|d� | �t |� W d |�� X | �t |� d S )NrU r1 )rU r r" rY r� r� rI r� )r rW r� Zreplacedr r r �test_patchobject_start_stop s
z%PatchTest.test_patchobject_start_stopc C sf ddi}|� � }tj|dgdd�}| �||� |�� z| �|ddi� W d |�� X | �||� d S )Nr1 r* )�spam�eggsT)r� r� r� )r� r r� rI r� r� )r �drW r� r r r �test_patch_dict_start_stop& s
z$PatchTest.test_patch_dict_start_stopc s� | �ddi� � � � }G � �fdd�dt�}tj� ddidd�|�}| �� |� |� }|�� | �� |� |�� | �� |� |� }|�� | �� |� |�� | �� |� d S ) Nr� r� c s( e Zd Z� �fdd�Z� �fdd�ZdS )z7PatchTest.test_patch_dict_class_decorator.<locals>.Testc s �� � ddi� d S )Nr1 r* )rI )r )r� �thisr r �
test_first: s zBPatchTest.test_patch_dict_class_decorator.<locals>.Test.test_firstc s �� � ddi� d S )Nr1 r* )rI )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 ri 9 s ri r1 r* T)r� )r� r" r r� rI r� r� )r rW ri rK r )r� r� r � test_patch_dict_class_decorator4 s z)PatchTest.test_patch_dict_class_decoratorc s� G dd� dt �}G dd� d�}xj|||� |fD ]X}t|�� t� � dd�� �fdd��}|� ��� jd� ��|jd� ��d� j� q.W d S ) Nc @ s e Zd ZdZdS )z0PatchTest.test_get_only_proxy.<locals>.Somethingr1 N)r r r r1 r r r r rG T s rG c @ s e Zd ZdZdS )z4PatchTest.test_get_only_proxy.<locals>.SomethingElser1 N)r r r r1 r r r r r[ V s r[ r1 r* c s �� � jd� d S )Nr* )rI r1 r )�proxyr r r rK \ s z+PatchTest.test_get_only_proxy.<locals>.test)r" r$ r rI r1 � assertNotIn�__dict__)r rG r[ �thingrK r )r� r r �test_get_only_proxyS s z PatchTest.test_get_only_proxyc s� G dd� dt �}G dd� d�}xn|||� |fD ]\}t|dd�� t� � dd�� �fd d
��}|� ��� jd� ��|jd� ��d� j� q.W d S ) Nc @ s e Zd ZdZdS )z6PatchTest.test_get_set_delete_proxy.<locals>.Somethingr1 N)r r r r1 r r r r rG f s rG c @ s e Zd ZdZdS )z:PatchTest.test_get_set_delete_proxy.<locals>.SomethingElser1 N)r r r r1 r r r r r[ h s r[ F)r# r1 r* c s �� � jd� d S )Nr* )rI r1 r )r� r r r rK n s z1PatchTest.test_get_set_delete_proxy.<locals>.test)r" r$ r rI r1 r� r� )r rG r[ r� rK r )r� r r �test_get_set_delete_proxye s z#PatchTest.test_get_set_delete_proxyc C sZ t dt� d�}ttf|�}|�� }|�� | �t |� | �|j� � d� | �
|jt� d S )N�! )� side_effectzfoo.bar.return_valuer1 ) �KeyErrorr r r� r� r� r> rI r1 r* rj )r �kwargsr� rc r r r �test_patch_keyword_argsw s
z!PatchTest.test_patch_keyword_argsc C s^ t dt� d�}tjtdf|�}|�� }|�� | �t |� | �|j �
� d� | � |j t� d S )Nr� )r� zfoo.bar.return_valuer1 r( ) r� r r r" r% r� r� r> rI r1 r* rj )r r� r� rc r r r � test_patch_object_keyword_args� s
z(PatchTest.test_patch_object_keyword_argsc C s` ddi}|� � }tj|dddd�}|�� z| �|tdddd�� W d |�� X | �||� d S )Nr1 r* r� � � )r1 r* Zbaz)r� r r� r� rI r� )r rW r� r� r r r � test_patch_dict_keyword_args� s
z&PatchTest.test_patch_dict_keyword_argsc s� G dd� dt �}�fdd�� �fdd��� ��fdd�}ttd d
�|�}|� }��t|� |� tjt }tj |d d d
�|�}|� }��t|� |� d S ) Nc @ s8 e Zd Zdd� Zdd� Zdd� ZdZG dd � d e�Zd
S ) z$PatchTest.test_autospec.<locals>.Booc S s d S )Nr )r r& r r r r' � s z-PatchTest.test_autospec.<locals>.Boo.__init__c S s d S )Nr )r r& r r r r( � s z&PatchTest.test_autospec.<locals>.Boo.fc S s d S )Nr )r r r r r) � s z&PatchTest.test_autospec.<locals>.Boo.gr* c @ s e Zd Zdd� ZdS )z(PatchTest.test_autospec.<locals>.Boo.Barc S s d S )Nr )r r r r r& � s z*PatchTest.test_autospec.<locals>.Boo.Bar.aN)r r r r& r r r r r0 � s r0 N) r r r r' r( r) r1 r" r0 r r r r �Boo� s
r� c s" | d� | � d� � �t| � d S )N� )�assert_called_withr> r? )rc )r r r �_test� s
z&PatchTest.test_autospec.<locals>._testc s | � d� | j �d� � �t| j � | �� | j�� � �t| jd� � �tt| d� | j�� | jj�� � �tt| jd� | � � | j �� | j �
� | j j
�� � �t| j j
d� | � � �
� | � � j
�� � �t| � � j
d� � �tt| j d� � �tt| � � d� d S )Nr� �hr* r4 ) r( r� r> r? r) ru r r1 �lowerr0 r& )rc )r r r �_test2� s(
z'PatchTest.test_autospec.<locals>._test2c s, � | � �| � �| d�� �� | t� | S )Nr� )r� r% )rc )r� r� r r r r5 � s
z)PatchTest.test_autospec.<locals>.functionT)r� r% )r" r r� r� r% �sys�modulesr )r r� r5 rK rc �moduler )r� r� r r �
test_autospec� s
zPatchTest.test_autospecc s& t dt dd�� fdd��}|� d S )Nz %s.functionT)r� c s� t �� � �tt j� � �tt j� t d� � �tt j� t �d� t �� t �� t dd� t �dd� � �tt � � �tt t d� d S )Nr� r� r� r1 )
r5 Zassert_not_calledr> �AssertionErrorZ
assert_calledZassert_called_oncer� r? ru r )rc )r r r rK � s
z.PatchTest.test_autospec_function.<locals>.test)r r )r rK r )r r �test_autospec_function� s z PatchTest.test_autospec_functionc C s0 t dt ddd�dd� �}|� }| �|d� d S )Nz %s.functionTr� )r� r` c S s
t dd�S )Nr� r� )r5 )Z
mock_functionr r r rK � s z.PatchTest.test_autospec_keywords.<locals>.test)r r rI )r rK �resultr r r �test_autospec_keywords� s
z PatchTest.test_autospec_keywordsc C s0 t dt dd��}t�� |�� W d Q R X d S )Nz%s.Foo.static_methodT)r� )r r r% r, �assert_called_once_with)r �methodr r r �test_autospec_staticmethod s z$PatchTest.test_autospec_staticmethodc C s0 t dt dd��}t�� |�� W d Q R X d S )Nz%s.Foo.class_methodT)r� )r r r% r/ r� )r r� r r r �test_autospec_classmethod s z#PatchTest.test_autospec_classmethodc C sN t dt ddd�}| �t|j� tjt }t j|dddd�}| �t|j� d S )Nz %s.functionr� T)�newr� r5 )r r r> r? r� r� r� r" )r r� r� r r r �test_autospec_with_new s
z PatchTest.test_autospec_with_newc C sR G dd� dt �}tt|d�}|�� }z | �||� | �|jt� W d |�� X d S )Nc @ s e Zd Zg ZdS )z0PatchTest.test_autospec_with_object.<locals>.BarN)r r r �extrar r r r r0 s r0 )r� )r% r r� r� rj r� �listr� )r r0 r� rc r r r �test_autospec_with_object s z#PatchTest.test_autospec_with_objectc C sH t }ttdd�}|�� }z | �||� | �|d�|� W d |�� X d S )NT)r� r� )r% r r� r� rj r� )r ZFooClassr� rc r r r �test_autospec_inherits( s z PatchTest.test_autospec_inheritsc C st t tdd�}|�� }zP| �dt|�� | �dt|j�� | �dt|d ��� | �dt|d �j�� W d |�� X d S )NT)r� z name='Foo'z
name='Foo.f'z
name='Foo()'z name='Foo().f')r r� r� r� �reprr( r� )r r� rc r r r �test_autospec_name3 s z PatchTest.test_autospec_namec C sb t �tdt� �dd� �}y
|� W n t�� }Y nX t�d d d�}|�|| �}| �d|� d S )Nr( c S s t �d S )N)r� r r r r rK A s z'PatchTest.test_tracebacks.<locals>.testr zraise AssertionError) r r" r% r� �exc_info�unittestZTextTestResultZ_exc_info_to_stringr� )r rK �errr� � tracebackr r r �test_tracebacks@ s
zPatchTest.test_tracebacksc C sX t ttd�}|�� }|�� |�� }|�� | �||� x||fD ]}| �|� qBW d S )N)� new_callable)r r� r r� r� r� rD )r r� �m1�m2rc r r r �test_new_callable_patchN s z!PatchTest.test_new_callable_patchc C s\ t jtdtd�}|�� }|�� |�� }|�� | �||� x||fD ]}| �|� qFW d S )Nr( )r )r r" r% r r� r� r� rD )r r� r r
rc r r r � test_new_callable_patch_object[ s z(PatchTest.test_new_callable_patch_objectc sf G � fdd�dt �� tt� ddd�}|�� }z*| �t|�� � | �� jtddd�� W d |� � X d S )Nc s e Zd ZdZ� fdd�ZdS )z:PatchTest.test_new_callable_keyword_arguments.<locals>.BarNc s
|� _ d S )N)r� )r r� )r0 r r r' k s zCPatchTest.test_new_callable_keyword_arguments.<locals>.Bar.__init__)r r r r� r' r )r0 r r r0 i s r0 r� r� )r �arg1�arg2)r
r )
r" r r� r� r� �typerI r� r� r� )r r� r� r )r0 r �#test_new_callable_keyword_argumentsh s z-PatchTest.test_new_callable_keyword_argumentsc s� G � fdd�dt �� tt� � d�}|�� z| �� jt� d�� W d |�� X tt� � d�}|�� z| �� jt� d�� W d |�� X d S )Nc s e Zd ZdZ� fdd�ZdS )z-PatchTest.test_new_callable_spec.<locals>.BarNc s
|� _ d S )N)r� )r r� )r0 r r r' z s z6PatchTest.test_new_callable_spec.<locals>.Bar.__init__)r r r r� r' r )r0 r r r0 x s r0 )r rp )rp )r r� )r� )r" r r� r� rI r� r� r� )r r� r )r0 r �test_new_callable_specw s
z PatchTest.test_new_callable_specc C sZ dt }t|td�}| �t|j� t|tdd�}|�� }z| j|dd� W d |�� X d S )Nz %s.weeeee)r T)r r� F)rB )r� r r r> ru r� rD r� )r Znon_existent_attr�pr� r r r �test_new_callable_create� s z"PatchTest.test_new_callable_createc C s4 | j tttt� td� | j ttjtdt� td� d S )N)r� r r( )r> �
ValueErrorr r� r" r r% )r r r r �'test_new_callable_incompatible_with_new� s
z1PatchTest.test_new_callable_incompatible_with_newc C s0 | j ttttdd� | j ttjtdtdd� d S )NT)r r� r( )r> 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� G dd� dt �}tttt |f}xd|D ]\}xVdD ]N}|di}ttfd|i|��}|�� }z|j}| �t t
|d� W d |� � X q.W q$W d S )Nc @ s e Zd ZdS )z>PatchTest.test_new_callable_inherit_for_mocks.<locals>.MockSubN)r r r r r r r �MockSub� s r )rp r� Tr �x) r
r r r r r� r� r` r> ru r r� ) r r Z MockClassesZKlassr� r� r r� r r r r �#test_new_callable_inherit_for_mocks� s
z-PatchTest.test_new_callable_inherit_for_mocksc C sh G dd� dt �}tt|dd�}|�� }z$| �t||�� | �tt|d� W d |� � X | �
|j t � d S )Nc @ s e Zd Zdd� ZdS )z>PatchTest.test_new_callable_inherit_non_mock.<locals>.NotAMockc S s
|| _ d S )N)rp )r rp 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 T)r rp r` )
r" r r� r� r@ r r> ru r r� rI rp r% )r r r r� r r r �"test_new_callable_inherit_non_mock� s
z,PatchTest.test_new_callable_inherit_non_mockc sR | �t � G � �fdd�dt�}tttd�|�}|� �� |� �� | �t � � d S )Nc s* e Zd Z� �fdd�Zdd� Zdd� ZdS )z>PatchTest.test_new_callable_class_decorating.<locals>.SomeTestc s( �� t� � ��t|� ��tt� d S )N)r� r% r� rj r )r �mock_foo)rW rK r r r� � s zDPatchTest.test_new_callable_class_decorating.<locals>.SomeTest._testc S s | � |� d S )N)r� )r r r r r �test_two� s zGPatchTest.test_new_callable_class_decorating.<locals>.SomeTest.test_twoc S s | � |� 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 )rW rK r r �SomeTest� s r )r )r% r" r r� r r r r� )r r r )rW rK r �"test_new_callable_class_decorating� s
z,PatchTest.test_new_callable_class_decoratingc s� t � t j}t j}tjtddd�}tjt ddd�}xx||fD ]l}|�� z,��t � � ��t jd� ��t jd� W d |� � X ��t � � ��t j|� ��t j|� q:W tjtddd�� �fdd��}|� d S )Nr� r� )r( r) r� r� c s, �� t� � ��tjd� ��tjd� d S )Nr� r� )r� r% rI r( r) r )� original_foor r r rK � s z+PatchTest.test_patch_multiple.<locals>.test)
r% r( r) r �multipler� r� r� rI r� )r �
original_f�
original_gZpatcher1Zpatcher2r� rK r )r! r r �test_patch_multiple� s"
z PatchTest.test_patch_multiplec C s$ | � ttjt� | � ttjt� d S )N)r> r r r" r� r% )r r r r � test_patch_multiple_no_kwargs s z'PatchTest.test_patch_multiple_no_kwargsc sV t � t j}t j}tjttdtd�� �fdd��}|� ��t j|� ��t j|� d S )Nr� )r( r) r1 c sZ �� t� � �� tj| � ��tjd� �� tj|� ��t| t�� ��t|t�� d S )Nr� ) r� r% r( rI r) r1 r@ r r )r( r1 )r! r r r rK s z8PatchTest.test_patch_multiple_create_mocks.<locals>.test)r% r( r) r r" r� r rI )r r# r$ rK r )r! r r � test_patch_multiple_create_mocks s z*PatchTest.test_patch_multiple_create_mocksc sr t j}t j}t�t dd�}d|_t�t dt�}d|_|g|_|� fdd��}|� � �t j|� � �t j|� d S )Nr( r� r) c s � � tj| � � �tjd� d S )Nr� )r� r% r) rI r( )r) )r r r rK * s zHPatchTest.test_patch_multiple_create_mocks_different_order.<locals>.test) r% r( r) r r" �attribute_namer �additional_patchersrI )r r# r$ r� r� rK r )r r �0test_patch_multiple_create_mocks_different_order s z:PatchTest.test_patch_multiple_create_mocks_different_orderc s� t �t j}t j}tjttd�tjttd�ttd �� fdd����}tjttd�ttd �tjttd�� fdd����}ttd �tjttd�tjttd�� fdd ����}��fd
d �� |� |� |� ��t j|� ��t j|� d S ) N)r( )r1 z.gc s � | f|� d S )Nr )r) r� )r� r r �test19 s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test1c s � | f|� d S )Nr )r) r� )r� r r �test2? s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test2c s � | f|� d S )Nr )r) r� )r� r r �test3E s z?PatchTest.test_patch_multiple_stacked_decorators.<locals>.test3c s� |� d�}|� d�}��|� ��t� � ��tj|� ��tj| � ��tj|� ��t|t �� ��t| t �� ��t|t �� d S )Nr( r1 )
�poprA r� r% r( r) r1 r@ r r )r) r� r( r1 )r! r r r r� K s
z?PatchTest.test_patch_multiple_stacked_decorators.<locals>._test)r% r( r) r r" r� r rI )r r# r$ r+ r, r- r )r� r! r r �&test_patch_multiple_stacked_decorators4 s$
z0PatchTest.test_patch_multiple_stacked_decoratorsc C s� t }t j}t j}tjttdtd�}|�� }zt|d }|d }| �t |�t ddg�� | �
t |� | �
t j|� | �
t j |� | � t
|t�� | � t
|t�� W d |�� X | �t j|� | �t j|� d S )Nr� )r( r) r1 r( r1 )r% r( r) r r" r� r r� rI �setr� r1 r@ r r r� )r r! r# r$ r� r� r( r1 r r r �(test_patch_multiple_create_mocks_patcher_ s"
z2PatchTest.test_patch_multiple_create_mocks_patcherc sv | �t � t j}t j}G � �fdd�dt�}tjttdtd�|�}|� }|�� |� � | �
t j|� | �
t j|� d S )Nc s* e Zd Z� �fdd�Zdd� Zdd� ZdS )z@PatchTest.test_patch_multiple_decorating_class.<locals>.SomeTestc sZ �� t� � �� tj|� ��tjd� �� tj|� ��t|t�� ��t|t�� d S )Nr� ) r� r% r( rI r) r1 r@ r r )r r( r1 )r! rK r r r� � s zFPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest._testc S s | � ||� d S )N)r� )r r( r1 r r r r � s zIPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest.test_twoc S s | � ||� d S )N)r� )r r( r1 r r r r � s zIPatchTest.test_patch_multiple_decorating_class.<locals>.SomeTest.test_oneN)r r r r� r r r )r! rK r r r ~ s r r� )r( r) r1 ) r% r( r) r" r r" r� r r r rI )r r# r$ r r� r )r! rK r �$test_patch_multiple_decorating_classx s z.PatchTest.test_patch_multiple_decorating_classc C sf t jtdd�}| �t|j� t jtddd�}|�� z| �tjd� W d |�� X | � t
td�� d S )N�blam)r3 T)r3 r� ) r r" r% r> ru r� rI r3 r� rA r� )r r� r r r �test_patch_multiple_create� s
z$PatchTest.test_patch_multiple_createc C s� t jttddgd�}|�� }z^| �tj|d � tj�d� tj�d� tjj� d� tjj� d� | �
t t tjdd � W d |�
� X d S )Nr& r4 )r1 r� r1 r� r� r� )r r" r% r r� rI r1 r& r4 r� r> ru r r� )r r� r� r r r � test_patch_multiple_spec_set� s z&PatchTest.test_patch_multiple_spec_setc C s� G dd� dt �}tjttt|d�}|�� }zT| �tj|d � | �tj|d � | � tj|� | � tj|� | �
tjtj� W d |� � X d S )Nc @ s e Zd ZdS )z9PatchTest.test_patch_multiple_new_callable.<locals>.ThingN)r r r r r r r �Thing� s r6 )r( r) r r( r) ) r" r r" r% r r� r� r( r) rj r� r� )r r6 r� r� r r r � test_patch_multiple_new_callable� s z*PatchTest.test_patch_multiple_new_callablec C s� t j}t j}t�t dd�t�t dd�t�t dd�dd� ���}t�t dd�t�t dd�t�t dd�dd� ���}t�t dd�t�t dd�t�t dd�d d
� ���}x:|||fD ],}| �t|� | �t j|� | �t j|� q�W d 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" r> ru rI )r r# r$ r9 r: r; �funcr r r �test_nested_patch_failure� s z#PatchTest.test_nested_patch_failurec C s t j}t j}t j}dd� }t�t dd�tjt d|d�t�t dd�dd � ���}tjt d|d�t�t dd�t�t dd�d
d � ���}t�t dd�t�t dd�tjt d|d�d d
� ���}xH|||fD ]:}| �t|� | �t j|� | �t j|� | �t j|� q�W d S )Nc S s t d��d S )N�crasher)r� r r r r r> � s z4PatchTest.test_new_callable_failure.<locals>.crasherr) r� r1 )r r( c S s d S )Nr r r r r r9 � 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) r1 r r" r> r� rI ) r r# r$ r! r> r9 r: r; r< r r r �test_new_callable_failure� s$ z#PatchTest.test_new_callable_failurec C s� t j}t j}t�t dd�}d|_t�t dd�}d|_t�t dd�}d|_xR||g||gfD ]>}||_|dd� �}| �t|� | � t j|� | � t j|� qZW d S )Nr( r� r) r8 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) r> ru rI )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}dd� }t�t dd�}d|_t�t dd�}d|_tjt d|d�}d|_x`||g||gfD ]L}||_|dd � �} | �t | � | �
t j|� | �
t j|� | �
t j|� qjW d S )
Nc S s t d��d S )Nr> )r� r r r r r> $ s zCPatchTest.test_patch_multiple_new_callable_failure.<locals>.crasherr( r� r) r1 )r c S s d S )Nr r r r r r< 3 s z@PatchTest.test_patch_multiple_new_callable_failure.<locals>.func) r% r( r) r1 r r" r( r) r> r� rI )
r r# r$ r! r> r� r@ rA rB r< r r r �(test_patch_multiple_new_callable_failure s" z2PatchTest.test_patch_multiple_new_callable_failurec sL t dtfddi�}|� � tj� dd�� �fdd��}|� ��� jd� d S )Nr% �fishZtastyz nearly gone)rE c s �� � jd� d S )Nz nearly gone)rI rE r )r1 r r r rK @ s z=PatchTest.test_patch_multiple_string_subclasses.<locals>.test)r �strr r" rI rE )r r% rK r )r1 r r �%test_patch_multiple_string_subclasses= s
z/PatchTest.test_patch_multiple_string_subclassesz unittest.mock.patch.TEST_PREFIXr1 c C sl G dd� dt �}t� |dd�|�}|� }| �|�� d� | �|�� d� | �|�� d� | �|�� d� d S )Nc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) z-PatchTest.test_patch_test_prefix.<locals>.FoorW c S s | j S )N)r� )r r r r �foo_oneM s z5PatchTest.test_patch_test_prefix.<locals>.Foo.foo_onec S s | j S )N)r� )r r r r �foo_twoO s z5PatchTest.test_patch_test_prefix.<locals>.Foo.foo_twoc S s | j S )N)r� )r r r r r Q s z6PatchTest.test_patch_test_prefix.<locals>.Foo.test_onec S s | j S )N)r� )r r r r r S s z6PatchTest.test_patch_test_prefix.<locals>.Foo.test_twoN)r r r r� rH rI r r r r r r r% J s
r% r� �changedrW )r" r rI rH rI r r )r r% r1 r r r �test_patch_test_prefixH s z PatchTest.test_patch_test_prefixr* c s� G � fdd�dt �}ddi� tj� dd�|�}|� }| �|�� ddi� | �|�� ddi� | �|�� ddi� | �|�� ddi� d S )Nc s< e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Zd S )
z2PatchTest.test_patch_dict_test_prefix.<locals>.Fooc s t � �S )N)r� )r )�the_dictr r �bar_oneb s z:PatchTest.test_patch_dict_test_prefix.<locals>.Foo.bar_onec s t � �S )N)r� )r )rL r r �bar_twod s z:PatchTest.test_patch_dict_test_prefix.<locals>.Foo.bar_twoc s t � �S )N)r� )r )rL r r r f s z;PatchTest.test_patch_dict_test_prefix.<locals>.Foo.test_onec s t � �S )N)r� )r )rL r r r h s z;PatchTest.test_patch_dict_test_prefix.<locals>.Foo.test_twoN)r r r rM rN r r r )rL r r r% a s r% �keyrW rJ )rO )r" r r� rI rM rN r r )r r% r1 r )rL r �test_patch_dict_test_prefix_ s
z%PatchTest.test_patch_dict_test_prefixc C s� x�dD ]x}t dt f|di�}|�� }zL| �dt|�� | �dt|j�� | �dt|� �� | �dt|� j�� W d |�� X qW d S )N)rp r� r� z %s.SomeClassTz name='SomeClass'z! name='SomeClass.class_attribute'z name='SomeClass()'z# name='SomeClass().class_attribute')r r r� r� r re r� )r r� r r� r r r � test_patch_with_spec_mock_repru s
z(PatchTest.test_patch_with_spec_mock_reprc C sJ t ddd��4}| �dt|j�� �� | �dt|�� �� �� W d Q R X d S )Nz unittest.test.testmock.supportT)r� z" name='support.SomeClass.wibble()'z$ name='support.SomeClass().wibble()')r r� r r rt )r r� r r r � test_patch_nested_autospec_repr� s
z)PatchTest.test_patch_nested_autospec_reprc C s x�dD ]�}t dt f|di�}|�� }z�|�� t�� g}| �|j|� | �|j|� | �|jjt� g� |� }|�t� � | �|j|� |�� |�t� �� � | �|j|� | �|jt�� g� | �|jjt� g� | �|jt�� g� W d |� � X qW d S )N)rp r� r� z %s.SomeClassT)
r r r� rt r rI Z
mock_callsZ method_calls�appendr� )r r� r r� Zkallsr� r r r �test_mock_calls_with_patch� s&
z$PatchTest.test_mock_calls_with_patchc C sp t d�}| �t|j� td��: t� }|tjd< d|_t d�}d|_|�� |� � W d Q R X | �
|jd� d S )Nz
squizz.squozz�squizz� r� ) r r> � ImportErrorr� r r
r� r� Zsquozzr� rI )r Zp1rU r r r �test_patch_imports_lazily� s
z#PatchTest.test_patch_imports_lazilyc s� G dd� d��G �fdd�dt �� � fdd�}|d�dd � �}td
��" t� }|tjd
< | �t|� W d Q R X | ��jd t� | � �jd d
� | � �jd d� d S )Nc @ s e Zd ZdZdS )z<PatchTest.test_patch_propogrates_exc_on_exit.<locals>.holder)NNNN)r r r r r r r r �holder� s rY c s e Zd Zd� fdd� ZeZdS )zBPatchTest.test_patch_propogrates_exc_on_exit.<locals>.custom_patchNc s t �| |||� |||f� _d S )N)r
�__exit__r )r �etype�val�tb)rY r r rZ � s zKPatchTest.test_patch_propogrates_exc_on_exit.<locals>.custom_patch.__exit__)NNN)r r r rZ r� r )rY r r � custom_patch� s r^ c
s$ t | �\}}� ||td dd d d i � S )NF)r r )r� �getterrF )r^ 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� )rc r r r rK � s z:PatchTest.test_patch_propogrates_exc_on_exit.<locals>.testrU r r� z exception value not propgatedr� z!exception traceback not propgated)
r
r r
r� r� r> r� r� r ZassertIsNotNone)r r` rK rU r )r^ rY r �"test_patch_propogrates_exc_on_exit� s
z,PatchTest.test_patch_propogrates_exc_on_exitc C st xndD ]f}t dt fddi|di��}| �t|j� | �tdd� � t tfddi|di��}|�� |�� qW d S )N)rp r� r� z%s.doesnotexistr� Tc S s t S )N)� doesnotexistr r r r rr � rs z1PatchTest.test_create_and_specs.<locals>.<lambda>)r r r> r? r� r� �MODNAMEr� )r �kwargr r r r �test_create_and_specs� s
z PatchTest.test_create_and_specsc C s� t }x>dD ]6}ttfddi|di��}| �t|j� | �t |� q
W x>dD ]6}ttfddi|di��}| �t|j� | �t |� qJW x>dD ]6}ttfddi|di��}| �t|j� | �t |� q�W d S )N)rp r� r� r )rp r� r� )r� r� rp )rY r rc r> r? r� r� )r rW rd r r r r �test_multiple_specs� s
z PatchTest.test_multiple_specsc C s8 t tdddd�}|�� }z|j d|_W d |�� X d S )NF)rp r� r� r� )r rc r� Zdoes_not_existr� )r r rc r r r � test_specs_false_instead_of_none� s
z*PatchTest.test_specs_false_instead_of_nonec C sJ xDdD ]<}t tf|di�}|�� }z| �tt|d� W d |�� X qW d S )N)rp r� r� r Z doesnotexit)r rc r� r> ru r r� )r rd r r� r r r �test_falsey_spec s
zPatchTest.test_falsey_specc C sd x^dD ]V}t tfddi|di��}|�� }z&| �tt|dd� | �tt|d� W d |�� X qW d S )N)rp r� r� Trb rU )r rc r� r> ru r r r� )r rd r r� r r r �test_spec_set_true
s
z PatchTest.test_spec_set_truec C s: d}t t|d�}|�� }z| �t|�� W d |�� X d S )N)�__call__)rp )r rc r� r@ �callabler� )r rp r r� r r r �test_callable_spec_as_list s z$PatchTest.test_callable_spec_as_listc C s: d}t t|d�}|�� }z| �t|�� W d |�� X d S )N)r1 r* )rp )r rc r� rA rk r� )r rp 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}tdt��� tdt��� td�� ��fdd��}|� ��t j|� d S )Nz os.unlinkzos.chdirzos.pathc s6 t �� ��tj| � ��tj�� ��tj� � d S )N)r �stopallr� r� �path�unlink�chdir)Z mock_path)rq r rp r r �patched4 s z-PatchTest.test_patch_stopall.<locals>.patched) r� rp rq ro r rU r� rZ r� )r ro rr r )rq r rp r �test_patch_stopall- s z PatchTest.test_patch_stopallc sR g �G dd� dt ����fdd�� � fdd�dD � t�� | ��dd d
g� d S ) Nc @ s e Zd Zd Z ZZdS )z*PatchTest.test_stopall_lifo.<locals>.thingN)r r r �one�two�threer r r r r� @ s r� c
s8 G � ��fdd�dt ����fdd�� d d dd d d i � S )Nc s" e Zd Z� ���fdd�Z� ZS )z?PatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.mypatchc s �� �� t�| ��� S )N)rS �superr� )r )� __class__rF �mypatch�stoppedr r r� E s
zDPatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.mypatch.stop)r r r r� �
__classcell__r )rF ry rz )rx r ry D s ry c s � S )Nr r )r� r r rr H rs z@PatchTest.test_stopall_lifo.<locals>.get_patch.<locals>.<lambda>F)r
)rF )rz r� )rF ry r � get_patchC s z.PatchTest.test_stopall_lifo.<locals>.get_patchc s g | ]}� |�� � �qS r )r� )�.0r\ )r| r r �
<listcomp>J s z/PatchTest.test_stopall_lifo.<locals>.<listcomp>)rt ru rv rv ru rt )r" r rn rI )r r )r| rz r� r �test_stopall_lifo> s zPatchTest.test_stopall_lifoc C s> ddd�}t �|dd�� | �|� d� W d Q R X | �|� d� t �|dd�� | �|jd� W d Q R X | �|jd � t �|d
d �� | �|jd � W d Q R X | �|jd � t �|d
tdg��� | �|jtdg�� W d Q R X | �|jt� � dd�dd�}t �|dtdg��� | �|� d� W d Q R X | �|� d� d S )Nr c S s | S )�TESTr )r r r r r1 Q s z)PatchTest.test_special_attrs.<locals>.foo� __defaults__)r� r� �__doc__ZFUNr� r Z
testpatch2z unittest.test.testmock.testpatch�__annotations__)�sr� )r c W s | S )Nr )r r& r r r r1 d s �__kwdefaults__)r r� )r )r r" rI r� r r� r� )r r1 r r r �test_special_attrsP s"
z PatchTest.test_special_attrsN)T)ir r r rD rL rN rP rT rX rd rg ro rw ry 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/ r1 r2 r4 r5 r7 r= r? rC rD rG r rK rP rQ rR rT rX ra re rf rg rh ri rl rm rs r r� r r r r r= [ s�
+ *
!
+
G
+! ! "
r= �__main__)T)(r� r� r Zunittest.test.testmockr Z unittest.test.testmock.supportr r Ztest.test_importlib.utilr Z
unittest.mockr r r r r
r r r
r r r ra r� r rY rc r$ rG rU r[ rZ r" r% r� r5 r6 ZTestCaser= �mainr r r r �<module> sB 4