Your IP : 216.73.216.213
3
� fr� � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZ G dd� de�ZG dd� de�ZG d d
� d
ej�Zed kr�ej� 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 �D/opt/alt/python36/lib64/python3.6/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 @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e j 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�S )��MockTestc C s t d� d S )Nzfrom unittest.mock import *)�exec)r r r r �test_all, s zMockTest.test_allc C s� t � }| j|jd� | j|jdd� | jt|jt �d� | j|jd d� | j|j g d� | j|j
g d� | j d|j d � | j
|jd
� | j
|jd � | j|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_constructor3 s(
zMockTest.test_constructorc C s t d d�}| j|jd� d S )N)r2 z(return value in constructor not honoured)r r8 r2 )r r r r r � test_return_value_in_constructorO s
z)MockTest.test_return_value_in_constructorc C s� t dd�}| jdt|�� | jdt|� t|�� t � dft dd�dfg}x�|D ]�\}}| jd| t|j�� | jd| t|j� �� | jd| t|j� j�� | jd | t|� �� | jd
| t|� � �� | jd | t|� � jjj� 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�idr@ r> Zbing�baz)r r �mocksr? r r r � test_reprU s
zMockTest.test_reprc C s� G dd� dt �}t|d�}| jdt|�� t|� d�}| jdt|�� t|d�}| jdt|�� t|� d�}| jdt|�� t|dd�}| jdt|�� | jd t|�� tdd
�}| jd t|�� t� }| jd t|�� tdgd�}| jd t|�� d S ) Nc @ s e Zd ZdS )z'MockTest.test_repr_with_spec.<locals>.XN)r r r r r r r �Xf s rG )�specz
spec='X' )�spec_setz spec_set='X' r> )rH r? z name='foo' )r? rH )�objectr rA rB r6 )r rG r r r r �test_repr_with_spece s$
z MockTest.test_repr_with_specc s� t � }dd� }||_| jt|dddd� |jdddd� dddg� � fdd�}||_| j|� |� |� gdddgd� t tjd �}| j|jtjd
� d d � }t |tjd
�}| j|� 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 � j � S )N)�popr )�resultsr r rO � 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 rV � s z.MockTest.test_side_effect.<locals>.side_effect)rV r2 ) r rV � assertRaisesrL �assert_called_withr/ r �
SideEffect�RETURN)r r rO rV r )rU r �test_side_effect� s
zMockTest.test_side_effectc s� dddg� � fdd�}dd� }t |�}dddg|_| j|� |� |� gdddgd� dddg� t |�}||_| j|� |� |� gdddgd � d S )
NrP rQ rR c s � j � S )N)rT r )rU r r rO � 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 rV r/ )r rO r\ r r )rU r �test_autospec_side_effect� s
z"MockTest.test_autospec_side_effectc C s, dd� }t |�}td�|_| jtd|� d S )Nc S s d S )Nr r r r r r\ � s z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r �
ValueErrorrV �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|jjd�d�}y|dddd� W n |jjk
rD Y n X | jd� |jdddd� d S ) Nr zBoom!)rV rP rQ rR )rS zjava exception not raised)ra r ZlangZRuntimeException�failrX )r ra 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}|j� tj|_|j}|� |j� | j |j
dd� | j |j |d� | j |j |d� | j
|jd� | j |jd d
� | j |jd d � | j |jg d � | j |jg d
|jg f � | j |jg � | j |jtjd� | j |j|d� | j
|jd� | j |jd|id� | j |j|d� | j
|jjd� d S )N� something�child)r? �parentrH )rd 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
SomethingElserd rY rV r2 �
reset_mockr/ �
_mock_namer9 r: r- r. r0 r3 r4 r5 �
mock_callsr; )r rf rH r rd r2 r r r �test_reset_mock� sD
zMockTest.test_reset_mockc C s t � }||_|j� d S )N)r r2 rg )r r r r r �test_reset_mock_recursion� s z"MockTest.test_reset_mock_recursionc C s t j� }|j� d S )N)r � mock_openrg )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 � }| jt|jt �d� |� }| j|� |d� |j� |tj�}| j|jd� | j|j dd� | j|j
tjfi fd� | j|j tjfi fgd� tj |_|tjtj
d�}| j|tj d � | j|j d
d � | j|j
tjfd tj
ifd� | j|j tjfi ftjfd tj
ifgd
� d S )Nz%Default return_value should be a Mockz'different result from consecutive callszcalled not setrP zcall_count incoreectzcall_args not setz(call_args_list not initialised correctly)�keyzincorrect return valuerQ zcall_count incorrectrn zcall_args_list not set)r r1 r r2 r/ rg r �Argr. r0 r3 r4 Z ReturnValueZKeyArg)r r �resultZret_valr r r � test_call� s4
zMockTest.test_callc C s� t � }|� |tj� |tjd� |tjtjd� | j|jf tjffdtjiftjfdtjifg� | j|jtjfdtjif� | j|jdk� d S )N)�kwrr za long sequence)r r ro ZKwargr/ r4 r3 r- )r r r r r �test_call_args_comparison s
z"MockTest.test_call_args_comparisonc C s� t j� }| j||k� | j||k� | j|t j� k� | j|t j� k� | j|t jk� | j|t jk� | jt j|k� | jt j|k� t jt j� �}t jt j�}| j||k� | j||k� | j||k� | j||k� d S )N)r r r1 r- �ANYr )r ZmmZcall1Zcall2r r r �test_calls_equal_with_any4 s z"MockTest.test_calls_equal_with_anyc C sd t � }|� |j� | jt|jd� |j� | jt|j� |dddddd� |jdddddd� d S )NrP rQ rR rS Znothing)r r! )r rX rW �AssertionErrorrg )r r r r r �test_assert_called_withI s z MockTest.test_assert_called_withc C s t � }|t � � |jtj� d S )N)r rX r rt )r �mr r r �test_assert_called_with_anyX s
z$MockTest.test_assert_called_with_anyc
C s� d dd�}t |d�}|dddd� |jddd� |jdddd� | jt|jdddd� | jt��}|jd d
� W d Q R X | j|jjt� d S ) Nc S s d S )Nr )r r! r" r# r r r r\ _ s z9MockTest.test_assert_called_with_function_spec.<locals>.f)rH rP rQ rR )r! r" )r r! r" � )�e)N)r rX rW rv �assertIsInstance� exception� __cause__� TypeError)r r\ 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� | j ddd� | j dddd� � jt| j dddd� d S )NrP rQ rR )r! r" )r r! r" )rX rW rv )r )r r r �_checkp s
z<MockTest.test_assert_called_with_method_spec.<locals>._check)rH )r r r$ r& r' )r r� r r )r r �#test_assert_called_with_method_speco s z,MockTest.test_assert_called_with_method_specc s� t � � � � � j� � � | jt� j� � j� | jt� j� � dddd� � jdddd� � j� � dddd� | jt� fdd�� d S )Nr> r@ rQ )rD c s � j dddd�S )N�bobr@ rQ )rD )�assert_called_once_withr )r r r �<lambda>� s z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r r� rW rv rg )r r )r r � test_assert_called_once_with� s z%MockTest.test_assert_called_once_withc
C s� ddd�}t |d�}|dddd� |jddd� |jdddd� | jt|jdddd� | jt��}|jd d
� W d Q R X | j|jjt� |d d d
� | jt|jddd� | jt|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)rH rP rQ rR )r! r" )r r! r" rz )r{ � � � )N)r r� rW rv 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}| jt|t �d� | j|j|d� t � }d|j_| j|j� dd� | j|jjd� d S )Nzattribute isn't a mockz+different attributes returned for same namerR zmethod returned wrong valuez!method didn't record being called)r rd r1 r r/ r2 r. )r r rd r r r �#test_attribute_access_returns_mocks� s
z,MockTest.test_attribute_access_returns_mocksc C s0 t � }|j}| j|jdd� | j|j|d� d S )Nrd z attribute name not set correctlyz"attribute parent not set correctly)r rd r/ rh r9 )r r rd 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� | j|jjdd
dtjifgd� | j|jdd dd ifd d dtjifgd� d S )
NrR )rS r� )�cakerd r� z#method calls not recorded correctlyrS zsomething_else.something)r� )rR )r� )r rd �something_elser ZCaker/ r5 )r r r r r �test_method_calls_recorded� s
z#MockTest.test_method_calls_recordedc C s� t � }|j� | j|jd g� | j|jdf i fg� t � }|jd� | j|jd g� | j|jdd i fg� t � }|jdd� | j|jdddifg� | j|jdf ddifg� t � }|jddd� | j|jdd
ddifg� d S )Nrd � differentrP )�xr� Zmore)�somer� )rd �r� )rd r� )r� )r� )r rd r/ r5 )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 | jtdt|d� qW q W d S )Nrd rH rI z-Mock object has no attribute 'something_else'r� )rd )rH rI )r rd r_ �AttributeError�getattr)r rH �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>.SomethingrR 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 � jtdt| d� � jtdt| 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_attributes)rH )rJ r )r r r� r )r r �test_from_spec s zMockTest.test_from_specc C sJ t � }t |d�}| j|� |� � |j� |dddd� |jdddd� d S )N)�wrapsrP rQ rR )rS )r r/ rg rX )r �realr r r r �test_wraps_calls s
zMockTest.test_wraps_callsc s8 G dd� dt �}|� }t|d�� | jt� 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 � j � S )N)Znew_attrr )r r r r� / s zJMockTest.test_wraps_prevents_automatic_creation_of_mocks.<locals>.<lambda>)rJ r rW 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|_| j|� d� | j|j� d S )N)r� rR )r r2 r/ r- r. )r r� r r r r �,test_wraps_call_with_nondefault_return_value2 s
z5MockTest.test_wraps_call_with_nondefault_return_valuec s� G dd� dt �}|� }t|d�� | j� j� |j� � | jt� fdd�� | j� j|j� � jjdddd �}|jjjdddd � | j||jj� � 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 r� )r� c s � j S )N)rS r )r r r r� D s z0MockTest.test_wraps_attributes.<locals>.<lambda>rP rQ rR )rS ) rJ r r/ r� rW r� �assertNotEqualZfrogrX )r r� r� rp r )r r �test_wraps_attributes< s
z MockTest.test_wraps_attributesc C sd G dd� dt �}|� }t|d�}tjtg|j_| j|j� tj� | j|j� tj� | j 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 �methodN s zbMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_default.<locals>.Real.methodN)r r r r� r r r r r� M s r� )r� ) rJ r r �VALUE1r r� rV r/ r� rW �
StopIteration)r r� r� r r r r �Dtest_customize_wrapped_object_with_side_effect_iterable_with_defaultL s
zMMockTest.test_customize_wrapped_object_with_side_effect_iterable_with_defaultc C sf G dd� dt �}|� }t|d�}tjtjg|j_| j|j� tj� | j|j� tj� | jt |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� )
rJ r r r� �VALUE2r� rV r/ rW r� )r r� r� r r r r �7test_customize_wrapped_object_with_side_effect_iterableZ s
z@MockTest.test_customize_wrapped_object_with_side_effect_iterablec C s: G dd� dt �}|� }t|d�}t|j_| jt|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� j s zVMockTest.test_customize_wrapped_object_with_side_effect_exception.<locals>.Real.methodN)r r r r� r r r r r� i s r� )r� )rJ r � RuntimeErrorr� rV rW )r r� r� r r r r �8test_customize_wrapped_object_with_side_effect_exceptionh s
zAMockTest.test_customize_wrapped_object_with_side_effect_exceptionc C sF G dd� dt �}dd� }|� }t|d�}||j_| j|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� v s zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.Real.methodN)r r r r� r r r r r� u s r� c S s t jS )N)r �VALUEr r r r rV y s zUMockTest.test_customize_wrapped_object_with_side_effect_function.<locals>.side_effect)r� )rJ r r� rV r/ r r� )r r� rV r� r r r r �7test_customize_wrapped_object_with_side_effect_functiont s
z@MockTest.test_customize_wrapped_object_with_side_effect_functionc C s@ G dd� dt �}|� }t|d�}tj|j_| j|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� )rJ r r r� r� r2 r/ )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_| j |j� tj� | j |j� tj� | j
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� ) rJ r r r� r� r� rV Z WRONG_VALUEr2 r/ rW 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_| j|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� � s z[MockTest.test_customize_wrapped_object_with_return_value_and_side_effect2.<locals>.<lambda>)rJ r r� rV r r� r2 r/ )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_| j |j� tj� | j |j� tj� | j
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� ) rJ r r r� r r� rV rZ r2 r/ rW 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�}| jt|� t td�d�}| jt|� d S )N)rV r> )r r� rW )r r r r r � test_exceptional_side_effect� s
z%MockTest.test_exceptional_side_effectc C s4 t td�}| jt|� t td�d�}| jt|� d S )N)rV r> )r �KeyboardInterruptrW )r r r r r � test_baseexceptional_side_effect� s
z)MockTest.test_baseexceptional_side_effectc C s t � }| jtd|j� d S )Nz
Not called)r r_ rv rX )r r r r r � test_assert_called_with_message� s z(MockTest.test_assert_called_with_messagec C s t dd�}| jtd|j� d S )NZgeoffrey)r? z'Expected 'geoffrey' to be called once\.)r r_ rv r� )r r r r r �$test_assert_called_once_with_message� s
z-MockTest.test_assert_called_once_with_messagec s2 t � � | jt� fdd�� d� _| j� jd� d S )Nc s � j S )N)r r )r r r r� � s z'MockTest.test__name__.<locals>.<lambda>r> )r rW r� r r/ )r r )r r � test__name__� s zMockTest.test__name__c C sJ G dd� dt �}t|dg�d�}|jd� |jjd� | jtt|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> )rH rR )�listr �appendrX rW 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�}| j||� t|� d�}| j||� | j|j|� | jt� jjd� t|d�}| j||� t|� d�}| j||� d S )Nc @ s e Zd ZdS )z#MockTest.test_spec_class.<locals>.XN)r r r r r r r rG � s rG )rH r )rI )rJ r r| �assertIs� __class__r/ r )r rG r r r r �test_spec_class� s
zMockTest.test_spec_classc sL G dd� dt �}t|d�� d� _t|d�� � fdd�}d� _| jt|� d S )Nc @ s e Zd ZdZdS )z8MockTest.test_setting_attribute_with_spec_set.<locals>.XrR N)r r r r� r r r r rG s rG )rH r> )rI 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)rJ r r� r� rW r� )r rG r� r )r r �$test_setting_attribute_with_spec_set� s
z-MockTest.test_setting_attribute_with_spec_setc C s4 t j� }| jt j|� t jtd�� tjt� � 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�� | j� jd� � fdd�}| jt|� � fdd�}| jt|� d S )Nc @ s& e Zd Zdd� Zdd� Zeee�ZdS )z8MockTest.test_subclass_with_properties.<locals>.SubClassc S s dS )NrR 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� )rI rR c s
d� _ d S )NrR )r� r )�sr r �test# s z4MockTest.test_subclass_with_properties.<locals>.testc s
d� _ d S )Nr@ )r> r )r� r r r� ' s )r r/ r� rW 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� |jd� | jt|dd� d S )Nc S s
| j |�S )N)Z
_mock_call)r r r r r �__call__. s z,MockTest.test_setting_call.<locals>.__call__�one�two)r �typer� rX rW r )r r r� r r r �test_setting_call, s
zMockTest.test_setting_callc C s� t � }tt|��}tdd� tt �D ��}| jt� || � |j|jf | jdt|�� | jdt|�� d |_|_| jdt|�� | jdt|�� dd� |_ | jd t|�� d S )
Nc S s g | ]}|j d �s|�qS )�_)�
startswith)�.0rx 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� K s z#MockTest.test_dir.<locals>.<lambda>r )
r �set�dirr/ r r! rA r" r# r )r r �attrs�
type_attrsr r r �test_dir8 s
zMockTest.test_dirc C sX t tjd�}tttj��}tt|��}| jt� || � d|_| jt|�jd�d� d S )N)rH rR �versionrP )r �unittest�TestCaser� r� r/ r� �count)r r Ztestcase_attrsr� r r r �test_dir_from_specO s zMockTest.test_dir_from_specc
C sV t jtdd�}|j� z0ttt� ��}ttt��}| jt� || � W d |j� X d S )NZ
FILTER_DIRF) r rJ r �startr� r� r r/ �stop)r Zpatcherr� r� r r r �test_filter_dir\ s zMockTest.test_filter_dirc C s� t dd�}| j|jd� tdd�}| j|jd� tdt� d�}t f |�}| jt|� | j|jj� d� | j|jt� t � }|jf |� | jt|� | j|jj� d� | j|jt� d S )Nr@ )r> �! )rV zfoo.bar.return_valuer> ) r r/ r> r �KeyErrorrW r@ r| Zconfigure_mock)r r rN r r r �test_configure_mocki s
z MockTest.test_configure_mockc O s\ y|||� W n$ t j� d }| j||� Y nX | jd|f � t|�}| j||� d S )NrP zException %r not raised)r� �exc_infor| rb �strr/ )r r} �message�funcrM rN �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 }| jt|||f |ddd� q�W x.|D ]&}d }d}d }| jt|||f |� q�W d S )Nzmock(1, '2', 3, bar='foo')z Expected call: %s
Not calledrP �2rR r> )r@ )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)rQ zfoo())r
r� rv rX r> r� )r r �expectedr� Z assertersr$ Zactualr r r �'test_assert_called_with_failure_message� sF
z0MockTest.test_assert_called_with_failure_messagec C s� t � }| j|jg kd� t � }|� df i fg}| j|j|� |j� |jtj� � | j|j|� | j|jjdf i fg� t � }|� jdddddd� df i fd dtddd�fg}| j|j|� | j|jjjddtddd�fg� | j|jjd
dtddd�fg� t � }|� jj � j
� df i fd f i fd f i fg}| j|j|� | j|� jtjj � j
� j � � �x:t� td
d�fD �]$}t f |�}t |j� df i fg}| j|j|� t f |�}|j
� � df i fdf i fg}| j|j|� | j|j
� jt� g� t f |�}|d�d�d� | j|jtd�d�d�j � � | j|� jtd�d�j � � | j|� � jtd�j � � t f |�}|d�d�d�j
jjd� | j|jtd�d�d�j
jjd�j � � | j|� jtd�d�j
jjd�j � � | j|� � jtd�j
jjd�j � � t f |�}t |� jj � j
� � df i f}| j|jd |� | j|� jtjj � j
� j� j � � | j|� jj � jtj
� j� j � � | j|� jj � j
jt� j� j � � �qtW d S )NT� rP rQ rR r� r� )r r! z().foor> z
().foo.barz().foo.bar().bazr@ )r? z foo.__int__r za()z().foo.bar().baz().__int__)rP rQ rR )rP rQ rR )rP rQ rR ���)r r� ri r/ r> r� r �dictr2 r@ rD � call_listr� r r! r" �__int__)r r r� rN Z last_callr r r �test_mock_calls� st
zMockTest.test_mock_callsc C sF t � }|� }|j� | j|jt� t� j� g� | j|jtj� g� d S )N)r Zwibbler/ ri r )r rx rp r r r �test_child_mock_call_equal s
z#MockTest.test_child_mock_call_equalc C sF t � }|j� j� | j|jd tj� j� � | j|jd tj� � d S )NrP r )r r> rd r� ri r r� r/ )r rx r r r � test_mock_call_not_equal_leaf s z&MockTest.test_mock_call_not_equal_leafc C sF t � }|j� j� | j|jd tj� j� � | j|jd tj� � d S )NrP r )r r> r@ r� ri r rD )r rx r r r �!test_mock_call_not_equal_non_leaf" s z*MockTest.test_mock_call_not_equal_non_leafc C s8 t � }|jdd�j� | j|jd tjdd�j� � d S )NrP )r� rQ )r r> r@ r/ ri r )r rx 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 � }|jj� | j|jd tjj� � d S )Nr )r r> r@ r� ri r rD )r rx r r r �&test_mock_call_not_equal_non_leaf_attr0 s
z/MockTest.test_mock_call_not_equal_non_leaf_attrc C s. t � }|jj� | j|jd tj� j� � d S )Nr )r r> r@ r� ri r )r rx r r r �2test_mock_call_not_equal_non_leaf_call_versus_attr6 s
z;MockTest.test_mock_call_not_equal_non_leaf_call_versus_attrc C s^ t � }|j� j� jj� | jt|jd �d� | jt|jd �d� | jt|jd �d� d S )Nr z
call.foo()rP zcall.foo().bar()rQ zcall.foo().bar().baz.bob())r r> r@ rD r� r/ rB ri )r rx r r r �test_mock_call_repr< s
z MockTest.test_mock_call_reprc C sh G dd� dt �}|� }| j|j|� | j|� |� G dd� dt �}|� }| j|j|� | j|� |� d S )Nc @ s e Zd ZdS )z+MockTest.test_subclassing.<locals>.SubclassN)r r r r r r r �SubclassE s r c @ s e Zd Zdd� ZdS )z+MockTest.test_subclassing.<locals>.Subclassc [ s
t f |�S )N)r )r rN r r r �_get_child_mockM s z;MockTest.test_subclassing.<locals>.Subclass._get_child_mockN)r r r r r r r r r L s )r r| r> �assertNotIsInstance)r r r r r r �test_subclassingD s zMockTest.test_subclassingc s� t � t� t� t� g}� fdd�}x�|D ]z}||� t|�rd|� |dd� |dd� |j� ||� |j� |jjddd� |jd�j� jd� |j� ||� q&W d S )Nc sD d}x:|D ]2}t | |�}� j|t� � j|t� � j|g � q
W d S )Nr4 r5 ri )r4 r5 ri )r� r| r
r� r/ )r �namesr? �attr)r r r � assert_attrs] s
z-MockTest.test_arg_lists.<locals>.assert_attrsrP rQ rR )r ) r r r
r �callablerg r> r@ rD )r rE r r r )r r �test_arg_listsU s$
zMockTest.test_arg_listsc C s� t � }|ddd� |ddd� | jt|j�d� |j\}}| j|d� | j|tdd�� d tdd�fd
tdd�fg}xPt||j�D ]@\}}| jt|�d� | j|d |d � | j|d |d � q�W d S ) NrP rR )r rQ r� )r! r )rQ )rP )rQ )r r/ �lenr3 r� �zipr4 )r r rM rN Z
expected_listr� r3 r r r �test_call_args_two_tuplex s
z!MockTest.test_call_args_two_tuplec C s t tdddg�d�}| j|� |� |� gdddg� | jt|� tdddgd�}| j|� |� |� gdddg� | jt|� t dd�}| j|� |� |� gd d
d g� | jt|� G d d
� d
t�}t|d�}| j|� |� t t� d�}| j|� |� |� |� gddddg� | jt|� d S )NrP rQ rR )rV 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 r/ rW r� r rJ 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�}| jt|� | j|� d� | jt|� | j|� d� q
W d S )NrR r� )rV )r r r^ r� rW r/ )r �Klass�iterablerx r r r �$test_side_effect_iterator_exceptions� s
z-MockTest.test_side_effect_iterator_exceptionsc C s� t � }tdddg�|_| j|� |� |� gdddg� | jt|� |j}| j|ttg ��� dddg|_| j|� |� |� gdddg� | jt|� |j}| j|ttg ��� t� }||_| j|� |� |� |� gddd d
g� | jt|� | j |j|� d S ) NrP rQ rR r r! r" r r r r )
r r rV r/ rW r� r| r� r r� )r r rV Z this_iterr r r �!test_side_effect_setting_iterator� s" z*MockTest.test_side_effect_setting_iteratorc C s4 t dd�}tdtg�|_| j|� |� gddg� d S )NrQ )r2 rP )r r r rV r/ )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�ddifdf ddifdd ddifd di fd f ddifg}x |D ]}|j|gd
d � q~W x<tdd �tdd�t� dd d
fD ]}| jt|j|gd
d � q�W tdd�tdd�gtdd�tdd�gtdd�tdd�gg}x |D ]}|j|d
d � �qW tdd�tdd�tdd�gtdd�tdd�gtdd�tdd�tdd�gtdd�tdd�tdd�tdd�tdd�gg}x"|D ]}| jt|j|d
d � �q�W d S )NrP rQ rR )r r� r� )r! r r� T)� any_orderr� r> r� � �rR r� )r �rP rQ )r� r! )rP rQ )r r �assert_has_callsrW rv )r r �kallsZkallZ
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�ddifd
t dd�ddd difg}t j� t jd�g}|jt j� jdd
�j� � |jt jt� i d �j dg�j� � g }xVt
� t � fD ]F}|dd� |dd
� |dd� |dd� |ddd� |j ||f� q�W t
� }|j� |jd� |j� jdd
� |jt� i d �j dg� |j ||f� xz|D ]r\}}xft
t|��D ]V}xNdD ]F}|||| � }|j|� t|�dk�rH| jt|jtt|��� �qHW �q>W �q(W d S )NrP rQ r rR r� r� )r! r� r! )r )r> �rR r� )r% )rP )rP rQ rR )r r> r@ �extendZspamrD r� Zbamr� rS r r r� �ranger r" rW rv r� �reversed) r Zkalls1Zkalls2rE r r# r �stepZtheser r r �test_assert_has_calls� s<
"
z MockTest.test_assert_has_callsc C s� ddd�}t |d�}|dddd� |dd d
d d � |d
ddd� ddi fdddd ifdi fg}|j|� |j|dd� |j|dd � � |j|dd � dd� |j|d d� � |j|d d� dd� tt|��}| jt�� |j|� W d Q R X |j|dd� | jt�� |j|dd � � W d Q R X |j|dd � dd� | jt�� |j|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\ s z<MockTest.test_assert_has_calls_with_function_spec.<locals>.f)rH rP rQ rR )r! r" r� r� r� r )r" r# �
� � )r" r� r# T)r )N)rP rQ rR )r� r� r� )r+ r, r- r� r� r� r� )r r" r� r( rW rv )r r\ r Zcallsr r r �(test_assert_has_calls_with_function_spec s0
z1MockTest.test_assert_has_calls_with_function_specc C s� t � }|dd� |dd� |ddd� |jdd� |jdd� |jddd� | jt|j� | jt|jdd� | jt|jdd� d S )NrP rQ rR )r r� )r! r� )r �assert_any_callrW rv )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 � |jddd� |jdddd
� |jdd d
d � |jdd d
d d� | jt|jdddd� | jt��}|jdd� W d Q R X | j|jjt� d S )Nc S s d S )Nr )r r! r" r# r r r r\ U s z;MockTest.test_assert_any_call_with_function_spec.<locals>.f)rH rP rQ rR )r! r" r� r� r� r )r" r# )r r! r" )r r! r" r# rz )r{ )N)r r/ rW rv r| r} r~ r )r r\ r r� r r r �'test_assert_any_call_with_function_specT s
z0MockTest.test_assert_any_call_with_function_specc C sj dd� }t � }||_t|�t|�jg}x>|D ]6}|dd� |dd� | j|jtdd�tdd�g� q,W d S )Nc S s d S )Nr )r r! r r r r\ i s z3MockTest.test_mock_calls_create_autospec.<locals>.frP rQ rR r� )r r\ r r/ ri r )r r\ �objZfuncsr� r r r � test_mock_calls_create_autospech s
z(MockTest.test_mock_calls_create_autospecc C s$ t jt� dd�}| jdt|�� d S )NZ
sweet_func)r? )r r rJ rA rB )r rx r r r � test_create_autospec_with_name{ s z'MockTest.test_create_autospec_with_namec C s` t � }| jt�� |j� W d Q R X | jt�� |j� W d Q R X t dd�}|j� |j� d S )NT)Zunsafe)r rW r� Zassert_foo_callZassret_foo_call)r rx r r r �test_mock_unsafe� s
zMockTest.test_mock_unsafec C s< t � }|jj� |j� | jt�� |jj� W d Q R X d S )N)r �helloZassert_not_calledrW rv )r rx r r r �test_assert_not_called� s
z MockTest.test_assert_not_calledc C sN t � }| jt�� |jj� W d Q R X |j� |jj� |j� |jj� d S )N)r rW rv r6 Z
assert_called)r rx r r r �test_assert_called� s
zMockTest.test_assert_calledc C sd t � }| jt�� |jj� W d Q R X |j� |jj� |j� | jt�� |jj� W d Q R X d S )N)r rW rv r6 Zassert_called_once)r rx r r r �test_assert_called_once� s
z MockTest.test_assert_called_oncec C s0 t � }|jddd� d}| jt|jj�|� d S )Nr6 Zhero)r? Zdaddyz call(daddy='hero', name='hello'))r r6 r/ rB r3 )r rx �textr r r �test_sorted_call_signature� s z#MockTest.test_sorted_call_signaturec C sV t j� }t jdd�}t� }|j� |jdd� | j|jd |� | j|jd |� d S )N� r6 r rP )r r� �indexr r/ r5 )r r" r rx r r r �test_override_tuple_methods� s z$MockTest.test_override_tuple_methodsc C s> t dddgd�}|jddd� | j|jt � | j|jd � d S )Nr+ rQ rR )r2 rV T)r rg r| r2 r/ rV )r rx r r r � test_reset_return_sideeffect� s z%MockTest.test_reset_return_sideeffectc C s< t dddgd�}|jdd� | j|jt � | j|jd � d S )Nr+ rQ rR )r2 rV T)r2 )r rg r| r2 r� rV )r rx r r r �test_reset_return� s zMockTest.test_reset_returnc C s< t dddgd�}|jdd� | j|jd� | j|jd � d S )Nr+ rQ rR )r2 rV T)rV )r rg r/ r2 rV )r rx 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}|j|f� fdd� � |j|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 | j t
t|d
� | j t
t|d� d|k�r:| j t
t|dd � |j |f|� | j t
t|d � |j | j t
t|d� d|kr�| j 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>._OnerP N)r r r r� r r r r �_One� s rB c @ s e Zd ZdZdS )z)MockTest.test_mock_add_spec.<locals>._TworQ N)r r r r� r r r r �_Two� s rC c @ s e Zd Zd Z ZZdS )z-MockTest.test_mock_add_spec.<locals>.AnythingZfourN)r r r r� r� �threer r r r �Anything� s rE c s
| � d�S )N)rH r )�K)rE r r r� � s z-MockTest.test_mock_add_spec.<locals>.<lambda>c s
| � d�S )N)rI r )rF )rE r r r� � s T)rI r� r� rD rI )rJ r r r
r r� r� r� r� r� rD �
mock_add_specrW r� r� �setattr) r rB rC Zklassesr rN r ZOneZTwor )rE r �test_mock_add_spec� s>
zMockTest.test_mock_add_specc s� xzt tfD ]n}|� � t� � � jt� | jtt� � |� � � d d� j_� jt� | j t� �d� | jt� fdd�� q
W d S )Nr> r� c s � d S )Nr> r r )r r r r�
s z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� rG rJ rW r r� r2 r/ )r r r )r r � test_mock_add_spec_magic_methods� s
z)MockTest.test_mock_add_spec_magic_methodsc C s� x�t tttfD ]�}|� }t� |_|j� | j|jtj� g� | j|jtj� g� |� }tdd�|_ |j � | j|jg � | j|jg � |� }t� � |_
|j
� | j|jg � | j|jg � qW d S )Nr? )r? ) r
r r r r> r/ r5 r ri r@ rD )r r r r r r �test_adding_child_mock s
z MockTest.test_adding_child_mockc C sD x>t tfD ]2}|� }t� |_|� � | j|jt� t� � g� q
W d S )N)r r r2 r/ ri 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� }tj |d�}tj |d�}|j� }| j|j� |j� }| j|j� |j|d� |j|d� |j� |j� | j |j
t j� t j� 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 2 s r r� r� ) rJ r r r� r� r� � attach_mockr� r� r/ ri r )r r ZmanagerZp1Zp2Zmock_oneZmock_twor r r �test_manager_mock1 s zMockTest.test_manager_mockc C sl xft tfD ]Z}|� }t dd�|_tdd�|_t|� t|� | j|jtj� tj� g� | j|j g � q
W d S )NrR )r2 g @)
r r r� � __float__r� �floatr/ ri r r5 )r r rx r r r � test_magic_methods_mock_callsG s z&MockTest.test_magic_methods_mock_callsc C s< t jdd�}|d�}|j� }|d�}|j� }| j||� d S )N�data)� read_dataza-namez another-name)r rl �readr/ )r Z mocked_open�f1Zf1_data�f2Zf2_datar r r � test_mock_open_reuse_issue_21750R s z)MockTest.test_mock_open_reuse_issue_21750c
C sX t jt jdd��}t jd|��0 |j}|j}td�|_dd� }| jt|� W d Q R X d S )NZJLV)r? ztempfile.NamedTemporaryFilez Test 2 Errorc S s t j� jd� d S )NZasd)�tempfileZNamedTemporaryFile�writer r r r �attempta s z.MockTest.test_mock_open_write.<locals>.attempt) r rl r r r2 rY �OSErrorrV rW )r Zmock_namedtempZmock_filehandleZ
mock_writerZ r r r �test_mock_open_writeZ s
z MockTest.test_mock_open_writec C sJ t jdd�}dd� |jj_|� j� }|� j� }| jd|� | jd|� d S )Nzfoo
barn)rS c W s dS )N�abcr )rM r r r r� g s z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>r] )r rl r2 �readlinerV r/ )r Zmopen�first�secondr r r � test_mock_open_alter_readlinee s
z&MockTest.test_mock_open_alter_readlinec C s� t jdd�}|d�}|j� | jd|j� � | jd|j� � | jd|j� � | jd|j� � | jg |j� � | jg |j� � d S )Nr> )rS r@ r� )r rl rT r/ r^ � readlines)r Z_openr r r r �test_mock_open_after_eofm s z!MockTest.test_mock_open_after_eofc C s �xt tfD ]�}|� }t|�}||_| j|� |� | jt|�|� |j� | j|� |� | jt|�|� |� }|j|_| j dt|j�� | j dt|j�� |j� | j dt|j�� | j dt|j�� |� }t|�}|� |_||j_| jt|�|� | jt|j� �|� q W d S )Nz
name='mock.a')
r r rB r2 r� r/ rg r r! rA )r r rx Z
original_reprr r r �test_mock_parentsy s, zMockTest.test_mock_parentsc C s� t tttf}x�|D ]�}x�|D ]z}|� }|dd�}|j|d� | j|j|� | jdt|�� |jj d� | j
|j t jj d�g� | j
|j
t jj d�g� q W qW d S )Nr> )r? r@ zname='mock.bar'rP )r r r r
rM r� r@ rA rB rD r/ ri r r5 )r �classesr �Klass2rx �m2r r r �test_attach_mock� s
zMockTest.test_attach_mockc C s� t tttf}xxt tfD ]l}xf|D ]^}|� }|dd�}|j|d� | j|� |� | jdt|�� |j� | j |j
t � j� j � � q W qW d S )Nr> )r? r2 z
name='mock()')
r r r r
rM r� rA rB r> r/ ri r r� )r re r rf rx rg r r r � test_attach_mock_return_value� s
z&MockTest.test_attach_mock_return_valuec C sj xdt � t� t� t� fD ]L}| jt|d�� |`| jt|d�� |`| jt|d�� | j t
t |d� qW d S )Nrx r\ ) r r r r
r1 �hasattrrx r- r\ rW r� r� )r r r r r �test_attribute_deletion� s z MockTest.test_attribute_deletionc C s, t � }d|_|`|j� | jt|d�� d S )NTre )r re rg r- rj )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 ](}| j|t� t|_| j|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 | j tj� | jttj�t� | jttj� j�t� d S )N)�assertIsNotNoner r? r/ r� r )r r r r �test_name_attribute_of_call� s z$MockTest.test_name_attribute_of_callc C s8 | j tj� | jttj�t� | jttj� j�t� d S )N)rn r rf r/ r� r )r r r r � test_parent_attribute_of_call� s z&MockTest.test_parent_attribute_of_callN)pr r r r, r< r= rF rK r[ r] r` r� Z
skipUnlessr� �platformrc rj rk rm rq 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. r0 r1 r3 r4 r5 r7 r8 r9 r; r> r? r@ rA rI rJ rK rL rN rQ rW r\ ra rc rd rh ri rk rl rm ro rp r r r r r* * s� )"
7N
#
,(!
5
r* �__main__)r� r� rX r� Z unittest.test.testmock.supportr r Z
unittest.mockr r r r r r r
r r r
r rJ r r r� r* r �mainr r r r �<module> s0 4
H