�?�Vp � @ s 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
d d l m Z m
Z
d d l m Z Gd d � d e � Z Gd d � d e j � Z Gd d
� d
e j � Z Gd d � d e j � Z Gd
d � d e j � Z e d k re j � d S)� N)�call�create_autospec� MagicMock�Mock�ANY�patch� PropertyMock)�_Call� _CallList)�datetimec @ s7 e Z d Z d d � Z d d � Z d d d � Z d S)� SomeClassc C s d S)N� )�self�a�br
r
�/testhelpers.py�one s z
SomeClass.onec C s d S)Nr
)r r
r
r �two s z
SomeClass.twoNc C s d S)Nr
)r r r
r
r �three s zSomeClass.three)�__name__�
__module__� __qualname__r r r 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 S)
�AnyTestc C s] | j t t � � t � } | t � | j t � t � } | d t � | j d t � d S)N�foo)� assertEqualr �objectr �assert_called_with)r �mockr
r
r �test_any s
zAnyTest.test_anyc C s0 | j t t � d � | j t t � d � d S)Nz<ANY>)r �reprr �str)r r
r
r � test_repr' s zAnyTest.test_reprc C s< t � } | t j � d t j � �| j t d t �d S)Nr )r r �nowr r )r r r
r
r �test_any_and_datetime, s z AnyTest.test_any_and_datetimec C s t � } t j � } Gd d � d t � } x� t j � | � f D]� } | j � | | d | d | �| j | d | d | �| � j d | d | � t t d t d t �t j t d t d t �t � t � j d t d t � g } | j | | j � | j | j | � qA Wd S) Nc @ s( e Z d Z d d � Z d d � Z d S)z9AnyTest.test_any_mock_calls_comparison_order.<locals>.Fooc S s d S)NFr
)r �otherr
r
r �__eq__7 s z@AnyTest.test_any_mock_calls_comparison_order.<locals>.Foo.__eq__c S s d S)NTr
)r r$ r
r
r �__ne__9 s z@AnyTest.test_any_mock_calls_comparison_order.<locals>.Foo.__ne__N)r r r r% r& r
r
r
r �Foo6 s r' r �barZzinga�alphaZa1Zz99)
r r r" r �
reset_mock�methodr r r �
mock_calls)r r �dr' �expectedr
r
r �$test_any_mock_calls_comparison_order3 s
$z,AnyTest.test_any_mock_calls_comparison_orderN)r r r r r! r# r/ 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
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# S)$�CallTestc C s� t � } | j | t � � | j | t d � � | j | t f f � � | j | t i f � � | j | t d f f � � | j | t d i f � � | j | t d f i f � � | j | t d � � | j | t d f f � � | j | t d i f � � | j | t d f i f � � t d � } | j | t d � � | j | t d � � | j | t d i f � � | j | t d d i f � � t d � } | j | t d � � | j | t d d i f � � t d � } | j | t d � � | j | t d d i f � � | j | t d � � | j | t d d! i f � � t d
d i f � } | j | t d f d
d i f � � | j | t d d
d i f � � | j | t f d
d i f � � | j | t d
d i f � � d S)"N� r r( �bazZspam� � � � r )r1 )�foo�r3 r4 r5 )r8 �r3 r4 r5 )r9 �r3 r4 r5 )r1 r: )r3 r4 r5 )r3 r4 r5 �r3 r4 r6 )r; �r3 r4 r5 )r1 r< )r3 r4 r5 �r3 r4 r6 )r7 r= �r3 r4 r6 )r1 r> )r3 r4 r6 �r3 r4 r6 )zbarr? )r3 r4 r6 )r r �assertNotEqual)r �kallr
r
r �test_call_with_callO s< %""z CallTest.test_call_with_callc C s� t � } | j | f � | j | d � | j | f f � | j | d f f � | j | d f i f � | j | d i f � | j | i f � d S)Nr )zfoo)r r )r �argsr
r
r �test_empty__Callt s zCallTest.test_empty__Callc C s� t d f i f � } | j | d � | j | d f f � | j | d f i f � | j | d i f � | j | f f � | j | f � | j | i f � | j | d � | j | d f f � | j | d i f � d S)Nr r( )zfoo)zbar)r r r@ )r rC r
r
r �test_named_empty_call� s z CallTest.test_named_empty_callc C se t d i f � } | j | d � | j | d � | j | d d
i f � | j | d i f � d S) Nr3 r4 r5 r )r3 r4 r5 �r3 r4 r5 )rF �r3 r4 r5 )zfoorG )r3 r4 r5 )r3 r4 r5 )r r )r rC r
r
r �test_call_with_args� s
z CallTest.test_call_with_argsc C sh t d d i f � } | j | d � | j | d d i f � | j | d
� | j | d i f � d S) Nr r3 r4 r5 )r3 r4 r5 �r3 r4 r5 )zfoorI )r3 r4 r5 �r3 r4 r5 )rJ )r3 r4 r5 )r r r@ )r rC r
r
r �test_named_call_with_args� s
z"CallTest.test_named_call_with_argsc C s� t f t d d d d � f � } | j | t d d d d � f � | j | d t d d d d � f � | j | d f t d d d d � f � | j | f t d d d d � f � d S)Nr r5 r r6 r )r �dictr )r rC r
r
r �test_call_with_kwargs� s
!"%(z CallTest.test_call_with_kwargsc C s� t d f t d d d d � f � } | j | d t d d d d � f � | j | d f t d d d d � f � | j | t d d d d � f � | j | f t d d d d � f � d S)Nr r r5 r r6 )r rL r r@ )r rC r
r
r �test_named_call_with_kwargs� s
$%("z$CallTest.test_named_call_with_kwargsc C sm t d i f � } | j | t d d d � � | j t d d d � | � | j t d d d � | g � d S)Nr3 r4 r5 )r3 r4 r5 )r r r �assertIn)r rC r
r
r �#test_call_with_args_call_empty_name� s z,CallTest.test_call_with_args_call_empty_namec C ss | j t d � t d d � � | j t d � t d d d � k � | j t d i f � t d d d � k � d S) Nr3 r4 r5 �r3 r4 r5 )rQ �r3 r4 r5 )rR )r3 r4 )r@ r r � assertFalseZ
assertTrue)r r
r
r � test_call_ne� s %zCallTest.test_call_nec C sP t d � } x= d d | t f D]) } | j | | � | j | | k � q Wd S)Nr3 r4 r5 �r3 r4 r5 )rU )r �intr@ rS )r rA �valuer
r
r �test_call_non_tuples� s z CallTest.test_call_non_tuplesc C s� | j t t � � d � | j t t d � � d � | j t t d d d i f � � d � | j t t d
d d d i f � � d � | j t t � d � | j t t � d � | j t t � � d � | j t t d � � d
� | j t t d d � � d � | j t t � j � d � | j t t d � j j d d � j � d � | j t t � j d d d d �� d � | j t t � � � d � | j t t d � d � � d � | j t t � � j � j j d � � d � d S)Nzcall()r z
call.foo()r3 r4 r5 r r zcall(1, 2, 3, a='b')r( zcall.bar(1, 2, 3, a='b')r zcall(1)�zz�thingzcall(zz='thing')z
call().foozcall().foo.bar().bingzcall().foo(1, 2, a=3)zcall()()z call()(2)zcall()().bar().baz.beep(1))zfoo)r3 r4 r5 )r3 r4 r5 )
r r r r r r r( �bingr2 �beep)r r
r
r r! � s, !$ ' "!zCallTest.test_reprc C s� | j t � d f i f � | j t d d d d d d �d d
d d d d i f � t � } | d d d � | d
d d d � | j | j t d d d � t d
d d d � g � d S)Nr1 r r( r r5 r r6 r3 r4 r r � )zfoozbar)r r r �call_args_list)r r r
r
r � test_call� s zCallTest.test_callc C s� | j t j d � d d i f � | j t j j d d � d f d d i f � t � } | j d d d � | j j d d d d
� | j | j t j d d d � t j j d d d d
� g � d S) Nr3 r �fishZeggszbar.bazr4 r5 r r r] )r3 )r r r r( r2 r � method_calls)r r r
r
r �test_attribute_call� s " z CallTest.test_attribute_callc C s� t d � j d � j d d d �} | j | d d t d d � f � t � } | d d d d d d �| j | j t d d d d d d �� | j | j t d d d � � | j | j t d d d d d d �g � | j | j t d d d d d d �g � t � } | j d � j � � j
j d d � t j d � j � � j
j d d � } | j | j d
| � | j | j | j � � d S) Nr3 r4 r5 r r6 z ().foo().barr r] )r5 ���)
r r r( r rL r � call_argsr@ r^ r, r2 r\ � call_list)r �resultr Z last_callr
r
r �test_extended_call s $" ( ++ %'zCallTest.test_extended_callc C sV t � } | d � | j t d � j � | j � t � } | d � j d � | j t d � j d � j � | j � t � } | d � j d � d � | j t d � j d � d � j � | j � t � } t | d � j d � d � j j j d � d � � t d � j d � d � j j j d � d � j
� } | j | j � | j � d S)Nr3 r4 r5 r6 � ) r r r re r, r+ rV r r( r2 �__int__)r r rA r
r
r �test_call_list s
$
46zCallTest.test_call_listc C sS | j t t � t � } t | � | j | j t g � | j t g | j � d S)N)r r r r rV r, )r �mr
r
r �
test_call_any) s
zCallTest.test_call_anyc C s� t d d d i f d d �} | j t | � d � | j | d d � | j | d d d i � t d
d d i f � } | j | | � d S) Nr3 r4 r r5 r Tr )r3 r4 )r3 r4 )r3 r4 )r r �len)r rC Z
other_argsr
r
r �test_two_args_call2 s zCallTest.test_two_args_callN)r r r rB rD rE rH rK rM rN rP rT rX r! r_ rb rg rj rl rn r
r
r
r r0 M s" %
r0 c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e j d d � � Z
d
d � Z e j e
j 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 e j e
j d" � d# d$ � � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z e j e
j d3 � d4 d5 � � Z! d6 d7 � Z" e j e
j d3 � d8 d9 � � Z# d: d; � Z$ d< d= � Z% d> d? � Z& d@ dA � Z' dB dC � Z( dD dE � Z) dF S)G�SpecSignatureTestc C s> | j t t | d � | j d d � | j j d d � | j t | j j d d � | j t | j d � | j � | j j � | j t | j j d � | j t | j d � | j � | j j � | j t | j j d � | j t | j d d � | j d � | j j d � | j d d � | j j d d � d S)Nr r3 r4 r5 r6 r ) � assertRaises�AttributeError�getattrr r �AssertionError� TypeErrorr r )r r r
r
r �_check_someclass_mock> s(
z'SpecSignatureTest._check_someclass_mockc C s9 t t � } | j | � t t � � } | j | � d S)N)r r ru )r r r
r
r �
test_basicY s
z SpecSignatureTest.test_basicc C sp d d � } t | d d �} | j | � d � Gd d � d t � } t | d d �} | j | � d � d S)Nc S s d S)Nr
r
r
r
r �fa s z>SpecSignatureTest.test_create_autospec_return_value.<locals>.fZ return_valuer c @ s e Z d Z d S)z@SpecSignatureTest.test_create_autospec_return_value.<locals>.FooN)r r r r
r
r
r r' f s r' )r r r )r rw r r' r
r
r �!test_create_autospec_return_value` s z3SpecSignatureTest.test_create_autospec_return_valuec C s: t t � } t | � | j � | j | j j d � d S)Nr )r rV r* r ri Z
call_count)r rk r
r
r �test_autospec_reset_mockm s
z*SpecSignatureTest.test_autospec_reset_mockc C sU Gd d � d t � } t j | d � } | j � } | � j d � | j d � d S)Nc @ s e Z d Z d d � Z d S)z;SpecSignatureTest.test_mocking_unbound_methods.<locals>.Fooc S s d S)Nr
)r r r
r
r r v s z?SpecSignatureTest.test_mocking_unbound_methods.<locals>.Foo.fooN)r r r r r
r
r
r r' u s r' r r3 )r r �startr r )r r' �pZmock_foor
r
r � test_mocking_unbound_methodst s
z.SpecSignatureTest.test_mocking_unbound_methodsc C su Gd d � d t � } t | � } | � } | j t | j d � | j d � | j j d � | j t | j � d S)Nc @ s e Z d Z d d � Z d S)zCSpecSignatureTest.test_create_autospec_unbound_methods.<locals>.Fooc S s d S)Nr
)r r
r
r r � s zGSpecSignatureTest.test_create_autospec_unbound_methods.<locals>.Foo.fooN)r r r r r
r
r
r r' � s r' r3 )r r rp rt r r )r r' �klass�instancer
r
r �$test_create_autospec_unbound_methods s
z6SpecSignatureTest.test_create_autospec_unbound_methodsc C s? Gd d � d t � } t | d d �} | j | j d � d S)Nc @ s e Z d Z d Z d S)zESpecSignatureTest.test_create_autospec_keyword_arguments.<locals>.Foor5 N)r r r r r
r
r
r r' � s r' r �3)r r r r )r r' rk r
r
r �&test_create_autospec_keyword_arguments� s z8SpecSignatureTest.test_create_autospec_keyword_argumentsz(Keyword only arguments Python 3 specificc C s� d } i } t | | � | d } t | � } | d � | j d � | j t | d d � | d d d �| j d d d �d S)Nz def foo(a, *, b=None):
pass
r r3 r4 r r5 )�execr r rp rt )r Zfunc_def� namespacer rk r
r
r �+test_create_autospec_keyword_only_arguments� s
z=SpecSignatureTest.test_create_autospec_keyword_only_argumentsc C sK t � } d d � } | | _ t | � } | j d � | j j d � d S)Nc S s d S)Nr
)r r
r
r rw � s z@SpecSignatureTest.test_function_as_instance_attribute.<locals>.fr[ )r rw r r )r �objrw r r
r
r �#test_function_as_instance_attribute� s
z5SpecSignatureTest.test_function_as_instance_attributec C s� t g � } | j d � | j j d � | j t t | d � Gd d � d t � } t | � } | j j d � | j j j d � | j t t | j d � d S)Nr c @ s e Z d Z g Z d S)z0SpecSignatureTest.test_spec_as_list.<locals>.FooN)r r r r r
r
r
r r' � s r' r5 )r �appendr rp rq rr r r )r r r' r
r
r �test_spec_as_list� s
z#SpecSignatureTest.test_spec_as_listc C sM Gd d � d t � } t | � } x$ | | j f D] } | j | � q2 Wd S)Nc @ s e Z d Z e � Z d S)z.SpecSignatureTest.test_attributes.<locals>.SubN)r r r r �attrr
r
r
r �Sub� s r� )r r r� ru )r r� Zsub_mockr r
r
r �test_attributes� s z!SpecSignatureTest.test_attributesc C s) Gd d � d t � } t | � } | j d � | j j d � | j t t | j d � | j d � | j j d � | j t | j d d � | j t t | j d � | j d d g � | j j d d g � | j t t | j d � | j
j d � | j
j j d � | j t t | j
d � d S)Nc @ s( e Z d Z d d � Z e Z i Z d S)zGSpecSignatureTest.test_builtin_functions_types.<locals>.BuiltinSubclassc S s d S)Nr
)r �argr
r
r r( � s zKSpecSignatureTest.test_builtin_functions_types.<locals>.BuiltinSubclass.barN)r r r r( �sortedr� r
r
r
r �BuiltinSubclass� s r� r5 r r( r3 r4 ) �listr r� r rp rq rr r( rt r� r� �pop)r r� r r
r
r � test_builtin_functions_types� s
z.SpecSignatureTest.test_builtin_functions_typesc C s Gd d � d t � } t | � } | j d d � | j � | j d � t j d d � t j � t j d � g } | j | j | � | j j d d � | j j � | j j d � | j t j j d d � t j j � t j j d � g � | j | j | � d S)Nc @ s e Z d Z e � Z d S)z0SpecSignatureTest.test_method_calls.<locals>.SubN)r r r r r� r
r
r
r r� � s r� r3 r4 r5 )
r r r r r r r ra r� �extend)r r� r r. r
r
r �test_method_calls� s
*
4z#SpecSignatureTest.test_method_callsc C s� Gd d � d t � } t | � } | j t | � g � | j t t | � | j t t | j � | j t | � g � | j | d t � | j | j d t � d S)Nc @ s e Z d Z i Z d S)z=SpecSignatureTest.test_magic_methods.<locals>.BuiltinSubclassN)r r r r� r
r
r
r r� s r� r ) r� r r rp rt rV r� �assertIsInstancer )r r� r r
r
r �test_magic_methods� s z$SpecSignatureTest.test_magic_methodsc C s� Gd d � d t � } xk | | � f D]Z } t | d d �} | j | � | j t t | d d � | j t t | j d d � q& Wd S)Nc @ s e Z d Z e � Z d S)z,SpecSignatureTest.test_spec_set.<locals>.SubN)r r r r r� r
r
r
r r� s r� Zspec_setTr r( )r r ru rp rq �setattrr� )r r� �specr r
r
r �
test_spec_set
s
z SpecSignatureTest.test_spec_setc C s� Gd d � d t � } Gd d � d | � } Gd d � d t | � } x{ | | � | | � | | � f D]X } t | � } | j d d � | j j d d � | j d d
� | j j d d
� qg Wd S) Nc @ s4 e Z d Z e d d � � Z e d d � � Z d S)z/SpecSignatureTest.test_descriptors.<locals>.Fooc S s d S)Nr
)�clsr r r
r
r rw s z1SpecSignatureTest.test_descriptors.<locals>.Foo.fc S s d S)Nr
)r r r
r
r �g s z1SpecSignatureTest.test_descriptors.<locals>.Foo.gN)r r r � classmethodrw � staticmethodr� r
r
r
r r' s r' c @ s e Z d Z d S)z/SpecSignatureTest.test_descriptors.<locals>.BarN)r r r r
r
r
r �Bar" s r� c @ s e Z d Z d S)z/SpecSignatureTest.test_descriptors.<locals>.BazN)r r r r
r
r
r �Baz% s r� r3 r4 r5 r6 )r r r rw �assert_called_once_withr� )r r' r� r� r� r r
r
r �test_descriptors s ( z"SpecSignatureTest.test_descriptorsz No old style classes in Python 3c s� Gd d � d � � G� f d d � d � � } x~ � � � | | � f D]d } t | � } | j d d � | j j d d � | j t t | d � | j t t | j d � qH W| j j d d � | j j j d d � | j t t | j d � d S)Nc @ s e Z d Z d d � Z d S)z5SpecSignatureTest.test_old_style_classes.<locals>.Fooc S s d S)Nr
)r r r r
r
r rw 4 s z7SpecSignatureTest.test_old_style_classes.<locals>.Foo.fN)r r r rw r
r
r
r r' 3 s r' c s e Z d Z � � Z d S)z5SpecSignatureTest.test_old_style_classes.<locals>.BarN)r r r r� r
)r' r
r r� 7 s r� r3 r4 r )r rw r� rp rq rr r� )r r� r� r r
)r' r �test_old_style_classes1 s z(SpecSignatureTest.test_old_style_classesc C s� Gd d � d t � } | | _ t | � } | � | j | j j � | j � | j j � | j | j t j � t j j � g � | j | j
| j � | j | j
| j � | j
j
� | j t | j j
j � d S)Nc @ s( e Z d Z d d � Z d Z e Z d S)z+SpecSignatureTest.test_recursive.<locals>.Ac S s d S)Nr
)r r
r
r r I s z-SpecSignatureTest.test_recursive.<locals>.A.az foo bar bazN)r r r r r r( r
r
r
r �AH s r� )r �Br rS Zcalledr r ra r ZassertIsr r( � assertIsNot�lowerrp rs r )r r� r r
r
r �test_recursiveG s
(
z SpecSignatureTest.test_recursivec C s� Gd d � d t � } t | � } | j | | � � xy | | � f D]h } | j d d � | j j d d � | j j d � | j t | j d d � | j t t | d � qE Wt | � � } | j d � | j j d � | j j d d � | j t | j d d � | j t t | d � | j t | � | j j
d � | j j
j d � | j j
j d d � | j t t | j d
� | j � j
d � | j � j
j d � | j � j
j d d � | j t t | j � d
� d S) Nc @ s2 e Z d Z d d � Z Gd d � d e � Z d S)z@SpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Fooc S s d S)Nr
)r �xr
r
r r ` s zBSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.ac @ s e Z d Z d d � Z d S)zDSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.Barc S s d S)Nr
)r �yr
r
r rw c s zFSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.Bar.fN)r r r rw r
r
r
r r� b s r� N)r r r r r r� r
r
r
r r' _ s r' r� rh r r( r r] r� r� ) r r r� r r rp rt rq rr r� rw )r r' Z
class_mockZ this_mockZ
instance_mockr
r
r �!test_spec_inheritance_for_classes^ s0
z3SpecSignatureTest.test_spec_inheritance_for_classesc C s� Gd d � d t � } | | _ t | � } | � } | j t t | d � | j � } | j t t | d � t | � � } | j t t | d � | j t | � | j � } | j t t | d � d S)Nc @ s e Z d Z d Z d S)z+SpecSignatureTest.test_inherit.<locals>.Foor5 N)r r r r r
r
r
r r' � s r' r )r r' r rp rq rr rt )r r' r r~ Z
attr_instanceZ call_resultr
r
r � test_inherit� s z SpecSignatureTest.test_inheritc C s� t d � t t � t d � t t � t i � t t � t g � t t � t t � � t t � t d � t t � t d � t t � t d � t d � d S)Nr3 r g �?y �?FT)r rV r rL r� �set�float�complex)r r
r
r �
test_builtins� s
z SpecSignatureTest.test_builtinsc C s� d d � } t | � } | j t | � | d d � | j d d � | j d d d �| j d d d d � | | _ t | � } | j t | j � | j d d � | j j d d � | j j d d d d � d S) Nc S s d S)Nr
)r r r
r
r rw � s z*SpecSignatureTest.test_function.<locals>.fr3 r4 r r r5 r6 )r rp rt r rw )r rw r r
r
r �
test_function� s
z SpecSignatureTest.test_functionc s� Gd d � d t � � G� f d d � d t � } t | � � | j t � f d d � � � j d d � | j t � f d d � � � j } | j | j f d S)
Nc @ s e Z d Z d d d � Z d S)z;SpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserNc S s | d k r t d � � d S)Nz$Can only be accessed via an instance)rq )r r� �typer
r
r �__get__� s zCSpecSignatureTest.test_skip_attributeerrors.<locals>.Raiser.__get__)r r r r� r
r
r
r �Raiser� s r� c s+ e Z d Z � � Z e d d � � Z d S)z@SpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserClassc S s | | S)Nr
)r r r
r
r �existing� s zISpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserClass.existingN)r r r �raiserr� r� r
)r� r
r � RaiserClass� s r� c s � j d d d � S)Nr3 r4 r5 )r� )r� )�sr
r �<lambda>� s z=SpecSignatureTest.test_skip_attributeerrors.<locals>.<lambda>r3 r4 c s � j S)N)Z nonexistingr
)r� r
r r� � s ) r r rp rt r� rq r� r r( )r r� r� r
)r� r� r �test_skip_attributeerrors� s z+SpecSignatureTest.test_skip_attributeerrorsc C sj Gd d � d t � } t | � } | j t | � | d � | j d � | d d � | j d d � d S)Nc @ s e Z d Z d d d � Z d S)z3SpecSignatureTest.test_signature_class.<locals>.Foor5 c S s d S)Nr
)r r r r
r
r �__init__� s z<SpecSignatureTest.test_signature_class.<locals>.Foo.__init__N)r r r r� r
r
r
r r' � s r' r3 r6 rh )r r rp rt r� r )r r' r r
r
r �test_signature_class� s
z&SpecSignatureTest.test_signature_classz no old style classes in Python 3c C s� Gd d � d � } t | � } | j t | � | d � | j d � | j d d � | j t | j d � | d d � | j d d � | j d d d d � | j t | j d d d d �d S) Nc @ s e Z d Z d d d � Z d S)z=SpecSignatureTest.test_signature_old_style_class.<locals>.Foor5 c S s d S)Nr
)r r r r
r
r r� � s zFSpecSignatureTest.test_signature_old_style_class.<locals>.Foo.__init__N)r r r r� r
r
r
r r' � s r' r3 r r4 r6 rh r )r rp rt r� rs r )r r' r r
r
r � test_signature_old_style_class� s
z0SpecSignatureTest.test_signature_old_style_classc C s$ Gd d � d t � } t | � d S)Nc @ s e Z d Z d S)z6SpecSignatureTest.test_class_with_no_init.<locals>.FooN)r r r r
r
r
r r' s r' )r r )r r' r
r
r �test_class_with_no_init s z)SpecSignatureTest.test_class_with_no_initc C s! Gd d � d � } t | � d S)Nc @ s e Z d Z d S)z@SpecSignatureTest.test_old_style_class_with_no_init.<locals>.FooN)r r r r
r
r
r r' s r' )r )r r' r
r
r �!test_old_style_class_with_no_init s z3SpecSignatureTest.test_old_style_class_with_no_initc C s5 Gd d � d t � } t | � } | d d � | j d d � | j d d d d � | j t | d � | d d � } | j t | � | d d � | j d � | j d d � | d � | j d � | j d d � t | d d � � } | d d � | j d d � | j t | � | d � | j d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z;SpecSignatureTest.test_signature_callable.<locals>.Callablec S s d S)Nr
)r r� r� r
r
r r� s zDSpecSignatureTest.test_signature_callable.<locals>.Callable.__init__c S s d S)Nr
)r r r
r
r �__call__ s zDSpecSignatureTest.test_signature_callable.<locals>.Callable.__call__N)r r r r� r� r
r
r
r �Callable s r� r3 r4 r� r� r )r r r� rp rt r )r r� r r~ r
r
r �test_signature_callable s(
z)SpecSignatureTest.test_signature_callablec C s� Gd d � d t � } t | � } | � } | j � | j t | d � | j t | � | j t | d � t | � � } | j t | � | j t | d � d S)Nc @ s e Z d Z d d � Z d S)zASpecSignatureTest.test_signature_noncallable.<locals>.NonCallablec S s d S)Nr
)r r
r
r r� 1 s zJSpecSignatureTest.test_signature_noncallable.<locals>.NonCallable.__init__N)r r r r� r
r
r
r � NonCallable0 s r� r )r r r� rp rt )r r� r r~ r
r
r �test_signature_noncallable/ s
z,SpecSignatureTest.test_signature_noncallablec C s\ Gd d � d t � } t | � } | j } | j | t d � � | j � | j j � d S)Nc @ s e Z d Z d Z d S)z8SpecSignatureTest.test_create_autospec_none.<locals>.FooN)r r r r( r
r
r
r r' A s r' )r r r( �assertNotIsInstancer� r r� )r r' r Znoner
r
r �test_create_autospec_none@ s
z+SpecSignatureTest.test_create_autospec_nonec C s� Gd d � d t � } t | � } | j d � | j j d � | j j d d � | j d d � | j j d � | j j d d � d S)Nc @ s e Z d Z d d � Z d S)zMSpecSignatureTest.test_autospec_functions_with_self_in_odd_place.<locals>.Fooc S s d S)Nr
)r r r
r
r rw N s zOSpecSignatureTest.test_autospec_functions_with_self_in_odd_place.<locals>.Foo.fN)r r r rw r
r
r
r r' M s r' �
r )r r rw r )r r' r r
r
r �.test_autospec_functions_with_self_in_odd_placeL s
z@SpecSignatureTest.test_autospec_functions_with_self_in_odd_placec C s� Gd d � d t � } t | � } | j } | j | t � | d d d � | j d d d � | j d d d � | j j d d d � d S) Nc @ s" e Z d Z e d d � � Z d S)z5SpecSignatureTest.test_autospec_property.<locals>.Fooc S s d S)Nr5 r
)r r
r
r r \ s z9SpecSignatureTest.test_autospec_property.<locals>.Foo.fooN)r r r �propertyr r
r
r
r r' [ s r' r3 r4 r5 r6 rh r] )r r r r� r �abcr� )r r' r Z
mock_propertyr
r
r �test_autospec_propertyZ s z(SpecSignatureTest.test_autospec_propertyc C s{ Gd d � d t � } t | � } | j } | d d d � | j d d d � | j d d d � | j j d d d � d S) Nc @ s e Z d Z d g Z d S)z2SpecSignatureTest.test_autospec_slots.<locals>.Foor N)r r r � __slots__r
r
r
r r' l s r' r3 r4 r5 r6 rh r] )r r r r� r� )r r' r Z mock_slotr
r
r �test_autospec_slotsk s z%SpecSignatureTest.test_autospec_slotsN)*r r r ru rv rx ry r| �unittestZexpectedFailurer r� Z
skipUnless�six�PY3r� r� r� r� r� r� r� r� r� �skipIfr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r ro < sB
!
! ( ! ! ro c @ sa e Z d Z d d � Z d d � Z e j e j d � d d � � Z d d � Z
d
d � Z d S)
� TestCallListc C s� t � } | j | j t � | d d � | d d � | d d � | d d � xQ t d d � t d d � t d d � t d d � f D] } | j | | j � q� Wt d d � t d d � g } | j | | j � t d d � t d d � g } | j | | j � t d d � t d d � g } | j | | j � t d d � g } | j | | j � | j t d � | j � | j t d � g | j � d S) Nr3 r4 r r5 r6 r r] r` )r r� r^ r
r rO Z assertNotIn)r r rA Zcallsr
r
r �!test_args_list_contains_call_list{ s$
= z.TestCallList.test_args_list_contains_call_listc C se t � } | d d � | j d d � | j j � j d d d �d } | j t | j � | � d S) Nr3 r4 r r5 r` �catZdogzU[call(1, 2),
call.foo(a=3),
call.foo.bar(),
call.foo.bar().baz('fish', cat='dog')])r r r( r2 r r r, )r r r. r
r
r �test_call_list_str� s
z TestCallList.test_call_list_strz)Unicode is properly handled with Python 3c C sU t � } Gd d � d t � } | t d � | � i � | j t | j � d � d S)Nc @ s e Z d Z d d � Z d S)z9TestCallList.test_call_list_unicode.<locals>.NonAsciiReprc S s d S)N� ér
)r r
r
r �__repr__� s zBTestCallList.test_call_list_unicode.<locals>.NonAsciiRepr.__repr__N)r r r r� r
r
r
r � NonAsciiRepr� s r� r u [call(a=é)])r r �unicoder r r, )r r r� r
r
r �test_call_list_unicode� s z#TestCallList.test_call_list_unicodec C s� t d t d t �} | j � } z t j | j � t � } | j | j � | j | j t
� t
� g � d | _ | j | j t
� t
� t
d � g � Wd | j � Xd S)Nz%s.SomeClass.oneZ new_callabler5 ) r r r rz r r r� r r r, r �stop)r r{ r r� r
r
r �test_propertymock� s
,z TestCallList.test_propertymockc C sX t � } t � } | t | � _ | j } | j � | j | t � | j | t � d S)N)r r r� r r� r� r� )r rk r{ Zreturnedr
r
r � test_propertymock_returnvalue� s
z*TestCallList.test_propertymock_returnvalueN) r r r r� r� r� r� r� r� r� r� r� r
r
r
r r� y s
! r� �__main__)r� Z unittest2r� r r r r r r r r Z mock.mockr r
r r r ZTestCaser r0 ro r� r �mainr
r
r
r �<module> s 4
3�� � ?U