Your IP : 216.73.216.213
�
j f�� � @ 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 � n 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/python34/lib64/python3.4/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 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- 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� 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_effect�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 i | d 6d � | 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
SomethingElser_ rX rU r1 �
reset_mockr. �
_mock_namer8 r9 r, r- r/ r2 r3 r4 �
mock_callsr: )r r` rG r r_ r1 r r r �test_reset_mock� sD
zMockTest.test_reset_mockc C s t � } | | _ | j � d S)N)r r1 ra )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_openra )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 i t j
d 6f d � | j | j t j f i f t j f i t j
d 6f 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. ra 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 i t j d 6f t j f i t j d 6f g � | j | j t j f i t j d 6f � | j | j d k � d S)N�kwza long sequence)r r ri ZKwargr. r3 r2 r, )r r 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)NrO rP rR r rQ r Znothing)r rW rV �AssertionErrorra )r r r r r �test_assert_called_with* 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 QX| 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 rn �assertIsInstance� exception� __cause__� TypeError)r rZ r �cmr r r �%test_assert_called_with_function_spec9 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 rn )r )r r r �_checkK s
z<MockTest.test_assert_called_with_method_spec.<locals>._checkrG )r r r# r% r& )r rx r r )r r �#test_assert_called_with_method_specJ 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>r s z7MockTest.test_assert_called_once_with.<locals>.<lambda>)r rz rV rn ra )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 QX| 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 w s z>MockTest.test_assert_called_once_with_function_spec.<locals>.frG rO r rP r! rR r rp rq � � � )r rz rV rn rr rs rt ru )r rZ r rv r r r �*test_assert_called_once_with_function_specv 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 r_ r0 r r. r1 r- )r r r_ 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)Nr_ z attribute name not set correctlyz"attribute parent not set correctly)r r_ r. rb r8 )r r r_ 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 i t j d 6f g d � | j | j d d i d d 6f d d
i t j d 6f g d � d S) NrR rQ r Zcaker_ z#method calls not recorded correctlyzsomething_else.something)r )rR )r )r r_ �something_elser ZCaker. r4 )r r r r r �test_method_calls_recorded� s z#MockTest.test_method_calls_recordedc C s; 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 i d d 6f g � | j | j d f i d d 6f g � t � } | j d d d �| j | j d d i d d 6f g � d S) Nr_ � different�xrO ZsomeZmore)� something�r� )r� r� )r� )r� )r r_ r. r4 )r r r r r � test_method_calls_compare_easily� s"
#& z)MockTest.test_method_calls_compare_easilyc C sb x[ d g d f D]J } xA d D]9 } t i | | 6� } | j | j t d t | d � q Wq Wd S)Nr_ rG rH z-Mock object has no attribute 'something_else'r� )z something)zspeczspec_set)r r_ �assertRaisesRegex�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. ra 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 �Real 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� rj r )r r �test_wraps_attributes
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_effect 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_effect% s z)MockTest.test_baseexceptional_side_effectc C s# t � } | j t d | j � d S)Nz
Not called)r r� rn rW )r r r r r � test_assert_called_with_message- 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� rn rz )r r r r r �$test_assert_called_once_with_message3 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{ < 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 �SubC s r� rG r> rR )�listr �appendrW rV r� r� )r r� r r r r �test_spec_list_subclassB 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 M s rF rG r rH )rI r rr �assertIs� __class__r. r )r rF r r r r �test_spec_classL 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 a s rF rG r> rH c s
d � _ d S)Nr> )r� r )r r r �set_attrh 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_copyo 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 �SubClassz 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_propertiesy 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 ru )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)�.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�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 � i t d 6d d 6t � d 6} 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? rr 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_inforr r] �strr. )r rs �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� rn rW r> rz )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� rc r. r> r� r �dictr1 r? rC � call_listr� r r r! �__int__)r r r� rM Z last_callr r r �test_mock_calls$ 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 �Subclasss 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� z s )r rr r> �assertNotIsInstance)r r� r r r r �test_subclassingr s zMockTest.test_subclassingc s� t � t � t � t � g } � f d d � } x� | D]� } | | � t | � r� | � | d d � | d d � | j � | | � n | 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 rc )zcall_args_listz method_callsz
mock_calls)r� rr 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 �callablera 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 rr 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 �
ValueErrorr� rV r. )r �Klass�iterabler� 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� rr 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 � i d d 6f d f i d d 6f d d i d d 6f d d i f d f i d d 6f 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
�qWt 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
�q\Wd 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 rn )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 s8 t d d � i d d 6f d t d d � d d i d d 6f 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]o } xf d
D]^ } | | | | � } | j | � t | � d k r�| j t | j t t | � � � q�q�Wq�Wq�Wd S)NrO rP rR r 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 rn r� �reversed) r Zkalls1Zkalls2rD r r r� �stepZtheser r r �test_assert_has_calls 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 i d
d 6f 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 QX| j | d d �| j t � � | j | d d � � Wd QX| j | d d � d d �| j t � � | j | d d � � Wd QX| j | d d � d d �d S)Nc S s d S)Nr )r r r! r" r r r rZ H 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 rn )r rZ r Zcallsr r r �(test_assert_has_calls_with_function_specG 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 rn )r r r r r �test_assert_any_callh 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 QX| 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 rp rq )r r rV rn rr rs rt ru )r rZ r rv 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. rc 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 r� r r r � test_create_autospec_with_name� s z'MockTest.test_create_autospec_with_namec
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 Wxx| D]p} xgt � t d
d � f D]M} | � } | 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 � n | j | | � | j t
t | d � | j | j t
t | d � d
| k r2| j t
t | d d � q2q2Wq
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{ � s z;MockTest.test_mock_add_spec_magic_methods.<locals>.<lambda>)
r r r� r rI rV ru 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 rc 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. rc r )r r� r 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� ) rI r r r� r� r� � attach_mockr� r� r. rc 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. rc r r4 )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)r rf �readr. )r Z mocked_open�f1Zf1_data�f2Zf2_datar r r � test_mock_open_reuse_issue_217505 s z)MockTest.test_mock_open_reuse_issue_21750c
C su t j t j d d � � } t j d | � �B | j } | j } t d � | _ d d � } | j t | � Wd QXd S)Nr= ZJLVztempfile.NamedTemporaryFilez Test 2 Errorc S s t j � j d � d S)NZasd)�tempfileZNamedTemporaryFile�writer r r r �attemptD s z.MockTest.test_mock_open_write.<locals>.attempt) r rf r r r1 r+ �OSErrorrU rV )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)Nr$ zfoo
barnc W s d S)N�abcr )rL r r r r{ J s z8MockTest.test_mock_open_alter_readline.<locals>.<lambda>r/ )r rf r1 �readlinerU r. )r Zmopen�first�secondr r r � test_mock_open_alter_readlineH 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 rA r1 r� r. ra r r r@ )r r� r� Z
original_reprr r r �test_mock_parentsP 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. rc r r4 )r �classesr� �Klass2r� �m2r r r �test_attach_mockm 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. rc r r� )r r5 r� r6 r� r7 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� rZ ) r r r r
r0 �hasattrr� 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� rr r> )r r r r r �test_class_assignable� s
z MockTest.test_class_assignableN)Pr r r r+ r; r< rE rJ rY r[ r� Z
skipUnlessr� �platformr^ rd re rg rk rm ro 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. r3 r4 r8 r9 r; r< r r r r r) * s� ) "
7 N #
, ( ! 5
r) �__main__)r� r� r* 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
� � � � �