U
��,a� � @ s� d Z dZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZ ddlmZmZ dd� ee�D �Zd ZeZd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Z d�dd�Z dd� Z dd � Z d d � Z d d � Z!d�d!d"�Z"d#d$� Z#d%d&� Z$d'd(� Z%G d)d*� d*e&�Z'G d+d,� d,e&�Z(e(� Z)e)j*Z*e)j+Z,e)j-Z.d-d.d/d0d1d2d3d4hZ/d5d6� Z0G d7d8� d8e1�Z2d9d:� Z3G d;d<� d<e&�Z4G d=d>� d>e&�Z5G d?d@� d@e5�Z6dAdB� Z7G dCdD� dDe5�Z8G dEdF� dFe8e6�Z9dGdH� Z:dIdJ� Z;G dKdL� dLe&�Z<dMdN� Z=e*dddddfdOdP�Z>d�dQdR�Z?e*dddddfdSdT�Z@G dUdV� dVe&�ZAdWdX� ZBdYdZ� ZCe>e@_&eAe@_De?e@_EeCe@_Fd[e@_Gd\ZHd]ZId^�Jd_d`� eI�K� D ��ZLd^�Jdad`� eI�K� D ��ZMdbdcdddedfdgdhdidjdkdldmdndodpdqdrdshZNdtdu� ZOdvd� d^�JeHeIeLeMg��K� D �ZPdwdxdyhZQdzhZReQeRB ZSePeNB ZTeTeSB ZUd{d|d}d~dd�d�d�hZVd�d�� d�d�� d�d�� d�d�� d��ZWeXeXeXeXd�dddd�d�d d�dd��
ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]eZe[e\e]d��Z^d�d�� Z_G d�d�� d�e5�Z`G d�d�� d�e`e6�ZaG d�d�� d�e`�ZbG d�d�� d�e`e9�ZcG d�d�� d�e5�ZdG d�d�� d�e5�ZeG d�d�� d�eeebe9�ZfG d�d�� d�e&�Zgeg� Zhd�d�� ZiG d�d�� d�ej�Zkekdd��Zld�d�d��Zmd�d�� ZnG d�d�� d�e&�Zoepem�epehjq�fZrdasd�d�� Ztd�d�d��ZuG d�d�� d�e9�Zvd�d�� ZwG d�d�� d��ZxdS )�)�Mock� MagicMock�patch�sentinel�DEFAULT�ANY�call�create_autospec� AsyncMock�
FILTER_DIR�NonCallableMock�NonCallableMagicMock� mock_open� PropertyMock�sealz1.0� N)�CodeType�
ModuleType�
MethodType)� safe_repr)�wraps�partialc C s h | ]}|� d �s|�qS ��_��
startswith)�.0�name� r �2/opt/alt/python38/lib64/python3.8/unittest/mock.py� <setcomp>( s
r Tc C s> t | �rt| t�sdS t| d�r*t| d�} t�| �p<t�| �S )NF�__func__) �_is_instance_mock�
isinstancer �hasattr�getattr�asyncio�iscoroutinefunction�inspectZ isawaitable��objr r r �
_is_async_obj0 s
r* c C s t | dd �rt�| �S dS d S )N�__code__F)r$ r% r& ��funcr r r �_is_async_func8 s
r. c C s t t| �t�S �N)�
issubclass�typer r( r r r r! ? s r! c C s t | t�p t | t�o t| t�S r/ )r" �
BaseExceptionr1 r0 r( r r r �
_is_exceptionE s
�r3 c C s" t | t�rt| d�r| jS | S d S �N�mock)r" �
FunctionTypesr# r5 r( r r r �
_extract_mockL s r7 c C s� t | t�r|s| j} d}n,t | t�sFz
| j} W n tk
rD Y d S X |rVt| d �}n| }z| t�|�fW S t k
r� Y d S X d S �NT)
r" r1 �__init__r6 �__call__�AttributeErrorr r' � signature�
ValueError)r- Z as_instanceZeat_selfZsig_funcr r r �_get_signature_objectU s
r> Fc sN t | ||�� � d krd S � \} � � fdd�}t| |� |t|�_� t|�_d S )Nc s � j ||� d S r/ ��bind��self�args�kwargs��sigr r �checksigw s z"_check_signature.<locals>.checksig)r> �_copy_func_detailsr1 �_mock_check_sig�
__signature__)r- r5 � skipfirst�instancerG r rE r �_check_signaturer s
rM c C s: dD ]0}zt ||t| |�� W q tk
r2 Y qX qd S )N)�__name__�__doc__�__text_signature__�
__module__� __defaults__�__kwdefaults__)�setattrr$ r; )r- �funcopy� attributer r r rH ~ s
rH c C s@ t | t�rdS t | tttf�r(t| j�S t| dd �d k r<dS dS )NTr: F)r" r1 � staticmethod� classmethodr � _callabler r$ r( r r r rY � s
rY c C s t | �ttfkS r/ )r1 �list�tupler( r r r �_is_list� s r\ c C sF t | t�st| dd �d k S | f| j D ]}|j�d�d k r& dS q&dS )Nr: TF)r" r1 r$ �__mro__�__dict__�get)r) �baser r r �_instance_callable� s
ra c s� t |t�}t|||�}|d kr"| S |\}� � fdd�}t||� |j}|�� sRd}|| d�}d| } t| |� || }
t|
| � � |
S )Nc s � j | |� d S r/ r? �rC rD rE r r rG � s z _set_signature.<locals>.checksigrU )Z
_checksig_r5 zYdef %s(*args, **kwargs):
_checksig_(*args, **kwargs)
return mock(*args, **kwargs))r" r1 r> rH rN � isidentifier�exec� _setup_func) r5 �originalrL rK �resultr- rG r �context�srcrU r rE r �_set_signature� s$
�
rj c s� �� _ �fdd�}�fdd�}�fdd�}�fdd�}�fd d
�}�fd d �}�fd
d�} � �fdd�}
d� _d� _d � _t� � _t� � _t� � _�j� _�j � _ �j
� _
|� _ |� _ |� _
| � _|
� _|� _|� _|� _|� _� �_d S )Nc s � j | |�S r/ )�assert_called_withrb �r5 r r rk � s z'_setup_func.<locals>.assert_called_withc s � j | |�S r/ )�
assert_calledrb rl r r rm � s z"_setup_func.<locals>.assert_calledc s � j | |�S r/ )�assert_not_calledrb rl r r rn � s z&_setup_func.<locals>.assert_not_calledc s � j | |�S r/ )�assert_called_oncerb rl r r ro � s z'_setup_func.<locals>.assert_called_oncec s � j | |�S r/ )�assert_called_once_withrb rl r r rp � s z,_setup_func.<locals>.assert_called_once_withc s � j | |�S r/ )�assert_has_callsrb rl r r rq � s z%_setup_func.<locals>.assert_has_callsc s � j | |�S r/ )�assert_any_callrb rl r r rr � s z$_setup_func.<locals>.assert_any_callc s: t � � _t � � _��� � j} t| �r6| �k r6| �� d S r/ )� _CallList� method_calls�
mock_calls�
reset_mock� return_valuer! )�ret�rU r5 r r rv � s z _setup_func.<locals>.reset_mockFr )r5 �called�
call_count� call_argsrs �call_args_listrt ru rw � side_effect�_mock_childrenrk rp rq rr rv rm rn ro rJ �_mock_delegate) rU r5 rF rk rm rn ro rp rq rr rv r ry r re � s8 re c sJ t jj� _d� _d � _t� � _� fdd�}dD ]}t� |t||�� q.d S )Nr c s t � j| �||�S r/ )r$ r5 )�attrrC rD rl r r �wrapper� s z"_setup_async_mock.<locals>.wrapper)�assert_awaited�assert_awaited_once�assert_awaited_with�assert_awaited_once_with�assert_any_await�assert_has_awaits�assert_not_awaited) r% �
coroutines�
_is_coroutine� await_count�
await_argsrs �await_args_listrT r )r5 r� rV r rl r �_setup_async_mock� s
r� c C s d| dd� | kS )N�__%s__� ���r �r r r r � _is_magic s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_SentinelObjectc C s
|| _ d S r/ r� �rB r r r r r9 s z_SentinelObject.__init__c C s
d| j S �Nz sentinel.%sr� �rB r r r �__repr__ s z_SentinelObject.__repr__c C s
d| j S r� r� r� r r r �
__reduce__ s z_SentinelObject.__reduce__N)rN rQ � __qualname__r9 r� r� r r r r r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _Sentinelc C s
i | _ d S r/ )�
_sentinelsr� r r r r9 # s z_Sentinel.__init__c C s |dkr t �| j�|t|��S )N� __bases__)r; r� �
setdefaultr� r� r r r � __getattr__& s z_Sentinel.__getattr__c C s dS )Nr r r� r r r r� , s z_Sentinel.__reduce__N)rN rQ r� r9 r� r� r r r r r� ! s r� rw �_mock_return_valuer~ �_mock_side_effect� _mock_parent�_mock_new_parent�
_mock_name�_mock_new_namec C s8 t �| � d| }| |fdd�}| |fdd�}t||�S )NZ_mock_c S s" | j }|d krt| |�S t||�S r/ )r� r$ )rB r � _the_namerF r r r �_getA s
z"_delegating_property.<locals>._getc S s* | j }|d kr|| j|< n t|||� d S r/ )r� r^ rT )rB �valuer r� rF r r r �_setF s z"_delegating_property.<locals>._set)�_allowed_names�add�property)r r� r� r� r r r �_delegating_property> s
r� c @ s e Zd Zdd� Zdd� ZdS )rs c C sl t |t�st�| |�S t|�}t| �}||kr2dS td|| d �D ]"}| ||| � }||krD dS qDdS )NFr � T)r" rZ � __contains__�len�range)rB r� Z len_valueZlen_self�iZsub_listr r r r� S s
z_CallList.__contains__c C s t �t| ��S r/ )�pprintZpformatrZ r� r r r r� a s z_CallList.__repr__N)rN rQ r� r� r� r r r r rs Q s rs c C s| t |�}t|�sdS |js4|js4|jd k s4|jd k r8dS | }|d k rX||krPdS |j}q<|rh| |_||_|rx| |_||_dS )NFT)r7 r! r� r� r� r� )�parentr� r �new_name�_parentr r r �_check_and_set_parente s* ��r� c @ s e Zd Zdd� Zdd� ZdS )� _MockIterc C s t |�| _d S r/ )�iterr) )rB r) r r r r9 � s z_MockIter.__init__c C s
t | j�S r/ )�nextr) r� r r r �__next__� s z_MockIter.__next__N)rN rQ r� r9 r� r r r r r� � s r� c @ s e Zd ZeZdZdd� ZdS )�BaseNc O s d S r/ r rA r r r r9 � s z
Base.__init__)rN rQ r� r r� r� r9 r r r r r� � s r� c @ s` e Zd Zdd� ZdKdd�Zdd � ZdLd
d �ZdMd d
�Zdd� Zdd� Z dZ
e ee e
�Z e dd� �Z
ed�Zed�Zed�Zed�Zed�Zdd� Zd d � Ze ee�ZdNddd �d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z dOd2d3�Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%d>d?� Z&d@dA� Z'dPdBdC�Z(dDdE� Z)dFdG� Z*dQdIdJ�Z+dS )Rr c O s� | f}t | t�s^t�tj�}|j| f|�|�j}dd� |�� D �}|r^t ||d �r^t
| f}t | j |d| j
i�}tt| ��|�}|S )Nc S s g | ]}|� d �r|�qS ��specr �r �argr r r �
<listcomp>� s
�z+NonCallableMock.__new__.<locals>.<listcomp>r rO )r0 r r' r< r r9 Z bind_partialZ arguments�keysr* �AsyncMockMixinr1 rN rO � _safe_super�__new__) �clsrC �kw�basesrF Z
bound_argsZspec_arg�newrL r r r r� � s
�zNonCallableMock.__new__N� Fc K s� |d kr |}| j }
||
d< ||
d< ||
d< ||
d< d|
d< |d k rJ|}d}|
d krZ|d k }
| �||| |
� i |
d< ||
d < d |
d
< d|
d < d |
d < d
|
d< t� |
d< t� |
d< t� |
d< | |
d< | r�| jf | � tt| ��||||||� d S )Nr� r� r� r� F� _mock_sealedTr � _mock_wrapsr� Z _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsrt � _mock_unsafe)r^ �_mock_add_specrs �configure_mockr� r r9 )rB r� r r �spec_setr� � _spec_state� _new_name� _new_parent�_spec_as_instance� _eat_selfZunsaferD r^ r r r r9 � sD
�zNonCallableMock.__init__c C s0 t |�}d |_d |_d|_d |_t| ||� d S )Nr� )r7 r� r� r� r� rT )rB r5 rV Z
inner_mockr r r � attach_mock� s zNonCallableMock.attach_mockc C s | � ||� d S r/ )r� �rB r� r� r r r �
mock_add_spec� s z NonCallableMock.mock_add_specc C s� d }d }g }t |�D ] }t�t||d ��r|�|� q|d k r~t|�s~t|t�rV|}nt|�}t|||�} | ot| d }t |�}| j }
||
d< ||
d< ||
d< ||
d< ||
d< d S )Nr� � _spec_class� _spec_set�_spec_signature�
_mock_methods� _spec_asyncs)
�dirr% r& r$ �appendr\ r" r1 r> r^ ) rB r� r� r� r� r� r� r� r� �resr^ r r r r� � s,
� z NonCallableMock._mock_add_specc C s8 | j }| jd k r| jj}|tkr4| j| dd�}|| _|S )N�()�r� r� )r� r� rw r �_get_child_mock)rB rx r r r Z__get_return_value s
�z"NonCallableMock.__get_return_valuec C s, | j d k r|| j _n|| _t| |d d� d S )Nr� )r� rw r� r� )rB r� r r r Z__set_return_value s
z"NonCallableMock.__set_return_valuez1The value to be returned when the mock is called.c C s | j d krt| �S | j S r/ )r� r1 r� r r r � __class__! s
zNonCallableMock.__class__rz r{ r| r} ru c C sN | j }|d kr| jS |j}|d k rJt|�sJt|t�sJt|�sJt|�}||_|S r/ )r� r� r~ �callabler" r� r3 )rB � delegatedZsfr r r Z__get_side_effect. s ��z!NonCallableMock.__get_side_effectc C s( t |�}| j}|d kr || _n||_d S r/ )� _try_iterr� r� r~ )rB r� r� r r r Z__set_side_effect9 s
z!NonCallableMock.__set_side_effect)rw r~ c C s� |d kr g }t | �|kr d S |�t | �� d| _d | _d| _t� | _t� | _t� | _|r^t | _
|rhd | _ | j �
� D ]"}t|t�sr|tkr�qr|�|� qr| j
}t|�r�|| k r�|�|� d S )NFr )�idr� rz r| r{ rs ru r} rt r r� r� r �valuesr" �
_SpecState�_deletedrv r! )rB Zvisitedrw r~ �childrx r r r rv D s, zNonCallableMock.reset_mockc K sX t |�� dd� d�D ]>\}}|�d�}|�� }| }|D ]}t||�}q6t|||� qd S )Nc S s | d � d�S )Nr �.)�count)�entryr r r �<lambda>o � z0NonCallableMock.configure_mock.<locals>.<lambda>)�keyr� )�sorted�items�split�popr$ rT )rB rD r� �valrC �finalr) r� r r r r� b s �
z NonCallableMock.configure_mockc C s� |dkrt |��n:| jd k r<|| jks.|tkrLt d| ��nt|�rLt |��| jsd|�d�rdt d��| j�|�}|tkr�t |��np|d kr�d }| j d k r�t
| j |�}| j | |||| d�}|| j|< n.t |t
�r�t|j|j|j|j|j�}|| j|< |S )N> r� r� z Mock object has no attribute %r)�assertZassretz1Attributes cannot start with 'assert' or 'assret')r� r r r� r� )r; r� � _all_magicsr� r� r r r_ r� r� r$ r� r" r� r r� r� rL r� r )rB r rg r r r r r� x sF
�
�
zNonCallableMock.__getattr__c C s� | j g}| j}| }d}|dgkr$d}|d k rZ|}|�|j | � d}|j dkrRd}|j}q$tt|��}|jpnd}t|�dkr�|d dkr�|d7 }||d< d�|�S )Nr� r� r� r5 r� )r� z().r )r� r� r� rZ �reversedr� r� �join)rB Z
_name_listr� Zlast�dotZ_firstr r r �_extract_mock_name� s(
z"NonCallableMock._extract_mock_namec C s^ | � � }d}|dkr d| }d}| jd k rDd}| jr8d}|| jj }dt| �j||t| �f S )Nr� )r5 zmock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r� r� r� rN r1 r� )rB r Z name_stringZ spec_stringr r r r� � s
�zNonCallableMock.__repr__c C sv t st�| �S | jpg }tt| ��}t| j�}dd� | j� � D �}dd� |D �}dd� |D �}t
t || | | ��S )Nc S s g | ]\}}|t k r|�qS r )r� )r Zm_nameZm_valuer r r r� � s �z+NonCallableMock.__dir__.<locals>.<listcomp>c S s g | ]}|� d �s|�qS r r �r �er r r r� � s
c S s" g | ]}|� d �rt|�r|�qS r )r r� r r r r r� � s
�) r
�object�__dir__r� r� r1 rZ r^ r r� r� �set)rB ZextrasZ from_typeZ from_dictZfrom_child_mocksr r r r � s
�zNonCallableMock.__dir__c sT |t krt��||�S �jrH�jd k rH|�jkrH|�jkrHtd| ��n�|tkrbd| }t|��n�|tkr�jd k r�|�jkr�td| ��t |�s�t
t ��|t ||�� |� � �fdd�}n(t
�|d |� t
t ��||� |�j|< n,|dkr�|�_d S t
�|||��r|�j|< �j�rFt�|��sF��� � d|� �}td|� ���t��||�S )Nz!Mock object has no attribute '%s'z.Attempting to set unsupported magic method %r.c s � �f| �|�S r/ r �rC r� �rf rB r r r� � r� z-NonCallableMock.__setattr__.<locals>.<lambda>r� r� z Cannot set )r� r � __setattr__r� r� r^ r; �_unsupported_magicsr� r! rT r1 � _get_methodr� r r� r� r# r� )rB r r� �msg� mock_namer r r r � s< ��
zNonCallableMock.__setattr__c C s� |t kr2|t| �jkr2tt| �|� || jkr2d S | j�|t�}|| jkr\tt| �� |� n|t
krlt |��|tk r|| j|= t
| j|< d S r/ ) r� r1 r^ �delattrr r_ �_missingr� r � __delattr__r� r; )rB r r) r r r r s
zNonCallableMock.__delattr__c C s | j pd}t|||�S r4 )r� �_format_call_signature�rB rC rD r r r r �_format_mock_call_signature s
z+NonCallableMock._format_mock_call_signaturer c C s. d}| � ||�}| j}| j |� }||||f S )Nz.expected %s not found.
Expected: %s
Actual: %s)r r| )rB rC rD �action�message�expected_stringr| Z
actual_stringr r r � _format_mock_failure_message s
z,NonCallableMock._format_mock_failure_messagec C sj |s
| j S d }|�dd��d�}| j}|D ]:}|�|�}|d ksJt|t�rP qfq*t|�}|j}|j }q*|S )Nr� r� r� )r� �replacer� r r_ r" r� r7 )rB r rF �namesZchildrenr� r r r � _get_call_signature_from_name' s
z-NonCallableMock._get_call_signature_from_namec
C s� t |t�r&t|�dkr&| �|d �}n| j}|d k r�t|�dkrNd}|\}}n
|\}}}z||j||�fW S tk
r� } z|�d � W Y �S d }~X Y q�X n|S d S )Nr� r r� )r" r[ r� r r� r@ � TypeError�with_traceback)rB �_callrF r rC rD r r r r �
_call_matcherH s
"z NonCallableMock._call_matcherc C s0 | j dkr,d| jpd| j | �� f }t|��d S )Nr z9Expected '%s' to not have been called. Called %s times.%sr5 �r{ r� � _calls_repr�AssertionError�rB r
r r r rn b s
��z!NonCallableMock.assert_not_calledc C s$ | j dkr d| jpd }t|��d S )Nr z"Expected '%s' to have been called.r5 )r{ r� r r r r r rm l s
�z NonCallableMock.assert_calledc C s0 | j dks,d| jpd| j | �� f }t|��d S )Nr� z:Expected '%s' to have been called once. Called %s times.%sr5 r r r r r ro t s
��z"NonCallableMock.assert_called_oncec s� �j d kr.��� ��}d}d||f }t|��� ��fdd�}��� �f�}���j �}||kr~t|t�rn|nd }t|� �|�d S )Nz not called.z0expected call not found.
Expected: %s
Actual: %sc s �� � ��} | S r/ �r �r
�rC rD rB r r �_error_message� s z:NonCallableMock.assert_called_with.<locals>._error_message)r| r r r r" � Exception)rB rC rD �expected�actualZ
error_messager$ �causer r# r rk ~ s
� z"NonCallableMock.assert_called_withc O s8 | j dks,d| jpd| j | �� f }t|��| j||�S )Nr� z3Expected '%s' to be called once. Called %s times.%sr5 )r{ r� r r rk �rB rC rD r
r r r rp � s
��z'NonCallableMock.assert_called_once_withc s � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|s�||kr�|d krXd}nd�dd� |D ��}t|� d t|�� � jd
d ��d �� ��|�d S t|�}g }|D ]2}z|�|� W q� t k
r� |�
|� Y q�X q�|�rtd
� j p�dt |�|f �|�d S )Nc s g | ]}� � |��qS r �r �r �cr� r r r� � s z4NonCallableMock.assert_has_calls.<locals>.<listcomp>c s s | ]}t |t�r|V qd S r/ �r" r% r r r r � <genexpr>� s
z3NonCallableMock.assert_has_calls.<locals>.<genexpr>c 3 s | ]}� � |�V qd S r/ r* r+ r� r r r. � s zCalls not found.z+Error processing expected calls.
Errors: {}c S s g | ]}t |t�r|nd �qS r/ r- r r r r r� � s ��
Expected: ZActual)�prefixr� z@%r does not contain all of %r in its call list, found %r insteadr5 )
r� rs ru �formatr r �rstriprZ �remover= r� r� r[ ) rB �calls� any_orderr& r( Z all_calls�problem� not_found�kallr r� r rq � sH
��"�� ����z NonCallableMock.assert_has_callsc sZ � � ||f�}� fdd�� jD �}||krVt|t�r8|nd }� �||�}td| �|�d S )Nc s g | ]}� � |��qS r r* r+ r� r r r� � s z3NonCallableMock.assert_any_call.<locals>.<listcomp>z%s call not found)r r} r" r% r r �rB rC rD r& r' r( r r r� r rr � s ��z NonCallableMock.assert_any_callc K s� |� d�}|| jd kr"tf |�S t| �}t|t�rB|tkrBt}nbt|t�rp|tksd| j rj|| j krjt}q�t}n4t|t
�s�t|t �r�t}q�t|t �r�t
}n
|jd }| jr�d|kr�d|d nd}| �� | }t|��|f |�S )Nr� r� r� r r� r� )r_ r^ r r1 r0 r �_async_method_magicsr� �_all_sync_magicsr� �
CallableMixinr r r r] r� r� r; )rB r� r� �_type�klassrV r r r r r� � s2
��
z NonCallableMock._get_child_mock�Callsc C s" | j s
dS d|� dt| j �� d�S )Nr� �
z: r� )ru r )rB r0 r r r r s zNonCallableMock._calls_repr) NNNNNNr� NFNF)F)FF)N)r )F)r? ),rN rQ r� r� r9 r� r� r� Z"_NonCallableMock__get_return_valueZ"_NonCallableMock__set_return_valueZ"_NonCallableMock__return_value_docr� rw r� r� rz r{ r| r} ru Z!_NonCallableMock__get_side_effectZ!_NonCallableMock__set_side_effectr~ rv r� r� r� r� r r r r r r r rn rm ro rk rp rq rr r� r r r r r r � sn �
-
�
�
''
!
-'r c C sL | d kr | S t | �r| S t| �r$| S z
t| �W S tk
rF | Y S X d S r/ )r3 rY r� r r( r r r r� s
r� c
@ sR e Zd Zddedddddddf
dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )r< Nr� c K s6 || j d< tt| �j||||||| |
f| � || _d S )Nr� )r^ r� r< r9 r~ ) rB r� r~ rw r r r� r� r� r� r� rD r r r r9 # s
��zCallableMixin.__init__c O s d S r/ r rA r r r rI / s z CallableMixin._mock_check_sigc O s$ | j ||� | j||� | j||�S r/ )rI �_increment_mock_call�
_mock_callrA r r r r: 4 s zCallableMixin.__call__c O s | j ||�S r/ )�_execute_mock_callrA r r r rB < s zCallableMixin._mock_callc O s d| _ | jd7 _t||fdd�}|| _| j�|� | jd k }| j}| j}|dk}| j �td||f�� | j
}|d k r�|r�|j �t|||f�� |jd k }|r�|jd | }t|||f�} |j �| � |jr�|r�d}
nd}
|jdk}|j|
| }|j
}qpd S )NTr� ��twor� r� r� ) rz r{ �_Callr| r} r� r� r� r� ru r� rt ) rB rC rD r Zdo_method_callsZmethod_call_nameZmock_call_nameZ is_a_callr� Zthis_mock_callr� r r r rA ? s4
z"CallableMixin._increment_mock_callc O s| | j }|d k rPt|�r |�n(t|�s:t|�}t|�rD|�n
|||�}|tk rP|S | jtk r`| jS | jd k rv| j||�S | jS r/ )r~ r3 rY r� r r� rw r� )rB rC rD �effectrg r r r rC m s
z CallableMixin._execute_mock_call)
rN rQ r� r r9 rI r: rB rA rC r r r r r< ! s �
.r< c @ s e Zd ZdS )r N�rN rQ r� r r r r r � s r c C s8 z t | |�W S tk
r2 t|� t | |� Y S X d S r/ )r$ r; �
__import__)�thing�comp� import_pathr r r � _dot_lookup� s
rM c C sB | � d�}|�d�}t|�}|D ] }|d| 7 }t|||�}q |S )Nr� r z.%s)r� r� rI rM )�targetZ
componentsrL rJ rK r r r � _importer� s
rO c @ sz e Zd ZdZg Zdd� Zdd� Zdd� Zdd � Ze j
d
d � �Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� ZdS )�_patchNc
C sn |d k r(|t k rtd��|d k r(td��|| _|| _|| _|| _|| _|| _d| _|| _ || _
| | _ g | _ d S )Nz,Cannot use 'new' and 'new_callable' togetherz1Cannot use 'autospec' and 'new_callable' togetherF)
r r= �getterrV r� � new_callabler� �createZ has_localr� �autospecrD �additional_patchers)
rB rQ rV r� r� rS r� rT rR rD r r r r9 � s( ��z_patch.__init__c
C sH t | j| j| j| j| j| j| j| j| j � }| j
|_
dd� | j D �|_ |S )Nc S s g | ] }|� � �qS r )�copy)r �pr r r r� � s z _patch.copy.<locals>.<listcomp>) rP rQ rV r� r� rS r� rT rR rD �attribute_namerU )rB �patcherr r r rV � s ��z _patch.copyc C s2 t |t�r| �|�S t�|�r(| �|�S | �|�S r/ )r" r1 �decorate_classr' r& �decorate_async_callable�decorate_callable�rB r- r r r r: � s
z_patch.__call__c C sN t |�D ]@}|�tj�sqt||�}t|d�s0q| �� }t||||�� q|S �Nr: )r� r r � TEST_PREFIXr$ r# rV rT )rB r> r� �
attr_valuerY r r r rZ s
z_patch.decorate_classc c sr g }t �� �\}|jD ]8}|�|�}|jd k r8|�|� q|jtkr|�|� q|t |�7 }||fV W 5 Q R X d S r/ )
�
contextlib� ExitStack� patchings�
enter_contextrX �updater� r r� r[ )rB �patchedrC �keywargs�
extra_args�
exit_stack�patchingr� r r r �decoration_helper s
z_patch.decoration_helperc s> t � d�r� j��� � S t� �� ��fdd����g�_�S )Nrc c
s4 �� �| |�� \}}� ||�W 5 Q R � S Q R X d S r/ �rk �rC rg ZnewargsZ newkeywargs�r- rf rB r r rf ( s �z)_patch.decorate_callable.<locals>.patched�r# rc r� r r] r rn r r\ " s
z_patch.decorate_callablec s> t � d�r� j��� � S t� �� ��fdd����g�_�S )Nrc c
� s: �� �| |��"\}}� ||�I d H W 5 Q R � S Q R X d S r/ rl rm rn r r rf 9 s �z/_patch.decorate_async_callable.<locals>.patchedro r] r rn r r[ 3 s
z _patch.decorate_async_callablec C s� | � � }| j}t}d}z|j| }W n$ ttfk
rH t||t�}Y nX d}|tkrft|t �rfd| _
| j
s�|tkr�td||f ��||fS )NFTz!%s does not have the attribute %r) rQ rV r r^ r; �KeyErrorr$ � _builtinsr" r rS )rB rN r rf �localr r r � get_originalD s
�z_patch.get_originalc C s� | j | j| j }}}| j| j }}| j}| �� | _|dkr@d }|dkrLd }|dkrXd }|d k rp|d k rptd��|d k s�|d k r�|dkr�td��| � � \}}|t
k�r||d k�r|d} |dkr�|}|dkr�|}d }n&|d k r�|dkr�|}d }n |dkr�|}|d k �s |d k �r.|t
k�r td��t |t ��r.d} |d k�rHt
|��rHt}
nt}
i } |d k �r`|}
n^|d k �st|d k �r�|} |d k �r�|} t| ��r�d| k}
n
t| � }
t
| ��r�t}
n
|
�r�t}
|d k �r�|| d< |d k �r�|| d < t |
t ��r t|
t��r | j�r | j| d
< | �|� |
f | �}| �r�t|��r�|} |d k �rB|} t| ��sZt| ��sZt}
| �d
� |
f |d d �| ��|_nl|d k �r�|t
k �r�td
��|t
k�r�td��t|�}|dk�r�|}t |f|| jd�|��}n|�r�td��|}|| _ || _ t � � | _!zrt"| j| j|� | j#d k �rpi }| j t
k�r:||| j#< | j$D ](}| j!�%|�}|j t
k�r@|�|� �q@|W S |W S | j&t'�(� � �s�� Y nX d S )NFz Can't specify spec and autospec)TNz6Can't provide explicit spec_set *and* spec or autospecTz!Can't use 'spec' with create=Truer: r� r� r r� r� zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=True)r� �_namez.Can't pass kwargs to a mock we aren't creating))r� r� r� rT rD rR rQ rN r rs r r" r1 r* r r r\ r� r r0 r rV re r! ra r� rw �boolr �
temp_original�is_localra rb � _exit_stackrT rX rU rd �__exit__�sys�exc_info)rB r� r� r� rT rD rR rf rr Zinherit�Klass�_kwargsZ this_specZ not_callableZnew_attrrh rj r� r r r � __enter__\ s�
�
��
�
�
�
��
z_patch.__enter__c G s� | j r$| jtk r$t| j| j| j� n>t| j| j� | jsbt| j| j�rP| jdkrbt| j| j| j� | `| ` | `| j }| ` |j
|� S )N)rO rQ rR �__annotations__rS ) rw rv r rT rN rV r rS r# rx ry )rB r{ ri r r r ry � s �z_patch.__exit__c C s | � � }| j�| � |S r/ )r~ �_active_patchesr� )rB rg r r r �start� s z _patch.startc C s6 z| j �| � W n tk
r& Y d S X | �d d d �S r/ )r� r3 r= ry r� r r r �stop� s
z _patch.stop)rN rQ r� rX r� r9 rV r: rZ ra �contextmanagerrk r\ r[ rs r~ ry r� r� r r r r rP � s
rP c sP z� � dd�\� }W n& ttfk
r: td� f ��Y nX � fdd�}||fS )Nr� r� z.Need a valid target to patch. You supplied: %rc s t � �S r/ �rO r �rN r r r� r� z _get_target.<locals>.<lambda>)�rsplitr r= )rN rV rQ r r� r � _get_target
s � r� c
s> t � �tkrt� �d���� fdd�} t| ||||||||� S )Nz3 must be the actual object to be patched, not a strc s � S r/ r r r� r r r� * r� z _patch_object.<locals>.<lambda>)r1 �strr rP )
rN rV r� r� rS r� rT rR rD rQ r r� r �
_patch_object s � �r� c
s� t � �tkr� fdd�}n � fdd�}|s2td��t|�� �}|d \} }
t|| |
|||||i � } | | _|dd � D ]2\} }
t|| |
|||||i � } | | _| j�| � qt| S )Nc s t � �S r/ r� r r� r r r� H r� z!_patch_multiple.<locals>.<lambda>c s � S r/ r r r� r r r� J r� z=Must supply at least one keyword argument with patch.multipler r� ) r1 r� r= rZ r� rP rX rU r� )
rN r� rS r� rT rR rD rQ r� rV r� rY Z this_patcherr r� r �_patch_multiple1 sH � � �r� c
K s$ t | �\}} t|| |||||||� S r/ )r� rP )
rN r� r� rS r� rT rR rD rQ rV r r r r b s F �r c @ sN e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z eZ
e Z dS )� _patch_dictr Fc K s, || _ t|�| _| j�|� || _d | _d S r/ )�in_dict�dictr� re �clear� _original)rB r� r� r� rD r r r r9 � s
z_patch_dict.__init__c s. t � t�r��� �S t� �� �fdd��}|S )Nc s& �� � z� | |�W �S ��� X d S r/ )r� �
_unpatch_dictr ��frB r r �_inner� s z$_patch_dict.__call__.<locals>._inner)r" r1 rZ r )rB r� r� r r� r r: � s
z_patch_dict.__call__c C sX t |�D ]J}t||�}|�tj�rt|d�rt| j| j| j �}||�}t
|||� q|S r^ ) r� r$ r r r_ r# r� r� r� r� rT )rB r> r� r` Z decoratorZ decoratedr r r rZ � s
�z_patch_dict.decorate_classc C s | � � | jS r/ )r� r� r� r r r r~ � s z_patch_dict.__enter__c C s� | j }t| jt�r t| j�| _| j}| j}z |�� }W n. tk
rd i }|D ]}|| ||< qNY nX || _|rxt |� z|�
|� W n* tk
r� |D ]}|| ||< q�Y nX d S r/ ) r� r" r� r� rO r� rV r; r� � _clear_dictre )rB r� r� r� rf r� r r r r� � s& z_patch_dict._patch_dictc C sR | j }| j}t|� z|�|� W n* tk
rL |D ]}|| ||< q6Y nX d S r/ )r� r� r� re r; )rB r� rf r� r r r r� s z_patch_dict._unpatch_dictc G s | � � dS �NF)r� )rB rC r r r ry s z_patch_dict.__exit__N)r F) rN rQ r� r9 r: rZ r~ r� r� ry r� r� r r r r r� � s
r� c C s> z | � � W n, tk
r8 t| �}|D ]
}| |= q(Y nX d S r/ )r� r; rZ )r� r� r� r r r r� & s r� c C s t tj�D ] } | �� q
d S r/ )r� rP r� r� )r r r r �_patch_stopall/ s r� Ztestz�lt le gt ge eq ne getitem setitem delitem len contains iter hash str sizeof enter exit divmod rdivmod neg pos abs invert complex int float index round trunc floor ceil bool next fspath aiter zHadd sub mul matmul div floordiv mod lshift rshift and xor or pow truediv� c c s | ]}d | V qdS )zi%sNr �r �nr r r r. N s r. c c s | ]}d | V qdS )zr%sNr r� r r r r. O s �__get__�__set__�
__delete__� __reversed__� __missing__r� �
__reduce_ex__Z__getinitargs__�__getnewargs__� __getstate__� __setstate__�
__getformat__Z
__setformat__r� r �__subclasses__�
__format__�__getnewargs_ex__c s � fdd�}| |_ |S )Nc s � | f|�|�S r/ r �rB rC r� r, r r �method` s z_get_method.<locals>.method)rN )r r- r� r r, r r ^ s r c C s h | ] }d | �qS )r� r )r r� r r r r f s �
__aenter__� __aexit__� __anext__� __aiter__r� r r9 r� � __prepare__�__instancecheck__�__subclasscheck__�__del__c C s
t �| �S r/ )r �__hash__r� r r r r� | r� r� c C s
t �| �S r/ )r �__str__r� r r r r� } r� c C s
t �| �S r/ )r �
__sizeof__r� r r r r� ~ r� c C s" t | �j� d| �� � dt| �� �S )N�/)r1 rN r� r� r� r r r r� r� )r� r� r� �
__fspath__r� y �?g �?)
�__lt__�__gt__�__le__�__ge__�__int__r� �__len__ry � __complex__� __float__�__bool__� __index__r� c s � fdd�}|S )Nc s$ � j j}|tk r|S � | kr dS tS r8 )�__eq__r� r �NotImplemented)�other�ret_valr� r r r� � s z_get_eq.<locals>.__eq__r )rB r� r r� r �_get_eq� s r� c s � fdd�}|S )Nc s � j jtk rtS � | kr dS tS r� )�__ne__r� r r� )r� r� r r r� � s
z_get_ne.<locals>.__ne__r )rB r� r r� r �_get_ne� s r� c s � fdd�}|S )Nc s � j j} | tkrtg �S t| �S r/ )�__iter__r� r r� �r� r� r r r� � s z_get_iter.<locals>.__iter__r )rB r� r r� r � _get_iter� s r� c s � fdd�}|S )Nc s( � j j} | tkr ttg ��S tt| ��S r/ )r� r� r �_AsyncIteratorr� r� r� r r r� � s z"_get_async_iter.<locals>.__aiter__r )rB r� r r� r �_get_async_iter� s r� )r� r� r� r� c C sb t �|t�}|tk r ||_d S t�|�}|d k rB|| �}||_d S t�|�}|d k r^|| �|_d S r/ )�_return_valuesr_ r rw �_calculate_return_value�_side_effect_methodsr~ )r5 r� r ZfixedZreturn_calculatorrw Z
side_effectorr r r �_set_return_value� s
r� c @ s e Zd Zdd� Zdd� ZdS )�
MagicMixinc O s& | � � tt| �j||� | � � d S r/ )�_mock_set_magicsr� r� r9 r� r r r r9 � s zMagicMixin.__init__c C s� t tB }|}t| dd �d k rX|�| j�}t� }|| }|D ] }|t| �jkr:t| |� q:|tt| �j� }t| �}|D ]}t ||t
|| �� qvd S )Nr� ) �_magicsr: r$ � intersectionr� r r1 r^ r rT �
MagicProxy)rB Z orig_magicsZ these_magicsZ
remove_magicsr� r= r r r r� � s zMagicMixin._mock_set_magicsN)rN rQ r� r9 r� r r r r r� � s r� c @ s e Zd Zddd�ZdS )r Fc C s | � ||� | �� d S r/ �r� r� r� r r r r� � s z"NonCallableMagicMock.mock_add_specN)F�rN rQ r� r� r r r r r � s r c @ s e Zd Zdd� ZdS )�AsyncMagicMixinc O s& | � � tt| �j||� | � � d S r/ )r� r� r� r9 r� r r r r9 � s zAsyncMagicMixin.__init__N�rN rQ r� r9 r r r r r� � s r� c @ s e Zd Zddd�ZdS )r Fc C s | � ||� | �� d S r/ r� r� r r r r� s zMagicMock.mock_add_specN)Fr� r r r r r s r c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) r� c C s || _ || _d S r/ �r r� )rB r r� r r r r9 s zMagicProxy.__init__c C s8 | j }| j}|j|||d�}t|||� t|||� |S )N)r r� r� )r r� r� rT r� )rB r� r� �mr r r � create_mock s � zMagicProxy.create_mockNc C s | � � S r/ )r� )rB r) r= r r r r� ( s zMagicProxy.__get__)N)rN rQ r� r9 r� r� r r r r r� s r� c s� e Zd Zed�Zed�Zed�Z� fdd�Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
ddd�Zdd� Z� fdd�Z� ZS )r� r� r� r� c s\ t � j||� tjj| jd< d| jd< d | jd< t� | jd< ttd�}t j
|_ || jd< d S )Nr� r Z_mock_await_countZ_mock_await_argsZ_mock_await_args_list�r� r+ ) �superr9 r% r� r� r^ rs r r r' Z CO_COROUTINE�co_flags)rB rC rD � code_mock�r� r r r9 1 s
zAsyncMockMixin.__init__c � s t ||fdd�}| jd7 _|| _| j�|� | j}|d k r�t|�rL|�nbt|�s�z t|�}W n t k
rx t
�Y nX t|�r�|�n&t � |�r�|||�I d H }n
|||�}|t
k r�|S | jt
k r�| jS | jd k r�t � | j�r�| j||�I d H S | j||�S | jS )NTrD r� )rF r� r� r� r� r~ r3 rY r� �
StopIteration�StopAsyncIterationr% r& r r� rw r� )rB rC rD r rG rg r r r rC A s6
z!AsyncMockMixin._execute_mock_callc C s( | j dkr$d| jpd� d�}t|��d S )Nr � Expected r5 z to have been awaited.�r� r� r r r r r r� i s
z AsyncMockMixin.assert_awaitedc C s0 | j dks,d| jpd� d| j � d�}t|��d S �Nr� r� r5 z$ to have been awaited once. Awaited � times.r� r r r r r� q s
z"AsyncMockMixin.assert_awaited_oncec sz �j d kr&��� ��}td|� d���� ��fdd�}��� �f�}���j �}||krvt|t�rf|nd }t|� �|�d S )NzExpected await: z
Not awaitedc s �j � �dd�} | S )N�await)r r! r" r# r r r$ � s z:AsyncMockMixin.assert_awaited_with.<locals>._error_message)r� r r r r" r% )rB rC rD r& r$ r' r( r r# r r� z s
z"AsyncMockMixin.assert_awaited_withc O s8 | j dks,d| jpd� d| j � d�}t|��| j||�S r� )r� r� r r� r) r r r r� � s
z'AsyncMockMixin.assert_awaited_once_withc sZ � � ||f�}� fdd�� jD �}||krVt|t�r8|nd }� �||�}td| �|�d S )Nc s g | ]}� � |��qS r r* r+ r� r r r� � s z3AsyncMockMixin.assert_any_await.<locals>.<listcomp>z%s await not found)r r� r" r% r r r9 r r� r r� � s ��z AsyncMockMixin.assert_any_awaitFc s� � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|s�||kr�|d krXd}nd�dd� |D ��}t|� d t|�� d
� j� ��|�d S t|�}g }|D ]2}z|�|� W q� tk
r� |�|� Y q�X q�|r�td t |�f �|�d S ) Nc s g | ]}� � |��qS r r* r+ r� r r r� � s z4AsyncMockMixin.assert_has_awaits.<locals>.<listcomp>c s s | ]}t |t�r|V qd S r/ r- r r r r r. � s
z3AsyncMockMixin.assert_has_awaits.<locals>.<genexpr>c 3 s | ]}� � |�V qd S r/ r* r+ r� r r r. � s zAwaits not found.z,Error processing expected awaits.
Errors: {}c S s g | ]}t |t�r|nd �qS r/ r- r r r r r� � s �r/ z
Actual: z %r not all found in await list)
r� rs r� r1 r rZ r3 r= r� r[ ) rB r4 r5 r& r( Z
all_awaitsr6 r7 r8 r r� r r� � s> ���� ��z AsyncMockMixin.assert_has_awaitsc C s0 | j dkr,d| jpd� d| j � d�}t|��d S )Nr r� r5 z# to not have been awaited. Awaited r� r� r r r r r� � s
z!AsyncMockMixin.assert_not_awaitedc s& t � j||� d| _d | _t� | _d S �Nr )r� rv r� r� rs r� rA r� r r rv � s zAsyncMockMixin.reset_mock)F)rN rQ r� r� r� r� r� r9 rC r� r� r� r� r� r� r� rv �
__classcell__r r r� r r� , s (
, r� c @ s e Zd ZdS )r NrH r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_ANYc C s dS r8 r �rB r� r r r r� s z _ANY.__eq__c C s dS r� r r� r r r r� s z _ANY.__ne__c C s dS )Nz<ANY>r r� r r r r� s z
_ANY.__repr__N)rN rQ r� r� r� r� r r r r r� s r� c C s` d| }d}d� dd� |D ��}d� dd� |�� D ��}|r@|}|rX|rP|d7 }||7 }|| S )Nz%s(%%s)r� z, c S s g | ] }t |��qS r )�reprr� r r r r� ! s z*_format_call_signature.<locals>.<listcomp>c S s g | ]\}}d ||f �qS )z%s=%rr )r r� r� r r r r� " s )r� r� )r rC rD r Zformatted_argsZ args_stringZ
kwargs_stringr r r r s
�r c @ s� e Zd Zd dd�Zd!dd �Zd
d � ZejZd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zd d � Zd d � ZdS )"rF r r� NFTc C s� d}i }t |�}|dkr$|\}}}nr|dkrd|\} }
t| t�rX| }t|
t�rR|
}qb|
}q�| |
}}n2|dkr�|\}t|t�r�|}nt|t�r�|}n|}|r�t�| ||f�S t�| |||f�S )Nr � r� r� )r� r" r� r[ r� ) r� r� r r� rE � from_kallrC rD �_len�first�secondr r r r� C s.
z
_Call.__new__c C s || _ || _|| _d S r/ )r� r� �_mock_from_kall)rB r� r r� rE r� r r r r9 c s z_Call.__init__c C sh |t kr dS z t|�}W n tk
r. Y dS X d}t| �dkrJ| \}}n
| \}}}t| dd �r|t|dd �r|| j|jkr|dS d}|dkr�di }}n�|dkr�|\}}}n�|d kr�|\} t| t�r�| }i }n"t| t�r�| }di }}nd}| }nV|dk�r@|\}
} t|
t��r4|
}t| t��r(| i }}n
d| }}n
|
| }}ndS |�rX||k�rXdS ||f||fkS )
NTFr� r� r� r r r� r� )r r� r r$ r� r" r[ r� ) rB r� Z len_otherZ self_nameZ self_argsZ self_kwargsZ
other_nameZ
other_argsZ other_kwargsr� r� r� r r r r� j sR
�
z _Call.__eq__c O s< | j d kr td||fdd�S | j d }t| j ||f|| d�S )Nr� r� r� r� �r� rF r r r r r: � s
z_Call.__call__c C s2 | j d krt|dd�S d| j |f }t|| dd�S )NF)r r� z%s.%s)r r� r� r� )rB r� r r r r r� � s
z_Call.__getattr__c C s |t jkrt�t �| |�S r/ )r[ r^ r; �__getattribute__)rB r� r r r r� � s
z_Call.__getattribute__c O s | � d�||�S )Nr� �r� rA r r r r� � s z _Call.countc O s | � d�||�S )N�indexr� rA r r r r� � s z _Call.indexc C s( t | �dkr| \}}n
| \}}}||fS )Nr� )r� r r r r �_get_call_arguments� s
z_Call._get_call_argumentsc C s | � � d S r� �r� r� r r r rC � s z
_Call.argsc C s | � � d S )Nr� r� r� r r r rD � s z _Call.kwargsc C s| | j s&| jpd}|�d�r"d| }|S t| �dkr@d}| \}}n0| \}}}|sTd}n |�d�shd| }nd| }t|||�S )Nr r� zcall%sr� zcall.%s)r� r� r r� r )rB r rC rD r r r r� � s
z_Call.__repr__c C s4 g }| }|d k r(|j r |�|� |j}qtt|��S r/ )r� r� r� rs r� )rB ZvalsrJ r r r � call_list� s
z_Call.call_list)r r� NFT)r NNFT)rN rQ r� r� r9 r� r r� r: r� r� r� r� r� r� rC rD r� r� r r r r rF 0 s( �
�
7
rF )r� c K sl t | �rt| �} t| t�}t| �}d| i}|r8d| i}n | d krDi }|rT|rTd|d< |�|� t} t�| �rri }n8|r�|r�td��t } n"t
| �s�t } n|r�|r�t | �s�t } |�
d|�}|}
|d kr�d}
| f |||
|d�|��} t| t��rt| | �} |�rt| � nt| | ||� |d k �r,|�s,| |j|< |�rV|�sVd |k�rVt| |dd
| d �| _t| �D �]} t| ��rr�q^zt| | �}
W n tk
�r� Y �q^Y nX d|
i}|�r�d|
i}t|
t��s�t|
|| | |�}|| j| < np| }t| t��r�| j}t| | |�}||d < t � |
��rt }nt}|f || | |d
�|��}|| j| < t|
||d� t|t��r^t | | |� �q^| S )Nr� r� Tr� zJInstance can not be True when create_autospec is mocking an async functionr r� )r� r� r� r rw r� )rL rt r� r� )r� r r� r� )rK ) r\ r1 r" r. re r r' Zisdatadescriptor� RuntimeErrorr rY r ra r� r6 rj r� rM r r rw r� r� r$ r; r� r5 �
_must_skipr% r& rT )r� r� rL r� rt rD �is_typeZ
is_async_funcr} r| r� r5 r� rf r� r� rK Z child_klassr r r r � s�
��
�
��
r c C sx t | t�s$|t| di �kr dS | j} | jD ]H}|j�|t�}|tkrFq*t |tt f�rZ dS t |t
�rl| S dS q*|S )Nr^ F) r" r1 r$ r� r] r^ r_ r rW rX r6 )r� r� r� r> rg r r r r� w
s
r� c @ s e Zd Zddd�ZdS )r� FNc C s( || _ || _|| _|| _|| _|| _d S r/ )r� �idsr� r� rL r )rB r� r� r� r r� rL r r r r9 �
s z_SpecState.__init__)FNNNFr� r r r r r� �
s
�r� c C s" t | t�rt�| �S t�| �S d S r/ )r" �bytes�io�BytesIO�StringIO)� read_datar r r �
_to_stream�
s
r r� c s& t ��}|d g���fdd�}��fdd�}� �fdd����fdd�� ��fd d
�}td kr�d d l}ttt|j���tt|j����a| d kr�t d t
d
�} t td����j _ d �j
_ d �j_ d �j_ d �j_ |�j_�� �d< �d �j_|�j_� �j_|�j_����fdd�}|| _�| _ | S )Nc s$ �j jd k r�j jS � d j | |�S r� )� readlinesrw rb ��_state�handler r �_readlines_side_effect�
s z)mock_open.<locals>._readlines_side_effectc s$ �j jd k r�j jS � d j | |�S r� )�readrw rb r r r �_read_side_effect�
s z$mock_open.<locals>._read_side_effectc ? s$ � � E d H �d j | |�V q d S r� )�readlinerb )�_iter_side_effectr r r �_readline_side_effect�
s z(mock_open.<locals>._readline_side_effectc 3 s0 �j jd k r�j jV q � d D ]
} | V q d S r� )r rw )�liner r r r �
s z$mock_open.<locals>._iter_side_effectc s �j jd k r�j jS t� d �S r� )r rw r� r r r r �_next_side_effect�
s z$mock_open.<locals>._next_side_effectr �open)r r� r� r� c s6 t ���d< �jj�d kr2� � �d< �d �j_tS )Nr r� )r r r~ r rb )r
r r r r r �
reset_data�
s
z mock_open.<locals>.reset_data)r � file_spec�_iorZ r r� �
TextIOWrapper�unionr r r r~ rw �writer r r r~ r� r� )r5 r Z
_read_datar r
r r r r )r r
r r r r r
�
s8 "
r
c @ s& e Zd Zdd� Zddd�Zdd� ZdS ) r c K s
t f |�S r/ )r )rB rD r r r r� s z PropertyMock._get_child_mockNc C s | � S r/ r )rB r) Zobj_typer r r r� s zPropertyMock.__get__c C s | |� d S r/ r )rB r) r� r r r r�
s zPropertyMock.__set__)N)rN rQ r� r� r� r� r r r r r �
s
r c C s^ d| _ t| �D ]J}zt| |�}W n tk
r8 Y qY nX t|t�sFq|j| krt|� qd S r8 )r� r� r$ r; r" r r� r )r5 r� r� r r r r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r� c C s& || _ ttd�}tj|_|| jd< d S )Nr� r+ )�iteratorr r r' ZCO_ITERABLE_COROUTINEr� r^ )rB r r� r r r r9 + s
z_AsyncIterator.__init__c C s | S r/ r r� r r r r� 1 s z_AsyncIterator.__aiter__c � s* z t | j�W S tk
r Y nX t�d S r/ )r� r r� r� r� r r r r� 4 s
z_AsyncIterator.__anext__N)rN rQ r� r9 r� r� r r r r r� ' s r� )F)F)NFNNN)FFNN)Nr� )y�__all__� __version__r% ra r� r' r� rz �builtins�typesr r r Z
unittest.utilr � functoolsr r r� rq r
r� r� r* r. r! r3 r7 r> rM rH rY r\ ra rj re r� r� r r� r� r r �MISSINGr
ZDELETEDr� r� r� rZ rs r� r� r� r r� r< r rM rO rP r� r� r� r r� r� r� r� ZmultipleZstopallr_ Z
magic_methodsZnumericsr� r� Zinplace�rightZ
_non_defaultsr r� r: Z_sync_async_magicsZ
_async_magicsr; r� r r� r� r� r� r� r� r� r� r� r� r r� r r� r� r r� r r r[ rF r r r� r� r1 r� r6 r r r
r r r� r r r r �<module> s~
1 � h4 < �
�
2 �
Mw �� � �
���
� 8+ B
�
�
N