Your IP : 216.73.216.213
o
0�i�) � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZ d dlm Z d dl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZ G dd� de�ZG dd � d e�ZG d
d � d �Zd d
� Z G dd� dej �Z e dkrre� � dS dS )� N)� ALWAYS_EQ)� is_instance��mock)
�call�DEFAULT�patch�sentinel� MagicMock�Mock�NonCallableMock�NonCallableMagicMock� AsyncMock�_Call� _CallList�create_autospec�InvalidSpecErrorc @ s( e Zd Zdd� Zdd� Zdd� ZeZdS )�Iterc C s t g d��| _d S )N��this�isZan�iter)r �thing��self� r �F/opt/alt/python310/lib64/python3.10/unittest/test/testmock/testmock.py�__init__ � z
Iter.__init__c C s | S �Nr r r r r �__iter__ � z
Iter.__iter__c C s
t | j�S r )�nextr r r r r r" �
z Iter.nextN)�__name__�
__module__� __qualname__r r r" �__next__r r r r r s
r c @ s2 e Zd Zddd�Zeddd��Zeddd��ZdS ) � SomethingNc C � d S r r �r �a�b�c�dr r r �meth � zSomething.methc C r) r r )�clsr+ r, r- r. r r r �cmeth" r! zSomething.cmethc C r) r r �r+ r, r- r. r r r �smeth% r! zSomething.smethr )r$ r% r& r/ � classmethodr2 � staticmethodr4 r r r r r( s
r( c @ s e Zd ZdZdZdZdS )�TyposN)r$ r% r& � autospect� auto_spec�set_specr r r r r7 ) s r7 c C r) r r �r+ r r r � something/ r0 r< 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��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��dS ( �MockTestc C s t d� d S )Nzfrom unittest.mock import *)�execr r r r �test_all4 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_constructor; s:
��
�
�
�
���
�zMockTest.test_constructorc C s t d d�}| �|jd� d S )N�rF z(return value in constructor not honoured)r rL rF rP r r r � test_return_value_in_constructorW s
�z)MockTest.test_return_value_in_constructorc C s* dd� }t |�}d|j_| �|� d� d S )Nc S r) r r r r r r �f^ r0 z9MockTest.test_change_return_value_via_delegate.<locals>.f� )r r rF rC �r rT r r r r �%test_change_return_value_via_delegate] s z.MockTest.test_change_return_value_via_delegatec C sP dd� }t |�}t� |j_| �t�� |� W d � d S 1 s!w Y d S )Nc S r) r r r r r r rT e r0 z8MockTest.test_change_side_effect_via_delegate.<locals>.f)r � TypeErrorr � side_effect� assertRaisesrV r r r �$test_change_side_effect_via_delegated s
"�z-MockTest.test_change_side_effect_via_delegatec C s� t dd�}| �dt|�� | �dt|� t|�� t � dft dd�dfg}|D ]P\}}| �d| t|j�� | �d| t|�� �� | �d| t|�� j�� | �d | t|� �� | �d
| t|� � �� | �d | t|� � jj�� j�� q'd 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�idr_ r\ Zbing�baz)r r �mocksr^ r r r � test_reprl 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 @ � e Zd ZdS )z'MockTest.test_repr_with_spec.<locals>.XN�r$ r% r& r r r r �X} � rh ��specz
spec='X' ��spec_setz spec_set='X' r\ )rk r^ z name='foo' r] rk )�objectr r` ra rJ �r rh r r r r �test_repr_with_spec| s$
z MockTest.test_repr_with_specc s� t � }dd� }||_| jt|dddd� |jdddd� g d�� � fdd�}||_| �|� |� |� gg d �d
� t tjd �}| �|jtjd � d
d� }t |tjd�}| �|� tj� d S )Nc _ � t d��)NZkablooie)� SystemError)�args�kwargsr r r �effect� � z)MockTest.test_side_effect.<locals>.effectrU � � ��fish�rU rw rx c � � � � S r ��popr �Zresultsr r ru � rv �rx rw rU z side effect not used correctly�rY z#side effect in constructor not usedc S � t S r �r r r r r rY � r! z.MockTest.test_side_effect.<locals>.side_effect)rY rF ) r rY rZ rr �assert_called_withrC r �
SideEffect�RETURN)r r ru rY r r r �test_side_effect� s$ � �zMockTest.test_side_effectc s� g d�� � fdd�}dd� }t |�}g d�|_| �|� |� |� gg d�d� g d�� t |�}||_| �|� |� |� gg d�d� d S ) Nr{ c r| r r} r r r r ru � rv z2MockTest.test_autospec_side_effect.<locals>.effectc S r) r r r r r r rT � r0 z-MockTest.test_autospec_side_effect.<locals>.fz1side effect not used correctly in create_autospecr� z'callable side effect not used correctly)r rY rC )r ru rT r r r r �test_autospec_side_effect� s
��z"MockTest.test_autospec_side_effectc C s, dd� }t |�}td�|_| �td|� d S )Nc S r) r r r r r r rT � r0 z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r �
ValueErrorrY �assertRaisesRegexrV r r r �#test_autospec_side_effect_exception� s
z,MockTest.test_autospec_side_effect_exceptionc C s� G dd� dt �}tj� |d��� | �td��
t|�j W d � n1 s&w Y | �td�� tjj |ddd��� W d � n1 sGw Y | �td�� tjj |jd dd��� W d � n1 siw Y | �td
�� tjj |d|jd ��� W d � n1 s�w Y | �td �� tjj |d|jd
��� W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )Nc @ s e Zd ZG dd� de�ZdS )z&MockTest.test_autospec_mock.<locals>.Ac @ s e Zd ZdZdS )z(MockTest.test_autospec_mock.<locals>.A.BN)r$ r% r& �Cr r r r �B� � r� N)r$ r% r& rn r� r r r r �A� s r� r� z8Cannot autospec attr 'B' from target <MagicMock spec='A'z(Cannot autospec attr 'B' from target 'A'T�Zautospecz-Cannot autospec attr 'C' as the patch target r� z!Cannot spec attr 'B' as the spec rj z%Cannot spec attr 'B' as the spec_set rl )rn r r r� r r r� �start)r r� r r r �test_autospec_mock� s< � �� �� �� �� ��
"�zMockTest.test_autospec_mockc 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 )Nr< �child)r^ �parentrk �r< 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
SomethingElser< r� rY rF �
reset_mockrC �
_mock_namerM rN rA rB rD rG rH rI �
mock_callsrO )r r� rk r r< rF r r r �test_reset_mock� sV
�
�
�
�� �
��
�zMockTest.test_reset_mockc C s t � }||_|�� d S r )r rF r� rP 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 setrU zcall_count incorrectzcall_args not setr@ )�keyzincorrect return valuerw r� zcall_args_list not set)r rE r rF rC r� r �ArgrB rD rG rs rt rH 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� ZKwargrC rH rG rs rt rA rP r r r �test_call_args_comparison@ s"
��z"MockTest.test_call_args_comparisonc C s2 t �� }| �||k� | �||k� | �|t �� k� | �|t �� k� | �|t jk� | �|t jk� | �t j|k� | �t j|k� | �|tk� | �|tk� t �t �� �}t �t j�}| �||k� | �||k� | �||k� | �||k� | �|tk� | �|tk� | �|dk� | �|dk� d S )NrU )r r
rE rA �ANYr r )r ZmmZcall1Zcall2r r r �test_calls_equal_with_anyV s* z"MockTest.test_calls_equal_with_anyc C sd t � }|� |�� | �t|jd� |�� | �t|j� |dddddd� |jdddddd� d S )NrU rw rx rz Znothing�r+ r, )r r� rZ �AssertionErrorr� rP r r r �test_assert_called_withr s z MockTest.test_assert_called_withc C s t � }|t � � |�tj� d S r )r
r� 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 d � n1 sAw Y | �|jjt� d S ) Nc S r) r r r3 r r r rT � r0 z9MockTest.test_assert_called_with_function_spec.<locals>.frj rU rw rx �r, r- �r+ r, r- � ��er )r r� rZ r� �assertIsInstance� exception� __cause__rX �r rT r �cmr r r �%test_assert_called_with_function_spec� 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 )NrU rw rx r� r� )r� rZ r� r r r r �_check� s
�z<MockTest.test_assert_called_with_method_spec.<locals>._checkrj )r r( r/ r2 r4 )r r� r r r r �#test_assert_called_with_method_spec� s z,MockTest.test_assert_called_with_method_specc C s� d}| � t|�d���
t� �� W d � n1 sw Y | � t|�d��� tdd��� W d � d S 1 s<w Y d S )Nz"Expected '{0}' to have been calledr Z test_namer] )r� 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 )Nr\ r_ rw �rc c s � j dddd�S )N�bobr_ rw r� ��assert_called_once_withr r r r �<lambda>� � z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r r� rZ 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 )NrU rw zCalls: [call(1), call(2)]c s
� � d�S )Nrw r� r �r� r r r� � �
zAMockTest.test_assert_called_once_with_call_list.<locals>.<lambda>)r r� 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 d � n1 sAw Y | �|jjt� |d d d
� | �t|jddd� | �t|jd d d
� d S )Nc S r) r r r3 r r r rT � r0 z>MockTest.test_assert_called_once_with_function_spec.<locals>.frj rU rw rx r� r� r� r� � � � r )r r� rZ r� r� r� r� rX r� r r r �*test_assert_called_once_with_function_spec� s&
� �
�
�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 namerx zmethod returned wrong valuez!method didn't record being called)r r< rE r rC rF rB �r r r< 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 )Nr< z attribute name not set correctlyz"attribute parent not set correctly)r r< rC r� rM 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 ) Nrx ry r� )�caker< )r� r� z#method calls not recorded correctly)rx rz zsomething_else.something)r r< �something_elser ZCakerC rI rP 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� r< r � different)r< �r� r� rU ��xr� Zmore)�somer� )r r< rC rI rP r r r � test_method_calls_compare_easily s&
�
�z)MockTest.test_method_calls_compare_easilyc C sH dgdfD ] }dD ]}t di ||i��}|j | �tdt|d� q qd S )Nr< r� )rk rm z-Mock object has no attribute 'something_else'r� r )r r< r� �AttributeError�getattr)r rk �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>.Somethingrx Nc S r) r r r r r r �y< r0 z,MockTest.test_from_spec.<locals>.Something.y)r$ r% r& r� �
__something__r� r r r r r( 9 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� r� r� r� r r r r �test_attributes> s ��z0MockTest.test_from_spec.<locals>.test_attributesrj )rn r )r r( r� r r r �test_from_spec8 s zMockTest.test_from_specc C sJ t � }t |d�}| �|� |� � |�� |dddd� |jdddd� d S )N��wrapsrU rw rx ry )r rC r� r� �r �realr r r r �test_wraps_callsR s
zMockTest.test_wraps_callsc s8 G dd� dt �}|� }t|d�� | �t� fdd�� d S )Nc @ rf )zFMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.RealNrg r r r r �Real_ ri r� r� c r| r )Znew_attrr r r r r� e � zJMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.<lambda>)rn r rZ 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� rx )r rF rC rA rB r� r r r �,test_wraps_call_with_nondefault_return_valueh 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� s s
r� r� c � � j S r ry r r r r r� z � z0MockTest.test_wraps_attributes.<locals>.<lambda>rU rw rx ry ) rn r rC r� rZ r� �assertNotEqualZfrogr� )r r� r� r� r r r �test_wraps_attributesr s
z MockTest.test_wraps_attributesc C sd G dd� dt �}|� }t|d�}tjtg|j_| �|�� tj� | �|�� tj� | � t
|j� d S )Nc @ � e Zd Zdd� ZdS )z[MockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Realc S � t jS r )r �ORIGINAL_VALUEr r r r �method� � zbMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Real.methodN�r$ r% r& r� r r r r r� � � r� r� ) rn r r �VALUE1r r� rY rC r� rZ �
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 @ r� )zNMockTest.test_customize_wrapped_object_with_side_effect_iterable.<locals>.Realc S r) r r r r r r r� � r0 zUMockTest.test_customize_wrapped_object_with_side_effect_iterable.<locals>.Real.methodNr� r r r r r� � r� r� r� )
rn r r r� �VALUE2r� rY rC rZ r� r� r r r �7test_customize_wrapped_object_with_side_effect_iterable� 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 @ r� )zOMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Realc S r) r r r r r r r� � r0 zVMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Real.methodNr� r r r r r� � r� r� r� )rn r � RuntimeErrorr� rY rZ r� r r r �8test_customize_wrapped_object_with_side_effect_exception� s
zAMockTest.test_customize_wrapped_object_with_side_effect_exceptionc C sF G dd� dt �}dd� }|� }t|d�}||j_| �|�� tj� d S )Nc @ r� )zNMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Realc S r) r r r r r r r� � r0 zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Real.methodNr� r r r r r� � r� r� c S r� r )r �VALUEr r r r rY � r� zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.side_effectr� )rn r r� rY rC r r� )r r� rY 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 @ r� )zFMockTest.test_customize_wrapped_object_with_return_value.<locals>.Realc S r) r r r r r r r� � r0 zMMockTest.test_customize_wrapped_object_with_return_value.<locals>.Real.methodNr� r r r r r� � r� r� r� )rn r r r� r� rF rC 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 @ r� )zVMockTest.test_customize_wrapped_object_with_return_value_and_side_effect.<locals>.Realc S r) r r r r r r r� � r0 z]MockTest.test_customize_wrapped_object_with_return_value_and_side_effect.<locals>.Real.methodNr� r r r r r� � r� r� r� ) rn r r r� r� r� rY Z WRONG_VALUErF rC rZ 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 @ r� )zWMockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.Realc S r) r r r r r r r� � r0 z^MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.Real.methodNr� r r r r r� � r� r� r� c S r� r r� r r r r r� � r0 z[MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.<lambda>)rn r r� rY r r� rF rC 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 @ r� )z^MockTest.test_customize_wrapped_object_with_return_value_and_side_effect_default.<locals>.Realc S r) r r r r r r r� � r0 zeMockTest.test_customize_wrapped_object_with_return_value_and_side_effect_default.<locals>.Real.methodNr� r r r r r� � r� r� r� ) rn r r r� r r� rY r� rF rC rZ 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 s� ddi}t |d�}| �|�d�d� | �|d t � | �d|v � d|j_| �|�d�d� tj|j_| �|�d�d� | �|�d�d � | �|d t � | �d|v � d|d< | �|�d�d� | �|d t � | �d|v � |d= | �|�d�d � d S )Nr\ r_ r� rF rc �spam)r
rC �getr� rA rF r r )r �dataZ wrapped_dictr r r � test_magic_method_wraps_dict� s$
z%MockTest.test_magic_method_wraps_dictc C sT G dd� d�}t |d�}|� }| �|�d�d� | �|d d� | �|�� d� d S )Nc @ � e Zd Zdd� Zdd� ZdS )z3MockTest.test_magic_method_wraps_class.<locals>.Fooc S s |S r r )r �indexr r r � __getitem__ r! z?MockTest.test_magic_method_wraps_class.<locals>.Foo.__getitem__c S � dS �Nr\ r r r r r �__custom_method__ r! zEMockTest.test_magic_method_wraps_class.<locals>.Foo.__custom_method__N)r$ r% r& r r r r r r �Foo
s r r� rw r\ )r
rC r r )r r �klass�objr r r � test_magic_method_wraps_class s
z&MockTest.test_magic_method_wraps_classc C �4 t td�}| �t|� t td�d�}| �t|� d S �Nr� r\ )r r� rZ rP r r r � test_exceptional_side_effect �
z%MockTest.test_exceptional_side_effectc C r r )r �KeyboardInterruptrZ rP r r r � test_baseexceptional_side_effect% r z)MockTest.test_baseexceptional_side_effectc C s t � }| �td|j� d S )Nz
not called)r r� r� r� rP r r r � test_assert_called_with_message- s �z(MockTest.test_assert_called_with_messagec C s t dd�}| �td|j� d S )NZgeoffreyr] z'Expected 'geoffrey' to be called once\.)r r� r� r� rP r r r �$test_assert_called_once_with_message3 s
�z-MockTest.test_assert_called_once_with_messagec s2 t � � | �t� fdd�� d� _| �� jd� d S )Nc r� r )r$ r r r r r� < r� z'MockTest.test__name__.<locals>.<lambda>r\ )r rZ r� r$ rC 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 @ rf )z-MockTest.test_spec_list_subclass.<locals>.SubNrg r r r r �SubC ri r r\ rj rx )�listr �appendr� rZ r� r� )r r r r r r �test_spec_list_subclassB 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 @ rf )z#MockTest.test_spec_class.<locals>.XNrg r r r r rh M ri rh rj r rl )rn r r� �assertIs� __class__rC r$ ro r r r �test_spec_classL s
zMockTest.test_spec_classc C s� G dd� d�}t |d�}| �||� t |� d�}| �||� | �|j|� | �t � jjd� t |d�}| �||� t |� d�}| �||� d S )Nc @ rf )z2MockTest.test_spec_class_no_object_base.<locals>.XNrg r r r r rh a ri rh rj r rl )r r� r r! rC r$ ro 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 @ � e Zd ZdZdS )z8MockTest.test_setting_attribute_with_spec_set.<locals>.Xrx N)r$ r% r& r� r r r r rh u r� rh rj r\ rl c �
d� _ d S r
r� r r r r �set_attr| r# z?MockTest.test_setting_attribute_with_spec_set.<locals>.set_attr)rn r r� r� rZ r� )r rh r& r r r �$test_setting_attribute_with_spec_sett 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 Zcurrentr 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 r �Nrx r r r r r �_get� r! z=MockTest.test_subclass_with_properties.<locals>.SubClass._getc S rq )Nz
strange error)� NameError)r �valuer r r �_set� rv z=MockTest.test_subclass_with_properties.<locals>.SubClass._setN)r$ r% r& r0 r3 �property�some_attributer r r r �SubClass� s r6 rl rx c r% r/ )r5 r ��sr r �test� r# z4MockTest.test_subclass_with_properties.<locals>.testc r% )Nr_ �r\ r r7 r r r9 � r# )r rC r5 rZ r1 r� )r r6 r9 r r7 r � test_subclass_with_properties� 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__� s
z,MockTest.test_setting_call.<locals>.__call__�one�two)r �typer< r� rZ rX )r r r< r r r �test_setting_call� 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>� s z%MockTest.test_dir.<locals>.<listcomp>r+ r, r- r. c S s t g �S r )r r7 r r r r� � r� z#MockTest.test_dir.<locals>.<lambda>r )
r �set�dirrC r+ r, r` r- r. r )r r �attrs�
type_attrsr r r �test_dir� s
zMockTest.test_dirc C sX t tjd�}tttj��}tt|��}| �t� || � d|_| �t|��d�d� d S )Nrj rx �versionrU )r �unittest�TestCaserE rF rC rJ �count)r r Ztestcase_attrsrG r r r �test_dir_from_spec� s zMockTest.test_dir_from_specc C s\ t �tdd�}|�� z ttt� ��}ttt��}| �t� || � W |�� d S |�� w )NZ
FILTER_DIRF) r rn r r� rE rF r rC �stop)r ZpatcherrG rH r r r �test_filter_dir� s zMockTest.test_filter_dirc C s6 t � }d|j_| �dt|�� |`| �dt|�� d S )NrU r� )r r� rF r` rF rJ rP 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 di |��}| �t|� | �|j�� d� | �|jt� t � }|jdi |�� | �t|� | �|j�� d� | �|jt� d S )Nr_ r: �! )rY zfoo.bar.return_valuer\ r ) r rC r\ r
�KeyErrorrZ r_ r� Zconfigure_mock)r r rt r r r �test_configure_mock� s
� z MockTest.test_configure_mockc O sR | � |��}||i |�� W d � n1 sw Y t|j�}| �||� d S r )rZ �strr� rC )r r� �message�funcrs rt �contextr� r r r �assertRaisesWithMsg� s
�
z MockTest.assertRaisesWithMsgc
C s t � }d}d}d}| jt|||f |jddddd� |jddddd � |jj|jjg}|D ]}d
}d }d}| jt|||f |ddddd� q-|D ]}d
}d }d}| jt|||f |dd� qH|D ]}d
}d
}d}| �t|||f |ddd� q`|D ]}d
}d}d}| �t|||f |� qyd S )Nz not called.zmock(1, '2', 3, bar='foo')z0expected call not found.
Expected: %s
Actual: %srU �2rx r\ )r_ r: zfoo(1, '2', 3, foo='foo')zfoo(1, '2', 3, bar='foo')zfoo(bar='foo')z foo(1, 2, 3)rw zfoo())r rY r� r� r\ r� )r r Zactual�expectedrV Z assertersr/ r r r �'test_assert_called_with_failure_message sV � �
� � ���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 di |��}t |j� ddi fg}| �|j|� t di |��}|�
� � ddi fddi fg}| �|j|� | �|�
� jt� g� t di |��}|d�d�d� | �|jtd�d�d�� � � | �|� jtd�d�� � � | �|� � jtd�� � � t di |��}|d�d�d�j
j�d� | �|jtd�d�d�j
j�d�� � � | �|� jtd�d�j
j�d�� � � | �|� � jtd�j
j�d�� � � t di |��}t |� j� � �
� � ddi f}| �|jd |� | �|� jtj� � �
� �� � � � | �|� j� � jt�
� �� � � � | �|� j� � j
jt� �� � � � q�d S )NT� r rU rw rx r� r� r� z().foor{ r\ z
().foo.barz().foo.bar().bazr_ r] z foo.__int__r+ za()z().foo.bar().baz().__int__���)r
r r� rC r\ r r �dictrF r_ rc � call_listr, r+ r, r- �__int__)r r r[ rt 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 ZwibblerC r� r )r r� r� r r r �test_child_mock_call_equal� s
z#MockTest.test_child_mock_call_equalc C sF t � }|�� �� | �|jd t�� �� � | �|jd t�� � d S �NrU r )r r\ r< r� r� r r� rC r� r r r � test_mock_call_not_equal_leaf� � z&MockTest.test_mock_call_not_equal_leafc C sF t � }|�� �� | �|jd t�� �� � | �|jd t�� � d S rd �r r\ r_ r� r� r rc r� r r r �!test_mock_call_not_equal_non_leaf� rf z*MockTest.test_mock_call_not_equal_non_leafc C s8 t � }|jdd��� | �|jd tjdd��� � d S )NrU r� rw )r r\ r_ rC r� r r� r r r �2test_mock_call_not_equal_non_leaf_params_different� s "z;MockTest.test_mock_call_not_equal_non_leaf_params_differentc C s, t � }|j�� | �|jd tj�� � d S �Nr rg r� r r r �&test_mock_call_not_equal_non_leaf_attr� s
z/MockTest.test_mock_call_not_equal_non_leaf_attrc C s. t � }|j�� | �|jd t�� �� � d S rj )r r\ r_ r� r� r r� r r r �2test_mock_call_not_equal_non_leaf_call_versus_attr� 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()rU zcall.foo().bar()rw zcall.foo().bar().baz.bob())r r\ r_ rc r� rC ra 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 r\ ra Z assertRegexr� r r r �test_mock_call_repr_loop� s z!MockTest.test_mock_call_repr_loopc C s t � }| �t� g|jv � d S r )r rA r r� r� r r r �test_mock_calls_contains� s z!MockTest.test_mock_calls_containsc C sh G dd� dt �}|� }| �|j|� | �|� |� G dd� dt �}|� }| �|j|� | �|� |� d S )Nc @ rf )�+MockTest.test_subclassing.<locals>.SubclassNrg r r r r �Subclass� ri rq c @ r� )rp c [ s t di |��S )Nr )r )r rt r r r �_get_child_mock� s z;MockTest.test_subclassing.<locals>.Subclass._get_child_mockN)r$ r% r& rr r r r r rq � r� )r r� r\ �assertNotIsInstance)r rq r r r r �test_subclassing� s zMockTest.test_subclassingc s� t � t� t� t� g}� fdd�}|D ]=}||� t|�r1|� |dd� |dd� |�� ||� |�� |jjddd� |�d��� �d� |�� ||� qd S )Nc s@ d}|D ]}t | |�}� �|t� � �|t� � �|g � qd S )N)rH rI r� )r� r� r r rC )r �namesr^ �attrr r r � assert_attrs� s
�z-MockTest.test_arg_lists.<locals>.assert_attrsrU rw rx r; ) r r
r r
�callabler� r\ r_ rc )r rd rw 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 )
NrU rx r; rw r� �r, )rw �rU r ) r rC �lenrG rs rt r_ �ziprH )r r Z
expected_listr[ rG r r r �test_call_args_two_tuple� s �z!MockTest.test_call_args_two_tuplec C s t tg d��d�}| �|� |� |� gg d�� | �t|� tg d�d�}| �|� |� |� gg d�� | �t|� t dd�}| �|� |� |� gg d�� | �t|� G dd� dt�}t|d�}| �|� |� t t� d�}| �|� |� |� |� gg d�� | �t|� d S ) Nr{ r� r� Zghi)�g�h�ic @ rf )z/MockTest.test_side_effect_iterator.<locals>.FooNrg r r r r r ri r r ) r r rC rZ r� r
rn r� r )r r r r r r �test_side_effect_iterator s"
�z"MockTest.test_side_effect_iteratorc C s\ t tfD ]'}tdtdf}||d�}| �t|� | �|� d� | �t|� | �|� d� qd S )Nrx r� r� )r r
r� rS rZ rC )r �Klass�iterabler� r r r �$test_side_effect_iterator_exceptions& s
�z-MockTest.test_side_effect_iterator_exceptionsc C s� t � }tg d��|_| �|� |� |� gg d�� | �t|� |j}| �|ttg ��� g d�|_| �|� |� |� gg d�� | �t|� |j}| �|ttg ��� t� }||_| �|� |� |� |� gg d�� | �t|� | � |j|� d S )Nr{ r� r )
r r rY rC rZ r� r� r? r r )r r rY Z this_iterr r r �!test_side_effect_setting_iterator0 s$
� z*MockTest.test_side_effect_setting_iteratorc C s4 t dd�}tdtg�|_| �|� |� gddg� d S )Nrw rR rU )r r r rY rC rP r r r �!test_side_effect_iterator_defaultE 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� qXtdd�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 )NrU rw rx r; r� r� rz r+ �)rx r� r )r] �rU rw r] r� T�Z any_orderrZ r\ r� � )r r �assert_has_callsrZ r� )r r �kalls�kallZ
kall_listsZ kall_listr r r � test_assert_has_calls_any_orderJ sF
�"��*���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 ]#}|dd� |dd � |dd
� |dd� |ddd� |� ||f� qDt
� }|�� |�d� |�� jdd � |jt� i d �� dg� |� ||f� |D ]1\}}t
t|��D ]&}dD ]!}|||| � }|�|� t|�dkr�| �t|jtt|��� q�q�q�d S )NrU rw r+ rx r� r� rz r] r{ r, r; r: r� r{ )r r\ r_ �extendr rc r` ZbamrE rz r r
r �ranger| r� rZ r� r �reversed) r Zkalls1Zkalls2rd r r� r� �stepZtheser r r �test_assert_has_callsv sH
�"
�����z MockTest.test_assert_has_callsc C sD G dd� d�}t |�}||� fD ]$}|jddddd� |�tjddddd�g� |�t�dddd�g� q|�� ||� fD ]6}| �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 r) r r r r r r r � r0 zFMockTest.test_assert_has_calls_nested_spec.<locals>.Something.__init__Nc S r) r r r* r r r r/ � r0 zBMockTest.test_assert_has_calls_nested_spec.<locals>.Something.methc @ r )zAMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Fooc S r) r r r� r r r r � r0 zJMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.__init__c S r) r r )r r+ r, r r r �meth1� r0 zGMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.meth1N)r$ r% r& r r� r r r r r � s r r )r$ r% r& r r/ r r r r r r( � s
r( rU rw rx �r. r� )
r r/ r� r r� rZ r� r r� Z non_existent)r r( Z
mock_classr� Z
invalid_callsr� r r r �!test_assert_has_calls_nested_spec� s0
" ���z*MockTest.test_assert_has_calls_nested_specc C sJ t � }|� �� �� �� |�� �� �� t�� �� �� �� }|� |� d S r )
r
r\ r_ rc r= r> �threer r` r� )r r� �callsr r r �)test_assert_has_calls_nested_without_spec� 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 d � n1 s~w Y |j|dd� | �t�� |�|dd � � W d � n1 s�w Y |j|dd � dd� | �t�� |�|d d� � W d � n1 s�w Y |j|d d� dd� d S )Nc S r) r r r3 r r r rT � r0 z<MockTest.test_assert_has_calls_with_function_spec.<locals>.frj rU rw rx r� r� r� r� r� �r- r. �
� � )r- r] r{ )r� r� r� r. )r� r� r� Tr� r^ r )r r� r r� rZ r� )r rT r r� r r r �(test_assert_has_calls_with_function_spec� s8
�
� � �z1MockTest.test_assert_has_calls_with_function_specc C s� d dd�}t |d�}|d� | �td�t�d����}|�t� g� W d � n1 s,w Y | �|j j
� | �td�t�d����}|�t� tdd�g� W d � n1 sZw Y | � |j j
t � d S )
Nc S r) r r r� r r r rT � r0 zAMockTest.test_assert_has_calls_not_matching_spec_error.<locals>.frj rU 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)]rw r )
r r� r� r� r� r� r� r rL r� r� r� rX r� r r r �-test_assert_has_calls_not_matching_spec_error� s2
�� ���� � 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 )NrU rw rx r; r� rz r� )r �assert_any_callrZ r� rP 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 d � n1 sZw Y | �|jjt� d S )Nc S r) r r r3 r r r rT # r0 z;MockTest.test_assert_any_call_with_function_spec.<locals>.frj rU rw rx r� r� r� r� r� r� r� r3 r� r� r )r r� rZ r� r� r� r� rX 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 ]}|dd� |dd� | �|jtdd�tdd�g� qd S )Nc S r) r r r� r r r rT 6 r0 z3MockTest.test_mock_calls_create_autospec.<locals>.frU rw rx r� )r rT r rC r� r )r rT r ZfuncsrW r r r � test_mock_calls_create_autospec5 s �
��z(MockTest.test_mock_calls_create_autospecc C s$ t jt� dd�}| �dt|�� d S )NZ
sweet_funcr] )r r rn r` ra r� r r r � test_create_autospec_with_nameG s z'MockTest.test_create_autospec_with_namec C st G dd� d�}dD ].}| j |d�� t�t||��}|� |�� | �t|d� W d � n1 s2w Y q d S )Nc @ s$ e Zd Zedd� �Zedd� �ZdS )zMMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClassc S r) r r )r1 r r r � class_methodN r! zZMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.class_methodc S r) r r r r r r �
static_methodQ r! z[MockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.static_methodN)r$ r% r& r5 r� r6 r� r r r r � TestClassM s
r� )r� r� )r� Z extra_arg)ZsubTestr r r� r� rZ rX )r r� r� Z mock_methodr r r �1test_create_autospec_classmethod_and_staticmethodL s ���z:MockTest.test_create_autospec_classmethod_and_staticmethodc C sD t � }d}| �t|�� |�� W d � n1 sw Y | �t|�� |�� W d � n1 s4w Y | �t|�� |�� W d � n1 sNw Y | �t|�� |�� W d � n1 shw Y | �t|�� |�� W d � n1 s�w Y t dd�}|�� |�� |�� |�� |�� d S )Nz1is not a valid assertion. Use a spec for the mockT)�unsafe)r r� r� Zassert_foo_callZassret_foo_callZasert_foo_callZaseert_foo_callZassrt_foo_call)r r� r� r r r �test_mock_unsafe[ s.
�
�
�
�
�
zMockTest.test_mock_unsafec C sB G dd� dt �}t|d�}|�� |�� |j�� |j�� d S )Nc @ r )z.MockTest.test_mock_safe_with_spec.<locals>.Fooc S r) r r r r r r �
assert_barr r! z9MockTest.test_mock_safe_with_spec.<locals>.Foo.assert_barc S r) r r r r r r �
assertSomeu r! z9MockTest.test_mock_safe_with_spec.<locals>.Foo.assertSomeN)r$ r% r& r� r� r r r r r q s r rj )rn r r� r� �assert_called_once�r r r� r r r �test_mock_safe_with_specp s
z!MockTest.test_mock_safe_with_specc C sR t � }|j�� |�� | �t�� |j�� W d � d S 1 s"w Y d S r )r �hello�assert_not_calledrZ 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 )NrU rw zCalls: [call(1, 2)])r r� r� r� r� r� r� r r r � test_assert_not_called_message� s
�z'MockTest.test_assert_not_called_messagec C sb t � }| �t��
|j�� W d � n1 sw Y |�� |j�� |�� |j�� d S r )r rZ r� r� r� r� r r r �test_assert_called� s �
zMockTest.test_assert_calledc C s� t � }| �t��
|j�� W d � n1 sw Y |�� |j�� |�� | �t�� |j�� W d � d S 1 s@w Y d S r )r rZ r� r� r� r� r r r �test_assert_called_once� s �
"�z MockTest.test_assert_called_oncec C s2 t � }|dd� |d� | �tt�d�|j� d S )NrU rw rx z Calls: [call(1, 2), call(3)])r r� r� r� r� r� r� r r r � test_assert_called_once_message� s
�z(MockTest.test_assert_called_once_messagec C sN t � }| �t�� }|�� W d � n1 sw Y | �dt|j�� d S )NzCalls:)r rZ r� r� rJ rU 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)r^ Zdaddyz call(name='hello', daddy='hero'))r r� rC ra rG )r r� �textr r r �test_ordered_call_signature� s z$MockTest.test_ordered_call_signaturec C sV t �� }t �dd�}t� }|�� |�dd� | �|jd |� | �|jd |� d S )N� r� r rU )r rM r
r rC rI )r r- r� r� r r r �test_override_tuple_methods� s z$MockTest.test_override_tuple_methodsc C s> t dddgd�}|jddd� | �|jt � | �|jd � d S )Nr� rw rx �rF rY T)r r� r� rF rC rY r� r r r � test_reset_return_sideeffect� s z%MockTest.test_reset_return_sideeffectc C s< t dddgd�}|jdd� | �|jt � | �|jd � d S )Nr� rw rx r� TrR )r r� r� rF r� rY r� r r r �test_reset_return� � zMockTest.test_reset_returnc C s< t dddgd�}|jdd� | �|jd� | �|jd � d S )Nr� rw rx r� Tr� )r r� rC rF rY r� r r r �test_reset_sideeffect� r� z MockTest.test_reset_sideeffectc C s@ t t dd�d�}| �|�� d� |jdd� | �|�� d� d S )NrU rR �rT T)r
rC rT r� r� r� r r r � test_reset_return_with_children� s z(MockTest.test_reset_return_with_childrenc C sD t t ddgd�d�}| �|jjd � |jdd� | �|jjd � d S )Nrw rx r� r� T)r
r� rT rY r� rC r� r r r �+test_reset_return_with_children_side_effect� s z4MockTest.test_reset_return_with_children_side_effectc s� G dd� dt �}G dd� dt �}G dd� dt �� ttttg}t|�D ]}|�|f� fdd� � |�|f� fd d� � q"|D ]�}t� td
d �fD ]y}|� }|j|j |j
f ||fd gd
gffD ]a\}}t� td
d �fD ]T}|j |fi |�� |j | � t
t|d
� | � t
t|d� d|v r�| � t
t|dd � |j |fi |�� | � t
t|d � |j | � t
t|d� d|v r�| � t
t|dd � qjq^qGq=d S )Nc @ r$ )z)MockTest.test_mock_add_spec.<locals>._OnerU N�r$ r% r& r= r r r r �_One� r� r� c @ r$ )z)MockTest.test_mock_add_spec.<locals>._Tworw N)r$ r% r& r> r r r r �_Two� r� r� c @ s e Zd Zd Z ZZdS )z-MockTest.test_mock_add_spec.<locals>.AnythingZfourN)r$ r% r& r= r> r� r r r r �Anything� s r� c �
| � d�S )Nrj r ��K�r� r r r� � r� z-MockTest.test_mock_add_spec.<locals>.<lambda>c r� )Nrl r r� r� r r r� � r� Trl r= r> r� rm )rn r r
r r
r r r_ r= r> r� �
mock_add_specrZ r� r� �setattr) r r� r� Zklassesr� rt r ZOneZTwor r� r �test_mock_add_spec� sV � ��
���
������zMockTest.test_mock_add_specc s| t tfD ]7}|� � t� � � �t� | �tt� � |� � � d d� j_� �t� | � t� �d� | �t� fdd�� qd S )Nr\ 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� rn rZ rX ra rF rC )r r� r r r � test_mock_add_spec_magic_methods s
�z)MockTest.test_mock_add_spec_magic_methodsc C s� t ttttfD ]V}|� }t� |_|�� | �|jt�� g� | �|j t�� g� |� }tdd�|_
|�
� | �|jg � | �|j g � |� }t� � |_ |� � | �|jg � | �|j g � qd S )Nr^ r] ) r r r
r
r r\ rC rI r r� r_ rc �r r� r r r r �test_adding_child_mock& s& �
�z MockTest.test_adding_child_mockc C s@ t tfD ]}|� }t� |_|� � | �|jt� t� � g� qd S r )r r
rF rC 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 r I s r r= r> ) rn r r r� r* rO � attach_mockr> r= rC r� r )r r �managerZp1Zp2Zmock_oneZmock_twor r r �test_manager_mockH s zMockTest.test_manager_mockc C sh t tfD ]-}|� }t dd�|_tdd�|_t|� t|� | �|jt�� t�� g� | �|j g � qd S )Nrx rR g @)
r r
ra � __float__r, �floatrC r� r rI )r r� 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 )Nr �Z read_data�a-namez another-name)r r� �readrC )r � mocked_open�f1Zf1_data�f2Zf2_datar r r � test_mock_open_reuse_issue_21750i 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 � g | ]}|�qS r r �rC �liner r r rD v r� z=MockTest.test_mock_open_dunder_iter_issue.<locals>.<listcomp>r z Remarkable
rU zNorwegian Blue)r r� rC r )r r� r� �linesr r r � test_mock_open_dunder_iter_issueq 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 d � d S 1 sJw Y d S )
Nz1st line
2nd line
3rd liner� r� c S r� r r r� r r r rD � r� z6MockTest.test_mock_open_using_next.<locals>.<listcomp>z 1st line
z 2nd line
r z3rd line)r r� r" r' rC r rZ r� )r r� r� Zline1Zline2r� r r r �test_mock_open_using_next{ s
"�z"MockTest.test_mock_open_using_nextc C s, t jdd�}d|jj_| �dt|� �� d S )N�foo
barnr� �abc)r r� rF �readlinerC r" )r �mopenr r r �3test_mock_open_next_with_readline_with_return_value� s
z<MockTest.test_mock_open_next_with_readline_with_return_valuec C sn t �t jdd��}t �d|�� |j}|j}td�|_dd� }| �t|� W d � d S 1 s0w Y d S )NZJLVr] ztempfile.NamedTemporaryFilez Test 2 Errorc S s t �� �d� d S )NZasd)�tempfileZNamedTemporaryFile�writer r r r �attempt� r z.MockTest.test_mock_open_write.<locals>.attempt) r r� r
r rF r� �OSErrorrY rZ )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 )Nr� r� c W r )Nr� r )rs r r r r� � r0 z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>r� )r r� rF r� rY rC )r r� �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 )Nr\ r� r_ r] )r r� r� rC r� � readlines)r Z_openr� 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
ra rF r rC r� r+ r, r` )r r� r� Z
original_reprr r r �test_mock_parents� s. �zMockTest.test_mock_parentsc C s� t tttf}|D ]B}|D ]=}|� }|dd�}|�|d� | �|j|� | �dt|�� |j� d� | �
|j t j� d�g� | �
|j
t j� d�g� q qd S )Nr\ r] r_ zname='mock.bar'rU )r r
r
r r� r r_ r` ra rc rC r� r rI �r Zclassesr� ZKlass2r� Zm2r r r �test_attach_mock� s
��zMockTest.test_attach_mockc C s� t tttf}t tfD ]4}|D ]/}|� }|dd�}|�|d� | �|� |� | �dt|�� |�� | � |j
t � �� � � � qq
d S )Nr\ r] rF z
name='mock()')
r r
r
r r� r r` ra r\ rC r� r r` r� r r r � test_attach_mock_return_value� s
��z&MockTest.test_attach_mock_return_valuec C s
t � }tjt� d�dd��n}| �|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 d � d S 1 s~w Y d S )
N�
.somethingTr� r< r� rU rw rx �
mock.child)
r r r r$ rC Z_extract_mock_namer� r� r< r r� r` ra )r r� Z mock_funcZ parent_callsZ child_callsr r r � test_attach_mock_patch_autospec� s
"�z(MockTest.test_attach_mock_patch_autospecc
C s� t jt� d�dd��I}t� }|�|d� t� }|jddddd � |�tj t j
ddddd �g� |j�tt j
ddddd �g� |�tt j
ddddd �g� W d � n1 sVw Y t jt� d
�dd��.}t� }|�|d � t d� |�t� d�g� t �td�g� |�td�g� W d � n1 s�w Y t jt� d �dd��M}t� }|�|d
� t� }|jddddd � |�t�
� t�
� jddddd �g� |j�tddddd �g� |�t� t� jddddd �g� W d � d S 1 s�w Y d S )N�.Something.methTr� � attach_methrU rw rx r� r� r� � attach_func�
.Something�
attach_obj)r r r$ r r� r( r/ r� r r� r� r< r� r )r Zmockedr� r r r r �)test_attach_mock_patch_autospec_signature� s8 � �
�""�z2MockTest.test_attach_mock_patch_autospec_signaturec C sf t � t� t� t� fD ]&}| �t|d�� |`| �t|d�� |`| �t|d�� | � t
t |d� q
d S )Nr� rT ) r r
r
r rE �hasattrr� rA rT rZ r� r� rP r r r �test_attribute_deletion s ��z MockTest.test_attribute_deletionc C s� t � t� t� t� fD ]:}d|_| �t|d�� | �|jd� |`| �t|d�� d|_| �t|d�� | �|jd� |`| �t|d�� q
d S )Nrx r\ r� ) r r
r
r r\ rE r rC rA rP 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 sP t � t� t� t� fD ]}|`| �t��
|`W d � n1 s w Y q
d S r )r r
r
r r\ rZ r� rP 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� rA r rP r r r �/test_reset_mock_does_not_raise_on_attr_deletionC s
z8MockTest.test_reset_mock_does_not_raise_on_attr_deletionc C s: t � t� fD ]}| �|t� t|_| �|t� |j qd S r )r r
rs r, r! r� r\ rP r r r �test_class_assignableM s �z MockTest.test_class_assignablec C �8 | � tj� | �ttj�t� | �tt�� j�t� d S r )�assertIsNotNoner r^ rC r? r r r r r �test_name_attribute_of_callU � z$MockTest.test_name_attribute_of_callc C r r )r
r r� rC r? r r r r r � test_parent_attribute_of_call\ r z&MockTest.test_parent_attribute_of_callc C sd dd� }t � }t|�|_|�dd� | �t|jd� | �|jt�dd�g� | �dt |jj
�� d S )Nc S r) r r r� r r r r\ f r0 zBMockTest.test_parent_propagation_with_create_autospec.<locals>.foorU rw r� ) r r r� rZ rX rC r� r r` ra r )r r\ r r r r �,test_parent_propagation_with_create_autospecd 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 r) r r r� r r r r\ r r0 zGMockTest.test_parent_propagation_with_autospec_attach_mock.<locals>.foor_ r] r� rU rw r� ) r r� r r� rZ rX rC r� r r` ra r )r r\ r� r r r �1test_parent_propagation_with_autospec_attach_mockp s z:MockTest.test_parent_propagation_with_autospec_attach_mockc s� t jj}| �dd� |� t�d��D tjd= � fdd�� | �tjt�� � t�� � ddl m
}m }m }m
} ||||tg}|D ]
}|td �}| �|t� q?W d � d S 1 sXw Y d S )
Nc S s t tjd| �S )Nr )r� rK r )r r r r r� � s z9MockTest.test_isinstance_under_settrace.<locals>.<lambda>z sys.modulesz
unittest.mockc s � S r r )�frameZeventr� ��tracer r r � r! z6MockTest.test_isinstance_under_settrace.<locals>.tracer )r r
r r
rj )rK r r r* r_ r( �modules�settrace�gettrace�
unittest.mockr r
r r
r r( r� ) r Z old_patchr r
r r
rd r r r r r � test_isinstance_under_settrace} s"
�
�
�"�z'MockTest.test_isinstance_under_settracec C s^ G dd� d�}|� }t jjj|dtd�� W d � n1 s w Y | �|jjjd� d S )Nc @ r� )zFMockTest.test_bool_not_called_when_passing_spec_arg.<locals>.Somethingc S s t j�� | _d S r )rK r r
�obj_with_bool_funcr r r r r � s zOMockTest.test_bool_not_called_when_passing_spec_arg.<locals>.Something.__init__N)r$ r% r& r r r r r r( � r� r( r rj r )rK r r rn rC r �__bool__rD )r r( r r r r �*test_bool_not_called_when_passing_spec_arg� s 4z3MockTest.test_bool_not_called_when_passing_spec_argc C s$ G dd� d�}| � t��! tt� d�dd�� W d � n1 s!w Y W d � n1 s0w Y | � t�� tj|ddd�� W d � n1 sNw Y W d � n1 s]w Y | � t��! tt� d�dd�� W d � n1 s|w Y W d � n1 s�w Y | � t�� tj|ddd�� W d � n1 s�w Y W d � n1 s�w Y | � t��! tt� d�dd�� W d � n1 s�w Y W d � n1 s�w Y | � t��! tj|ddd�� W d � n 1 �sw Y W d � n 1 �sw Y | � t�� t|dd�}W d � n 1 �s1w Y | � t��$ tjt� d �tdd
�� W d � n 1 �sSw Y W d � n 1 �scw Y | � t��$ tjt� d �tdd �� W d � n 1 �s�w Y W d � n 1 �s�w Y | � t��$ tjt� d �tdd �� W d � n 1 �s�w Y W d � n 1 �s�w Y tt� d�ddd
�� W d � n 1 �s�w Y tj|dddd
�� W d � n 1 �s�w Y tt� d�ddd�� W d � n 1 �sw Y tj|dddd�� W d � n 1 �s1w Y tt� d�ddd�� W d � n 1 �sLw Y tj|dddd�� W d � n 1 �sfw Y t|ddd�}tjt� d�dddd�� W d � d S 1 �s�w Y d S )Nc @ r$ )z/MockTest.test_misspelled_arguments.<locals>.Foor= Nr� r r r r r � r� r r� T)r8 r= )r9 )r: r )r/ r8 )r/ r9 )r/ r: )r� r8 )r� r9 )r� r: )r: r� z.Typos)r8 r: r9 ) rZ r� r r$ rn r r� Zmultipler r� r r r �test_misspelled_arguments� s� 2� � 0� � 2� � 0� � 2� � 2� � � � �� � � �� � � �� � �4 �4 �4�$�z"MockTest.test_misspelled_argumentsN)�r$ r% r& r? rQ rS rW r[ re rp 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@ rI rN rP rQ rT rY r\ rb rc re rh ri rk rl rm rn ro rt ry 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 r r r= 2 s )&
!
8N
#
,((
5
'
r= �__main__)!r- r� r( r� Z test.supportr rK Z unittest.test.testmock.supportr r r r r r r r
r r r
r r r r r rn r r( r7 r< rL r= r$ �mainr r r r �<module> sD <
< �