f
��_p� c @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ d dlmZ d dlmZ G dd� de�ZG dd� dej�ZG d d
� d
ej�ZG d d � d ej�ZG d
d� dej�ZG dd� dej�Zedkr�e�� dS )� N)
�call�_Call�create_autospec� MagicMock�Mock�ANY� _CallList�patch� PropertyMock� _callable)�datetime��partialc @ s& e Zd Zdd� Zdd� Zddd�ZdS ) � SomeClassc C � d S �N� ��self�a�br r �;/usr/lib64/python3.10/unittest/test/testmock/testhelpers.py�one � z
SomeClass.onec C r r r �r r r r �two r z
SomeClass.twoNc C r r r �r r r r r �three r zSomeClass.threer )�__name__�
__module__� __qualname__r r r r r r r r s r c @ �< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�AnyTestc C sF | � tt� � t� }|t� |�t� t� }|td� |jtd� d S �N��foo)� assertEqualr �objectr �assert_called_with�r �mockr r r �test_any s
zAnyTest.test_anyc C s$ | � tt�d� | � tt�d� d S )Nz<ANY>)r% �reprr �strr r r r � test_repr" s zAnyTest.test_reprc C s, t � }|t�� t�� d� |jttd� d S r" )r r �nowr' r r( r r r �test_any_and_datetime' s z AnyTest.test_any_and_datetimec C s� t � }G dd� dt�}t�� |� fD ]�}|�� ||||d� |j|||d� |� j||d� ttttd�tjtttd�t� t� jttd�g}| �||j � | �|j |� q$d S )Nc @ � e Zd Zdd� Zdd� ZdS )z9AnyTest.test_any_mock_calls_comparison_order.<locals>.Fooc S r r r �r �otherr r r �__eq__1 r z@AnyTest.test_any_mock_calls_comparison_order.<locals>.Foo.__eq__c S r r r r1 r r r �__ne__2 r z@AnyTest.test_any_mock_calls_comparison_order.<locals>.Foo.__ne__N)r r r r3 r4 r r r r �Foo0 � r5 �r$ �bar)ZzingaZalpha)Za1Zz99)
r r& r r. �
reset_mock�methodr r r% �
mock_calls)r r) r5 Zd�expectedr r r �$test_any_mock_calls_comparison_order. s �z,AnyTest.test_any_mock_calls_comparison_orderc C sN G dd� d�}t � }||� d� |�ttd�g� |�td� |�td� d S )Nc @ � e Zd Zdd� ZdS )z%AnyTest.test_any_no_spec.<locals>.Fooc S r r r r1 r r r r3 F r z,AnyTest.test_any_no_spec.<locals>.Foo.__eq__N�r r r r3 r r r r r5 E � r5 � �r �assert_has_callsr r r' �assert_any_call�r r5 r) r r r �test_any_no_specC s zAnyTest.test_any_no_specc C sR G dd� d�}t |d�}||� d� |�ttd�g� |�td� |�td� d S )Nc @ r> )z*AnyTest.test_any_and_spec_set.<locals>.Fooc S r r r r1 r r r r3 Q r z1AnyTest.test_any_and_spec_set.<locals>.Foo.__eq__Nr? r r r r r5 P r@ r5 )�specrA rB rE r r r �test_any_and_spec_setN s
z AnyTest.test_any_and_spec_setN) r r r r* r- r/ r= rF rH r r r r r! s r! c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�CallTestc C s t � }| �|t � � | �|t d�� | �|t d�� | �|t i f�� | �|t d�� | �|t di f�� | �|t ddi f�� | �|t d�� | �|t d�� | �|t di f�� | �|t d di f�� t d
�}| �|t d
�� | �|t d �� | �|t d i f�� | �|t dd i f�� t d
�}| �|t d �� | �|t dd i f�� t d�}| �|t d�� | �|t ddi f�� | �|t d�� | �|t ddi f�� t ddif�}| �|t ddddif�� | �|t dddif�� | �|t dddif�� | �|t ddif�� d S )N)� �r )rJ r rJ r r# �r8 r �bazZspam��rA � � )rJ rO rO )�rA rP � )r$ rR )rJ rR rR )r8 rR r8 r rQ �r r% �assertNotEqual)r �kallr r r �test_call_with_call\ s< z CallTest.test_call_with_callc C sj t � }| �|d� | �|d� | �|d� | �|d� | �|ddi f� | �|di f� | �|i f� d S )Nr r# rK �r$ r r$ )r r% �r �argsr r r �test_empty__Call� s zCallTest.test_empty__Callc C s� t ddi f�}| �|d� | �|d� | �|ddi f� | �|di f� | �|d� | �|d� | �|i f� | �|d� | �|d� | �|di f� d S ) Nr$ r r# rX rK )r8 rL r8 rT rY r r r �test_named_empty_call� s z CallTest.test_named_empty_callc C sf t di f�}| �|d� | �|d� | �|ddi f� | �|di f� | �|jd� | �|ji � d S )NrO rN �r$ rO r$ )r r% rZ �kwargsrY r r r �test_call_with_args� s z CallTest.test_call_with_argsc C sh t ddi f�}| �|d� | �|ddi f� | �|jd� | �|ji � | �|d� | �|di f� d S )Nr$ rO r] rN )r r% rZ r^ rU rY r r r �test_named_call_with_args� s z"CallTest.test_named_call_with_argsc C s� t dtddd�f�}| �|tddd�f� | �|dtddd�f� | �|ddtddd�f� | �|dtddd�f� | �|jd� | �|jtddd�� d S )Nr rQ rS �r r r$ )r �dictr% rZ r^ rY r r r �test_call_with_kwargs� s z CallTest.test_call_with_kwargsc C s� t ddtddd�f�}| �|dtddd�f� | �|ddtddd�f� | �|jd� | �|jtddd�� | �|tddd�f� | �|dtddd�f� d S )Nr$ r rQ rS ra )r rb r% rZ r^ rU rY r r r �test_named_call_with_kwargs� s z$CallTest.test_named_call_with_kwargsc C sN t di f�}| �|tddd�� | �tddd�|� | �tddd�|g� d S )NrO rA rP rQ )r r% r �assertInrY r r r �#test_call_with_args_call_empty_name� s z,CallTest.test_call_with_args_call_empty_namec C sR | � td�tdd�� | �td�tddd�k� | �tdi f�tddd�k� d S )NrN rA rP rQ �rA rP )rU r r � assertFalse�
assertTruer r r r � test_call_ne� s zCallTest.test_call_nec C s8 t d�}dd | tfD ] }| �||� | �||k� qd S )NrN rA )r �intrU rh )r rV �valuer r r �test_call_non_tuples� s z CallTest.test_call_non_tuplesc C s^ | � tt� �d� | � ttd��d� | � ttdddif��d� | � ttddddif��d � | � tt�d
� | � tt�d
� | � tt� �d� | � ttd ��d � | � ttd
d��d� | � tt� j�d� | � ttd �jjdd�j�d� | � tt� jd ddd��d� | � tt� � �d� | � ttd �d��d� | � tt� � �� j� d ��d� d S )Nzcall()r# z
call.foo()rO r r zcall(1, 2, 3, a='b')r8 zcall.bar(1, 2, 3, a='b')r rA zcall(1)Zthing)Zzzzcall(zz='thing')z
call().foorQ �r zcall().foo.bar().bingrP zcall().foo(1, 2, a=3)zcall()()z call()(2)zcall()().bar().baz.beep(1))
r% r+ r r r, r$ r8 �bingrM �beepr r r r r- � s6 �����zCallTest.test_reprc C s| | � t� ddi f� | � tddddd�ddddd�f� t� }|d d
d� |dd d � | � |jtd d
d�tdd d �g� d S )
NrJ r r$ r8 rQ rS )r r r7 rA rP � ra )r% r r �call_args_listr( r r r � test_call� s � �zCallTest.test_callc C s� | � t�d�ddi f� | � tjjdd�ddddif� t� }|�dd d
� |jjd
d d � | � |jt�dd d
�tjjd
d d �g� d S )
NrA r$ )rA Zeggs)�fishzbar.bazr rt rP rQ rq ra )r% r r$ r8 rM r � method_callsr( r r r �test_attribute_call
s � �z CallTest.test_attribute_callc C s t d��d�jddd�}| �|ddtdd�f� t� }|ddddd� | �|jt ddddd�� | �|jt ddd�� | �|jt ddddd�g� | �|j t ddddd�g� t� }|�d��� � j
j d d� t �d��� � j
j d d�}| �|j d
|� | �|j |� � � d S ) NrA rP rQ rS rn z ().foo().bar�rQ ra rq i����)
r r$ r8 r% rb r Z call_argsrU rr r; rM rp � call_list)r Zresultr) Z last_callr r r �test_extended_call s zCallTest.test_extended_callc C s@ t dd�j}t dd�j}| �||� | �||� | �||� d S �NrA ��xrP )r r$ r% rU r r r r �test_extended_not_equal* s
z CallTest.test_extended_not_equalc C sH t dd��� j}t dd��� j}| �||� | �||� | �||� d S rz )r r$ r8 r% rU r r r r �test_nested_calls_not_equal2 s
z$CallTest.test_nested_calls_not_equalc C s� t � }|d� | �td��� |j� t � }|d��d� | �td��d��� |j� t � }|d��d�d� | �td��d�d��� |j� t � }t|d��d�d�jj� d�d�� td��d�d�jj� d�d��
� }| �|�� |j� d S )NrA rP rQ rS � ) r r% r rx r; r: rk r$ r8 rM �__int__)r r) rV r r r �test_call_list: s" ��$$zCallTest.test_call_listc C s> | � tt� t� }t|� | � |jtg� | � tg|j� d S r )r% r r r rk r; �r �mr r r �
test_call_anyO s
zCallTest.test_call_anyc C sh t dddifdd�}| �t|�d� | �|d d� | �|d ddi� t dddif�}| �||� d S ) Nrg r rQ T)r rP r rA )r r% Zlen)r rZ Z
other_argsr r r �test_two_args_callX s zCallTest.test_two_args_callc C sJ | � tdd�d d� | � td�d d� | � tdddif�d d� d S ) Nr r$ r )�r8 ZbarzrJ r� ZhelloZworld)r% r r r r r �test_call_with_namea s z CallTest.test_call_with_namec C s� t � }|� �� d � | �|jt� t� �� t� �� �d�t� �� �� � g� t � }d|� �� d< | �|jt� t� �� t� �� �dd�g� t � }t|� �� � | �|jt� t� �� t� �� �� g� d S )Nr8 rA ) r r$ r% r; r Z __getitem__Z __setitem__ZiterZ__iter__r� r r r �test_dunder_callf s$ *� ��zCallTest.test_dunder_callN)r r r rW r[ r\ r_ r` rc rd rf rj rm r- rs rv ry r} r~ r� r� r� r� r� r r r r rI Z s* % rI c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMS )N�SpecSignatureTestc C s� | � tt|d� |�dd� |j�dd� | � t|jjdd� | � t|jd� |�� |j�� | � t|jjd� | � t|jd� |�� |j�� | � t|jjd� | � t|jdd� |�d� |j�d� |jdd� |jjdd� d S )Nr$ rA rP rQ rS rn ) � assertRaises�AttributeError�getattrr r' �AssertionError� TypeErrorr r r( r r r �_check_someclass_mock} s.
�
�
�
z'SpecSignatureTest._check_someclass_mockc C s* t t�}| �|� t t� �}| �|� d S r )r r r� r( r r r �
test_basic� s
z SpecSignatureTest.test_basicc C sP dd� }t |dd�}| �|� d� G dd� dt�}t |dd�}| �|� d� d S )Nc S r r r r r r r �f� r z>SpecSignatureTest.test_create_autospec_return_value.<locals>.fr$ �� return_valuec @ � e Zd ZdS )z@SpecSignatureTest.test_create_autospec_return_value.<locals>.FooN�r r r r r r r r5 � r@ r5 )r r% r& )r r� r) r5 r r r �!test_create_autospec_return_value� s z3SpecSignatureTest.test_create_autospec_return_valuec C s, t t�}t|� |�� | �|jjd� d S )Nr )r rk r9 r% r� Z
call_countr� r r r �test_autospec_reset_mock� s z*SpecSignatureTest.test_autospec_reset_mockc C s> G dd� dt �}t� |d�}|�� }|� �d� |�d� d S )Nc @ r> )z;SpecSignatureTest.test_mocking_unbound_methods.<locals>.Fooc S r r r )r r$ r r r r$ � r z?SpecSignatureTest.test_mocking_unbound_methods.<locals>.Foo.fooN�r r r r$ r r r r r5 � r@ r5 r$ rA )r& r �startr$ r' )r r5 �pZmock_foor r r � test_mocking_unbound_methods� s
z.SpecSignatureTest.test_mocking_unbound_methodsc C s. G dd� dt �}t|dd�}| �|jd� d S )Nc @ � e Zd ZdZdS )zESpecSignatureTest.test_create_autospec_keyword_arguments.<locals>.FoorQ N�r r r r r r r r r5 � r@ r5 Z3rn )r& r r% r )r r5 r� r r r �&test_create_autospec_keyword_arguments� s z8SpecSignatureTest.test_create_autospec_keyword_argumentsc C sV d d�dd�}t |�}|d� |�d� | �t|dd� |ddd� |jddd� d S )N�r c S r r r ra r r r r$ � r zJSpecSignatureTest.test_create_autospec_keyword_only_arguments.<locals>.foorA rP rQ )r r' r� r� )r r$ r� r r r �+test_create_autospec_keyword_only_arguments� s
z=SpecSignatureTest.test_create_autospec_keyword_only_argumentsc C s6 t � }dd� }||_t|�}|�d� |j�d� d S )Nc S r r r rn r r r r� � r z@SpecSignatureTest.test_function_as_instance_attribute.<locals>.fro )r r� r r' )r �objr� r) r r r �#test_function_as_instance_attribute� s
z5SpecSignatureTest.test_function_as_instance_attributec C sv t g �}|�d� |j�d� | �tt|d� G dd� dt�}t |�}|j�d� |jj�d� | �tt|jd� d S )Nr$ c @ s e Zd Zg ZdS )z0SpecSignatureTest.test_spec_as_list.<locals>.FooNr� r r r r r5 � r@ r5 rQ )r �appendr' r� r� r� r& r$ )r r) r5 r r r �test_spec_as_list� s
z#SpecSignatureTest.test_spec_as_listc C s6 G dd� dt �}t|�}||jfD ]}| �|� q"d S )Nc @ � e Zd Ze� ZdS )z.SpecSignatureTest.test_attributes.<locals>.SubN�r r r r �attrr r r r �Sub� r@ r� )r r r� r� )r r� Zsub_mockr) r r r �test_attributes� s z!SpecSignatureTest.test_attributesc s� G dd� dt �� G � fdd�dt �}t|�}|�d� | �t�� |�� W d � n1 s^0 Y | �t�� |�dd� W d � n1 s�0 Y d S )Nc @ r> )zVSpecSignatureTest.test_spec_has_descriptor_returning_function.<locals>.CrazyDescriptorc S s |d u rdd� S d S )Nc S r r r r{ r r r �<lambda>� r zpSpecSignatureTest.test_spec_has_descriptor_returning_function.<locals>.CrazyDescriptor.__get__.<locals>.<lambda>r )r r� Ztype_r r r �__get__� � z^SpecSignatureTest.test_spec_has_descriptor_returning_function.<locals>.CrazyDescriptor.__get__N�r r r r� r r r r �CrazyDescriptor� � r� c s e Zd Z� � ZdS )zNSpecSignatureTest.test_spec_has_descriptor_returning_function.<locals>.MyClassN)r r r � some_attrr �r� r r �MyClass� r� r� rA rP )r& r r� r� r� )r r� r) r r� r �+test_spec_has_descriptor_returning_function� s
& z=SpecSignatureTest.test_spec_has_descriptor_returning_functionc s� G � fdd�dt �� � � }| �t�� |j W d � n1 s@0 Y | �|�d�d� t|�}|�d� | �t�� |�� W d � n1 s�0 Y | �t�� |�dd� W d � n1 s�0 Y d S )Nc s& e Zd Z�� fdd�Zdd� Z� ZS )zISpecSignatureTest.test_spec_has_function_not_in_bases.<locals>.CrazyClassc s t � | ��� dg S )N�crazy)Zsuper�__dir__r )�
CrazyClass� __class__r r r�
� zQSpecSignatureTest.test_spec_has_function_not_in_bases.<locals>.CrazyClass.__dir__c S s |dkrdd� S t |��d S )Nr� c S � | S r r r{ r r r r� r zgSpecSignatureTest.test_spec_has_function_not_in_bases.<locals>.CrazyClass.__getattr__.<locals>.<lambda>�r� )r Zitemr r r � __getattr__ � zUSpecSignatureTest.test_spec_has_function_not_in_bases.<locals>.CrazyClass.__getattr__)r r r r� r� Z
__classcell__r �r� )r� r r� s r� �* rA rP )r& r� r� r2 r% r� r r� )r Zinstr) r r� r �#test_spec_has_function_not_in_bases s
$
& z5SpecSignatureTest.test_spec_has_function_not_in_basesc C s� G dd� dt �}t|�}|�d� |j�d� | �tt|jd� |�d� |j�d� | �t|jdd� | �tt|jd� |� ddg� |j �ddg� | �tt|j d� |j
� d� |j
j �d� | �tt|j
d� d S )Nc @ s e Zd Zdd� ZeZi ZdS )zGSpecSignatureTest.test_builtin_functions_types.<locals>.BuiltinSubclassc S r r r )r Zargr r r r8 ' r zKSpecSignatureTest.test_builtin_functions_types.<locals>.BuiltinSubclass.barN)r r r r8 �sortedr� r r r r �BuiltinSubclass& � r� rQ r$ r8 rA rP ) �listr r� r' r� r� r� r8 r� r� r� Zpop�r r� r) r r r � test_builtin_functions_types" s
z.SpecSignatureTest.test_builtin_functions_typesc C s� G dd� dt �}t|�}|�dd� |�� |�d� t�dd�t�� t�d�g}| �|j|� |j�dd� |j�� |j�d� |� tj�dd�tj�� tj�d�g� | �|j|� d S )Nc @ r� )z0SpecSignatureTest.test_method_calls.<locals>.SubNr� r r r r r� ? r@ r� rA rP rQ )
r r r r r r r% ru r� Zextend)r r� r) r< r r r �test_method_calls> s
�z#SpecSignatureTest.test_method_callsc C s| G dd� dt �}t|�}| �t |�g � | �tt|� | �tt|j� | �t |�g � | �|d t� | �|jd t� d S )Nc @ s e Zd Zi ZdS )z=SpecSignatureTest.test_magic_methods.<locals>.BuiltinSubclassN)r r r r� r r r r r� U r@ r� r$ ) r� r r% r� r� rk r� �assertIsInstancer r� r r r �test_magic_methodsT s z$SpecSignatureTest.test_magic_methodsc C s` G dd� dt �}||� fD ]@}t|dd�}| �|� | �tt|dd� | �tt|jdd� qd S )Nc @ r� )z,SpecSignatureTest.test_spec_set.<locals>.SubNr� r r r r r� c r@ r� T)Zspec_setr$ r8 )r r r� r� r� Zsetattrr� )r r� rG r) r r r �
test_spec_setb s
z SpecSignatureTest.test_spec_setc C s� G dd� dt �}G dd� d|�}G dd� dt|�}||� ||� ||� fD ]@}t|�}|�dd� |j�dd� |�d d
� |j�d d
� qHd S ) Nc @ s$ e Zd Zedd� �Zedd� �ZdS )z/SpecSignatureTest.test_descriptors.<locals>.Fooc S r r r )�clsr r r r r r� p r� z1SpecSignatureTest.test_descriptors.<locals>.Foo.fc S r r r ra r r r �gr r� z1SpecSignatureTest.test_descriptors.<locals>.Foo.gN)r r r � classmethodr� � staticmethodr� r r r r r5 o s
r5 c @ r� )z/SpecSignatureTest.test_descriptors.<locals>.BarNr� r r r r �Baru r r� c @ r� )z/SpecSignatureTest.test_descriptors.<locals>.BazNr� r r r r �Bazw r r� rA rP rQ rS )r& r r r� �assert_called_once_withr� )r r5 r� r� rG r) r r r �test_descriptorsn s z"SpecSignatureTest.test_descriptorsc C s� G dd� dt �}||_t|�}|� | �|jj� |�� |j�� | �|jt�� tj�� g� | � |j
|j � | � |j
|j � |j
�
� | �t|j j
j� d S )Nc @ s e Zd Zdd� ZdZeZdS )z+SpecSignatureTest.test_recursive.<locals>.Ac S r r r r r r r r � r z-SpecSignatureTest.test_recursive.<locals>.A.az foo bar bazN)r r r r r$ r8 r r r r �A� r� r� )r& ZBr rh Zcalledr r% ru r ZassertIsr$ r8 � assertIsNotZlowerr� r� r' )r r� r) r r r �test_recursive� s
z SpecSignatureTest.test_recursivec C s^ G dd� dt �}t|�}| �||� � ||� fD ]L}|jdd� |jjdd� |j�d� | �t|jdd� | �tt|d� q0t|� �}|�d� |j�d� |jjdd� | �t|jdd� | �tt|d� | �t|� |j �
d� |j j
�d� |j j
jdd � | �tt|j d
� |� � �
d� |� � j
�d� |� � j
jdd � | �tt|� � d
� d S ) Nc @ s$ e Zd Zdd� ZG dd� de�ZdS )z@SpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Fooc S r r r )r r| r r r r � r zBSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.ac @ r> )zDSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.Barc S r r r )r �yr r r r� � r zFSpecSignatureTest.test_spec_inheritance_for_classes.<locals>.Foo.Bar.fN�r r r r� r r r r r� � r@ r� N)r r r r r& r� r r r r r5 � r6 r5 r r{ r$ r8 r rq )r� r� ) r& r r� r r' r� r� r� r� r� r� )r r5 Z
class_mockZ this_mockZ
instance_mockr r r �!test_spec_inheritance_for_classes� s0
z3SpecSignatureTest.test_spec_inheritance_for_classesc C s� G dd� dt �}||_t|�}|� }| �tt|d� |�� }| �tt|d� t|� �}| �tt|d� | �t|� |�� }| �tt|d� d S )Nc @ r� )z+SpecSignatureTest.test_inherit.<locals>.FoorQ Nr� r r r r r5 � r@ r5 r )r& r5 r r� r� r� r� )r r5 r) �instanceZ
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 )NrA r$ g �?y �?FT)r rk r, rb r� ZsetZfloatZcomplexr r r r �
test_builtins� s
z SpecSignatureTest.test_builtinsc C s� dd� }t |�}| �t|� |dd� |�dd� |jddd� |jddd� ||_t |�}| �t|j� |�dd� |j�dd� |jjddd� d S ) Nc S r r r ra r r r r� � r z*SpecSignatureTest.test_function.<locals>.frA rP r� ra rQ rS )r r� r� r' r� �r r� r) r r r �
test_function� s
z SpecSignatureTest.test_functionc s� G dd� dt �� G � fdd�dt �}| �|�dd�d� t|��| �t�fdd �� | ���dd��jj� | �t�fd
d �� �j}|j |j
f d S ) Nc @ s e Zd Zddd�ZdS )z;SpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserNc S s |d u rt d��d S )Nz$Can only be accessed via an instancer� )r r� �typer r r r� � r� zCSpecSignatureTest.test_skip_attributeerrors.<locals>.Raiser.__get__r r� r r r r �Raiser� r@ r� c s e Zd Z� � Zedd� �ZdS )z@SpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserClassc S s | | S r r ra r r r �existing s zISpecSignatureTest.test_skip_attributeerrors.<locals>.RaiserClass.existingN)r r r �raiserr� r� r )r� r r � RaiserClass s r� rA rP rQ c s � � ddd�S )NrA rP rQ )r� r{ ��sr r r� r z=SpecSignatureTest.test_skip_attributeerrors.<locals>.<lambda>c s � j S r )Z nonexistingr r� r r r� r ) r& r% r� r r� r� r� r� r� r$ r8 )r r� r� r )r� r� r �test_skip_attributeerrors� s z+SpecSignatureTest.test_skip_attributeerrorsc C s� G dd� dt �}t|�}| �t|� |d� |�d� |jdd� | �t|jd� |dd� |�dd� |jddd� | jt|jddd� d S ) Nc @ s e Zd Zddd�ZdS )z3SpecSignatureTest.test_signature_class.<locals>.FoorQ c S r r r r r r r �__init__ r z<SpecSignatureTest.test_signature_class.<locals>.Foo.__init__Nrw �r r r r� r r r r r5 r@ r5 rA rn rP rS r ra )r& r r� r� r� r� r' rE r r r �test_signature_class s
z&SpecSignatureTest.test_signature_classc C s G dd� dt �}t|� d S )Nc @ r� )z6SpecSignatureTest.test_class_with_no_init.<locals>.FooNr� r r r r r5 * r@ r5 )r& r )r r5 r r r �test_class_with_no_init' s z)SpecSignatureTest.test_class_with_no_initc C s� G dd� dt �}t|�}|dd� |�dd� |jddd� | �t|d� |dd�}| �t|� |dd� |�d� |jdd� |d� |�d� |jdd� t|dd��}|dd� |jdd� | �t|� |d� |�d� d S )Nc @ r0 )z;SpecSignatureTest.test_signature_callable.<locals>.Callablec S r r r )r r| r� r r r r� 1 r zDSpecSignatureTest.test_signature_callable.<locals>.Callable.__init__c S r r r r r r r �__call__2 r zDSpecSignatureTest.test_signature_callable.<locals>.Callable.__call__N)r r r r� r� r r r r �Callable0 r6 r� rA rP �r| r� r rn )r& r r� r� r� r' )r r� r) r� r r r �test_signature_callable/ s(
z)SpecSignatureTest.test_signature_callablec C sv G dd� dt �}t|�}|� }|�� | �t|d� | �t|� | �t|d� t|� �}| �t|� | �t|d� d S )Nc @ r> )zASpecSignatureTest.test_signature_noncallable.<locals>.NonCallablec S r r r r r r r r� M r� zJSpecSignatureTest.test_signature_noncallable.<locals>.NonCallable.__init__Nr� r r r r � NonCallableL r@ r� r )r& r r� r� r� )r r� r) r� r r r �test_signature_noncallableK s
z,SpecSignatureTest.test_signature_noncallablec C sD G dd� dt �}t|�}|j}| �|td �� |�� |j�� d S )Nc @ s e Zd ZdZdS )z8SpecSignatureTest.test_create_autospec_none.<locals>.FooN)r r r r8 r r r r r5 ] r@ r5 )r& r r8 �assertNotIsInstancer� r$ r� )r r5 r) Znoner r r �test_create_autospec_none\ s z+SpecSignatureTest.test_create_autospec_nonec C sf G dd� dt �}t|�}|�d� |j�d� |jjdd� |jdd� |j�d� |jjdd� d S )Nc @ r> )zMSpecSignatureTest.test_autospec_functions_with_self_in_odd_place.<locals>.Fooc S r r r )r r r r r r� j r zOSpecSignatureTest.test_autospec_functions_with_self_in_odd_place.<locals>.Foo.fNr� r r r r r5 i r@ r5 �
r )r& r r� r' )r r5 r r r r �.test_autospec_functions_with_self_in_odd_placeh s
z@SpecSignatureTest.test_autospec_functions_with_self_in_odd_placec sv G dd� dt �� G dd� dt��G � �fdd�dt �}t|�}�fdd�}||j� ||j� ||j� ||j� d S ) Nc @ s& e Zd Zdd� Zddd�Zdd� ZdS ) zCSpecSignatureTest.test_autospec_data_descriptor.<locals>.Descriptorc S s
|| _ d S r )rl )r rl r r r r� w r� zLSpecSignatureTest.test_autospec_data_descriptor.<locals>.Descriptor.__init__Nc S r� r r )r r� r� r r r r� z r� zKSpecSignatureTest.test_autospec_data_descriptor.<locals>.Descriptor.__get__c S r r r )r r� rl r r r �__set__} r zKSpecSignatureTest.test_autospec_data_descriptor.<locals>.Descriptor.__set__r )r r r r� r� r� r r r r �
Descriptorv s
r� c @ r� )zCSpecSignatureTest.test_autospec_data_descriptor.<locals>.MyPropertyNr� r r r r �
MyProperty r@ r� c s2 e Zd ZdgZedd� �Z�dd� �Z� d�ZdS )z<SpecSignatureTest.test_autospec_data_descriptor.<locals>.Foo�slotc S r r r r r r r �prop� r� zASpecSignatureTest.test_autospec_data_descriptor.<locals>.Foo.propc S r r r r r r r �subprop� r� zDSpecSignatureTest.test_autospec_data_descriptor.<locals>.Foo.subpropr� N)r r r Z __slots__�propertyr� r� �descr )r� r� r r r5 � s
r5 c sH � � | t� | ddd� | �ddd� | �ddd� | j�ddd� d S )NrA rP rQ rS r rq )r� r Zabcr� )Z mock_attrr r r �check_data_descriptor� s
zNSpecSignatureTest.test_autospec_data_descriptor.<locals>.check_data_descriptor)r& r� r r� r� r� r� )r r5 r$ r� r )r� r� r r � test_autospec_data_descriptoru s
z/SpecSignatureTest.test_autospec_data_descriptorc C s\ t �tjt�� �}| �|� t� t|�}|� |�� |�� |ddd� |�ddd� d S )NrS r rq ) �typesZ
MethodType�timeZctimer� r, r r� r9 )r ZmethZmockedr r r �'test_autospec_on_bound_builtin_function� s z9SpecSignatureTest.test_autospec_on_bound_builtin_functionc C s0 G dd� d�}|� }t |�}| �t|d�� d S )Nc @ r> )zESpecSignatureTest.test_autospec_getattr_partial_function.<locals>.Fooc S s t dd� |�S )Nc S r� r r )Znamer r r r� � r zcSpecSignatureTest.test_autospec_getattr_partial_function.<locals>.Foo.__getattr__.<locals>.<lambda>r )r Z attributer r r r� � r� zQSpecSignatureTest.test_autospec_getattr_partial_function.<locals>.Foo.__getattr__N)r r r r� r r r r r5 � r� r5 r )r rh Zhasattr)r r5 ZproxyZautospecr r r �&test_autospec_getattr_partial_function� s z8SpecSignatureTest.test_autospec_getattr_partial_functionc C sp dd� }t |�}|dd� |ddd� | �t�|�t�|�� | �|jtdd�tddd�g� | �t|d� d S )Nc S r r r r� r r r �myfunc� r z=SpecSignatureTest.test_spec_inspect_signature.<locals>.myfuncrA rP r� )r r% �inspect� signaturer; r r� r� )r r r) r r r �test_spec_inspect_signature� s
z-SpecSignatureTest.test_spec_inspect_signaturec C s� d ddddd�dd�}| � |dddd �d
� t|�}|dddd � |ddd � | � t�|�t�|�� | � |jtdddd �tddd �g� | �t|d� | jt|dddd d � d S )
Nr� rk )r r �cZreturnc S s | | | S r r )r r r r r r r$ � r� zFSpecSignatureTest.test_spec_inspect_signature_annotations.<locals>.foorA rP rQ )r rq rS )r� )r% r r r r; r r� r� )r r$ r) r r r �'test_spec_inspect_signature_annotations� s $z9SpecSignatureTest.test_spec_inspect_signature_annotationsc C s" dd� }t |�}| �|jd� d S )Nc S s dS )NZnoper r r r r r� � r z>SpecSignatureTest.test_spec_function_no_name.<locals>.<lambda>Zfuncopy)r r% r )r Zfuncr) r r r �test_spec_function_no_name� r� z,SpecSignatureTest.test_spec_function_no_namec C sf dd� }t |�}|d� |�td�g� | �t�� |�td�g� W d � n1 sX0 Y d S )Nc S r r r rn r r r r� � r z@SpecSignatureTest.test_spec_function_assert_has_calls.<locals>.frA rP )r rC r r� r� r� r r r �#test_spec_function_assert_has_calls� s z5SpecSignatureTest.test_spec_function_assert_has_callsc C sZ dd� }t |�}|d� |�d� | �t�� |�d� W d � n1 sL0 Y d S )Nc S r r r rn r r r r� � r z?SpecSignatureTest.test_spec_function_assert_any_call.<locals>.frA rP )r rD r� r� r� r r r �"test_spec_function_assert_any_call� s
z4SpecSignatureTest.test_spec_function_assert_any_callc C sv dd� }t � }t||d�}|d�d� | �|jtd�g� | �|jtd�g� |�� | �|jg � | �|jg � d S )Nc S r r r rn r r r r� � r z:SpecSignatureTest.test_spec_function_reset_mock.<locals>.fr� rA rP )r r r% r; r r9 )r r� Zrvr) r r r � test_spec_function_reset_mock� s z/SpecSignatureTest.test_spec_function_reset_mockN))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� { sL
&
,
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestCallListc C s t � }| �|jt� |dd� |dd� |dd� |dd� tdd�tdd�tdd�tdd�fD ]}| �||j� q`tdd�tdd�g}| �||j� tdd�tdd�g}| �||j� tdd�tdd�g}| �||j� tdd�g}| �||j� | �td�|j� | �td�g|j� d S ) NrA rP rQ rn rS rq r� rt )r r� rr r r re Z assertNotIn)r r) rV Zcallsr r r �!test_args_list_contains_call_list� s$
( z.TestCallList.test_args_list_contains_call_listc C sJ t � }|dd� |jdd� |j�� jddd� d}| �t|j�|� d S ) NrA rP rQ rn rt Zdog)ZcatzU[call(1, 2),
call.foo(a=3),
call.foo.bar(),
call.foo.bar().baz('fish', cat='dog')])r r$ r8 rM r% r, r; )r r) r< r r r �test_call_list_str s
�z TestCallList.test_call_list_strc C s� t dt td�}|�� }zftj |�� t� }|j |�� | �|j t
� t
� g� d|_| �|j t
� t
� t
d�g� W |� � n
|� � 0 d S )Nz%s.SomeClass.one)Z new_callablerQ ) r r r r� r r r� r' r% r; r Zstop)r r� r) r� r r r �test_propertymock& s z TestCallList.test_propertymockc C s@ t � }t� }|t|�_|j}|�� | �|t � | �|t� d S r )r r r� r$ r� r� r� )r r� r� Zreturnedr r r � test_propertymock_returnvalue8 s
z*TestCallList.test_propertymock_returnvalueN)r r r r r
r r r r r r r � s r c @ r )�TestCallablePredicatec C s( t tttttfD ]}| �t|�� qd S r )r, Zbytesrk r� Ztupler ri r
)r r� r r r � test_typeE s z TestCallablePredicate.test_typec C s& G dd� d�}|� }| � t|�� d S )Nc @ r> )z>TestCallablePredicate.test_call_magic_method.<locals>.Callablec S r r r r r r r r� K r zGTestCallablePredicate.test_call_magic_method.<locals>.Callable.__call__N)r r r r� r r r r r� J r@ r� )ri r
)r r� r� r r r �test_call_magic_methodI s z,TestCallablePredicate.test_call_magic_methodc C �" G dd� d�}| � t|j�� d S )Nc @ � e Zd Zedd� �ZdS )zATestCallablePredicate.test_staticmethod.<locals>.WithStaticMethodc S r r r r r r r �
staticfuncQ r� zLTestCallablePredicate.test_staticmethod.<locals>.WithStaticMethod.staticfuncN)r r r r� r r r r r �WithStaticMethodP � r )ri r
r )r r r r r �test_staticmethodO � z'TestCallablePredicate.test_staticmethodc C r )Nc @ � e Zd Zed�ZdS )zMTestCallablePredicate.test_non_callable_staticmethod.<locals>.BadStaticMethodN)r r r r� � not_callabler r r r �BadStaticMethodV r@ r �rh r
r )r r r r r � test_non_callable_staticmethodU � z4TestCallablePredicate.test_non_callable_staticmethodc C r )Nc @ r )z?TestCallablePredicate.test_classmethod.<locals>.WithClassMethodc S r r r )r� r r r � classfunc\ r� zITestCallablePredicate.test_classmethod.<locals>.WithClassMethod.classfuncN)r r r r� r r r r r �WithClassMethod[ r r! )ri r
r )r r! r r r �test_classmethodZ r z&TestCallablePredicate.test_classmethodc C r )Nc @ r )zKTestCallablePredicate.test_non_callable_classmethod.<locals>.BadClassMethodN)r r r r� r r r r r �BadClassMethoda r@ r# r )r r# r r r � test_non_callable_classmethod` r z3TestCallablePredicate.test_non_callable_classmethodN) r r r r r r r r" r$ r r r r r C s r Z__main__) r r� r� ZunittestZ
unittest.mockr r r r r r r r r r
r Z functoolsr
r& r ZTestCaser! rI r� r r r Zmainr r r r Z<module> s* 0 E # G#