Your IP : 216.73.216.213
��YfO� � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z m Z m
Z
m Z m Z m
Z
m Z m Z m Z m Z Gd d � d e � Z Gd d � d e � Z Gd d
� d
e j � Z e d k r� e j � d S) � N)� is_instance)�mock)
�call�DEFAULT�patch�sentinel� MagicMock�Mock�NonCallableMock�NonCallableMagicMock� _CallList�create_autospecc @ 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 �D/opt/alt/python35/lib64/python3.5/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 @ 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 �cmeth! s zSomething.cmethc C s d S)Nr )r r r! r" r r r �smeth% s zSomething.smeth)r r r r# � classmethodr% � staticmethodr& r r r r r s
r c @ s� 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+ dM dN � Z, dO dP � Z- dQ dR � Z. dS dT � Z/ dU dV � Z0 dW dX � Z1 dY dZ � Z2 d[ d\ � Z3 d] d^ � Z4 d_ d` � Z5 da db � Z6 dc dd � Z7 de df � Z8 dg dh � Z9 di dj � Z: dk dl � Z; dm dn � Z< do dp � Z= dq dr � Z> ds dt � Z? du dv � Z@ dw dx � ZA dy dz � ZB d{ d| � ZC d} d~ � ZD d 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� 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 | 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 r r r r �test_constructor3 s( zMockTest.test_constructorc C s& t d d � } | j | j d � d S)Nr1 z(return value in constructor not honoured)r r7 r1 )r r r r r � test_return_value_in_constructorO 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�name�fooz'%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 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 �Xf s rF �specz
spec='X' �spec_setz spec_set='X' r= r> z name='foo' )�objectr r@ rA r5 )r rF r r r r �test_repr_with_spece 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 rN � 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 rU � s z.MockTest.test_side_effect.<locals>.side_effectr1 ) r rU � assertRaisesrK �assert_called_withr. r �
SideEffectZRETURN)r r rN rU r )rT 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)
NrO rP rR c s
� j � S)N)rS r )rT r r rN � 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
rU r. )r rN rZ r r )rT 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 rZ � s z7MockTest.test_autospec_side_effect_exception.<locals>.fzBazinga!)r
�
ValueErrorrU �assertRaisesRegex)r rZ 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 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 rU zBoom!rO rP rQ rR zjava exception not raised)r_ r ZlangZRuntimeException�failrW )r r_ r 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� somethingr= Zchild�parentrG 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
SomethingElserb rX rU r1 �
reset_mockr. �
_mock_namer8 r9 r, r- r/ r2 r3 r4 �
mock_callsr: )r rc rG r rb r1 r r r �test_reset_mock� sD
zMockTest.test_reset_mockc C s t � } | | _ | j � d S)N)r r1 rd )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_openrd )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 setrO zcall_count incoreectzcall_args not setz(call_args_list not initialised correctly�keyzincorrect return valuerP zcall_count incorrectzcall_args_list not set)r r0 r r1 r. rd r �Argr- r/ r2 r3 Z ReturnValueZKeyArg)r r �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 rl ZKwargr. r3 r2 r, )r r r r r �test_call_args_comparison s
z"MockTest.test_call_args_comparisonc C s3 t j � } | j | | k � | j | | k � | j | t j � k � | j | t j � k � | j | t j k � | j | t j k � | j t j | k � | j t j | k � t j t j � � } t j t j � } | j | | k � | j | | k � | j | | k � | j | | k � d S)N)r r r0 r, �ANYr )r ZmmZcall1Zcall2r r r �test_calls_equal_with_any4 s z"MockTest.test_calls_equal_with_anyc 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)NrO rP rR r rQ r Znothing)r rW rV �AssertionErrorrd )r r r r r �test_assert_called_withI s
z MockTest.test_assert_called_withc C s* t � } | t � � | j t j � d S)N)r rW r rq )r �mr r r �test_assert_called_with_anyX s
z$MockTest.test_assert_called_with_anyc
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 QRX| j | j j t � d S) Nc S s d S)Nr )r r r! r" r r r rZ _ s z9MockTest.test_assert_called_with_function_spec.<locals>.frG rO r rP r! rR r �e� )r rW rV rs �assertIsInstance� exception� __cause__� TypeError)r rZ r �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)NrO r rP r! rR r )rW rV rs )r )r r r �_checkp s
z<MockTest.test_assert_called_with_method_spec.<locals>._checkrG )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 � � � | 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? rC rP c s � j d d d d �S)NZbobr? rC rP )�assert_called_once_withr )r r r �<lambda>� s z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r r� rV rs rd )r r )r 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 QRX| 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 rZ � s z>MockTest.test_assert_called_once_with_function_spec.<locals>.frG rO r rP r! rR r rw rx � � � )r r� rV rs ry rz r{ r| )r rZ r 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 namerR zmethod returned wrong valuez!method didn't record being called)r rb r0 r r. r1 r- )r r rb 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)Nrb z attribute name not set correctlyz"attribute parent not set correctly)r rb r. re r8 )r r rb 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) NrR rQ r� Zcakerb z#method calls not recorded correctlyzsomething_else.something)r� )rR )r� )r rb �something_elser ZCaker. r4 )r r 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) Nrb � different�xrO ZsomeZmore)rb �r� )rb r� )r� )r� )r rb r. r4 )r r 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)Nrb rG rH z-Mock object has no attribute 'something_else'r� )rb )rG rH )r rb r] �AttributeError�getattr)r rG �argr 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>.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 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� r] r� r� )r )r r r �test_attributes s
z0MockTest.test_from_spec.<locals>.test_attributesrG )rI 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�wrapsrO rP rQ rR )r r. rd rW )r �realr r r r �test_wraps_calls s
zMockTest.test_wraps_callsc C sH t � } t d | � } d | _ | j | � d � | j | j � d S)Nr� rR )r r1 r. r, r- )r r� r 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 � attributer r r r �Real3 s r� r� c s � j S)N)rQ r )r r r r� : s z0MockTest.test_wraps_attributes.<locals>.<lambda>rO rP rQ rR ) rI r r. r� rV r� ZassertNotEqualZfrogrW )r r� r� rm r )r 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)NrU r> )r r� rV )r r 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)NrU r> )r �KeyboardInterruptrV )r r 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 r] rs rW )r r 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)Nr= Zgeoffreyz'Expected 'geoffrey' to be called once\.)r r] rs r� )r r 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 )r r r r� a s z'MockTest.test__name__.<locals>.<lambda>r> )r rV r� r r. )r r )r 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� rG r> rR )�listr �appendrW rV r� r� )r r� r 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 rF r s rF rG r rH )rI r ry �assertIs� __class__r. r )r rF r 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>.XrR N)r r r r� r r r r rF � s rF rG r> rH 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)rI r r� r� rV r� )r rF r� r )r 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 Zcurrentr r r � test_copy� s zMockTest.test_copyc 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)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� rH 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� rV 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� rW rV r| )r r 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)�.0ru 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� � s z#MockTest.test_dir.<locals>.<lambda>r )
r �set�dirr. r r r@ r! r" r )r r �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)NrG rR �versionrO )r �unittest�TestCaser� r� r. r� �count)r r 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 rI r �startr� r� r r. �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? rU zfoo.bar.return_value�! ) r r. r> r �KeyErrorrV r? ry Zconfigure_mock)r r rM 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)NrO zException %r not raised)r� �exc_infory r` �strr. )r rz �message�funcrL rM �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 calledrO �2rR 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)rP zfoo())r
r� rs rW 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 | 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� rO rP rR r r� r r� z().foor> z
().foo.barz().foo.bar().bazr= r? z foo.__int__za()z().foo.bar().baz().__int__)rO rP rR )rO rP rR )rO rP rR ���)r r� rf r. r> r� r �dictr1 r? rC � call_listr� r r r! �__int__)r r r� rM Z last_callr r r �test_mock_callsI 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 �Subclass� 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 rM 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 ry r> �assertNotIsInstance)r r� r 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)Nr3 r4 rf )r3 r4 rf )r� ry r r� r. )r �namesr= �attr)r r r � assert_attrs� s
z-MockTest.test_arg_lists.<locals>.assert_attrsrO rP r rR ) r r r
r �callablerd r> r? rC )r rD r� r 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) NrO r rR rP r r� r )rP )rO )rP )r r. �lenr2 r� �zipr3 )r r rL rM Z
expected_listr� r2 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)NrU rO rP rR 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 �Foo� s r� r r r r ) r r r. rV �
StopIterationr rI ry r )r r r� r r r �test_side_effect_iterator� s +++!z"MockTest.test_side_effect_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)NrR r� rU )r r r\ r� rV r. )r �Klass�iterableru r r r �$test_side_effect_iterator_exceptions� s z-MockTest.test_side_effect_iterator_exceptionsc 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) NrO rP rR r r r! r r r r )
r r rU r. rV r� ry r� r r� )r r rU Z this_iterr r r �!test_side_effect_setting_iterator� s" + + !z*MockTest.test_side_effect_setting_iteratorc C sJ t d d � } t d t g � | _ | j | � | � g d d g � d S)Nr1 rP rO )r r r rU r. )r r r r r �!test_side_effect_iterator_default 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)NrO rP r rR r� r r� r� � any_orderTr� r> r� � �rR r� )r �rO rP )r� r )rO rP )r r �assert_has_callsrV rs )r r �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)NrO rP r rR r� r r� r� r> �rR r� )r )rO )rO rP rR )r r> r? �extendZspamrC r� Zbamr� rQ r r r� �ranger� r rV rs r� �reversed) r Zkalls1Zkalls2rD r r r� �stepZtheser r r �test_assert_has_callsD 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 rZ m s z<MockTest.test_assert_has_calls_with_function_spec.<locals>.frG rO r rP r! rR r� r� r� r" r �
� � r� r T)rO rP rR )r� r� r� )r
r r r� r� r� r� )r r r� r
rV rs )r rZ r Zcallsr r r �(test_assert_has_calls_with_function_specl 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)NrO rP r rR r r� r� )r �assert_any_callrV rs )r r 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 QRX| j | j j t � d S)Nc S s d S)Nr )r r r! r" r r r rZ � s z;MockTest.test_assert_any_call_with_function_spec.<locals>.frG rO r rP r! rR r� r� r� r" r r rw rx )r r rV rs ry rz r{ r| )r rZ r 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 rZ � s z3MockTest.test_mock_calls_create_autospec.<locals>.frO rP rR r� )r rZ r
r. rf r )r rZ �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)Nr= Z
sweet_func)r r
rI r@ rA )r ru 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 rV r� Zassert_foo_callZassret_foo_call)r ru 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_calledrV rs )r ru r r r �test_assert_not_called� s
z MockTest.test_assert_not_calledc C sE t � } | j d d d d � d } | j t | j j � | � d S)Nr= r ZdaddyZheroz call(daddy='hero', name='hello'))r r r. rA r2 )r ru �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 rO )r r� �indexr r. r4 )r r! r� ru 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>._OnerO N)r r r r� r r r r �_One� s r c @ s e Z d Z d Z d S)z)MockTest.test_mock_add_spec.<locals>._TworP N)r r r r� r r r r �_Two� 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 �Anything� s r" c s
| d � � S)NrG r )�K)r" r r r� s z-MockTest.test_mock_add_spec.<locals>.<lambda>c s
| d � � S)NrH r )r# )r" r r r� s rH Tr� r� r! )rI r r r
r r� r� r� r� r� r! �
mock_add_specrV r� r� �setattr) r r r Zklassesr� rM r ZOneZTwor )r" r �test_mock_add_spec� 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 )r r r r� 9 s z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� r$ rI rV r| r� r1 r. )r r� r )r r � test_mock_add_spec_magic_methods+ 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)Nr= ) r
r r r r> r. r4 r rf r? rC )r r� r r r r �test_adding_child_mock< 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 r1 r. rf r )r r� r r r r � test_adding_return_value_mockT 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� ) rI r r r� r� r� � attach_mockr� r� r. rf r )r r� ZmanagerZp1Zp2Zmock_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)Nr1 rR g @)
r r r� � __float__r� �floatr. rf r r4 )r r� ru r r r � test_magic_methods_mock_callss 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)r ri �readr. )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)Nr= ZJLVztempfile.NamedTemporaryFilez Test 2 Errorc S s t j � j d � d S)NZasd)�tempfileZNamedTemporaryFile�writer r r r �attempt� s z.MockTest.test_mock_open_write.<locals>.attempt) r ri r r r1 r6 �OSErrorrU rV )r Zmock_namedtempZmock_filehandleZ
mock_writer7 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)Nr/ zfoo
barnc W s d S)N�abcr )rL r r r r� � s z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>r: )r ri r1 �readlinerU r. )r Zmopen�first�secondr r r � test_mock_open_alter_readline� s z&MockTest.test_mock_open_alter_readlinec C s� t j d d � } | d � } | j � | j d | j � � | j d | j � � | j d | j � � | j d | j � � | j g | j � � | j g | j � � d S)Nr/ r> r? r� )r ri r1 r. r; � readlines)r Z_openr� r r r �test_mock_open_after_eof� s
z!MockTest.test_mock_open_after_eofc 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 rA r1 r� r. rd r r r@ )r r� ru 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)Nr= r> r? zname='mock.bar'rO )r r r r
r* r� r? r@ rA rC r. rf r r4 )r �classesr� �Klass2ru �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)Nr= r> r1 z
name='mock()')
r r r r
r* r� r@ rA r> r. rf r r� )r rB r� rC ru rD 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)Nru rZ ) r r r r
r0 �hasattrru r, rZ rV r� r� )r r 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� ry r> )r r r r r �test_class_assignable� s
z MockTest.test_class_assignableN)Xr r r r+ r; r< rE rJ rY r[ r^ r� Z
skipUnlessr� �platformra rg rh rj rn rp rr rt rv 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. r4 r9 r> r@ rA rE rF rH rI r r r r r) * s� ) "
7 N #
, ( ! 5
r) �__main__)r� r� r5 r� Z unittest.test.testmock.supportr r Z
unittest.mockr r r r r r r
r r r
rI r r r� r) r �mainr r r r �<module> s F
� � � � � �