�0�V&� � @ sn d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m
Z
m Z m Z m
Z
m Z m Z m Z d d l m Z d d l m Z m Z m Z y e Wn e k
r� e Z Yn XGd d � d e � Z Gd d � d e � Z Gd d
� d
e � Z Gd d � d e � Z Gd
d � d e j � Z e! d k rje j" � d S)� N) �call�DEFAULT�patch�sentinel� MagicMock�Mock�NonCallableMock�NonCallableMagicMock�create_autospec)� _CallList)�callable� is_instance�nextc @ s: e Z d Z d d � Z d d � Z d d � Z e Z d S)�Iterc C s t d d d d g � | _ d S)N�this�is�an�iter)r �thing)�self� r � /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)r r )r r r r r ' s z Iter.nextN)�__name__�
__module__� __qualname__r r r �__next__r r r r r s r c @ sI e Z d Z d d d � Z e d d d � � Z e d d d � � Z d S)� 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 �cmeth1 s zSomething.cmethc C s d S)Nr )r r r! r" r r r �smeth5 s zSomething.smeth)r r r r# � classmethodr% � staticmethodr& r r r r r - s
r c @ s e Z d Z d S)�SubclassN)r r r r r r r r) : s r) c @ s e Z d Z d Z d Z d S)�Thing� �barN)r r r � attribute�foor r r r r* > s r* c @ sB e Z 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
e j d e
j k d � 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 d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z! d9 d: � Z" d; d< � Z# d= d> � Z$ d? d@ � Z% dA dB � Z& dC dD � Z' dE dF � Z( dG dH � Z) dI dJ � Z* dK dL � Z+ e j, e- j. dM � dN dO � � Z/ e j, e- j. dM � dP dQ � � Z0 dR dS � Z1 dT dU � Z2 dV dW � Z3 dX dY � Z4 dZ d[ � Z5 d\ d] � Z6 d^ d_ � Z7 d` da � Z8 db dc � Z9 dd de � Z: df dg � Z; dh di � Z< dj dk � Z= dl dm � Z> dn do � Z? dp dq � Z@ dr ds � ZA dt du � ZB dv dw � ZC dx dy � ZD dz d{ � ZE d| d} � ZF d~ d � ZG d� d� � ZH d� d� � ZI d� d� � ZJ d� d� � ZK d� d� � ZL d� d� � ZM d� d� � ZN d� d� � ZO d� d� � ZP d� d� � ZQ d� d� � ZR d� d� � ZS d� d� � ZT d� d� � ZU d� d� � ZV d� d� � ZW d� d� � ZX d� d� � ZY d� d� � ZZ d� d� � Z[ e j\ d� d� � � Z] d� S)��MockTestc C s t d � d S)Nzfrom mock import *)�exec)r r r r �test_allF s zMockTest.test_allc C s� t � } | j | j d � | j | j d d � | j t | j t � d � | j | j d d � | j | j g d � | j | j
g d � | j d | j d � | j
| j d
� | j
| j d � | j | j i 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 �mockr r r �test_constructorM s( zMockTest.test_constructorc C s& t d d � } | j | j d � d S)Nr7 z(return value in constructor not honoured)r r= r7 )r rA r r r � test_return_value_in_constructori s z)MockTest.test_return_value_in_constructorc C sP t d d � } | j d t | � � | j d t | � t | � � t � d f t d d � d f g } x� | D]� \ } } | j d | t | j � � | j d | t | j � � � | j d | t | j � j � � | j d | t | � � � | j d
| t | � � � � | j d | t | � � j j j � j � � qp Wd S) N�namer. z'%s'rA r, z%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 rA �mocksrD r r r � test_repro s $ #
zMockTest.test_reprc C sa Gd d � d t � } t d | � } | j d t | � � t d | � � } | j d t | � � t d | � } | j d t | � � t d | � � } | j d t | � � t d | d d � } | j d t | � � | j d t | � � t d d � } | j d t | � � t � } | j d t | � � t d d g � } | j d t | � � d S)
Nc @ s e Z d Z d S)z'MockTest.test_repr_with_spec.<locals>.XN)r r r r r r r �X� s rK �specz
spec='X' �spec_setz spec_set='X' rD r. z name='foo' )�objectr rE rF r; )r rK rA r r r �test_repr_with_spec s$ z MockTest.test_repr_with_specc s t � } d d � } | | _ | j t | d d d d �| j d d d d �d d d g � � f d d � } | | _ | j | � | � | � g d d d g d � t d t j � } | j | j t j d
� d d � } t d | d
t j � } | j | � t j � d S)Nc _ s t d � � d S)NZkablooie)� SystemError)�args�kwargsr r r �effect� s z)MockTest.test_side_effect.<locals>.effect� � �fish� c s
� j � S)N)�popr )�resultsr r rS � 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 rZ � s z.MockTest.test_side_effect.<locals>.side_effectr7 ) r rZ � assertRaisesrP �assert_called_withr4 r �
SideEffectZRETURN)r rA rS rZ r )rY r �test_side_effect� s ' zMockTest.test_side_effectc s� d d d g � � f d d � } d d � } t | � } d d d g | _ | j | � | � | � g d d d g d � d d d g � t | � } | | _ | j | � | � | � g d d d g d � d S)
NrT rU rW c s
� j � S)N)rX r )rY r r rS � 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
rZ r4 )r rS r_ rA r )rY r �test_autospec_side_effect� s ' 'z"MockTest.test_autospec_side_effectc C s> d d � } t | � } t d � | _ | j t d | � d S)Nc S s d S)Nr r r r r r_ � s z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r
�
ValueErrorrZ �assertRaisesRegex)r r_ rA r r r �#test_autospec_side_effect_exception� s z,MockTest.test_autospec_side_effect_exception�javaz This test only applies to Jythonc
C s� d d l } t d | j j d � � } y | d d d d �Wn | j j k
rX Yn X| j d � | j d d d d �d S) Nr rZ zBoom!rT rU rV rW zjava exception not raised)rd r ZlangZRuntimeException�failr\ )r rd rA r r r � test_java_exception_side_effect� s
z(MockTest.test_java_exception_side_effectc C s� t � } d g } t d d d | d | � } | t j d t j �| j } | j � t j | _ | j } | � | j � | j | j
d d � | j | j | d � | j | j | d � | j
| j d � | j | j d
d � | j | j d d � | j | j g d
� | j | j g d | j g f � | j | j g � | j | j t j d � | j | j | d � | j
| j d � | j | j d | i d � | j | j | d � | j
| j j d � d S)N� somethingrD Zchild�parentrL 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
SomethingElserg r] rZ r7 �
reset_mockr4 �
_mock_namer> r? r2 r3 r5 r8 r9 r: �
mock_callsr@ )r rh rL rA rg r7 r r r �test_reset_mock� sD
zMockTest.test_reset_mockc C s t � } | | _ | j � d S)N)r r7 ri )r rA r r r �test_reset_mock_recursion
s z"MockTest.test_reset_mock_recursionc C s t j � } | j � d S)N)rA � 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 � } | j t | j t � d � | � } | j | � | d � | j � | t j � } | j | j d � | j | j d d � | j | j
t j f i f d � | j | j t j f i f g d � t j | _ | t j d t j
�} | j | t j d � | j | j d
d � | j | j
t j f d t j
i f d � | j | j t j f i f t j f d t j
i f g d � d S)
Nz%Default return_value should be a Mockz'different result from consecutive callszcalled not setrT zcall_count incoreectzcall_args not setz(call_args_list not initialised correctly�keyzincorrect return valuerU zcall_count incorrectzcall_args_list not set)r r6 r
r7 r4 ri r �Argr3 r5 r8 r9 Z ReturnValueZKeyArg)r rA �resultZret_valr r r � test_call s4
zMockTest.test_callc C s� t � } | � | t j � | d t j � | t j d t j �| j | j f t j f f d t j i f t j f d t j i f g � | j | j t j f d t j i f � | j | j d k � d S)N�kwza long sequence)r r rq ZKwargr4 r9 r8 r2 )r rA r r r �test_call_args_comparison: s
z"MockTest.test_call_args_comparisonc C s� t � } | � | j � | j t | j d � | j � | j t | j � | d d d d d d d �| j d d d d d d d �d S)NrT rU rW r rV r Znothing)r r\ r[ �AssertionErrorri )r rA r r r �test_assert_called_withM s
z MockTest.test_assert_called_withc
C s� d d d � } t d | � } | d d d d d �| j d d d � | j d d d d d d � | j t | j d d d d d �| j t � � } | j d
d � Wd QRXt | j d � r� | j | j j t � 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>.frL rT r rU r! rW r �e� � __cause__) r r\ r[ rv �hasattr� exception�assertIsInstancerz � TypeError)r r_ rA �cmr r r �%test_assert_called_with_function_spec\ s z.MockTest.test_assert_called_with_function_specc s� � f d d � } t d t � j � } | | � t d t j � } | | � t d t � j � } | | � t d t j � } | | � t d t � j � } | | � d S)Nc sk | d d d d d �| j d d d � | j d d d d d d � � j t | j d d d d d �d S)NrT r rU r! rW r )r\ r[ rv )rA )r r r �_checko s
z<MockTest.test_assert_called_with_method_spec.<locals>._checkrL )r r r# r% r& )r r� rA r )r r �#test_assert_called_with_method_specn s
z,MockTest.test_assert_called_with_method_specc s� t � � � � � j � � � | j t � j � � j � | j t � j � � d d d d �� j d d d d �� j � � d d d d �| j t � f d d � � d S)Nr. r, rH rU c s � j d d d d �S)NZbobr, rH rU )�assert_called_once_withr )rA r r �<lambda>� s z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r r� r[ rv ri )r r )rA r � test_assert_called_once_with� s
z%MockTest.test_assert_called_once_withc
C s" d d d � } t d | � } | d d d d d �| j d d d � | j d d d d d d � | j t | j d d d d d �| j t � � } | j d
d � Wd QRXt | j d � r� | j | j j t � | d
d d � | j t | j d d d � | j t | j d
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>.frL rT r rU r! rW r rx ry rz � � r+ ) r r� r[ rv r{ r| r} rz r~ )r r_ rA r r r r �*test_assert_called_once_with_function_spec� s
z3MockTest.test_assert_called_once_with_function_specc C s� t � } | j } | j t | t � d � | j | j | d � t � } d | j _ | j | j � d d � | j | j j d � d S)Nzattribute isn't a mockz+different attributes returned for same namerW zmethod returned wrong valuez!method didn't record being called)r rg r6 r
r4 r7 r3 )r rA rg r r r �#test_attribute_access_returns_mocks� s z,MockTest.test_attribute_access_returns_mocksc C sB t � } | j } | j | j d d � | j | j | d � d S)Nrg z attribute name not set correctlyz"attribute parent not set correctly)r rg r4 rj r> )r rA rg r r r �(test_attributes_have_name_and_parent_set� s z1MockTest.test_attributes_have_name_and_parent_setc C s� t � } | j d d d �| j j d d t j �| j | j j d d d t j i f g d � | j | j d d d d i f d d
d t j i f g d � d S) NrW rV r+ Zcakerg z#method calls not recorded correctlyzsomething_else.something)r+ )rW )r+ )r rg �something_elser ZCaker4 r: )r rA r r r �test_method_calls_recorded� s z#MockTest.test_method_calls_recordedc C s8 t � } | j � | j | j d g � | j | j d f i f g � t � } | j d � | j | j d g � | j | j d d
i f g � t � } | j d d � | j | j d d d i f g � | j | j d f d d i f g � t � } | j d d d �| j | j d d d d i f g � d S) Nrg � different�xrT ZsomeZmore)� something�r� )r� r� )r� )r� )r rg r4 r: )r rA r r r � test_method_calls_compare_easily� s"
"% z)MockTest.test_method_calls_compare_easilyc C sa xZ d g d f D]I } x@ d D]8 } t | | i � } | j | j t d t | d � q Wq Wd S)Nrg rL rM z-Mock object has no attribute 'something_else'r� )z something)zspeczspec_set)r rg rb �AttributeError�getattr)r rL �argrA r r r � test_only_allowed_methods_exist� s
z(MockTest.test_only_allowed_methods_existc sU Gd d � d t � } � f d d � } | t d | � � | t d | � � � d S)Nc @ s( e Z d Z d Z d Z d d � Z d S)z*MockTest.test_from_spec.<locals>.SomethingrW 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 sK | j | j | j � j t d t | d � � j t d t | d � d S)Nz Mock object has no attribute 'z'�zz)Mock object has no attribute '__foobar__'Z
__foobar__)r� r� r� rb r� r� )rA )r r r �test_attributes s
z0MockTest.test_from_spec.<locals>.test_attributesrL )rN r )r r r� r )r r �test_from_spec s zMockTest.test_from_specc C se t � } t d | � } | j | � | � � | j � | d d d d �| j d d d d �d S)N�wrapsrT rU rV rW )r r4 ri r\ )r �realrA r r r �test_wraps_calls s
zMockTest.test_wraps_callsc C sH t � } t d | � } d | _ | j | � d � | j | j � d S)Nr� rW )r r7 r4 r2 r3 )r r� rA r r r �,test_wraps_call_with_nondefault_return_value( s
z5MockTest.test_wraps_call_with_nondefault_return_valuec s� Gd d � d t � } | � } t d | � � | j � j � | j � � | j t � f d d � � | j � j | j � � j j d d d d �} | j j j d d d d �| j | | j j � � d S)
Nc @ s e Z d Z e � Z d S)z,MockTest.test_wraps_attributes.<locals>.RealN)r r r r r- r r r r �Real3 s r� r� c s � j S)N)rV r )rA r r r� : s z0MockTest.test_wraps_attributes.<locals>.<lambda>rT rU rV rW ) rN r r4 r- r[ r� ZassertNotEqualZfrogr\ )r r� r� rr r )rA r �test_wraps_attributes2 s z MockTest.test_wraps_attributesc C sH t d t � } | j t | � t d t d � � } | j t | � d S)NrZ r. )r r� r[ )r rA r r r � test_exceptional_side_effectB s z%MockTest.test_exceptional_side_effectc C sH t d t � } | j t | � t d t d � � } | j t | � d S)NrZ r. )r �KeyboardInterruptr[ )r rA r r r � test_baseexceptional_side_effectJ s z)MockTest.test_baseexceptional_side_effectc C s# t � } | j t d | j � d S)Nz
Not called)r rb rv r\ )r rA r r r � test_assert_called_with_messageR s z(MockTest.test_assert_called_with_messagec C s) t d d � } | j t d | j � d S)NrD Zgeoffreyz'Expected 'geoffrey' to be called once\.)r rb rv r� )r rA r r r �$test_assert_called_once_with_messageX s z-MockTest.test_assert_called_once_with_messagec sE t � � | j t � f d d � � d � _ | j � j d � d S)Nc s � j S)N)r r )rA r r r� a s z'MockTest.test__name__.<locals>.<lambda>r. )r r[ r� r r4 )r r )rA r � test__name___ s zMockTest.test__name__c C se Gd d � d t � } t d | d g � � } | j d � | j j d � | j t t | d � d S)Nc @ s e Z d Z d S)z-MockTest.test_spec_list_subclass.<locals>.SubN)r r r r r r r �Subh s r� rL r. rW )�listr �appendr\ r[ r� r� )r r� rA r r r �test_spec_list_subclassg s
z MockTest.test_spec_list_subclassc C s� Gd d � d t � } t d | � } | j | | � t d | � � } | j | | � | j | j | � | j t � j j d � t d | � } | j | | � t d | � � } | j | | � d S)Nc @ s e Z d Z d S)z#MockTest.test_spec_class.<locals>.XN)r r r r r r r rK r s rK rL r rM )rN r r} �assertIs� __class__r4 r )r rK rA r r r �test_spec_classq s zMockTest.test_spec_classc sl Gd d � d t � } t d | � � d � _ t d | � � � f d d � } d � _ | j t | � d S)Nc @ s e Z d Z d Z d S)z8MockTest.test_setting_attribute_with_spec_set.<locals>.XrW N)r r r r� r r r r rK � s rK rL r. rM c s
d � _ d S)Nr. )r� r )rA r r �set_attr� s z?MockTest.test_setting_attribute_with_spec_set.<locals>.set_attr)rN r r� r� r[ r� )r rK r� r )rA r �$test_setting_attribute_with_spec_set� s z-MockTest.test_setting_attribute_with_spec_setc C sF t j � } | j t j | � t j t d � � t j t � � d S)Ng e��A)�sys�getrecursionlimit�
addCleanup�setrecursionlimit�int�copyr )r �currentr r r � test_copy� s zMockTest.test_copyz no old style classes in Python 3c s� Gd d � d � } t d | � � d � _ | j t � f d d � � t d | � � � d � _ | j t � f d d � � d S)Nc @ s e Z d Z d Z d S)z1MockTest.test_spec_old_style_classes.<locals>.Foo� N)r r r r, r r r r �Foo� s r� rL r+ c s � j S)N)r. r )rA r r r� � s z6MockTest.test_spec_old_style_classes.<locals>.<lambda>c s � j S)N)r. r )rA r r r� � s )r r, r[ r� )r r� r )rA r �test_spec_old_style_classes� s z$MockTest.test_spec_old_style_classesc s� Gd d � d � } t d | � � d � _ | j t � f d d � � � f d d � } | j t | � t d | � � � d � _ | j t � f d d � � � f d
d � } | j t | � d S) Nc @ s e Z d Z d Z d S)z5MockTest.test_spec_set_old_style_classes.<locals>.Foor� N)r r r r, r r r r r� � s r� rM r+ c s � j S)N)r. r )rA r r r� � s z:MockTest.test_spec_set_old_style_classes.<locals>.<lambda>c s
d � _ d S)NrW )r. r )rA r r �_set� s z6MockTest.test_spec_set_old_style_classes.<locals>._setc s � j S)N)r. r )rA r r r� � s c s
d � _ d S)NrW )r. r )rA r r r� � s )r r, r[ r� )r r� r� r )rA r � test_spec_set_old_style_classes� s z(MockTest.test_spec_set_old_style_classesc s� Gd d � d t � } | d | � � | j � j d � � f d d � } | j t | � � f d d � } | j t | � d S)Nc @ s7 e Z d Z d d � Z d d � Z e e e � Z d S)z8MockTest.test_subclass_with_properties.<locals>.SubClassc S s d S)NrW 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 r� � s z=MockTest.test_subclass_with_properties.<locals>.SubClass._setN)r r r r� r� �property�some_attributer r r r �SubClass� s r� rM rW c s
d � _ d S)NrW )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 r4 r� r[ r� r� )r r� r� r )r� r � test_subclass_with_properties� s z&MockTest.test_subclass_with_propertiesc C sU t � } d d � } | t | � _ | d � | j d � | j t | d d � 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� r\ r[ r~ )r rA r� r r r �test_setting_call� s
zMockTest.test_setting_callc C s� t � } t t | � � } t d d � t t � D� � } | j t � | | � | j | j f | j d t | � � | j d t | � � d | _ | _ | j d t | � � | j d t | � � d d � | _ | j d t | � � d S)
Nc S s% g | ] } | j d � s | � q S)�_)�
startswith)�.0�mr 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� � s z#MockTest.test_dir.<locals>.<lambda>r )
r �set�dirr4 r r rE r! r" r )r rA �attrs�
type_attrsr r r �test_dir� s zMockTest.test_dirc C s| t d t j � } t t t j � � } t t | � � } | j t � | | � d | _ | j t | � j d � d � d S)NrL rW �versionrT )r �unittest�TestCaser� r� r4 r� �count)r rA Ztestcase_attrsr� r r r �test_dir_from_spec� s zMockTest.test_dir_from_specc
C ss t j t d d � } | j � zB t t t � � � } t t t � � } | j t � | | � Wd | j � Xd S)NZ
FILTER_DIRF) r rN rA �startr� r� r r4 �stop)r Zpatcherr� r� r r r �test_filter_dir s
zMockTest.test_filter_dirc C s� t d d � } | j | j d � t d d � } | j | j d � d t d d d t � i } t | � } | j t | � | j | j j � d � | j | j t � t � } | j | � | j t | � | j | j j � d � | j | j t � d S)Nr. r, rZ zfoo.bar.return_value�! ) r r4 r. r �KeyErrorr[ r, r} Zconfigure_mock)r rA rR r r r �test_configure_mock s
z MockTest.test_configure_mockc O sp y | | | � Wn( t j � d } | j | | � Yn X| j d | f � t | � } | j | | � d S)NrT zException %r not raised)r� �exc_infor} re �strr4 )r r| �message�funcrQ rR �instance�msgr r r �assertRaisesWithMsg( s z MockTest.assertRaisesWithMsgc
C s� t � } d } d } | j t | | f | j d d d d d �| j d d d d d �| j j | j j g } xL | D]D } d } d } d
} | j t | | | f | d d d d d �qy WxC | D]; } d } d } d
} | j t | | | f | d d �q� WxF | D]> } d } d } d
} | j t | | | f | d d
d � qWx= | D]5 } d } d } d
} | j t | | | f | � qWWd S)Nzmock(1, '2', 3, bar='foo')z Expected call: %s
Not calledrT �2rW 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)rU zfoo())r r� rv r\ r. r� )r rA �expectedr� Z assertersr# �actualr r r �'test_assert_called_with_failure_message6 sF
z0MockTest.test_assert_called_with_failure_messagec C s t � } | j | j g k d � t � } | � d f i f g } | j | j | � | j � | j t j � � | j | j | � | j | j j d f i f g � t � } | � j d d d d d d d �d f i f d
d t d d d d � f g } | j | j | � | j | j j j d d t d d d d � f g � | j | j j d d t d d d d � f g � t � } | � j j � j
� d f i f d f i f d
f i f g } | j | j | � | j | � j t j j � j
� j � � xt � t d d � f D]�} t | � } t | j � d f i f g } | j | j | � t | � } | j
� � d f i f d f i f g } | j | j | � | j | j
� j t � g � t | � } | d � d � d � | j | j t d � d � d � j � � | j | � j t d � d � j � � | j | � � j t d � j � � t | � } | d � d � d � j
j j d � | j | j t d � d � d � j
j j d � j � � | j | � j t d � d � j
j j d � j � � | j | � � j t d � j
j j d � j � � t | � } t | � j j � j
� � d f i f } | j | j d | � | j | � j t j j � j
� j � j � � | j | � j j � j t j
� j � j � � | j | � j j � j
j t � j � j � � qWd S)NT� rT rU rW r r� r r� z().foor. z
().foo.barz().foo.bar().bazrD r, z foo.__int__za()z().foo.bar().baz().__int__)rT rU rW )rT rU rW )rT rU rW ���)r r� rk r4 r. r� r �dictr7 r, rH � call_listr� r r r! �__int__)r rA r� rR Z last_callr r r �test_mock_callsm st
" "-""
+(% % .(" "zMockTest.test_mock_callsc C s� Gd d � d t � } | � } | j | j | � | j | � | � Gd d � d t � } | � } | j | j | � | j | � | � d S)Nc @ s e Z d Z d S)z+MockTest.test_subclassing.<locals>.SubclassN)r r r r r r r r) � s r) c @ s e Z d Z d d � Z d S)z+MockTest.test_subclassing.<locals>.Subclassc [ s
t | � S)N)r )r rR r r r �_get_child_mock� s z;MockTest.test_subclassing.<locals>.Subclass._get_child_mockN)r r r r� r r r r r) � s )r r} r. �assertNotIsInstance)r r) rA r r r �test_subclassing� s zMockTest.test_subclassingc s� t � t � t � t � g } � f d d � } x� | D]� } | | � t | � r� | � | d d � | d d � | j � | | � | j � | j j d d d �| j d � j � j d � | j � | | � q7 Wd S)Nc sZ d } xM | D]E } t | | � } � j | t � � j | t � � j | g � q
Wd S)Nr9 r: rk )zcall_args_listz method_callsz
mock_calls)r� r} r r� r4 )rA �namesrD �attr)r r r � assert_attrs� s
z-MockTest.test_arg_lists.<locals>.assert_attrsrT rU r rW ) r r r r r ri r. r, rH )r rI r� rA r )r r �test_arg_lists� s$
zMockTest.test_arg_listsc C s t � } | d d d �| d d d �| j t | j � d � | j \ } } | j | d � | j | t d d � � d t d d � f d
t d d � f g } xf t | | j � D]R \ } } | j t | � d � | j | d | d � | j | d | d � q� Wd S) NrT r rW rU r r� r )rU )rT )rU )r r4 �lenr8 r� �zipr9 )r rA rQ rR Z
expected_listr� r8 r r r �test_call_args_two_tuple� s * z!MockTest.test_call_args_two_tuplec C s� t d t d d d g � � } | j | � | � | � g d d d g � | j t | � t d d d d g � } | j | � | � | � g d d d g � | j t | � t d d � } | j | � | � | � g d d
d g � | j t | � Gd d
� d
t � } t d | � } | j | � | � t d t � � } | j | � | � | � | � g d d d d g � | j t | � d S)NrZ rT rU rW r r r! Zghi�g�h�ic @ s e Z d Z d S)z/MockTest.test_side_effect_iterator.<locals>.FooN)r r r r r r r r�
s r� r r r r ) r r r4 r[ �
StopIterationr rN r} r )r rA r� r r r �test_side_effect_iterator s +++!z"MockTest.test_side_effect_iteratorc C s` t � } t d d d g � | _ | j | � | � | � g d d d g � | j t | � | j } | j | t t g � � � d d d g | _ | j | � | � | � g d d d g � | j t | � | j } | j | t t g � � � t � } | | _ | j | � | � | � | � g d d d d
g � | j t | � | j | j | � d S) NrT rU rW r r r! r r r r )
r r rZ r4 r[ r� r} r� r r� )r rA rZ Z this_iterr r r �!test_side_effect_setting_iterator s" + + !z*MockTest.test_side_effect_setting_iteratorc C s� x{ t t f D]m } t d t d f } | d | � } | j t | � | j | � d � | j t | � | j | � d � q
Wd S)NrW r+ rZ )r r ra r� r[ r4 )r �Klass�iterabler� r r r �$test_side_effect_iterator_exceptions. s z-MockTest.test_side_effect_iterator_exceptionsc C sJ t d d � } t d t g � | _ | j | � | � g d d g � d S)Nr7 rU rT )r r r rZ r4 )r rA r r r �!test_side_effect_iterator_default8 s z*MockTest.test_side_effect_iterator_defaultc C s� t � } | d d � | d d � | d d � | d d � | d d � t d d � d d i f d f d d i f d d d d i f d d i f d f d d i f g } x$ | D] } | j | g d d
�q� WxT t d d � t d d � t � d d d f D]% } | j t | j | g d d
�q� Wt d d � t d d � g t d d � t d d � g t d d � t d d � g g } x! | D] } | j | d d
�q�Wt d d � t d d � t d d � g t d d � t d d � g t d d � t d d � t d
d � g t d d � t d d � t d d � t d d � t d d � g g } x* | D]" } | j t | j | d d
�qXWd S)NrT rU r rW r� r r+ r� � any_orderTr� r. r� r� �rW r� )r �rT rU )r� r )rT rU )r r �assert_has_callsr[ rv )r rA �kallsZkallZ
kall_listsZ kall_listr r r � test_assert_has_calls_any_order= s:
$
4 !
''E
z(MockTest.test_assert_has_calls_any_orderc C s3 t d d � d d i f d t d d � d d d d i f g } t j � t j d � g } | j t j � j d d � j � � | j t j t � d i �j d g � j � � g } xq t
� t � f D]] } | d d � | d d � | d d � | d d � | d d d �| j | | f � q� Wt
� } | j � | j d � | j � j d d � | j t � d i �j d g � | j | | f � x� | D]� \ } } x� t
t | � � D]l } xc d
D][ } | | | | � } | j | � t | � d k r�| j t | j t t | � � � q�Wq�Wq�Wd S)NrT rU r rW r� r r+ r� r. �rW r� )r )rT )rT rU rW )r r. r, �extendZspamrH r� Zbamr� rV r r r� �ranger� r r[ rv r� �reversed) r Zkalls1Zkalls2rI rA r r� �stepZtheser r r �test_assert_has_callsi s< %1
"
z MockTest.test_assert_has_callsc C s� d d d � } t d | � } | d d d d d �| d d
d d d d
�| d d d d �d d i f d d d d
i f d i f g } | j | � | j | d d �| j | d d � � | j | d d � d d �| j | d d � � | j | d d � d d �t t | � � } | j t � � | j | � Wd QRX| j | d d �| j t � � | j | d d � � Wd QRX| j | d d � d d �| j t � � | j | d d � � Wd QRX| j | d d � d d �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>.frL rT r rU r! rW r� r� r+ r" r� �
� � r� r T)rT rU rW )r� r� r+ )r r r r� r� r� r� )r r r� r r[ rv )r r_ rA Zcallsr r r �(test_assert_has_calls_with_function_spec� s0
z1MockTest.test_assert_has_calls_with_function_specc C s� t � } | d d � | d d � | d d d �| j d d � | j d d � | j d d d �| j t | j � | j t | j d d � | j t | j d d �d S)NrT rU r rW r r+ r� )r �assert_any_callr[ rv )r rA r r r �test_assert_any_call� s$
z MockTest.test_assert_any_callc
C s+ d d d � } t d | � } | d d d d d �| d d
d d d d
�| j d d d � | j d d d d d d � | j d d
d d
� | j d d d d
d d d d
� | j t | j d d d d d �| j t � � } | j d d � Wd QRXt | j d � r'| j | j j t � 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>.frL rT r rU r! rW r� r� r+ r" r� r rx ry rz ) r r r[ rv r{ r| r} rz r~ )r r_ rA r r r r �'test_assert_any_call_with_function_spec� s "z0MockTest.test_assert_any_call_with_function_specc C s� d d � } t � } | | _ t | � t | � j g } xS | D]K } | d d � | d d � | j | j t d d � t d d � g � q@ Wd S)Nc S s d S)Nr )r r r r r r_ � s z3MockTest.test_mock_calls_create_autospec.<locals>.frT rU rW r� )r r_ r
r4 rk r )r r_ �objZfuncsr� r r r � test_mock_calls_create_autospec� s
z(MockTest.test_mock_calls_create_autospecc C s2 t j t � d d �} | j d t | � � d S)NrD Z
sweet_func)rA r
rN rE rF )r r� r r r � test_create_autospec_with_name� s z'MockTest.test_create_autospec_with_namec C sr t � } | j t � � | j � Wd QRX| j t � � | j � Wd QRXt d d � } | j � | j � d S)NZunsafeT)r r[ r� Zassert_foo_callZassret_foo_call)r r� r r r �test_mock_unsafe� s
zMockTest.test_mock_unsafec C sH t � } | j j � | j � | j t � � | j j � Wd QRXd S)N)r �helloZassert_not_calledr[ rv )r r� r r r �test_assert_not_called s
z MockTest.test_assert_not_calledc C s_ t � } | j t � � | j j � Wd QRX| j � | j j � | j � | j j � d S)N)r r[ rv r Z
assert_called)r r� r r r �test_assert_called s
zMockTest.test_assert_calledc C sv t � } | j t � � | j j � Wd QRX| j � | j j � | j � | j t � � | j j � Wd QRXd S)N)r r[ rv r Zassert_called_once)r r� r r r �test_assert_called_once s
z MockTest.test_assert_called_oncec C sE t � } | j d d d d � d } | j t | j j � | � d S)NrD r ZdaddyZheroz call(daddy='hero', name='hello'))r r r4 rF r8 )r r� �textr r r �test_sorted_call_signature! s z#MockTest.test_sorted_call_signaturec C ss t j � } t j d d � } t � } | j � | j d d � | j | j d | � | j | j d | � d S)N� r r rT )r r� �indexr r4 r: )r r! r� r� r r r �test_override_tuple_methods( s
z$MockTest.test_override_tuple_methodsc
s Gd d � d t � } Gd d � d t � } Gd d � d t � � t t t t g } xL t | � D]> } | j | � f d d � � | j | � f d d � � qa Wxr| D]j} xat � t d
d � f D]G} | � } | j | j | j
f x | | f d g d
g f g D]� \ } } x� t � t d
d � f D]� } | j | | � | j | j t
t | d
� | j t
t | d � d
| k r�| j t
t | d d � | j | | � | j t
t | d � | j | j t
t | d � d
| k r2| j t
t | d d � q2Wq
Wq� Wq� Wd S)Nc @ s e Z d Z d Z d S)z)MockTest.test_mock_add_spec.<locals>._OnerT N)r r r r� r r r r �_One2 s r% c @ s e Z d Z d Z d S)z)MockTest.test_mock_add_spec.<locals>._TworU N)r r r r� r r r r �_Two4 s r& c @ s e Z d Z d Z Z Z d S)z-MockTest.test_mock_add_spec.<locals>.AnythingZfourN)r r r r� r� �threer r r r �Anything6 s r( c s
| d � � S)NrL r )�K)r( r r r� = s z-MockTest.test_mock_add_spec.<locals>.<lambda>c s
| d � � S)NrM r )r) )r( r r r� > s rM Tr� r� r' )rN r r r r r� r� r� r� r� r' �
mock_add_specr[ r� r� �setattr) r r% r& Zklassesr rR rA ZOneZTwor )r( r �test_mock_add_spec1 s>
+ zMockTest.test_mock_add_specc s� x� t t f D]� } | � � t � � � j t � | j t t � � | � � � d d � j _ � j t � | j t � � d � | j t � f d d � � q
Wd S)Nr. r� c s � d S)Nr. r r )rA r r r� t s z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� r* rN r[ r~ r� r7 r4 )r r r )rA r � test_mock_add_spec_magic_methodsf s
z)MockTest.test_mock_add_spec_magic_methodsc C s xt t t t f D]� } | � } t � | _ | j � | j | j t j � g � | j | j t j � g � | � } t d d � | _ | j � | j | j g � | j | j g � | � } t � � | _
| j
� | j | j g � | j | j g � q Wd S)NrD ) r r r r r. r4 r: r rk r, rH )r r rA r r r �test_adding_child_mockw s
z MockTest.test_adding_child_mockc C s\ xU t t f D]G } | � } t � | _ | � � | j | j t � t � � g � q
Wd S)N)r r r7 r4 rk r )r r rA r r r � test_adding_return_value_mock� s
z&MockTest.test_adding_return_value_mockc C s� Gd d � d t � } t � } t j | d � } t j | 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 Z d Z d Z d Z d S)z'MockTest.test_manager_mock.<locals>.Foor� r� N)r r r r� r� r r r r r� � s r� r� r� ) rN r r r� r� r� � attach_mockr� r� r4 rk r )r r� �managerZp1Zp2Zmock_oneZmock_twor r r �test_manager_mock� s
zMockTest.test_manager_mockc C s� x� t t f D] } | � } t d d � | _ t d d � | _ t | � t | � | j | j t j � t j � g � | j | j g � q
Wd S)Nr7 rW g @)
r r r� � __float__r� �floatr4 rk r r: )r r r� r r r � test_magic_methods_mock_calls� s
%z&MockTest.test_magic_methods_mock_callsc C sV t j d d � } | d � } | j � } | d � } | j � } | j | | � d S)N� read_data�dataza-namez another-name)rA rn �readr4 )r Z 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 j t j d d � � } t j d | � �B | j } | j } t d � | _ d d � } | j t | � Wd QRXd S)NrD ZJLVztempfile.NamedTemporaryFilez Test 2 Errorc S s t j � j d � d S)NZasd)�tempfile�NamedTemporaryFile�writer r r r �attempt� s z.MockTest.test_mock_open_write.<locals>.attempt) rA rn r r r7 r> �OSErrorrZ r[ )r Zmock_namedtempZmock_filehandleZ
mock_writer? r r r �test_mock_open_write� s z MockTest.test_mock_open_writec C si t j d d � } d d � | j j _ | � j � } | � j � } | j d | � | j d | � d S)Nr6 zfoo
barnc W s d S)N�abcr )rQ r r r r� � s z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>rB )rA rn r7 �readlinerZ r4 )r Zmopen�first�secondr r r � test_mock_open_alter_readline� s z&MockTest.test_mock_open_alter_readlinec C sw xpt t f D]b} | � } t | � } | | _ | j | � | � | j t | � | � | j � | j | � | � | j t | � | � | � } | j | _ | j d t | j � � | j d t | j � � | j � | j d t | j � � | j d t | j � � | � } t | � } | � | _ | | j _ | j t | � | � | j t | j � � | � q
Wd S)Nz
name='mock.a')
r r rF r7 r� r4 ri r r rE )r r r� Z
original_reprr r r �test_mock_parents� s,
zMockTest.test_mock_parentsc C s� t t t t f } x� | D]� } x� | D]� } | � } | d d � } | j | d � | j | j | � | j d t | � � | j j d � | j
| j t j j d � g � | j
| j
t j j d � g � q& Wq Wd S)NrD r. r, zname='mock.bar'rT )r r r r r0 r� r, rE rF rH r4 rk r r: )r �classesr �Klass2r� �m2r r r �test_attach_mock� s
"zMockTest.test_attach_mockc C s� t t t t f } x� t t f D]� } x� | D]� } | � } | d d � } | j | d � | j | � | � | j d t | � � | j � | j | j
t � j � j � � q, Wq Wd S)NrD r. r7 z
name='mock()')
r r r r r0 r� rE rF r. r4 rk r r� )r rH r rI r� rJ r r r � test_attach_mock_return_value s
z&MockTest.test_attach_mock_return_valuec C s� x� t � t � t � t � f D]j } | j t | d � � | ` | j t | d � � | ` | j t | d � � | j t
t | d � q Wd S)Nr� r_ ) r r r r r6 r{ r� r2 r_ r[ r� r� )r rA r r r �test_attribute_deletion s z MockTest.test_attribute_deletionc C sQ xJ t � t � f D]6 } | j | t � t | _ | j | t � | j q Wd S)N)r r r� r� r� r} r. )r rA r r r �test_class_assignable s
z MockTest.test_class_assignablec C s� x� t t t t f D]� } | d d d d � } | j d d d � t j | � } t j | � } | j j d d d � | j | j
� | j t | | � � | j
| t � | j d t | � � | j | j d � q Wd S)NrD r. r- rW rT rU z
name="foo")r r r) r r. �pickle�dumps�loadsr� r2 r3 r6 r
r} r* rE rF r4 r- )r r rA r7 �newr r r � test_pickle( s zMockTest.test_pickleN)^r r r r1 rB rC rJ rO r^ r` rc r� Z
skipUnlessr� �platformrf rl rm ro rs ru rw r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ZskipIf�sixZPY3r� 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/ r2 r5 r; rA rF rG rK rL rM rN ZexpectedFailurerS r r r r r/ D s� ) "
!!
7 N #
, ( !
5
r/ �__main__)#r� rO r� r< rU Z unittest2r� rA r r r r r r r r r
Z mock.mockr Zmock.tests.supportr r
r �unicoder� r� rN r r r) r* r� r/ r �mainr r r r �<module> s2 @
� � � � � �