File: //usr/local/lib/python3.7/unittest/test/testmock/__pycache__/testmock.cpython-37.pyc
B
"��g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ 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�_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 �;/usr/local/lib/python3.7/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 )N)�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 )Nr )r �a�b�c�dr r r �meth s zSomething.methc C s d S )Nr )�clsr r! r" r# r r r �cmeth" s zSomething.cmethc C s d S )Nr )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 )Nr )r r r r � something+ � r* c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e � de
jkd�dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� 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� Z0dWdX� Z1dYdZ� Z2d[d\� Z3d]d^� Z4d_d`� Z5dadb� Z6dcdd� Z7dedf� Z8dgdh� Z9didj� Z:dkdl� Z;dmdn� Z<dodp� Z=dqdr� Z>dsdt� Z?dudv� Z@dwdx� ZAdydz� ZBd{d|� ZCd}d~� ZDdd�� 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�S )��MockTestc C s t d� d S )Nzfrom unittest.mock import *)�exec)r r r r �test_all0 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 correctlyz(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_constructor7 s(
zMockTest.test_constructorc C s t d d�}| �|jd� d S )N)r4 z(return value in constructor not honoured)r r: r4 )r r r r r � test_return_value_in_constructorS s
z)MockTest.test_return_value_in_constructorc C s� t dd�}| �dt|�� | �dt|� t|�� t � dft dd�dfg}x�|D ]�\}}| �d| t|j�� | �d| t|�� �� | �d| t|�� j�� | �d | t|� �� | �d
| t|� � �� | �d | t|� � jj�� j�� qPW 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�idrB r@ Zbing�baz)r r �mocksrA r r r � test_reprY 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 �Xj s rI )�specz
spec='X' )�spec_setz spec_set='X' r@ )rJ rA z name='foo' )rA rJ )�objectr rC rD r8 )r rI r r r r �test_repr_with_speci 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>.effect� � � )�fishc s � � � S )N)�popr )�resultsr r rQ � s z side effect not used correctly)� side_effectz#side effect in constructor not usedc S s t S )N)r r r r r rX � s z.MockTest.test_side_effect.<locals>.side_effect)rX r4 ) r rX � assertRaisesrN �assert_called_withr1 r �
SideEffect�RETURN)r r rQ rX r )rW r �test_side_effect� s
zMockTest.test_side_effectc s� dddg� � fdd�}dd� }t |�}dddg|_| �|� |� |� gdddgd� dddg� t |�}||_| �|� |� |� gdddgd � d S )
NrR rS rT c s � � � S )N)rV r )rW r r rQ � s z2MockTest.test_autospec_side_effect.<locals>.effectc S s d S )Nr r r r r �f� s z-MockTest.test_autospec_side_effect.<locals>.fz1side effect not used correctly in create_autospecz'callable side effect not used correctly)r rX r1 )r rQ r^ r r )rW 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 )Nr r r r r r^ � s z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r �
ValueErrorrX �assertRaisesRegex)r r^ r r r r �#test_autospec_side_effect_exception� s
z,MockTest.test_autospec_side_effect_exception�javaz This test only applies to Jythonc C sd dd l }t|j�d�d�}y|dddd� W n |jjk
rD Y n X | �d� |jdddd� d S ) Nr zBoom!)rX rR rS rT )rU zjava exception not raised)rc r �langZRuntimeException�failrZ )r rc r r r r � test_java_exception_side_effect� s
z(MockTest.test_java_exception_side_effectc 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)rA �parentrJ )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[ rX r4 �
reset_mockr1 �
_mock_namer; r< r/ r0 r2 r5 r6 r7 �
mock_callsr= )r rh rJ r r* r4 r r r �test_reset_mock� sD
zMockTest.test_reset_mockc C s t � }||_|�� d S )N)r r4 ri )r r r r r �test_reset_mock_recursion� s z"MockTest.test_reset_mock_recursionc C s t �� }|�� d S )N)r � mock_openri )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 tjfi fgd� tj |_|tjtj
d�}| �|tj d � | �|j d
d � | �|j
tjfd tj
ifd� | �|j tjfi ftjfd tj
ifgd
� d S )Nz%Default return_value should be a Mockz'different result from consecutive callszcalled not setrR zcall_count incoreectzcall_args not setz(call_args_list not initialised correctly)�keyzincorrect return valuerS zcall_count incorrectrp zcall_args_list not set)r r3 r r4 r1 ri r �Argr0 r2 r5 r6 Z ReturnValueZKeyArg)r r �resultZret_valr r r � test_call s4
zMockTest.test_callc C s� t � }|� |tj� |tjd� |tjtjd� | �|jdtjffdtjiftjfdtjifg� | �|jtjfdtjif� | �|jdk� d S )N)�kwr rt za long sequence)r r rq ZKwargr1 r6 r5 r/ )r r 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 )N)r r r3 r/ �ANYr )r ZmmZcall1Zcall2r r r �test_calls_equal_with_any8 s z"MockTest.test_calls_equal_with_anyc C sd t � }|� |�� | �t|jd� |�� | �t|j� |dddddd� |jdddddd� d S )NrR rS rT rU Znothing)r r! )r rZ rY �AssertionErrorri )r r r r r �test_assert_called_withM s z MockTest.test_assert_called_withc C s t � }|t � � |�tj� d S )N)r rZ r rv )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 Q R X | �|jjt� d S ) Nc S s d S )Nr )r r! r" r# r r r r^ c s z9MockTest.test_assert_called_with_function_spec.<locals>.f)rJ rR rS rT )r! r" )r r! r" � )�e)N)r rZ rY rx �assertIsInstance� exception� __cause__� TypeError)r r^ r �cmr r r �%test_assert_called_with_function_specb 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 )NrR rS rT )r! r" )r r! r" )rZ rY rx )r )r r r �_checkt s
z<MockTest.test_assert_called_with_method_spec.<locals>._check)rJ )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 s� t � � � � � �� � � | �t� j� � �� | �t� j� � dddd� � jdddd� � �� � dddd� | �t� fdd�� d S )Nr@ rB rS )rF c s � j dddd�S )N�bobrB rS )rF )�assert_called_once_withr )r r r �<lambda>� r+ z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r r� rY rx ri )r r )r r � test_assert_called_once_with� s z%MockTest.test_assert_called_once_withc C s� ddd�}t |d�}|dddd� |�ddd� |jdddd� | jt|jdddd� | �t��}|jd d
� W d Q R X | �|jjt� |d d d
� | �t|jddd� | �t|jd d d
� d S )Nc S s d S )Nr )r r! r" r# r r r r^ � s z>MockTest.test_assert_called_once_with_function_spec.<locals>.f)rJ rR rS rT )r! r" )r r! r" r| )r} � � � )N)r r� rY rx r~ r r� r� )r r^ r 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 namerT zmethod returned wrong valuez!method didn't record being called)r r* r3 r r1 r4 r0 )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* r1 rj r; )r r 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 ) NrT )rU r� )�caker* )r� r� z#method calls not recorded correctly)rT rU zsomething_else.something)r r* �something_elser ZCaker1 r7 )r r 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 )
N)r* r* r � different)r* )r� )r� rR )�xr� Zmore)�somer� )r r* r1 r7 )r r r r r � test_method_calls_compare_easily� s"
z)MockTest.test_method_calls_compare_easilyc C sL xFdgdfD ]8}x2dD ]*}t f ||i�}|j | �tdt|d� qW q W d S )Nr* )r* )rJ rK z-Mock object has no attribute 'something_else'r� )r r* ra �AttributeError�getattr)r rJ �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>.SomethingrT Nc S s d S )Nr )r r r r �y s 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� ra r� r� )r )r r r �test_attributes s
z0MockTest.test_from_spec.<locals>.test_attributes)rJ )rL 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)�wrapsrR rS rT )rU )r r1 ri rZ )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>.RealN)r r r r r r r �Real- s r� )r� c s � � � S )N)Znew_attrr )r r r r� 3 r+ zJMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.<lambda>)rL r rY 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 )N)r� rT )r r4 r1 r/ r0 )r r� r r r r �,test_wraps_call_with_nondefault_return_value6 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� A s r� )r� c s � j S )N)rU r )r r r r� H r+ z0MockTest.test_wraps_attributes.<locals>.<lambda>rR rS rT )rU ) rL r r1 r� rY r� �assertNotEqualZfrogrZ )r r� r� rr r )r r �test_wraps_attributes@ 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 )N)r �ORIGINAL_VALUE)r r r r �methodR s zbMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Real.methodN)r r r r� r r r r r� Q s r� )r� ) rL r r �VALUE1r r� rX r1 r� rY �
StopIteration)r r� r� r r r r �Dtest_customize_wrapped_object_with_side_effect_iterable_with_defaultP 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
t � �d S )N)�NotImplementedError)r r r r r� ` s zUMockTest.test_customize_wrapped_object_with_side_effect_iterable.<locals>.Real.methodN)r r r r� r r r r r� _ s r� )r� )
rL r r r� �VALUE2r� rX r1 rY r� )r r� 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 @ s e Zd Zdd� ZdS )zOMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Realc S s
t � �d S )N)r� )r r r r r� n s zVMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Real.methodN)r r r r� r r r r r� m s r� )r� )rL r � RuntimeErrorr� rX rY )r r� r� r r r r �8test_customize_wrapped_object_with_side_effect_exceptionl 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
t � �d S )N)r� )r r r r r� z s zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Real.methodN)r r r r� r r r r r� y s r� c S s t jS )N)r �VALUEr r r r rX } s zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.side_effect)r� )rL r r� rX r1 r r� )r r� rX r� r r r r �7test_customize_wrapped_object_with_side_effect_functionx 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
t � �d S )N)r� )r r r r r� � s zMMockTest.test_customize_wrapped_object_with_return_value.<locals>.Real.methodN)r r r r� r r r r r� � s r� )r� )rL r r r� r� r4 r1 )r r� r� 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
t � �d S )N)r� )r r r r r� � s z]MockTest.test_customize_wrapped_object_with_return_value_and_side_effect.<locals>.Real.methodN)r r r r� r r r r r� � s r� )r� ) rL r r r� r� r� rX Z WRONG_VALUEr4 r1 rY r� )r r� 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
t � �d S )N)r� )r r r r r� � s z^MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.Real.methodN)r r r r� r r r r r� � s r� )r� c S s t S )N)r r r r r r� � r+ z[MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.<lambda>)rL r r� rX r r� r4 r1 )r r� r� 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
t � �d S )N)r� )r r r r r� � s zeMockTest.test_customize_wrapped_object_with_return_value_and_side_effect_default.<locals>.Real.methodN)r r r r� r r r r r� � s r� )r� ) rL r r r� r r� rX r\ r4 r1 rY r� )r r� 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 )N)rX r@ )r r� rY )r r 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 )N)rX r@ )r �KeyboardInterruptrY )r r 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 ra rx rZ )r r r r r � test_assert_called_with_message� s z(MockTest.test_assert_called_with_messagec C s t dd�}| �td|j� d S )NZgeoffrey)rA z'Expected 'geoffrey' to be called once\.)r ra rx r� )r r 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 )N)r r )r r r r� � r+ z'MockTest.test__name__.<locals>.<lambda>r@ )r rY r� r r1 )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>.SubN)r r r r r r r �Sub� s r� r@ )rJ rT )�listr �appendrZ rY 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>.XN)r r r r r r r rI � s rI )rJ r )rK )rL r r~ �assertIs� __class__r1 r )r rI r r r r �test_spec_class� s
zMockTest.test_spec_classc 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>.XrT N)r r r r� r r r r rI s rI )rJ r@ )rK c s
d� _ d S )Nr@ )r� r )r r r �set_attr s z?MockTest.test_setting_attribute_with_spec_set.<locals>.set_attr)rL r r� r� rY r� )r rI 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 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 s dS )NrT r )r r r r �_get 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 �SubClass s r� )rK rT c s
d� _ d S )NrT )r� r )�sr r �test' s z4MockTest.test_subclass_with_properties.<locals>.testc s
d� _ d S )NrB )r@ r )r� r r r� + s )r r1 r� rY r� r� )r r� r� r )r� 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 )N)Z
_mock_call)r r r r r �__call__2 s z,MockTest.test_setting_call.<locals>.__call__�one�two)r �typer� rZ rY r� )r r r� r r r �test_setting_call0 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)�.0rz r r r �
<listcomp>? s z%MockTest.test_dir.<locals>.<listcomp>r r! r" r# c S s t g �S )N)r )r� r r r r� O r+ z#MockTest.test_dir.<locals>.<lambda>r )
r �set�dirr1 r r! rC 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 )N)rJ rT �versionrR )r �unittest�TestCaser� r� r1 r� �count)r r Ztestcase_attrsr� r r r �test_dir_from_specS s zMockTest.test_dir_from_specc C sV t �tdd�}|�� z0ttt� ��}ttt��}| �t� || � W d |�� X d S )NZ
FILTER_DIRF) r rL r �startr� r� r r1 �stop)r Zpatcherr� r� r r r �test_filter_dir` s zMockTest.test_filter_dirc C s6 t � }d|j_| �dt|�� |`| �dt|�� d S )NrR rg )r rg r4 rC r� r8 )r r r r r �,test_dir_does_not_include_deleted_attributesm 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 )NrB )r@ �! )rX zfoo.bar.return_valuer@ ) r r1 r@ r �KeyErrorrY rB r~ Zconfigure_mock)r r rP r r r �test_configure_mockv s
z MockTest.test_configure_mockc O s\ y|||� W n$ t �� d }| �||� Y nX | �d|f � t|�}| �||� d S )NrR zException %r not raised)r� �exc_infor~ re �strr1 )r r �message�funcrO rP �instance�msgr r r �assertRaisesWithMsg� s z MockTest.assertRaisesWithMsgc
C s( t � }d}d}| jt||f |jddddd� |jddddd� |jj|jjg}x8|D ]0}d }d
}d }| jt|||f |ddddd� qVW x2|D ]*}d }d }d }| jt|||f |dd� q�W x4|D ],}d }d
}d }| �t|||f |ddd� q�W x.|D ]&}d }d}d }| �t|||f |� q�W d S )Nzmock(1, '2', 3, bar='foo')z Expected call: %s
Not calledrR �2rT r@ )rB )r@ zfoo(1, '2', 3, foo='foo')zfoo(1, '2', 3, bar='foo')z!Expected call: %s
Actual call: %szfoo(bar='foo')z foo(1, 2, 3)rS zfoo())r
r� rx rZ r@ r� )r r �expectedr� Z assertersr$ �actualr r r �'test_assert_called_with_failure_message� sF
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� � �
� � � � �x: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� �� � � � �qtW d S )NT� r rR rS rT r� r� )r r! z().foo)rR rS rT r@ z
().foo.barz().foo.bar().bazrB )rA z foo.__int__r za()z().foo.bar().baz().__int__���)r r� rk r1 r@ r� r �dictr4 rB rF � call_listr� r r! r" �__int__)r r r� rP Z last_callr r r �test_mock_calls� st
zMockTest.test_mock_callsc C sF t � }|� }|�� | �|jt� t� �� g� | �|jt�� g� d S )N)r Zwibbler1 rk r )r rz rr 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 )NrR r )r r@ r* r� rk r r� r1 )r rz 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 )NrR r )r r@ rB r� rk r rF )r rz r r r �!test_mock_call_not_equal_non_leaf/ s z*MockTest.test_mock_call_not_equal_non_leafc C s8 t � }|jdd��� | �|jd tjdd��� � d S )NrR )r� rS )r r@ rB r1 rk r )r rz r r r �2test_mock_call_not_equal_non_leaf_params_different6 s z;MockTest.test_mock_call_not_equal_non_leaf_params_differentc C s, t � }|j�� | �|jd tj�� � d S )Nr )r r@ rB r� rk r rF )r rz 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 )Nr )r r@ rB r� rk r )r rz r r r �2test_mock_call_not_equal_non_leaf_call_versus_attrC 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()rR zcall.foo().bar()rS zcall.foo().bar().baz.bob())r r@ rB rF r� r1 rD rk )r rz r r r �test_mock_call_reprI s
z MockTest.test_mock_call_reprc C sh G dd� dt �}|� }| �|j|� | �|� |� G dd� dt �}|� }| �|j|� | �|� |� d S )Nc @ s e Zd ZdS )z+MockTest.test_subclassing.<locals>.SubclassN)r r r r r r r �SubclassR s r c @ s e Zd Zdd� ZdS )z+MockTest.test_subclassing.<locals>.Subclassc [ s
t f |�S )N)r )r rP r r r �_get_child_mockZ s z;MockTest.test_subclassing.<locals>.Subclass._get_child_mockN)r r r r r r r r r Y s )r r~ r@ �assertNotIsInstance)r r r r r r �test_subclassingQ s zMockTest.test_subclassingc s� t � t� t� t� g}� fdd�}x�|D ]z}||� t|�rd|� |dd� |dd� |�� ||� |�� |jjddd� |�d��� �d� |�� ||� q&W d S )Nc sD d}x:|D ]2}t | |�}� �|t� � �|t� � �|g � q
W d S )N)r6 r7 rk )r� r~ r
r� r1 )r �namesrA �attr)r r r � assert_attrsj s
z-MockTest.test_arg_lists.<locals>.assert_attrsrR rS rT )r ) r r r
r �callableri r@ rB rF )r rG r r r )r r �test_arg_listsb s$
zMockTest.test_arg_listsc C s� t � }|ddd� |ddd� | �t|j�d� |j\}}| �|d� | �|tdd�� dtdd�fdtdd�fg}xPt||j�D ]@\}}| �t|�d� | �|d |d � | �|d |d � q�W d S )
NrR rT )r rS r� )r! )rS )rR r )r r1 �lenr5 r �zipr6 )r r rO rP Z
expected_listr� r5 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 )NrR rS rT )rX r r! r" Zghi�g�h�ic @ s e Zd ZdS )z/MockTest.test_side_effect_iterator.<locals>.FooN)r r r r r r r �Foo� s r r r r r ) r r r1 rY r� r rL r~ r )r r r r r r �test_side_effect_iterator� s
z"MockTest.test_side_effect_iteratorc C s` xZt tfD ]N}tdtdf}||d�}| �t|� | �|� d� | �t|� | �|� d� q
W d S )NrT r� )rX )r r r` r� rY r1 )r �Klass�iterablerz 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 ) NrR rS rT r r! r" r r r r )
r r rX r1 rY r� r~ r� r r� )r r rX 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 )NrS )r4 rR )r r r rX r1 )r r 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}x |D ]}|j|gdd� q~W x<tdd�tdd�t� dd dfD ]}| jt|j|gdd� q�W tdd�tdd�gtdd�tdd�gtdd�tdd�gg}x |D ]}|j|dd� �qW tdd�tdd�tdd�gtdd�tdd�gtdd�tdd�tdd�gtdd�tdd�tdd�tdd�tdd�gg}x"|D ]}| jt|j|dd� �q�W d S )NrR rS rT )r r� r� )r! r ))rT r� r )r� )rR rS r� )rR rS T)� any_orderr� r@ r� � )r r �assert_has_callsrY rx )r r �kalls�kallZ
kall_listsZ kall_listr r r � test_assert_has_calls_any_order� s:
$
.
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 }xVt
� t � fD ]F}|dd� |dd � |dd
� |dd� |ddd� |� ||f� q�W t
� }|�� |�d� |�� jdd � |jt� i d �� dg� |� ||f� xz|D ]r\}}xft
t|��D ]V}xNdD ]F}|||| � }|�|� t|�dk�rH| �t|jtt|��� �qHW �q>W �q(W d S )NrR rS r rT ))rT r� r� )r! r� )rR r! )r )r@ r� )rR rS rT )r r@ rB �extendZspamrF r Zbamr� rU r r r� �ranger r# rY rx r� �reversed) r Zkalls1Zkalls2rG r r$ r �stepZtheser r r �test_assert_has_calls� s<
"
z MockTest.test_assert_has_callsc C sR G dd� d�}t |�}xV||� gD ]H}|jddddd� |�tjddddd�g� |�t�dddd�g� q"W |�� xz||� gD ]l}| �t|jt�� g� |�d��dd� |�t�d�t�d��dd�g� |j�td�t� �dd�g� q�W |�� t�d�t� d�t�� � d�t�� �dddd�g}x |D ]}| �t|j|g� �q2W 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 )Nr )r r r r r ) r+ zFMockTest.test_assert_has_calls_nested_spec.<locals>.Something.__init__Nc S s d S )Nr )r r r! r" r# r r r r$ * 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 )Nr )r r r r r r . r+ zJMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.__init__c S s d S )Nr )r r r! r r r �meth1/ r+ zGMockTest.test_assert_has_calls_nested_spec.<locals>.Something.Foo.meth1N)r r r r r, r r r r r , s r )N)r r r r r$ r r r r r r ' s
r rR rS rT )r# r� )
r r$ r# r ri rY rx r r, Z non_existent)r r Z
mock_classrz Z
invalid_callsr% r r r �!test_assert_has_calls_nested_spec& s*
""
z*MockTest.test_assert_has_calls_nested_specc C sJ t � }|� �� �� �� |�� �� �� t�� �� �� �� }|� |� d S )N)
r r@ rB rF r� r� �threer r r# )r rz �callsr r r �)test_assert_has_calls_nested_without_specN 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 Q R X |j|dd� | �t�� |�|dd � � W d Q R X |j|dd � dd� | �t�� |�|d d� � W d Q R X |j|d d� dd� d S )Nc S s d S )Nr )r r! r" r# r r r r^ W s z<MockTest.test_assert_has_calls_with_function_spec.<locals>.f)rJ rR rS rT )r! r" r� r� r� r" )r" r# �
� � )r" r� )rR rS rT )r� r� r� r# )r1 r2 r3 T)r! r� )N)r r# r� r) rY rx )r r^ r r/ r r r �(test_assert_has_calls_with_function_specV s0
z1MockTest.test_assert_has_calls_with_function_specc C s� d dd�}t |d�}|d� | �td�t�d����}|�t� g� W d Q R X | �|j j
� | �td�t�d���� }|�t� tdd�g� W d Q R X | � |j j
t � d S )
Nc S s d S )Nr )r� r r r r^ w r+ zAMockTest.test_assert_has_calls_not_matching_spec_error.<locals>.f)rJ rR 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)]rS )N)
r ra rx �format�re�escaper# r r: r r� r~ r� )r r^ r r� r r r �-test_assert_has_calls_not_matching_spec_errorv 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 )NrR rS rT )r r� )r! r� )r �assert_any_callrY rx )r r 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 Q R X | �|jjt� d S )Nc S s d S )Nr )r r! r" r# r r r r^ � s z;MockTest.test_assert_any_call_with_function_spec.<locals>.f)rJ rR rS rT )r! r" r� r� r� r" )r" r# )r r! r" )r r! r" r# r| )r} )N)r r9 rY rx r~ r r� r� )r r^ r r� r r r �'test_assert_any_call_with_function_spec� s
z0MockTest.test_assert_any_call_with_function_specc C sj dd� }t � }||_t|�t|�jg}x>|D ]6}|dd� |dd� | �|jtdd�tdd�g� q,W d S )Nc S s d S )Nr )r r! r r r r^ � s z3MockTest.test_mock_calls_create_autospec.<locals>.frR rS rT r� )r r^ r r1 rk r )r r^ �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_func)rA )r r rL rC rD )r rz r r r � test_create_autospec_with_name� s z'MockTest.test_create_autospec_with_namec
C sd G dd� d�}xPdD ]H}| j |d��2 t�t||��}|� |�� | �t|d� W d Q R X qW d S )Nc @ s$ e Zd Zedd� �Zedd� �ZdS )zMMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClassc S s d S )Nr )r% r r r � class_method� s zZMockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.class_methodc S s d S )Nr r r r r �
static_method� s z[MockTest.test_create_autospec_classmethod_and_staticmethod.<locals>.TestClass.static_methodN)r r r r( r? r) r@ r r r r � TestClass� s rA )r? r@ )r� Z extra_arg)ZsubTestr r r� r� rY r� )r rA r� Z mock_methodr r r �1test_create_autospec_classmethod_and_staticmethod� s
z:MockTest.test_create_autospec_classmethod_and_staticmethodc C s` t � }| �t�� |�� W d Q R X | �t�� |�� W d Q R X t dd�}|�� |�� d S )NT)Zunsafe)r rY r� Zassert_foo_callZassret_foo_call)r rz r r r �test_mock_unsafe� s
zMockTest.test_mock_unsafec C s< t � }|j�� |�� | �t�� |j�� W d Q R X d S )N)r �helloZassert_not_calledrY rx )r rz r r r �test_assert_not_called� s
z MockTest.test_assert_not_calledc C sN t � }| �t�� |j�� W d Q R X |�� |j�� |�� |j�� d S )N)r rY rx rD Z
assert_called)r rz r r r �test_assert_called� s
zMockTest.test_assert_calledc C sd t � }| �t�� |j�� W d Q R X |�� |j�� |�� | �t�� |j�� W d Q R X d S )N)r rY rx rD Zassert_called_once)r rz r r r �test_assert_called_once s
z MockTest.test_assert_called_oncec C s0 t � }|jddd� d}| �t|jj�|� d S )NrD Zhero)rA Zdaddyz call(daddy='hero', name='hello'))r rD r1 rD r5 )r rz �textr r r �test_sorted_call_signature s z#MockTest.test_sorted_call_signaturec C sV t �� }t �dd�}t� }|�� |�dd� | �|jd |� | �|jd |� d S )N� rD r rR )r r� �indexr r1 r7 )r r" r rz r r r �test_override_tuple_methods s z$MockTest.test_override_tuple_methodsc C s> t dddgd�}|jddd� | �|jt � | �|jd � d S )Nr1 rS rT )r4 rX T)r ri r~ r4 r1 rX )r rz r r r � test_reset_return_sideeffect s z%MockTest.test_reset_return_sideeffectc C s< t dddgd�}|jdd� | �|jt � | �|jd � d S )Nr1 rS rT )r4 rX T)r4 )r ri r~ r4 r� rX )r rz r r r �test_reset_return& s zMockTest.test_reset_returnc C s< t dddgd�}|jdd� | �|jd� | �|jd � d S )Nr1 rS rT )r4 rX T)rX )r ri r1 r4 rX )r rz r r r �test_reset_sideeffect, s z MockTest.test_reset_sideeffectc s� G dd� dt �}G dd� dt �}G dd� dt �� ttttg}x<t|�D ]0}|�|f� fdd� � |�|f� fd d� � qFW �x|D �]}�xt� td
d �fD ]�}|� }|j|j |j
f x�||fd gd
gfgD ]�\}}x�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�W q�W q�W q�W d S )Nc @ s e Zd ZdZdS )z)MockTest.test_mock_add_spec.<locals>._OnerR N)r r r r� r r r r �_One3 s rP c @ s e Zd ZdZdS )z)MockTest.test_mock_add_spec.<locals>._TworS N)r r r r� r r r r �_Two5 s rQ 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 �Anything7 s rR c s
| � d�S )N)rJ r )�K)rR r r r� > r+ z-MockTest.test_mock_add_spec.<locals>.<lambda>c s
| � d�S )N)rK r )rS )rR r r r� ? r+ T)rK r� r� r. rK )rL r r r
r r� r� r r� r� r. �
mock_add_specrY r� r� �setattr) r rP rQ Zklassesr rP r ZOneZTwor )rR r �test_mock_add_spec2 s>
zMockTest.test_mock_add_specc s� xzt tfD ]n}|� � t� � � �t� | �tt� � |� � � d d� j_� �t� | � t� �d� | �t� fdd�� q
W d S )Nr@ r� c s � d S )Nr@ r r )r r r r� u r+ z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� rT rL rY r� r r4 r1 )r r r )r r � test_mock_add_spec_magic_methodsg s
z)MockTest.test_mock_add_spec_magic_methodsc C s� x�t tttfD ]�}|� }t� |_|�� | �|jt�� g� | �|jt�� g� |� }tdd�|_ |� � | �|jg � | �|jg � |� }t� � |_
|�
� | �|jg � | �|jg � qW d S )NrA )rA ) r
r r r r@ r1 r7 r rk rB rF )r r r r r r �test_adding_child_mockx s
z MockTest.test_adding_child_mockc C sD x>t tfD ]2}|� }t� |_|� � | �|jt� t� � g� q
W d S )N)r r r4 r1 rk r )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 � s r r� r� ) rL r r r� r� r� � attach_mockr� r� r1 rk r )r r �managerZp1Zp2Zmock_oneZmock_twor r r �test_manager_mock� s zMockTest.test_manager_mockc C sl xft tfD ]Z}|� }t dd�|_tdd�|_t|� t|� | �|jt�� t�� g� | �|j g � q
W d S )NrT )r4 g @)
r r r � __float__r� �floatr1 rk r r7 )r r rz 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)� read_dataza-namez another-name)r rn �readr1 )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 Blue)ra za-namec 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
rR zNorwegian Blue)r rn r1 r� )r rc rd �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 d Q R X d S )
Nz1st line
2nd line
3rd line)ra za-namec S s g | ]}|�qS r r )r� rg r r r r� � s z6MockTest.test_mock_open_using_next.<locals>.<listcomp>z 1st line
z 2nd line
r z3rd line)r rn r r r1 r� rY r� )r rc rd Zline1Zline2rh 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 d Q R X d S )NZJLV)rA ztempfile.NamedTemporaryFilez Test 2 Errorc S s t �� �d� d S )NZasd)�tempfileZNamedTemporaryFile�writer r r r �attempt� s z.MockTest.test_mock_open_write.<locals>.attempt) r rn r r r4 rl �OSErrorrX rY )r Zmock_namedtempZmock_filehandleZ
mock_writerm 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
barn)ra c W s dS )N�abcr )rO r r r r� � r+ z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>rp )r rn r4 �readlinerX r1 )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 )Nr@ )ra rB r� )r rn rb r1 rq � readlines)r Z_openr r r r �test_mock_open_after_eof� s z!MockTest.test_mock_open_after_eofc C s �xt tfD ]�}|� }t|�}||_| �|� |� | �t|�|� |�� | �|� |� | �t|�|� |� }|j|_| � dt|j�� | � dt|j�� |�� | � dt|j�� | � dt|j�� |� }t|�}|� |_||j_| �t|�|� | �t|�� �|� q W d S )Nz
name='mock.a')
r r rD r4 r� r1 ri r r! rC )r r rz Z
original_reprr r r �test_mock_parents� s, zMockTest.test_mock_parentsc C s� t tttf}x�|D ]�}x�|D ]z}|� }|dd�}|�|d� | �|j|� | �dt|�� |j� d� | �
|j t j� d�g� | �
|j
t j� d�g� q W qW d S )Nr@ )rA rB zname='mock.bar'rR )r r r r
rZ r� rB rC rD rF r1 rk r r7 )r �classesr �Klass2rz �m2r r r �test_attach_mock s
zMockTest.test_attach_mockc C s� t tttf}xxt tfD ]l}xf|D ]^}|� }|dd�}|�|d� | �|� |� | �dt|�� |�� | � |j
t � �� � � � q W qW d S )Nr@ )rA r4 z
name='mock()')
r r r r
rZ r� rC rD r@ r1 rk r r )r rx r ry rz rz r r r � test_attach_mock_return_value& 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 d Q R X d S )
Nz
.somethingT)�autospecr* rg rR rS rT z
mock.child)
r r r r r1 Z_extract_mock_namerZ rg r* r rk rC rD )r rh Z mock_funcZ parent_callsZ child_callsr r r � test_attach_mock_patch_autospec6 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 d Q R X t jt� d
�dd��R}t� }|�|d � t d� |�t� d�g� t �td�g� |�td�g� W d Q R X t jt� d �dd���}t� }|�|d
� t� }|jddddd � |�t�
� t�
� jddddd �g� |j�tddddd �g� |�t� t� jddddd �g� W d Q R X d S )Nz.Something.methT)r} � attach_methrR rS rT r� )r# z
.something� attach_funcz
.Something�
attach_obj)r r r r rZ r r$ r# r r rv r* r� r� )r Zmockedr[ r< r r r �)test_attach_mock_patch_autospec_signatureJ s0 &
z2MockTest.test_attach_mock_patch_autospec_signaturec C sj xdt � t� t� t� fD ]L}| �t|d�� |`| �t|d�� |`| �t|d�� | � t
t |d� qW d S )Nrz r^ ) r r r r
r3 �hasattrrz r/ r^ rY r� r� )r r r r r �test_attribute_deletiong s z MockTest.test_attribute_deletionc C s� x�t � t� t� t� fD ]t}d|_| �t|d�� | �|jd� |`| �t|d�� d|_| �t|d�� | �|jd� |`| �t|d�� qW d S )NrT r@ r� ) r r r r
r@ r3 r� r1 r/ )r r r r r �7test_mock_does_not_raise_on_repeated_attribute_deletiont s z@MockTest.test_mock_does_not_raise_on_repeated_attribute_deletionc
C s@ x:t � t� t� t� fD ]"}|`| �t��
|`W d Q R X qW d S )N)r r r r
r@ rY r� )r r 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 )NTrg )r rg ri r/ r� )r r 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> x8t � t� fD ](}| �|t� t|_| �|t� |j qW d S )N)r r r
r� r� r~ r@ )r r r r r �test_class_assignable� s
z MockTest.test_class_assignablec C s8 | � tj� | �ttj�t� | �tt�� j�t� d S )N)�assertIsNotNoner rA r1 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 )N)r� r rh r1 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 )Nr )r r! r r r r@ � s zBMockTest.test_parent_propagation_with_create_autospec.<locals>.foorR rS z
mock.child) r r rg rY r� r1 rk r rC rD r )r r@ 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 )Nr )r r! r r r r@ � r+ zGMockTest.test_parent_propagation_with_autospec_attach_mock.<locals>.foorB )rA rg rR rS z
mock.child) r rZ r rg rY r� r1 rk r rC rD r )r r@ rh 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��z tjd= � fdd�� t�� � | �tjd � ddlm }m
}m }m } ||||g}x"|D ]}|t
d �}| �|t
� qzW W d Q R X d S )
Nc S s t tjd| �S )Nr )rU r� r )r r r r r� � r+ z9MockTest.test_isinstance_under_settrace.<locals>.<lambda>z sys.modulesz
unittest.mockc s � S )Nr )�frameZeventr� )�tracer r r� � s z6MockTest.test_isinstance_under_settrace.<locals>.tracer )r r r
r )rJ )r� r r r� r r� �modules�settrace�
unittest.mockr r r
r r r~ ) r Z old_patchr r r
r rG r r< r )r� r � test_isinstance_under_settrace� s
z'MockTest.test_isinstance_under_settraceN)~r r r r. r> r? rH rM r] r_ rb r� Z
skipUnlessr� �platformrf rl rm ro rs ru rw 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- r0 r4 r8 r: r; r= r> rB rC rE rF rG rI rL rM rN rO rV rW rX rY r\ r_ rf ri rj ro rt rv rw r{ r| r~ r� r� r� r� r� r� r� r� r� r� r� r r r r r, . s� )"
7N
#
,((
5
r, �__main__) r� r6 r� rk r� Z unittest.test.testmock.supportr r r� r r r r r r r
r r r
r rL r r r* r� r, r �mainr r r r �<module> s8 4
R