File: //opt/python38/lib/python3.8/unittest/test/testmock/__pycache__/testmock.cpython-38.pyc
U
�i�_x � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ G dd� de�ZG dd� de�Zd d
� ZG d d � d ej�Zed
kr�e� � dS )� N)� is_instance��mock) �call�DEFAULT�patch�sentinel� MagicMock�Mock�NonCallableMock�NonCallableMagicMock� AsyncMock�_Call� _CallList�create_autospecc @ s( e Zd Zdd� Zdd� Zdd� ZeZdS )�Iterc C s t ddddg�| _d S )N�this�is�an�iter)r �thing��self� r �>/opt/python38/lib/python3.8/unittest/test/testmock/testmock.py�__init__ s z
Iter.__init__c C s | S �Nr r r r r �__iter__ s z
Iter.__iter__c C s
t | j�S r )�nextr r r r r r s z Iter.nextN)�__name__�
__module__� __qualname__r r r �__next__r r r r r s r c @ s2 e Zd Zddd�Zed dd��Zed
dd��ZdS ) � SomethingNc C s d S r r �r �a�b�c�dr r r �meth � zSomething.methc C s d S r r )�clsr% r& r' r( r r r �cmeth! s zSomething.cmethc C s d S r r �r% r& r' r( r r r �smeth$ s zSomething.smeth)N)N)N)r r r! r) � classmethodr, � staticmethodr. r r r r r# s
r# c C s d S r r �r% r r r � something( r* r2 c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:dqdr� Z;dsdt� Z<dudv� Z=dwdx� Z>dydz� Z?d{d|� Z@d}d~� ZAdd�� 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�� ZVd�d�� ZWd�d�� ZXd�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�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� Zvd�d� Zwd�d� Zxd�d� Zyd�d� Zzd�d� Z{d�d� Z|d�d�� Z}d�d�� Z~d�d�� Zd�d�� Z�d�d�� Z�d��d � Z��dS ( �MockTestc C s t d� d S )Nzfrom unittest.mock import *)�execr r r r �test_all- s zMockTest.test_allc C s� t � }| �|jd� | �|jdd� | �t|jt �d� | �|jd d� | �|j g d� | �|j
g d� | � d|j d � | �
|jd
� | �
|jd � | �|ji d � d S )
Nz called not initialised correctlyr z$call_count not initialised correctlyz&return_value not initialised correctlyz#call_args not initialised correctly�(call_args_list not initialised correctlyz&method_calls not initialised correctlyZ_itemsz/default mock should not have '_items' attributez parent not initialised correctlyz!methods not initialised correctlyz$children not initialised incorrectly)r
� assertFalse�called� assertEqual�
call_count�
assertTruer � return_value� call_args�call_args_list� method_calls� assertNotIn�__dict__� assertIsNone� _mock_parent�
_mock_methods�_mock_children�r r r r r �test_constructor4 s:
��
�
�
�
���
�zMockTest.test_constructorc C s t d d�}| �|jd� d S )N�r< z(return value in constructor not honoured)r
rB r< rF r r r � test_return_value_in_constructorP s
�z)MockTest.test_return_value_in_constructorc C s* dd� }t |�}d|j_| �|� d� d S )Nc S s d S r r r r r r �fW r* z9MockTest.test_change_return_value_via_delegate.<locals>.f� )r r r< r9 �r rJ r r r r �%test_change_return_value_via_delegateV s z.MockTest.test_change_return_value_via_delegatec C s: dd� }t |�}t� |j_| �t�� |� W 5 Q R X d S )Nc S s d S r r r r r r rJ ^ r* z8MockTest.test_change_side_effect_via_delegate.<locals>.f)r � TypeErrorr � side_effect� assertRaisesrL r r r �$test_change_side_effect_via_delegate] s
z-MockTest.test_change_side_effect_via_delegatec C s� t dd�}| �dt|�� | �dt|� t|�� t � dft dd�dfg}|D ]�\}}| �d| t|j�� | �d| t|�� �� | �d| t|�� j�� | �d | t|� �� | �d
| t|� � �� | �d | t|� � jj�� j�� qNd S ) N�foo��namez'%s'r �barz%s.barz%s.foo()z
%s.foo().bingz%s()z%s()()z%s()().foo.bar.baz().bing)r
�assertIn�repr�idrU rR Zbing�baz)r r �mocksrT r r r � test_repre s
�zMockTest.test_reprc C s� G dd� dt �}t|d�}| �dt|�� t|� d�}| �dt|�� t|d�}| �dt|�� t|� d�}| �dt|�� t|dd�}| �dt|�� | �d t|�� tdd
�}| �d t|�� t� }| �d t|�� tdgd�}| �d t|�� d S ) Nc @ s e Zd ZdS )z'MockTest.test_repr_with_spec.<locals>.XN�r r r! r r r r �Xv s r] ��specz
spec='X' ��spec_setz spec_set='X' rR )r_ rT z name='foo' rS r_ )�objectr
rV rW r@ �r r] r r r r �test_repr_with_specu s$
z MockTest.test_repr_with_specc s� t � }dd� }||_| jt|dddd� |jdddd� dddg� � fdd�}||_| �|� |� |� gdddgd� t tjd �}| �|jtjd
� d d � }t |tjd
�}| �|� tj� d S )Nc _ s t d��d S )NZkablooie)� SystemError)�args�kwargsr r r �effect� s z)MockTest.test_side_effect.<locals>.effectrK � � ��fishc s � � � S r ��popr ��resultsr r rh � s z side effect not used correctly�rO z#side effect in constructor not usedc S s t S r �r r r r r rO � s z.MockTest.test_side_effect.<locals>.side_effect)rO r< ) r
rO rP re �assert_called_withr9 r �
SideEffect�RETURN)r r rh rO r ro r �test_side_effect� s$
� �zMockTest.test_side_effectc s� dddg� � fdd�}dd� }t |�}dddg|_| �|� |� |� gdddgd� dddg� t |�}||_| �|� |� |� gdddgd � d S )
NrK ri rj c s � � � S r rm r ro r r rh � s z2MockTest.test_autospec_side_effect.<locals>.effectc S s d S r r r r r r rJ � r* z-MockTest.test_autospec_side_effect.<locals>.fz1side effect not used correctly in create_autospecz'callable side effect not used correctly)r rO r9 )r rh rJ r r ro r �test_autospec_side_effect� s
�
�z"MockTest.test_autospec_side_effectc C s, dd� }t |�}td�|_| �td|� d S )Nc S s d S r r r r r r rJ � r* z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r �
ValueErrorrO �assertRaisesRegexrL r r r �#test_autospec_side_effect_exception� s
z,MockTest.test_autospec_side_effect_exceptionc C sR t � }dg}t d||d�}|tjtjd� |j}|�� tj|_|j}|� |�� | � |j
dd� | � |j |d� | � |j |d� | �
|jd� | � |jd d
� | � |jd d � | � |jg d � | � |jg d
|jg f � | � |jg � | � |jtjd� | � |j|d� | �
|jd� | � |jd|id� | � |j|d� | �
|jjd� d S )Nr2 �child)rT �parentr_ �r2 zname incorrectly resetzparent incorrectly resetzmethods incorrectly resetzcalled not resetr zcall_count not resetzcall_args not resetzcall_args_list not resetz0method_calls not initialised correctly: %r != %rz side_effect incorrectly resetz return_value incorrectly resetzreturn value mock not resetzchildren reset incorrectlyz children incorrectly clearedzchild not reset)r
r r# Z
SomethingElser2 rt rO r< �
reset_mockr9 �
_mock_namerC rD r7 r8 r: r= r> r? �
mock_callsrE )r r| r_ r r2 r< r r r �test_reset_mock� sV
�
�
�
�� �
��
�zMockTest.test_reset_mockc C s t � }||_|�� d S r )r
r< r~ rF r r r �test_reset_mock_recursion� s z"MockTest.test_reset_mock_recursionc C s t �� }|�� d S r )r � mock_openr~ �r r% r r r �(test_reset_mock_on_mock_open_issue_18622� s z1MockTest.test_reset_mock_on_mock_open_issue_18622c C s> t � }| �t|jt �d� |� }| �|� |d� |�� |tj�}| �|jd� | �|j dd� | �|j
tjfi fd� | �|j
j tjfd� | �|j
j i d� | �|j
tjfi fgd� tj|_|tjtjd�}| �|tjd � | �|j d
d � | �|j
tjfd tjifd� | �|j
tjfi ftjfd tjifgd
� d S )Nz%Default return_value should be a Mockz'different result from consecutive callszcalled not setrK zcall_count incoreectzcall_args not setr6 )�keyzincorrect return valueri zcall_count incorrectr� zcall_args_list not set)r
r; r r< r9 r~ r �Argr8 r: r= rf rg r> Z ReturnValueZKeyArg)r r �resultZret_valr r r � test_call� sP �
�
�� ��
��
��zMockTest.test_callc C s� t � }|� |tj� |tjd� |tjtjd� | �|jdtjffdtjiftjfdtjifg� | �|jtjfdtjif� | �|jjtjf� | �|jjdtji� | � |jdk� d S )N)�kwr r� za long sequence)
r
r r� ZKwargr9 r> r= rf rg r7 rF r r r �test_call_args_comparison# s"
��z"MockTest.test_call_args_comparisonc C s� t �� }| �||k� | �||k� | �|t �� k� | �|t �� k� | �|t jk� | �|t jk� | �t j|k� | �t j|k� t �t �� �}t �t j�}| �||k� | �||k� | �||k� | �||k� d S r )r r r; r7 �ANYr )r ZmmZcall1Zcall2r r r �test_calls_equal_with_any9 s z"MockTest.test_calls_equal_with_anyc C sd t � }|� |�� | �t|jd� |�� | �t|j� |dddddd� |jdddddd� d S )NrK ri rj rl Znothing�r% r& )r
rs rP �AssertionErrorr~ rF r r r �test_assert_called_withN s z MockTest.test_assert_called_withc C s t � }|t � � |�tj� d S r )r rs r r� �r �mr r r �test_assert_called_with_any] s
z$MockTest.test_assert_called_with_anyc C s� d dd�}t |d�}|dddd� |�ddd� |jdddd� | jt|jdddd� | �t��}|jd d
� W 5 Q R X | �|jjt� d S ) Nc S s d S r r r- r r r rJ d r* z9MockTest.test_assert_called_with_function_spec.<locals>.fr^ rK ri rj �r& r' �r% r& r' � ��e)N)r
rs rP r� �assertIsInstance� exception� __cause__rN �r rJ r �cmr r r �%test_assert_called_with_function_specc s
� z.MockTest.test_assert_called_with_function_specc sz � fdd�}t t� jd�}||� t tjd�}||� t t� jd�}||� t tjd�}||� t t� jd�}||� d S )Nc sF | dddd� | � ddd� | j dddd� � jt| j dddd� d S )NrK ri rj r� r� )rs rP r� r r r r �_checkt s
�z<MockTest.test_assert_called_with_method_spec.<locals>._checkr^ )r
r# r) r, r. )r r� r r r r �#test_assert_called_with_method_specs s z,MockTest.test_assert_called_with_method_specc C s\ d}| � t|�d��� t� �� W 5 Q R X | � t|�d��� tdd��� W 5 Q R X d S )Nz"Expected '{0}' to have been calledr Z test_namerS )ry r� �formatr
�
assert_called)r �msgr r r �$test_assert_called_exception_message� s
z-MockTest.test_assert_called_exception_messagec s� t � � � � � �� � � | �t� j� � �� | �t� j� � dddd� � jdddd� � �� � dddd� | �t� fdd�� d S )NrR rU ri �rY c s � j dddd�S )N�bobrU ri r� ��assert_called_once_withr r r r �<lambda>� r* z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r
r� rP r� r~ r r r r � test_assert_called_once_with� s
�z%MockTest.test_assert_called_once_withc s6 t � � � d� � d� | �tt�d�� fdd�� d S )NrK ri zCalls: [call(1), call(2)]c s
� � d�S )Nri r� r �r� r r r� � r* zAMockTest.test_assert_called_once_with_call_list.<locals>.<lambda>)r
ry r� �re�escaper r r� r �&test_assert_called_once_with_call_list� s
�z/MockTest.test_assert_called_once_with_call_listc C s� ddd�}t |d�}|dddd� |�ddd� |jdddd� | jt|jdddd� | �t��}|jd d
� W 5 Q R X | �|jjt� |d d d
� | �t|jddd� | �t|jd d d
� d S )Nc S s d S r r r- r r r rJ � r* z>MockTest.test_assert_called_once_with_function_spec.<locals>.fr^ rK ri rj r� r� r� r� � � � )N)r
r� rP r� r� r� r� rN r� r r r �*test_assert_called_once_with_function_spec� s0
�
�
�z3MockTest.test_assert_called_once_with_function_specc C sb t � }|j}| �t|t �d� | �|j|d� t � }d|j_| �|�� dd� | �|jjd� d S )Nzattribute isn't a mockz+different attributes returned for same namerj zmethod returned wrong valuez!method didn't record being called)r
r2 r; r r9 r< r8 �r r r2 r r r �#test_attribute_access_returns_mocks� s
�
�z,MockTest.test_attribute_access_returns_mocksc C s0 t � }|j}| �|jdd� | �|j|d� d S )Nr2 z attribute name not set correctlyz"attribute parent not set correctly)r
r2 r9 r rC r� r r r �(test_attributes_have_name_and_parent_set� s
�
�z1MockTest.test_attributes_have_name_and_parent_setc C st t � }|jdd d� |jjdtjd� | �|jjdddtjifgd� | �|jdd d
d ifd ddtjifgd� d S ) Nrj rk r� )�caker2 )r� r� z#method calls not recorded correctly)rj rl zsomething_else.something)r
r2 �something_elser ZCaker9 r? rF r r r �test_method_calls_recorded� s
� ��z#MockTest.test_method_calls_recordedc C s� t � }|�� | �|jdg� | �|jddi fg� t � }|�d� | �|jdg� | �|jddi fg� t � }|jdd� | �|jdd difg� | �|jddd difg� t � }|jdd
d � | �|jddd d
ifg� d S )
Nr} r2 r � different)r2 �r� r� rK ��xr� Zmore)�somer� )r
r2 r9 r? rF r r r � test_method_calls_compare_easily� s&
� �z)MockTest.test_method_calls_compare_easilyc C sD dgdfD ]4}dD ]*}t f ||i�}|j | �tdt|d� qq
d S )Nr2 r} )r_ ra z-Mock object has no attribute 'something_else'r� )r
r2 ry �AttributeError�getattr)r r_ �argr r r r � test_only_allowed_methods_exist s �z(MockTest.test_only_allowed_methods_existc s> G dd� dt �}� fdd�}|t|d�� |t|� d�� d S )Nc @ s e Zd ZdZdZdd� ZdS )z*MockTest.test_from_spec.<locals>.Somethingrj Nc S s d S r r r r r r �y r* z,MockTest.test_from_spec.<locals>.Something.y)r r r! r� �
__something__r� r r r r r# s r# c s: | j | j | j � �tdt| d� � �tdt| d� d S )Nz Mock object has no attribute 'z'�zz)Mock object has no attribute '__foobar__'Z
__foobar__)r� r� r� ry r� r� r r r r �test_attributes s" � �z0MockTest.test_from_spec.<locals>.test_attributesr^ )rb r
)r r# r� r r r �test_from_spec s zMockTest.test_from_specc C sJ t � }t |d�}| �|� |� � |�� |dddd� |jdddd� d S )N��wrapsrK ri rj rk )r
r9 r~ rs �r �realr r r r �test_wraps_calls. s
zMockTest.test_wraps_callsc s8 G dd� dt �}|� }t|d�� | �t� fdd�� d S )Nc @ s e Zd ZdS )zFMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.RealNr\ r r r r �Real; s r� r� c s � � � S r )Znew_attrr r r r r� A r* zJMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.<lambda>)rb r
rP r� )r r� r� r r r �/test_wraps_prevents_automatic_creation_of_mocks: s
z8MockTest.test_wraps_prevents_automatic_creation_of_mocksc C s4 t � }t |d�}d|_| �|� d� | �|j� d S )Nr� rj )r
r< r9 r7 r8 r� r r r �,test_wraps_call_with_nondefault_return_valueD s
z5MockTest.test_wraps_call_with_nondefault_return_valuec s� G dd� dt �}|� }t|d�� | �� �� |�� � | �t� fdd�� | �� j|j� � jjdddd �}|jjjdddd � | �||j�� � d S )
Nc @ s e Zd Ze� ZdS )z,MockTest.test_wraps_attributes.<locals>.RealN)r r r! r
� attributer r r r r� O s r� r� c s � j S r rk r r r r r� V r* z0MockTest.test_wraps_attributes.<locals>.<lambda>rK ri rj rk ) rb r
r9 r� rP r� �assertNotEqualZfrogrs )r r� r� r� r r r �test_wraps_attributesN s
z MockTest.test_wraps_attributesc C sd G dd� dt �}|� }t|d�}tjtg|j_| �|�� tj� | �|�� tj� | � t
|j� d S )Nc @ s e Zd Zdd� ZdS )z[MockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Realc S s t jS r )r �ORIGINAL_VALUEr r r r �method` s zbMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Real.methodN�r r r! r� r r r r r� _ s r� r� ) rb r
r �VALUE1r r� rO r9 r� rP �
StopIteration�r r� r� r r r r �Dtest_customize_wrapped_object_with_side_effect_iterable_with_default^ s
zMMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_defaultc C sf G dd� dt �}|� }t|d�}tjtjg|j_| �|�� tj� | �|�� tj� | �t |j� d S )Nc @ s e Zd Zdd� ZdS )zNMockTest.test_customize_wrapped_object_with_side_effect_iterable.<locals>.Realc S s d S r r r r r r r� n r* zUMockTest.test_customize_wrapped_object_with_side_effect_iterable.<locals>.Real.methodNr� r r r r r� m s r� r� )
rb r
r r� �VALUE2r� rO r9 rP r� r� r r r �7test_customize_wrapped_object_with_side_effect_iterablel s
z@MockTest.test_customize_wrapped_object_with_side_effect_iterablec C s: G dd� dt �}|� }t|d�}t|j_| �t|j� d S )Nc @ s e Zd Zdd� ZdS )zOMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Realc S s d S r r r r r r r� { r* zVMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Real.methodNr� r r r r r� z s r� r� )rb r
� RuntimeErrorr� rO rP r� r r r �8test_customize_wrapped_object_with_side_effect_exceptiony s
zAMockTest.test_customize_wrapped_object_with_side_effect_exceptionc C sF G dd� dt �}dd� }|� }t|d�}||j_| �|�� tj� d S )Nc @ s e Zd Zdd� ZdS )zNMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Realc S s d S r r r r r r r� � r* zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Real.methodNr� r r r r r� � s r� c S s t jS r )r �VALUEr r r r rO � s zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.side_effectr� )rb r
r� rO r9 r r� )r r� rO r� r r r r �7test_customize_wrapped_object_with_side_effect_function� s
z@MockTest.test_customize_wrapped_object_with_side_effect_functionc C s@ G dd� dt �}|� }t|d�}tj|j_| �|�� tj� d S )Nc @ s e Zd Zdd� ZdS )zFMockTest.test_customize_wrapped_object_with_return_value.<locals>.Realc S s d S r r r r r r r� � r* zMMockTest.test_customize_wrapped_object_with_return_value.<locals>.Real.methodNr� r r r r r� � s r� r� )rb r
r r� r� r< r9 r� r r r �/test_customize_wrapped_object_with_return_value� s
z8MockTest.test_customize_wrapped_object_with_return_valuec C sp G dd� dt �}|� }t|d�}tjtjg|j_tj|j_| � |�� tj� | � |�� tj� | �
t |j� d S )Nc @ s e Zd Zdd� ZdS )zVMockTest.test_customize_wrapped_object_with_return_value_and_side_effect.<locals>.Realc S s d S r r r r r r r� � r* z]MockTest.test_customize_wrapped_object_with_return_value_and_side_effect.<locals>.Real.methodNr� r r r r r� � s r� r� ) rb r
r r� r� r� rO Z WRONG_VALUEr< r9 rP r� r� r r r �?test_customize_wrapped_object_with_return_value_and_side_effect� s
zHMockTest.test_customize_wrapped_object_with_return_value_and_side_effectc C sL G dd� dt �}|� }t|d�}dd� |j_tj|j_| �|�� tj� d S )Nc @ s e Zd Zdd� ZdS )zWMockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.Realc S s d S r r r r r r r� � r* z^MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.Real.methodNr� r r r r r� � s r� r� c S s t S r rr r r r r r� � r* z[MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.<lambda>)rb r
r� rO r r� r< r9 r� r r r �@test_customize_wrapped_object_with_return_value_and_side_effect2� s
zIMockTest.test_customize_wrapped_object_with_return_value_and_side_effect2c C sn G dd� dt �}|� }t|d�}tjtg|j_tj|j_| � |�� tj� | � |�� tj� | �
t |j� d S )Nc @ s e Zd Zdd� ZdS )z^MockTest.test_customize_wrapped_object_with_return_value_and_side_effect_default.<locals>.Realc S s d S r r r r r r r� � r* zeMockTest.test_customize_wrapped_object_with_return_value_and_side_effect_default.<locals>.Real.methodNr� r r r r r� � s r� r� ) rb r
r r� r r� rO ru r< r9 rP r� r� r r r �Gtest_customize_wrapped_object_with_return_value_and_side_effect_default� s
zPMockTest.test_customize_wrapped_object_with_return_value_and_side_effect_defaultc C s4 t td�}| �t|� t td�d�}| �t|� d S �Nrq rR )r
r� rP rF r r r � test_exceptional_side_effect� s
z%MockTest.test_exceptional_side_effectc C s4 t td�}| �t|� t td�d�}| �t|� d S r� )r
�KeyboardInterruptrP rF r r r � test_baseexceptional_side_effect� s
z)MockTest.test_baseexceptional_side_effectc C s t � }| �td|j� d S )Nz
not called)r
ry r� rs rF r r r � test_assert_called_with_message� s �z(MockTest.test_assert_called_with_messagec C s t dd�}| �td|j� d S )NZgeoffreyrS z'Expected 'geoffrey' to be called once\.)r
ry r� r� rF r r r �$test_assert_called_once_with_message� s
�z-MockTest.test_assert_called_once_with_messagec s2 t � � | �t� fdd�� d� _| �� jd� d S )Nc s � j S r )r r r r r r� � r* z'MockTest.test__name__.<locals>.<lambda>rR )r
rP r� r r9 r r r r � test__name__� s zMockTest.test__name__c C sJ G dd� dt �}t|dg�d�}|�d� |j�d� | �tt|d� d S )Nc @ s e Zd ZdS )z-MockTest.test_spec_list_subclass.<locals>.SubNr\ r r r r �Sub� s r� rR r^ rj )�listr
�appendrs rP r� r� )r r� r r r r �test_spec_list_subclass� s
z MockTest.test_spec_list_subclassc C s� G dd� dt �}t|d�}| �||� t|� d�}| �||� | �|j|� | �t� jjd� t|d�}| �||� t|� d�}| �||� d S )Nc @ s e Zd ZdS )z#MockTest.test_spec_class.<locals>.XNr\ r r r r r] � s r] r^ r
r` )rb r
r� �assertIs� __class__r9 r rc r r r �test_spec_class� s
zMockTest.test_spec_classc C s� G dd� d�}t |d�}| �||� t |� d�}| �||� | �|j|� | �t � jjd� t |d�}| �||� t |� d�}| �||� d S )Nc @ s e Zd ZdS )z2MockTest.test_spec_class_no_object_base.<locals>.XNr\ r r r r r]
s r] r^ r
r` )r
r� r� r� r9 r rc r r r � test_spec_class_no_object_base s
z'MockTest.test_spec_class_no_object_basec sL G dd� dt �}t|d�� d� _t|d�� � fdd�}d� _| �t|� d S )Nc @ s e Zd ZdZdS )z8MockTest.test_setting_attribute_with_spec_set.<locals>.Xrj N)r r r! r� r r r r r] s r] r^ rR r` c s
d� _ d S �NrR r� r r r r �set_attr% s z?MockTest.test_setting_attribute_with_spec_set.<locals>.set_attr)rb r
r� r� rP r� )r r] r� r r r �$test_setting_attribute_with_spec_set s
z-MockTest.test_setting_attribute_with_spec_setc C s4 t �� }| �t j|� t �td�� t�t� � d S )Ng e��A)�sys�getrecursionlimit�
addCleanup�setrecursionlimit�int�copyr
)r �currentr r r � test_copy, s zMockTest.test_copyc s\ G dd� dt �}||d�� | �� jd� � fdd�}| �t|� � fdd�}| �t|� d S )Nc @ s& e Zd Zdd� Zdd� Zeee�ZdS )z8MockTest.test_subclass_with_properties.<locals>.SubClassc S s dS �Nrj r r r r r �_get8 s z=MockTest.test_subclass_with_properties.<locals>.SubClass._getc S s t d��d S )Nz
strange error)� NameError)r �valuer r r �_set: s z=MockTest.test_subclass_with_properties.<locals>.SubClass._setN)r r r! r� r �property�some_attributer r r r �SubClass7 s r r` rj c s
d� _ d S r� )r r ��sr r �testA s z4MockTest.test_subclass_with_properties.<locals>.testc s
d� _ d S )NrU �rR r r r r r E s )r
r9 r rP r r� )r r r r r r � test_subclass_with_properties6 s
z&MockTest.test_subclass_with_propertiesc C s> t � }dd� }|t|�_|d� |�d� | �t|dd� d S )Nc S s | � |� | �|�S r )Z_increment_mock_callZ
_mock_callr� r r r �__call__L s
z,MockTest.test_setting_call.<locals>.__call__�one�two)r
�typer rs rP rN )r r r r r r �test_setting_callJ s
zMockTest.test_setting_callc C s� t � }tt|��}tdd� tt �D ��}| �t� || � |j|jf | �dt|�� | �dt|�� d |_|_| �dt|�� | �dt|�� dd� |_ | �d t|�� d S )
Nc S s g | ]}|� d �s|�qS )�_)�
startswith)�.0r� r r r �
<listcomp>Z s
z%MockTest.test_dir.<locals>.<listcomp>r% r& r' r( c S s t g �S r )r r r r r r� j r* z#MockTest.test_dir.<locals>.<lambda>r )
r
�set�dirr9 r% r& rV r' r( r )r r �attrs�
type_attrsr r r �test_dirW s
zMockTest.test_dirc C sX t tjd�}tttj��}tt|��}| �t� || � d|_| �t|��d�d� d S )Nr^ rj �versionrK )r
�unittest�TestCaser r r9 r �count)r r Ztestcase_attrsr r r r �test_dir_from_specn s zMockTest.test_dir_from_specc C sV t �tdd�}|�� z0ttt� ��}ttt��}| �t� || � W 5 |�� X d S )NZ
FILTER_DIRF) r rb r �start�stopr r r
r9 )r Zpatcherr r r r r �test_filter_dir{ s zMockTest.test_filter_dirc C s6 t � }d|j_| �dt|�� |`| �dt|�� d S )NrK r{ )r
r{ r< rV r r@ rF r r r �,test_dir_does_not_include_deleted_attributes� s
z5MockTest.test_dir_does_not_include_deleted_attributesc C s� t dd�}| �|jd� tdd�}| �|jd� tdt� d�}t f |�}| �t|� | �|j�� d� | �|jt� t � }|jf |� | �t|� | �|j�� d� | �|jt� d S )NrU r �! )rO zfoo.bar.return_valuerR ) r
r9 rR r �KeyErrorrP rU r� Zconfigure_mock)r r rg r r r �test_configure_mock� s
�
z MockTest.test_configure_mockc O s: | � |��}|||� W 5 Q R X t|j�}| �||� d S r )rP �strr� r9 )r r� �message�funcrf rg �contextr� r r r �assertRaisesWithMsg� s
z MockTest.assertRaisesWithMsgc
C s t � }d}d}d}| jt|||f |jddddd� |jddddd � |jj|jjg}|D ]0}d
}d }d}| jt|||f |ddddd� qZ|D ]*}d
}d }d}| jt|||f |dd� q�|D ],}d
}d
}d}| �t|||f |ddd� q�|D ]&}d
}d}d}| �t|||f |� q�d S )Nz not called.zmock(1, '2', 3, bar='foo')z0expected call not found.
Expected: %s
Actual: %srK �2rj rR )rU r zfoo(1, '2', 3, foo='foo')zfoo(1, '2', 3, bar='foo')zfoo(bar='foo')z foo(1, 2, 3)ri zfoo())r r) r� rs rR r� )r r �actual�expectedr& Z assertersr) r r r �'test_assert_called_with_failure_message� sz
� �
�
�
�
�z0MockTest.test_assert_called_with_failure_messagec C s� t � }| �|jg kd� t � }|� ddi fg}| �|j|� |�� |�t�� � | �|j|� | �|jjddi fg� t � }|� jdddddd � ddi fd
d tddd �fg}| �|j|� | �|jjjdd tddd �fg� | �|jjd d tddd �fg� t � }|� j� � �
� ddi fd
di fddi fg}| �|j|� | �|� jtj� � �
� � � � t� tdd�fD �]$}t f |�}t |j� ddi fg}| �|j|� t f |�}|�
� � ddi fddi fg}| �|j|� | �|�
� jt� g� t f |�}|d�d�d� | �|jtd�d�d�� � � | �|� jtd�d�� � � | �|� � jtd�� � � t f |�}|d�d�d�j
j�d� | �|jtd�d�d�j
j�d�� � � | �|� jtd�d�j
j�d�� � � | �|� � jtd�j
j�d�� � � t f |�}t |� j� � �
� � ddi f}| �|jd |� | �|� jtj� � �
� �� � � � | �|� j� � jt�
� �� � � � | �|� j� � j
jt� �� � � � �qpd S )NT� r rK ri rj r� r� r� z().foo�rK ri rj rR z
().foo.barz().foo.bar().bazrU rS z foo.__int__r% za()z().foo.bar().baz().__int__���)r r� r� r9 rR r� r �dictr< rU rY � call_listr� r% r& r' �__int__)r r r, rg Z last_callr r r �test_mock_calls� s� � �
� �
�
�
� �
�� �zMockTest.test_mock_callsc C sF t � }|� }|�� | �|jt� t� �� g� | �|jt�� g� d S r )r
Zwibbler9 r� r )r r� r� r r r �test_child_mock_call_equal4 s
z#MockTest.test_child_mock_call_equalc C sF t � }|�� �� | �|jd t�� �� � | �|jd t�� � d S �NrK r )r
rR r2 r� r� r r� r9 r� r r r � test_mock_call_not_equal_leaf> s z&MockTest.test_mock_call_not_equal_leafc C sF t � }|�� �� | �|jd t�� �� � | �|jd t�� � d S r6 �r
rR rU r� r� r rY r� r r r �!test_mock_call_not_equal_non_leafE s z*MockTest.test_mock_call_not_equal_non_leafc C s8 t � }|jdd��� | �|jd tjdd��� � d S )NrK r� ri )r
rR rU r9 r� r r� r r r �2test_mock_call_not_equal_non_leaf_params_differentL s z;MockTest.test_mock_call_not_equal_non_leaf_params_differentc C s, t � }|j�� | �|jd tj�� � d S �Nr r8 r� r r r �&test_mock_call_not_equal_non_leaf_attrS s
z/MockTest.test_mock_call_not_equal_non_leaf_attrc C s. t � }|j�� | �|jd t�� �� � d S r; )r
rR rU r� r� r r� r r r �2test_mock_call_not_equal_non_leaf_call_versus_attrY s
z;MockTest.test_mock_call_not_equal_non_leaf_call_versus_attrc C s^ t � }|�� �� j�� | �t|jd �d� | �t|jd �d� | �t|jd �d� d S )Nr z
call.foo()rK zcall.foo().bar()ri zcall.foo().bar().baz.bob())r
rR rU rY r� r9 rW r� r� r r r �test_mock_call_repr_ s
z MockTest.test_mock_call_reprc C s0 t � }||_t|�� � | �t|�� �d� d S )Nz <Mock name='mock\(\)' id='\d+'>)r
rR rW Z assertRegexr� r r r �test_mock_call_repr_loopg s z!MockTest.test_mock_call_repr_loopc C s t � }| �t� g|jk� d S r )r
r7 r r� r� r r r �test_mock_calls_containsn s z!MockTest.test_mock_calls_containsc C sh G dd� dt �}|� }| �|j|� | �|� |� G dd� dt �}|� }| �|j|� | �|� |� d S )Nc @ s e Zd ZdS )�+MockTest.test_subclassing.<locals>.SubclassNr\ r r r r �Subclasst s rB c @ s e Zd Zdd� ZdS )rA c [ s
t f |�S r )r
)r rg r r r �_get_child_mock| s z;MockTest.test_subclassing.<locals>.Subclass._get_child_mockN)r r r! rC r r r r rB { s )r
r� rR �assertNotIsInstance)r rB r r r r �test_subclassings s zMockTest.test_subclassingc s� t � t� t� t� g}� fdd�}|D ]z}||� t|�rb|� |dd� |dd� |�� ||� |�� |jjddd� |�d��� �d� |�� ||� q$d S )Nc s@ d}|D ]2}t | |�}� �|t� � �|t� � �|g � qd S )N)r> r? r� )r� r� r r� r9 )r �namesrT �attrr r r � assert_attrs� s
z-MockTest.test_arg_lists.<locals>.assert_attrsrK ri rj r1 ) r
r r r �callabler~ rR rU rY )r rZ rH r r r r �test_arg_lists� s& �
zMockTest.test_arg_listsc C s� t � }|ddd� |ddd� | �t|j�d� | �|jjd� | �|jjtdd�� dtdd�fdtdd�fg}t||j�D ]@\}}| �t|�d� | �|d |d � | �|d |d � q~d S )
NrK rj r1 ri r� �r& )ri �rK r ) r
r9 �lenr= rf rg r1 �zipr> )r r Z
expected_listr, r= r r r �test_call_args_two_tuple� s z!MockTest.test_call_args_two_tuplec C s t tdddg�d�}| �|� |� |� gdddg� | �t|� tdddgd�}| �|� |� |� gdddg� | �t|� t dd�}| �|� |� |� gd d
d g� | �t|� G d d
� d
t�}t|d�}| �|� |� t t� d�}| �|� |� |� |� gddddg� | �t|� d S )NrK ri rj rq r% r& r' Zghi�g�h�ic @ s e Zd ZdS )z/MockTest.test_side_effect_iterator.<locals>.FooNr\ r r r r �Foo� s rS r r r r ) r
r r9 rP r� r rb r� r )r r rS r r r �test_side_effect_iterator� s"
�z"MockTest.test_side_effect_iteratorc C s\ t tfD ]N}tdtdf}||d�}| �t|� | �|� d� | �t|� | �|� d� qd S )Nrj r� rq )r
r rx r# rP r9 )r �Klass�iterabler� r r r �$test_side_effect_iterator_exceptions� s
z-MockTest.test_side_effect_iterator_exceptionsc C s� t � }tdddg�|_| �|� |� |� gdddg� | �t|� |j}| �|ttg ��� dddg|_| �|� |� |� gdddg� | �t|� |j}| �|ttg ��� t� }||_| �|� |� |� |� gddd d
g� | �t|� | � |j|� d S ) NrK ri rj r% r& r' r r r r )
r
r rO r9 rP r� r� r r r� )r r rO Z this_iterr r r �!test_side_effect_setting_iterator� s$
� z*MockTest.test_side_effect_setting_iteratorc C s4 t dd�}tdtg�|_| �|� |� gddg� d S )Nri rH rK )r
r r rO r9 rF r r r �!test_side_effect_iterator_default� s
z*MockTest.test_side_effect_iterator_defaultc C s� t � }|dd� |dd� |dd� |dd� |dd� tdd�ddifd d
ddifd d ddifd d
i fd d
ddifg}|D ]}|j|gdd� q|tdd�tdd�t� dd dfD ]}| jt|j|gdd� q�tdd�tdd�gtdd�tdd�gtdd�tdd�gg}|D ]}|j|dd� �q
tdd�tdd�tdd�gtdd�tdd�gtdd�tdd�tdd�gtdd�tdd�tdd�tdd�tdd�gg}|D ]}| jt|j|dd� �q�d S )NrK ri rj r1 r� r� rK r% �)rj r� r )r. �rK ri r. r[ T�Z any_orderr* rR r� � )r
r �assert_has_callsrP r� )r r �kalls�kallZ
kall_listsZ kall_listr r r � test_assert_has_calls_any_order� sT
�" ��*� �z(MockTest.test_assert_has_calls_any_orderc
C s� t dd�ddifdt dd�dd d
difg}t �� t �d�g}|�t �� jdd ��� � |�t jt� i d �� dg��� � g }t
� t � fD ]F}|dd� |dd � |dd
� |dd� |ddd� |� ||f� q�t
� }|�� |�d� |�� jdd � |jt� i d �� dg� |� ||f� |D ]j\}}t
t|��D ]R}dD ]F}|||| � }|�|� t|�dk�r>| �t|jtt|��� �q>�q6�q"d S )NrK ri r% rj rZ r� rK r. rL r& r1 r r� r/ )r rR rU �extendZspamrY r2 Zbamr rl r
r r� �rangerM r^ rP r� r� �reversed) r Zkalls1Zkalls2rZ r r_ rR �stepZtheser r r �test_assert_has_calls sD �"
�z MockTest.test_assert_has_callsc C sF G dd� d�}t |�}||� fD ]H}|jddddd� |�tjddddd�g� |�t�dddd�g� q |�� ||� fD ]l}| �t|jt�� g� |�d��dd� |�t�d�t�d��dd�g� |j�td�t� �dd�g� q||�� t�d�t� d�t�� � d�t�� �dddd�g}|D ]}| �t|j|g� �q(d S )Nc @ s, e Zd Zdd� Zddd�ZG dd� d�ZdS ) z=MockTest.test_assert_has_calls_nested_spec.<locals>.Somethingc S s d S r r r r r r r J r* zFMockTest.test_assert_has_calls_nested_spec.<locals>.Something.__init__Nc S s d S r r r$ r r r r) K r* zBMockTest.test_assert_has_calls_nested_spec.<locals>.Something.methc @ s e Zd Zdd� Zdd� ZdS )zAMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Fooc S s d S r r r� r r r r O r* zJMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.__init__c S s d S r r )r r% r& r r r �meth1P r* zGMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.meth1N)r r r! r rg r r r r rS M s rS )N)r r r! r r) rS r r r r r# H s
r# rK ri rj �r( r� )
r r) r^ r r~ rP r� rS rg Z non_existent)r r# Z
mock_classr� Z
invalid_callsr` r r r �!test_assert_has_calls_nested_specG s.
" ��z*MockTest.test_assert_has_calls_nested_specc C sJ t � }|� �� �� �� |�� �� �� t�� �� �� �� }|� |� d S r )
r rR rU rY r r
�threer r2 r^ )r r� �callsr r r �)test_assert_has_calls_nested_without_speco s
z2MockTest.test_assert_has_calls_nested_without_specc C s� ddd�}t |d�}|dddd� |dd d
d d � |d
ddd� ddi fdddd ifdi fg}|�|� |j|dd� |�|dd � � |j|dd � dd� |�|d d� � |j|d d� dd� tt|��}| �t�� |�|� W 5 Q R X |j|dd� | �t�� |�|dd � � W 5 Q R X |j|dd � dd� | �t�� |�|d d� � W 5 Q R X |j|d d� dd� d S )Nc S s d S r r r- r r r rJ x r* z<MockTest.test_assert_has_calls_with_function_spec.<locals>.fr^ rK ri rj r� r� r� r� r] �r' r( �
� � )r' r. r/ )r� r� r� r( )rn ro rp Tr\ r0 )N)r
r^ r� rd rP r� )r rJ r rk r r r �(test_assert_has_calls_with_function_specw s2
�
z1MockTest.test_assert_has_calls_with_function_specc C s� d dd�}t |d�}|d� | �td�t�d����}|�t� g� W 5 Q R X | �|j j
� | �td�t�d���� }|�t� tdd�g� W 5 Q R X | � |j j
t � d S )
Nc S s d S r r r� r r r rJ � r* zAMockTest.test_assert_has_calls_not_matching_spec_error.<locals>.fr^ rK z^{}$z5Calls not found.
Expected: [call()]
Actual: [call(1)]z�Error processing expected calls.
Errors: [None, TypeError('too many positional arguments')]
Expected: [call(), call(1, 2)]
Actual: [call(1)]ri )N)
r
ry r� r� r� r� r^ r rB r� r� r� rN r� r r r �-test_assert_has_calls_not_matching_spec_error� s.
����� z6MockTest.test_assert_has_calls_not_matching_spec_errorc C s� t � }|dd� |dd� |ddd� |�dd� |jdd� |jddd� | �t|j� | �t|jdd� | jt|jdd� d S )NrK ri rj r1 r� rK r� )r
�assert_any_callrP r� rF r r r �test_assert_any_call� s,
� ��z MockTest.test_assert_any_callc C s� ddd�}t |d�}|dddd� |dd d
d d � |�ddd� |jdddd
� |�dd d
d � |jdd d
d d� | jt|jdddd� | �t��}|jdd� W 5 Q R X | �|jjt� d S )Nc S s d S r r r- r r r rJ � r* z;MockTest.test_assert_any_call_with_function_spec.<locals>.fr^ rK ri rj r� r� r� r� r] rm r� r- r� r� )N)r
rs rP r� r� r� r� rN r� r r r �'test_assert_any_call_with_function_spec� s
� z0MockTest.test_assert_any_call_with_function_specc C sf dd� }t � }||_t|�t|�jg}|D ]6}|dd� |dd� | �|jtdd�tdd�g� q*d S )Nc S s d S r r r� r r r rJ � r* z3MockTest.test_mock_calls_create_autospec.<locals>.frK ri rj r� )r rJ r r9 r� r )r rJ �objZfuncsr' r r r � test_mock_calls_create_autospec� s �
�z(MockTest.test_mock_calls_create_autospecc C s$ t jt� dd�}| �dt|�� d S )NZ
sweet_funcrS )r r rb rV rW r� r r r � test_create_autospec_with_name� s z'MockTest.test_create_autospec_with_namec
C s` G dd� d�}dD ]H}| j |d��2 t�t||��}|� |�� | �t|d� W 5 Q R X qd S )Nc @ s$ e Zd Zedd� �Zedd� �ZdS )zMMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClassc S s d S r r )r+ r r r � class_method� s zZMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.class_methodc S s d S r r r r r r �
static_method� s z[MockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.static_methodN)r r r! r/ ry r0 rz r r r r � TestClass� s
r{ )ry rz )r� Z extra_arg)ZsubTestr r r� r� rP rN )r r{ r� Z mock_methodr r r �1test_create_autospec_classmethod_and_staticmethod� s z:MockTest.test_create_autospec_classmethod_and_staticmethodc C sh t � }d}| �t|�� |�� W 5 Q R X | �t|�� |�� W 5 Q R X t dd�}|�� |�� d S )Nz1Attributes cannot start with 'assert' or 'assret'T)Zunsafe)r
ry r� Zassert_foo_callZassret_foo_call)r r� r� r r r �test_mock_unsafe s
zMockTest.test_mock_unsafec C s< t � }|j�� |�� | �t�� |j�� W 5 Q R X d S r )r
�hello�assert_not_calledrP r� r� r r r �test_assert_not_called s
z MockTest.test_assert_not_calledc C s* t � }|dd� | �tt�d�|j� d S )NrK ri zCalls: [call(1, 2)])r
ry r� r� r� r r� r r r � test_assert_not_called_message s
�z'MockTest.test_assert_not_called_messagec C sN t � }| �t�� |j�� W 5 Q R X |�� |j�� |�� |j�� d S r )r
rP r� r~ r� r� r r r �test_assert_called s
zMockTest.test_assert_calledc C sd t � }| �t�� |j�� W 5 Q R X |�� |j�� |�� | �t�� |j�� W 5 Q R X d S r )r
rP r� r~ �assert_called_oncer� r r r �test_assert_called_once( s
z MockTest.test_assert_called_oncec C s2 t � }|dd� |d� | �tt�d�|j� d S )NrK ri rj z Calls: [call(1, 2), call(3)])r
ry r� r� r� r� r� r r r � test_assert_called_once_message3 s
�z(MockTest.test_assert_called_once_messagec C s: t � }| �t��}|�� W 5 Q R X | �dt|j�� d S )NzCalls:)r
rP r� r� r@ r% r� )r r� r� r r r �*test_assert_called_once_message_not_called; s z3MockTest.test_assert_called_once_message_not_calledc C s0 t � }|jddd� d}| �t|jj�|� d S )Nr~ Zhero)rT Zdaddyz call(name='hello', daddy='hero'))r
r~ r9 rW r= )r r� �textr r r �test_ordered_call_signatureB s z$MockTest.test_ordered_call_signaturec C sV t �� }t �dd�}t� }|�� |�dd� | �|jd |� | �|jd |� d S )N� r~ r rK )r r �indexr
r9 r? )r r' rR r� r r r �test_override_tuple_methodsI s z$MockTest.test_override_tuple_methodsc C s> t dddgd�}|jddd� | �|jt � | �|jd � d S )Nrn ri rj �r< rO T)r
r~ r� r< r9 rO r� r r r � test_reset_return_sideeffectR s z%MockTest.test_reset_return_sideeffectc C s< t dddgd�}|jdd� | �|jt � | �|jd � d S )Nrn ri rj r� TrH )r
r~ r� r< r� rO r� r r r �test_reset_returnX s zMockTest.test_reset_returnc C s< t dddgd�}|jdd� | �|jd� | �|jd � d S )Nrn ri rj r� Trq )r
r~ r9 r< rO r� r r r �test_reset_sideeffect^ s z MockTest.test_reset_sideeffectc s� G dd� dt �}G dd� dt �}G dd� dt �� ttttg}t|�D ]0}|�|f� fdd� � |�|f� fd d� � qD|D �]}t� td
d �fD ]�}|� }|j|j |j
f ||fd gd
gffD ]�\}}t� td
d �fD ]�}|j |f|� |j | � t
t|d
� | � t
t|d� d|k�r*| � t
t|dd � |j |f|� | � t
t|d � |j | � t
t|d� d|kr�| � t
t|dd � q�q�q�qzd S )Nc @ s e Zd ZdZdS )z)MockTest.test_mock_add_spec.<locals>._OnerK N)r r r! r r r r r �_Onee s r� c @ s e Zd ZdZdS )z)MockTest.test_mock_add_spec.<locals>._Twori N)r r r! r
r r r r �_Twog s r� c @ s e Zd Zd Z ZZdS )z-MockTest.test_mock_add_spec.<locals>.AnythingZfourN)r r r! r r
rj r r r r �Anythingi s r� c s
| � d�S )Nr^ r ��K�r� r r r� p r* z-MockTest.test_mock_add_spec.<locals>.<lambda>c s
| � d�S )Nr` r r� r� r r r� q r* Tr` r r
rj ra )rb r
r r r r� r� r1 r r
rj �
mock_add_specrP r� r� �setattr) r r� r� ZklassesrU rg r ZOneZTwor r� r �test_mock_add_specd sz �
� �
� � � �zMockTest.test_mock_add_specc s| t tfD ]n}|� � t� � � �t� | �tt� � |� � � d d� j_� �t� | � t� �d� | �t� fdd�� qd S )NrR r� c s � d S r� r r r r r r� � r* z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� r� rb rP rN r3 r< r9 )r rU r r r � test_mock_add_spec_magic_methods� s
z)MockTest.test_mock_add_spec_magic_methodsc C s� t ttttfD ]�}|� }t� |_|�� | �|jt�� g� | �|j t�� g� |� }tdd�|_
|�
� | �|jg � | �|j g � |� }t� � |_ |� � | �|jg � | �|j g � qd S )NrT rS ) r r
r r r
rR r9 r? r r� rU rY �r rU r r r r �test_adding_child_mock� s$ �
z MockTest.test_adding_child_mockc C s@ t tfD ]2}|� }t� |_|� � | �|jt� t� � g� qd S r )r
r r< r9 r� r r� r r r � test_adding_return_value_mock� s
z&MockTest.test_adding_return_value_mockc C s� G dd� dt �}t� }t� |d�}t� |d�}|�� }| �|j� |�� }| �|j� |�|d� |�|d� |�� |�� | � |j
t �� t �� g� d S )Nc @ s e Zd ZdZdZdS )z'MockTest.test_manager_mock.<locals>.Foor r
N)r r r! r r
r r r r rS � s rS r r
) rb r
r r r� r � attach_mockr
r r9 r� r )r rS �managerZp1Zp2Zmock_oneZmock_twor r r �test_manager_mock� s zMockTest.test_manager_mockc C sh t tfD ]Z}|� }t dd�|_tdd�|_t|� t|� | �|jt�� t�� g� | �|j g � qd S )Nrj rH g @)
r
r r3 � __float__r� �floatr9 r� r r? )r rU r� r r r � test_magic_methods_mock_calls� s z&MockTest.test_magic_methods_mock_callsc C s< t jdd�}|d�}|�� }|d�}|�� }| �||� d S )N�data�Z read_data�a-namez another-name)r r� �readr9 )r � mocked_open�f1Zf1_data�f2Zf2_datar r r � test_mock_open_reuse_issue_21750� s z)MockTest.test_mock_open_reuse_issue_21750c C sV t jdd�}|d�}dd� |D �}| �|d d� | �|d d � | �t|�g � d S )
NzRemarkable
Norwegian Bluer� r� c S s g | ]}|�qS r r �r �liner r r r � s z=MockTest.test_mock_open_dunder_iter_issue.<locals>.<listcomp>r z Remarkable
rK zNorwegian Blue)r r� r9 r� )r r� r� �linesr r r � test_mock_open_dunder_iter_issue� s z)MockTest.test_mock_open_dunder_iter_issuec C s� t jdd�}|d�}t|�}|�� }dd� |D �}| �|d� | �|d� | �|d d � | �t|�g � | �t�� t|� W 5 Q R X d S )
Nz1st line
2nd line
3rd liner� r� c S s g | ]}|�qS r r r� r r r r s z6MockTest.test_mock_open_using_next.<locals>.<listcomp>z 1st line
z 2nd line
r z3rd line)r r� r r" r9 r� rP r� )r r� r� Zline1Zline2r� r r r �test_mock_open_using_next� s z"MockTest.test_mock_open_using_nextc C sX t �t jdd��}t �d|��0 |j}|j}td�|_dd� }| �t|� W 5 Q R X d S )NZJLVrS ztempfile.NamedTemporaryFilez Test 2 Errorc S s t �� �d� d S )NZasd)�tempfile�NamedTemporaryFile�writer r r r �attempt s z.MockTest.test_mock_open_write.<locals>.attempt) r r� r r r< r� �OSErrorrO rP )r Zmock_namedtempZmock_filehandleZ
mock_writer� r r r �test_mock_open_write s
z MockTest.test_mock_open_writec C sJ t jdd�}dd� |jj_|� �� }|� �� }| �d|� | �d|� d S )Nzfoo
barnr� c W s dS )N�abcr )rf r r r r� r* z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>r� )r r� r< �readlinerO r9 )r Zmopen�first�secondr r r � test_mock_open_alter_readline s
z&MockTest.test_mock_open_alter_readlinec C s� t jdd�}|d�}|�� | �d|�� � | �d|�� � | �d|�� � | �d|�� � | �g |�� � | �g |�� � d S )NrR r� rU r. )r r� r� r9 r� � readlines)r �_openrQ r r r �test_mock_open_after_eof s z!MockTest.test_mock_open_after_eofc C s
t tfD ]�}|� }t|�}||_| �|� |� | �t|�|� |�� | �|� |� | �t|�|� |� }|j|_| � dt|j�� | � dt|j�� |�� | � dt|j�� | � dt|j�� |� }t|�}|� |_||j_| �t|�|� | �t|�� �|� qd S )Nz
name='mock.a')
r
r rW r< r� r9 r~ r% r& rV )r rU r� Z
original_reprr r r �test_mock_parents+ s, zMockTest.test_mock_parentsc C s� t tttf}|D ]�}|D ]z}|� }|dd�}|�|d� | �|j|� | �dt|�� |j� d� | �
|j t j� d�g� | �
|j
t j� d�g� qqd S )NrR rS rU zname='mock.bar'rK )r
r r r r� r� rU rV rW rY r9 r� r r? �r �classesrU ZKlass2r� Zm2r r r �test_attach_mockH s
zMockTest.test_attach_mockc C s� t tttf}t tfD ]h}|D ]^}|� }|dd�}|�|d� | �|� |� | �dt|�� |�� | � |j
t � �� � � � q qd S )NrR rS r< z
name='mock()')
r
r r r r� r� rV rW rR r9 r� r r2 r� r r r � test_attach_mock_return_valueY s
z&MockTest.test_attach_mock_return_valuec C s� t � }tjt� d�dd���}| �|j�� d� |�|d� |�d� td� |d� t �d�t �d�t �d�g}t d�t d�t d�g}| �|j
|� | �|jj
|� | �tj
|� | �|j
|� | � d t |jj�� | �|j�� d � W 5 Q R X d S )
N�
.somethingT�Zautospecr2 r{ rK ri rj �
mock.child)
r
r r r r9 Z_extract_mock_namer� r{ r2 r r� rV rW )r r| Z mock_funcZ parent_callsZ child_callsr r r � test_attach_mock_patch_autospeci s
z(MockTest.test_attach_mock_patch_autospecc
C s� t jt� d�dd���}t� }|�|d� t� }|jddddd � |�tj t j
ddddd �g� |j�tt j
ddddd �g� |�tt j
ddddd �g� W 5 Q R X t jt� d
�dd��R}t� }|�|d � t d� |�t� d�g� t �td�g� |�td�g� W 5 Q R X t jt� d �dd���}t� }|�|d
� t� }|jddddd � |�t�
� t�
� jddddd �g� |j�tddddd �g� |�t� t� jddddd �g� W 5 Q R X d S )Nz.Something.methTr� � attach_methrK ri rj r� rh r� � attach_funcz
.Something�
attach_obj)r r r r
r� r# r) r^ r r� r� r2 r� r� )r Zmockedr� rv r r r �)test_attach_mock_patch_autospec_signature} s2 &
�z2MockTest.test_attach_mock_patch_autospec_signaturec C sf t � t� t� t� fD ]L}| �t|d�� |`| �t|d�� |`| �t|d�� | � t
t |d� qd S )Nr� rJ ) r
r r r r; �hasattrr� r7 rJ rP r� r� rF r r r �test_attribute_deletion� s �z MockTest.test_attribute_deletionc C s� t � t� t� t� fD ]t}d|_| �t|d�� | �|jd� |`| �t|d�� d|_| �t|d�� | �|jd� |`| �t|d�� qd S )Nrj rR r� ) r
r r r rR r; r� r9 r7 rF r r r �7test_mock_does_not_raise_on_repeated_attribute_deletion� s �z@MockTest.test_mock_does_not_raise_on_repeated_attribute_deletionc
C s< t � t� t� t� fD ]"}|`| �t��
|`W 5 Q R X qd S r )r
r r r rR rP r� rF r r r �4test_mock_raises_when_deleting_nonexistent_attribute� s � z=MockTest.test_mock_raises_when_deleting_nonexistent_attributec C s, t � }d|_|`|�� | �t|d�� d S )NTr{ )r
r{ r~ r7 r� rF r r r �/test_reset_mock_does_not_raise_on_attr_deletion� s
z8MockTest.test_reset_mock_does_not_raise_on_attr_deletionc C s: t � t� fD ](}| �|t� t|_| �|t� |j q d S r )r
r rD r� r� r� rR rF r r r �test_class_assignable� s
z MockTest.test_class_assignablec C s8 | � tj� | �ttj�t� | �tt�� j�t� d S r )�assertIsNotNoner rT r9 r r r r r r �test_name_attribute_of_call� s z$MockTest.test_name_attribute_of_callc C s8 | � tj� | �ttj�t� | �tt�� j�t� d S r )r� r r| r9 r r r r r r � test_parent_attribute_of_call� s z&MockTest.test_parent_attribute_of_callc C sd dd� }t � }t|�|_|�dd� | �t|jd� | �|jt�dd�g� | �dt |jj
�� d S )Nc S s d S r r r� r r r rR � r* zBMockTest.test_parent_propagation_with_create_autospec.<locals>.foorK ri r� ) r
r r{ rP rN r9 r� r rV rW r )r rR r r r r �,test_parent_propagation_with_create_autospec� s
z5MockTest.test_parent_propagation_with_create_autospecc C sp dd� }t � }|�t|dd�d� |�dd� | �t|jd� | �|jjt�dd�g� | � dt
|jj �� d S ) Nc S s d S r r r� r r r rR � r* zGMockTest.test_parent_propagation_with_autospec_attach_mock.<locals>.foorU rS r{ rK ri r� ) r
r� r r{ rP rN r9 r� r rV rW r )r rR r| r r r �1test_parent_propagation_with_autospec_attach_mock� s z:MockTest.test_parent_propagation_with_autospec_attach_mockc s� t jj}| �dd� |� t�d��| tjd= � fdd�� | �tjt�� � t�� � ddl m
}m }m }m
} ||||tg}|D ]}|td �}| �|t� q~W 5 Q R X d S )
Nc S s t tjd| �S )Nr )r� r r )r r r r r�
r* z9MockTest.test_isinstance_under_settrace.<locals>.<lambda>z sys.modulesz
unittest.mockc s � S r r )�frame�eventr� ��tracer r r� s z6MockTest.test_isinstance_under_settrace.<locals>.tracer )r
r r r r^ )r r r r� r1 r� �modules�settrace�gettrace�
unittest.mockr
r r r r
r# r� ) r Z old_patchr
r r r rZ r rv r r� r � test_isinstance_under_settrace� s&
�
�
z'MockTest.test_isinstance_under_settraceN)�r r r! r5 rG rI rM rQ r[ rd rv rw 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- r4 r5 r7 r9 r: r< r= r> r? r@ rE rJ rO rT rW rX rY ra rf ri rl rq rr rt ru rw rx 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 r3 + s )&
8N
#
,((
5
r3 �__main__) r� r� r� r� r Z unittest.test.testmock.supportr r r� r r r r r r
r r r
r r r rb r r# r2 r r3 r �mainr r r r �<module> s: 8