a
XC?h�� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl 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.h d-�Z/d.d/� Z0G d0d1� d1e1�Z2d2d3� Z3G d4d5� d5e&�Z4G d6d7� d7e&�Z5G d8d9� d9e5�Z6e�7e6j8�Z9G d:d;� d;e1�Z:dd?� d?e5�ZdDdE� Z?G dFdG� dGe&�Z@dHdI� ZAe*dddddfdJdK�ZBd�dLdM�ZCe*dddddfdNdO�ZDG dPdQ� dQe&�ZEdRdS� ZFdTdU� ZGeBeD_&eEeD_HeCeD_IeGeD_JdVeD_KdWZLdXZMdY�NdZd[� eM�O� D ��ZPdY�Nd\d[� eM�O� D ��ZQh d]�ZRd^d_� ZSd`d� dY�NeLeMePeQg��O� D �ZTh da�ZUdbhZVeUeVB ZWeTeRB ZXeXeWB ZYh dc�ZZddde� dfde� dgde� dhde� di�Z[e\e\e\e\djddddkdld djddm�
Z]dndo� Z^dpdq� Z_drds� Z`dtdu� Zae^e_e`eadv�Zbdwdx� ZcG dydz� dze5�ZdG d{d|� d|ede6�ZeG d}d~� d~ed�ZfG dd�� d�ede=�ZgG d�d�� d�e5�ZhG d�d�� d�e5�ZiG d�d�� d�eiefe=�ZjG d�d�� d�e&�Zkek� Zld�d�� ZmG d�d�� d�en�Zoeodd��Zpd�d�d��Zqd�d�� ZrG d�d�� d�e&�Zseteq�etelju�fZvdawd�d�� Zxd�d�d��ZyG d�d�� d�e=�Zzd�d�� Z{G d�d�� d��Z|dS )�)�Mock� MagicMock�patch�sentinel�DEFAULT�ANY�call�create_autospec� AsyncMock�
FILTER_DIR�NonCallableMock�NonCallableMagicMock� mock_open� PropertyMock�seal� N)�iscoroutinefunction)�CodeType�
ModuleType�
MethodType)� safe_repr)�wraps�partialc C s h | ]}|� d �s|�qS ��_��
startswith)�.0�name� r �2/opt/alt/python39/lib64/python3.9/unittest/mock.py� ' � r Tc C s< t | �rt| t�sdS t| d�r*t| d�} t| �p:t�| �S )NF�__func__)�_is_instance_mock�
isinstancer �hasattr�getattrr �inspectZ isawaitable��objr r r �
_is_async_obj/ s
r* c C s t | dd �rt| �S dS d S )N�__code__F)r& r ��funcr r r �_is_async_func7 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_exceptionD s
�r3 c C s" t | t�rt| d�r| jS | S d S �N�mock)r$ �
FunctionTypesr% r5 r( r r r �
_extract_mockK s r7 c C s� t | t�r|s| j} d}n*t | t�sDz
| j} W n tyB Y d S 0 |rTt| d �}n| }z| t�|�fW S t y| Y d S 0 d S �NT)
r$ r1 �__init__r6 �__call__�AttributeErrorr r' � signature�
ValueError)r- Z as_instanceZeat_selfZsig_funcr r r �_get_signature_objectT s
r> Fc sN t | ||�� � d u rd S � \} � � fdd�}t| |� |t|�_� t|�_d S )Nc s � j |i |�� d S r/ ��bind��self�args�kwargs��sigr r �checksigv s z"_check_signature..checksig)r> �_copy_func_detailsr1 �_mock_check_sig�
__signature__)r- r5 � skipfirst�instancerG r rE r �_check_signatureq s
rM c C s8 dD ].}zt ||t| |�� W q ty0 Y q0 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 ur.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 | i |��S r/ )�assert_called_withrb �r5 r r rk � s z'_setup_func..assert_called_withc s � j | i |��S r/ )�
assert_calledrb rl r r rm � s z"_setup_func..assert_calledc s � j | i |��S r/ )�assert_not_calledrb rl r r rn � s z&_setup_func..assert_not_calledc s � j | i |��S r/ )�assert_called_oncerb rl r r ro � s z'_setup_func..assert_called_oncec s � j | i |��S r/ )�assert_called_once_withrb rl r r rp � s z,_setup_func..assert_called_once_withc s � j | i |��S r/ )�assert_has_callsrb rl r r rq � s z%_setup_func..assert_has_callsc s � j | i |��S r/ )�assert_any_callrb rl r r rr � s z$_setup_func..assert_any_callc s: t � � _t � � _��� � j} t| �r6| �ur6| �� d S r/ )� _CallList� method_calls�
mock_calls�
reset_mock� return_valuer# )�ret�rU r5 r r rv � s z _setup_func..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| �|i |��S r/ )r& r5 )�attrrC rD rl r r �wrapper� s z"_setup_async_mock..wrapper)�assert_awaited�assert_awaited_once�assert_awaited_with�assert_awaited_once_with�assert_any_await�assert_has_awaits�assert_not_awaited) �asyncio�
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� > �_mock_side_effect� _mock_parent�_mock_return_value�
_mock_name�_mock_new_parent�_mock_new_namerw r~ c C s8 t �| � d| }| |fdd�}| |fdd�}t||�S )NZ_mock_c S s" | j }|d u rt| |�S t||�S r/ )r� r& )rB r � _the_namerF r r r �_get@ s
z"_delegating_property.._getc S s* | j }|d u r|| j|< n t|||� d S r/ )r� r^ rT )rB �valuer r� rF r r r �_setE s z"_delegating_property.._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� R s
z_CallList.__contains__c C s t �t| ��S r/ )�pprintZpformatrZ r� r r r r� ` s z_CallList.__repr__N)rN rQ r� r� r� r r r r rs P s rs c C s| t |�}t|�sdS |js4|js4|jd us4|jd ur8dS | }|d urX||u rPdS |j}q<|rh| |_||_|rx| |_||_dS )NFT)r7 r# r� r� r� r� )�parentr� r �new_name�_parentr r r �_check_and_set_parentd 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$dd?� Z&d@dA� Z'dPdBdC�Z(dDdE� Z)dFdG� Z*dQdIdJ�Z+dS )Rr c O s| | f}t | t�sTtj| g|�R i |��j}|�d|�d��}|d urTt|�rTt| f}t| j|d| j i�}t
t | �� |�}|S )N�spec_set�specrO )
r0 �AsyncMockMixin� _MOCK_SIGZ bind_partialZ argumentsr_ r* r1 rN rO � _safe_superr �__new__)�clsrC �kw�basesZ
bound_argsZspec_arg�newrL r r r r� � s
zNonCallableMock.__new__N� Fc K s� |d u r |}| j }
||
d< ||
d< ||
d< ||
d< d|
d< |d urJ|}d}|
d u rZ|d u}
| �||| |
� i |
d< ||
d < d |
d
< d|
d < d |
d < d
|
d< t� |
d< t� |
d< t� |
d< | |
d< | r�| jf i | �� 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 r� r� � _spec_state� _new_name� _new_parent�_spec_as_instance� _eat_selfZunsaferD r^ r r r r9 � s<
�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 ] }tt||d ��r|�|� q|d ur|t|�s|t|t�rT|}nt|�}t|||�} | or| d }t |�}| j}
||
d< ||
d< ||
d< ||
d< ||
d< d S )Nr� � _spec_class� _spec_set�_spec_signature�
_mock_methods� _spec_asyncs) �dirr 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 ur| jj}|tu r4| 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 ur|| 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 u rt| �S | j S r/ )r� r1 r� r r r � __class__ s
zNonCallableMock.__class__rz r{ r| r} ru c C sN | j }|d u r| jS |j}|d urJt|�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 u r || _n||_d S r/ )� _try_iterr� r� r~ )rB r� r� r r r Z__set_side_effect2 s
z!NonCallableMock.__set_side_effect�rw r~ c C s� |d u r g }t | �|v r d S |�t | �� d| _d | _d| _t� | _t� | _t� | _|r^t | _
|rhd | _ | j �
� D ](}t|t�sr|tu r�qr|j|||d� qr| j
}t|�r�|| ur�|�|� d S )NFr r� )�idr� rz r| r{ rs ru r} rt r r� r� r �valuesr$ �
_SpecState�_deletedrv r# )rB Zvisitedrw r~ �childrx r r r rv = 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 �h r! z0NonCallableMock.configure_mock..)�keyr� )�sorted�items�split�popr& rT )rB rD �arg�valrC �finalr) r� r r r r� [ s �
z NonCallableMock.configure_mockc C s� |dv rt |��n:| jd ur<|| jvs.|tv rLt d| ��nt|�rLt |��| jsd|�d�rdt d��| j�|�}|tu r�t |��np|d u r�d }| j d ur�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� q s:
�
�
zNonCallableMock.__getattr__c C s� | j g}| j}| }d}|dgkr$d}|d urZ|}|�|j | � d}|j dkrRd}|j}q$tt|��}|jpnd}t|�dkr�|d dvr�|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}|dvr d| }d}| jd urDd}| 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 ur|�qS r )r� )r Zm_nameZm_valuer r r �
� s �z+NonCallableMock.__dir__..c S s g | ]}|� d �s|�qS r r �r �er r r r� � r! 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 v rt��||�S �jrH�jd urH|�jvrH|�jvrHtd| ��n�|tv rbd| }t|��n�|tv r�jd ur�|�jvr�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 � �g| �R i |��S r/ r �rC r� �rf rB r r r� � r! z-NonCallableMock.__setattr__..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 v r2|t| �jv r2tt| �|� || jvr2d S | j�|t�}|| jv r\tt| �� |� n|t
u rlt |��|tur|| 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 u sJt|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 ur�t|�dkrNd}|\}}n
|\}}}z"|j|i |��}t||j|j�W S t y� } z|�
d �W Y d }~S d }~0 0 n|S d S )Nr� r r� ) r$ r[ r� r r� r@ r rC rD � TypeError�with_traceback)rB �_callrF r rC rD Z
bound_callr r r r �
_call_matcherA 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 \ 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 f 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 n s
��z"NonCallableMock.assert_called_oncec s� �j d u r.��� ��}d}d||f }t|��� ��fdd�}��t� �fdd��}���j �}||kr�t|t�rv|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.._error_messageT��two)r| r r r �_Callr$ � Exception)rB rC rD �expected�actualZ
error_messager# �causer r" r rk x s
� z"NonCallableMock.assert_called_withc O s< | j dks,d| jpd| j | �� f }t|��| j|i |��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�||vr�|d u rXd}nd�dd� |D ��}t|� d t|�� � jd
d ��d �� ��|�d S t|�}g }|D ]0}z|�|� W q� t y� |�
|� Y q�0 q�|r�td
� j p�dt |�|f �|�d S )Nc s g | ]}� � |��qS r �r �r �cr� r r r� � r! z4NonCallableMock.assert_has_calls..c s s | ]}t |t�r|V qd S r/ �r$ r' r� r r r � � r! z3NonCallableMock.assert_has_calls..c 3 s | ]}� � |�V qd S r/ r, r- r� r r r0 � r! 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 � sN
������ ����z NonCallableMock.assert_has_callsc sj � � t||fdd��}t|t�r$|nd }� fdd�� jD �}|sL|t|�vrf� �||�}td| �|�d S )NTr$ c s g | ]}� � |��qS r r, r- r� r r r� � r! z3NonCallableMock.assert_any_call..z%s call not found)r r& r$ r' r} � _AnyComparerr r �rB rC rD r( r* r) r r r� r rr � s ��z NonCallableMock.assert_any_callc K s� |� d�}|| jd v r&tf i |��S | jrZd|v rBd|d � �nd}| �� | }t|��t| �}t|t�rz|t v rzt}nbt|t
�r�|t v s�| j r�|| j v r�t}q�t}n4t|t
�s�t|t�r�t}q�t|t�r�t}n
|jd }|f i |��S )Nr� r� r r� r� r� )r_ r^ r r� r� r; r1 r0 r �_async_method_magicsr� �_all_sync_magicsr� �
CallableMixinr r r r] )rB r� r� rV r
�_type�klassr 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 r2 r r r r � s zNonCallableMock._calls_repr) NNNNNNr� NFNF)F)FF)N)r )F)rB ),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 � s^ �
-
�
�
''
!
-&r c @ s e Zd Zdd� ZdS )r; c C s, | D ]"}t dd� t||�D ��r dS qdS )Nc S s g | ]\}}||k�qS r r )r r( r) r r r r� s �z-_AnyComparer.__contains__..TF)�all�zip)rB �itemr r r r r� s �z_AnyComparer.__contains__N)rN rQ r� r� r r r r r;
s r; c C sJ | d u r | S t | �r| S t| �r$| S z
t| �W S tyD | Y S 0 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 s: || j d< tt| �j||||||| |
fi | �� || _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 s0 | j |i |�� | j|i |�� | j|i |��S r/ )rI �_increment_mock_call�
_mock_callrA r r r r: ? s zCallableMixin.__call__c O s | j |i |��S r/ )�_execute_mock_callrA r r r rH G s zCallableMixin._mock_callc O s d| _ | jd7 _t||fdd�}|| _| j�|� | jd u}| j}| j}|dk}| j �td||f�� | j
}|d ur�|r�|j �t|||f�� |jd u}|r�|jd | }t|||f�} |j �| � |jr�|r�d}
nd}
|jdk}|j|
| }|j
}qpd S )NTr� r$ r� r� r� ) rz r{ r& r| 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 rG J s4
z"CallableMixin._increment_mock_callc O s� | j }|d urTt|�r |�n,t|�s:t|�}t|�rH|�n||i |��}|turT|S | jturd| jS | jd ur~| j|i |��S | jS r/ )r~ r3 rY r� r r� rw r� )rB rC rD �effectrg r r r rI x s
z CallableMixin._execute_mock_call)
rN rQ r� r r9 rI r: rH rG rI 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 s6 z t | |�W S ty0 t|� t | |� Y S 0 d S r/ )r& r; �
__import__)�thing�comp� import_pathr r r � _dot_lookup� s
rP c C sB | � d�}|�d�}t|�}|D ] }|d| 7 }t|||�}q |S )Nr� r z.%s)r� r� rL rP )�targetZ
componentsrO rM rN r r r � _importer� s
rR 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 ur(|t urtd��|d ur(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 rT rV r� r� rV r� rW rU 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..) rS rT rV r� r� rV r� rW rU rD �attribute_namerX )rB �patcherr r r rY � 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% rY rT )rB rA r� �
attr_valuer\ r r r r] s
z_patch.decorate_classc c s� g }t �� �f}|jD ]8}|�|�}|jd ur8|�|� q|jtu r|�|� q|t |�7 }||fV W d � n1 sx0 Y d S r/ )
�
contextlib� ExitStack� patchings�
enter_contextr[ �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 )Nrf c sD �� �| |��"\}}� |i |��W d � S 1 s60 Y d S r/ �rn �rC rj ZnewargsZ newkeywargs�r- ri rB r r ri 3 s �z)_patch.decorate_callable..patched�r% rf r� r r` r rq r r_ - s
z_patch.decorate_callablec s> t � d�r� j��� � S t� �� ��fdd����g�_�S )Nrf c � sJ �� �| |��(\}}� |i |��I d H W d � S 1 s<0 Y d S r/ ro rp rq r r ri D s �z/_patch.decorate_async_callable..patchedrr r` r rq r r^ > s
z _patch.decorate_async_callablec C s� | � � }| j}t}d}z|j| }W n" ttfyF t||t�}Y n0 d}|tv rdt|t �rdd| _
| j
s�|tu r�td||f ��||fS )NFTz!%s does not have the attribute %r) rT rV r r^ r; �KeyErrorr& � _builtinsr$ r rV )rB rQ r rf �localr r r � get_originalO s
�z_patch.get_originalc C s� | j | j| j }}}| j| j }}| j}| �� | _|du r@d }|du rLd }|du rXd }|d urp|d urptd��|d us�|d ur�|dvr�td��| � � \}}|t
u �r�|d u �r�d} |du r�|}|du r�|}d }n&|d ur�|du r�|}d }n |du r�|}|d u�s |d u�r.|t
u �r td��t |t ��r.d} |d u �rHt
|��rHt}
nt}
i } |d u�r`|}
n^|d u�st|d u�r�|} |d u�r�|} t| ��r�d| v}
n
t| � }
t
| ��r�t}
n
|
�r�t}
|d u�r�|| d< |d u�r�|| d < t |
t ��r t|
t��r | j�r | j| d
< | �|� |
f i | ��}| �r�t|��r�|} |d u�rF|} t| ��s^t| ��s^t}
| �d
� |
f |d d �| ��|_nl|d u�r�|t
u�r�td
��|t
u �r�td��t|�}|du �r�|}t |f|| jd�|��}n|�r�td��|}|| _ || _ t � � | _!zrt"| j| j|� | j#d u�rti }| j t
u �r>||| j#< | j$D ](}| j!�%|�}|j t
u �rD|�|� �qD|W S |W S | j&t'�(� � �s�� Y n0 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� rW rD rU rT rQ r rv r r$ r1 r* r r r\ r� r r0 r rV rh r# ra r� rw �boolr �
temp_original�is_localrd re � _exit_stackrT r[ rX rg �__exit__�sys�exc_info)rB r� r� r� rW rD rU rf ru Zinherit�Klass�_kwargsZ this_specZ not_callableZnew_attrrk rm r� r r r � __enter__g s�
�
��
�
�
�
��
z_patch.__enter__c G s� | j r$| jtur$t| j| j| j� n>t| j| j� | jsbt| j| j�rP| jdv rbt| j| j| j� | `| ` | `| j }| ` |j
|� S )N)rO rQ rR �__annotations__rS ) rz ry r rT rQ rV r rV r% r{ r| )rB r~ rl r r r r| � s �z_patch.__exit__c C s | � � }| j�| � |S r/ )r� �_active_patchesr� �rB rg r r r �start s z _patch.startc C s4 z| j �| � W n ty$ Y d S 0 | �d d d �S r/ )r� r5 r= r| r� r r r �stop s
z _patch.stop)rN rQ r� r[ r� r9 rY r: r] rd �contextmanagerrn r_ r^ rv r� r| r� r� r r r r rS � s
rS c
sP z� � dd�\� }W n& tttfy: td� ����Y n0 � fdd�}||fS )Nr� r� z,Need a valid target to patch. You supplied: c s t � �S r/ �rR r �rQ r r r� r! z _get_target..)�rsplitr r= r; )rQ rV rT r r� r � _get_target s �
r� c
s> t � �tu rt� �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� 5 r! z _patch_object..)r1 �strr rS )
rQ rV r� r� rV r� rW rU rD rT r r� r �
_patch_object s �
�r� c
s� t � �tu r� 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� S r! z!_patch_multiple..c s � S r/ r r r� r r r� U r! z=Must supply at least one keyword argument with patch.multipler r� ) r1 r� r= rZ r� rS r[ rX r� )
rQ r� rV r� rW rU rD rT r� rV r� r\ Z this_patcherr r� r �_patch_multiple< s, � � �r� c
K s$ t | �\}} t|| |||||||� S r/ )r� rS )
rQ r� r� rV r� rW rU rD rT rV r r r r m s
G
�r c @ sV e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dS )� _patch_dictr Fc K s, || _ t|�| _| j�|� || _d | _d S r/ )�in_dict�dictr� rh �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 s0 �� � z� | i |��W ��� S ��� 0 d S r/ )r� �
_unpatch_dictr ��frB r r �_inner� s �z$_patch_dict.__call__.._inner)r$ r1 r] 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 ra ) r� r& r r rb r% r� r� r� r� rT )rB rA r� rc � decoratorZ decoratedr r r r] � 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, tyb i }|D ]}|| ||< qLY n0 || _|rvt |� z|�
|� W n( ty� |D ]}|| ||< q�Y n0 d S r/ ) r� r$ r� r� rR r� rY r; r� � _clear_dictrh )rB r� r� r� rf r� r r r r� s& z_patch_dict._patch_dictc C sP | j }| j}t|� z|�|� W n( tyJ |D ]}|| ||< q4Y n0 d S r/ )r� r� r� rh r; )rB r� rf r� r r r r� s z_patch_dict._unpatch_dictc G s | j d ur| �� dS �NF)r� r� )rB rC r r r r| ) s
z_patch_dict.__exit__c C s | � � }tj�| � |S r/ )r� rS r� r� r� r r r r� 0 s z_patch_dict.startc C s4 zt j�| � W n ty$ Y d S 0 | �d d d �S r/ )rS r� r5 r= r| r� r r r r� 7 s
z_patch_dict.stopN)r F) rN rQ r� r9 r: r] r� r� r� r| r� r� r r r r r� � s
r� c C s< z | � � W n* ty6 t| �}|D ]
}| |= q&Y n0 d S r/ )r� r; rZ )r� �keysr� r r r r� B s r� c C s t tj�D ] } | �� q
d S r/ )r� rS r� r� )r r r r �_patch_stopallK s r� �testz�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 r0 j r! r0 c c s | ]}d | V qdS )zr%sNr r� r r r r0 k r! > Z__getinitargs__�__getnewargs_ex__�
__delete__�
__format__r r� � __getstate__�__getnewargs__�
__reduce_ex__�
__getformat__� __setstate__�__get__� __reversed__�__subclasses__�
__setformat__� __missing__r� �__set__c s � fdd�}| |_ |S )Nc s � | g|�R i |��S r/ r �rB rC r� r, r r �method| s z_get_method..method)rN )r r- r� r r, r r z s r c C s h | ] }d | �qS )r� r )r r� r r r r � s > �
__aenter__� __aexit__� __anext__� __aiter__> r9 r� � __prepare__r �__del__�__subclasscheck__r� �__instancecheck__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__r| � __complex__� __float__�__bool__� __index__r� c s � fdd�}|S )Nc s$ � j j}|tur|S � | u r dS tS r8 )�__eq__r� r �NotImplemented)�other�ret_valr� r r r� � s z_get_eq..__eq__r )rB r� r r� r �_get_eq� s r� c s � fdd�}|S )Nc s � j jturtS � | u r dS tS r� )�__ne__r� r r� )r� r� r r r� � s
z_get_ne..__ne__r )rB r� r r� r �_get_ne� s r� c s � fdd�}|S )Nc s � j j} | tu rtg �S t| �S r/ )�__iter__r� r r� �r� r� r r r� � s z_get_iter..__iter__r )rB r� r r� r � _get_iter� s r� c s � fdd�}|S )Nc s( � j j} | tu r ttg ��S tt| ��S r/ )r� r� r �_AsyncIteratorr� r� r� r r r� � s z"_get_async_iter..__aiter__r )rB r� r r� r �_get_async_iter� s r� )r� r� r� r� c C sb t �|t�}|tur ||_d S t�|�}|d urB|| �}||_d S t�|�}|d ur^|| �|_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|i |�� | � � d S r/ )�_mock_set_magicsr� r� r9 r� r r r r9 � s zMagicMixin.__init__c C s� t tB }|}t| dd �d urX|�| j�}t� }|| }|D ] }|t| �jv r: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|i |�� | � � 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 7 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� D s zMagicProxy.__get__)N)rN rQ r� r9 r� r� r r r r r� 6 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|i |�� 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 M s
zAsyncMockMixin.__init__c � s t ||fdd�}| jd7 _|| _| j�|� | j}|d ur�t|�rL|�nft|�s�z t|�}W n t yv t
�Y n0 t|�r�|�n,t |�r�||i |��I d H }n||i |��}|t ur�|S | j
t ur�| jS | jd u�r
t | j�r�| j|i |��I d H S | j|i |��S | jS )NTr$ r� )r& r� r� r� r� r~ r3 rY r� �
StopIteration�StopAsyncIterationr r r� rw r� )rB rC rD r rJ rg r r r rI ] 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� � 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� � s
�z"AsyncMockMixin.assert_awaited_oncec s� �j d u r&��� ��}td|� d���� ��fdd�}��t� �fdd��}���j �}||kr~t|t�rn|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.._error_messageTr$ )r� r r r r& r$ r' )rB rC rD r( r# r) r* r r" r r� � s
z"AsyncMockMixin.assert_awaited_withc O s< | j dks,d| jpd� d| j � d�}t|��| j|i |��S r� )r� r� r r� r+ r r r r� � s
�z'AsyncMockMixin.assert_awaited_once_withc sj � � t||fdd��}t|t�r$|nd }� fdd�� jD �}|sL|t|�vrf� �||�}td| �|�d S )NTr$ c s g | ]}� � |��qS r r, r- r� r r r� � r! z3AsyncMockMixin.assert_any_await..z%s await not found)r r& r$ r' r� r; r r r< r r� r r� � s ��z AsyncMockMixin.assert_any_awaitFc s� � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|s�||vr�|d u rXd}nd�dd� |D ��}t|� d t|�� d
� j� ��|�d S t|�}g }|D ]0}z|�|� W q� ty� |�|� Y q�0 q�|r�td t |�f �|�d S ) Nc s g | ]}� � |��qS r r, r- r� r r r� � r! z4AsyncMockMixin.assert_has_awaits..c s s | ]}t |t�r|V qd S r/ r/ r� r r r r0 � r! z3AsyncMockMixin.assert_has_awaits..c 3 s | ]}� � |�V qd S r/ r, r- r� r r r0 � r! 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 �r1 z
Actual: z %r not all found in await list)
r� rs r� r3 r rZ r5 r= r� r[ ) rB r6 r7 r( r* Z
all_awaitsr8 r9 r: r r� r r� � sF ������ ��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|i |�� 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 rI r� r� r� r� r� r� r� rv �
__classcell__r r r� r r� H s (
, r� c @ s e Zd ZdS )r NrK 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� 0 s z _ANY.__ne__c C s dS )Nzr r� r r r r� 3 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 )�repr)r r� r r r r� = r! z*_format_call_signature..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 @ sv e Zd Zd dd�Zd dd �Zd
d � ZejZd d
� Zdd� Z dd� Z
dd� Z e dd� �Z
e dd� �Zdd� Zdd� ZdS ) r& 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� r% � from_kallrC rD �_len�first�secondr r r r� _ s.
z
_Call.__new__c C s || _ || _|| _d S r/ )r� r� �_mock_from_kall)rB r� r r� r% r� r r r r9 s z_Call.__init__c C s\ z t |�}W n ty" t Y S 0 d}t | �dkr>| \}}n
| \}}}t| dd �rpt|dd �rp| j|jkrpdS d}|dkr�di }}n�|dkr�|\}}}n�|dkr�|\} t| t�r�| }i }n"t| t�r�| }di }}nd}| }nV|dk�r4|\}
} t|
t��r(|
}t| t��r | i }}n
d| }}n
|
| }}ndS |�rL||k�rLdS ||f||fkS ) Nr� r� r� Fr 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� � sN
�
z _Call.__eq__c O s< | j d u r td||fdd�S | j d }t| j ||f|| d�S )Nr� r� r� r� �r� r& r r r r r: � s
z_Call.__call__c C s2 | j d u rt|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 jv rt�t �| |�S r/ )r[ r^ r; �__getattribute__)rB r� r r r r� � s
z_Call.__getattribute__c 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 ur(|j r |�|� |j}qtt|��S r/ )r� r� r� rs r� )rB ZvalsrM 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� rC rD r� r� r r r r r& L s$ �
�
5
r& )r� c K sh t | �rt| �} t| t�}t| �}d| i}|r8d| i}n | d u rDi }|rT|rTd|d< |�|� t} t�| �rri }n8|r�|r�td��t } n"t
| �s�t } n|r�|r�t | �s�t } |�
d|�}|}
|d u r�d}
| f |||
|d�|��} t| t��rt| | �} |�rt| � nt| | ||� |d u�r,|�s,| |j|< |�rV|�sVd |v�rVt| |dd
| d �| _t| �D �]} t| ��rr�q^zt| | �}
W n t�y� Y �q^Y n0 d|
i}|�r�d|
i}t|
t��s�t|
|| | |�}|| j| < nn| }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 rw r� r� )r� r r� r� )rK ) r\ r1 r$ r. rh r r' Zisdatadescriptor� RuntimeErrorr rY r ra r� r6 rj r� rM r r rw r� r� r& r; r� r5 �
_must_skipr rT )r� r� rL r� rw 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 �v r dS | j} | jD ]H}|j�|t�}|tu rFq*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 rA rg r r r r �
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 u r�d d l}ttt|j���tt|j����a| d u r�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 ur�j jS � d j | i |��S r� )� readlinesrw rb ��_state�handler r �_readlines_side_effect�
s z)mock_open.._readlines_side_effectc s( �j jd ur�j jS � d j | i |��S r� )�readrw rb r r r �_read_side_effect�
s z$mock_open.._read_side_effectc ? s( � � E d H �d j | i |��V q d S r� )�readlinerb )�_iter_side_effectr r r �_readline_side_effect�
s z(mock_open.._readline_side_effectc 3 s0 �j jd ur�j jV q � d D ]
} | V q d S r� )r rw )�liner r r r �
s z$mock_open.._iter_side_effectc s �j jd ur�j jS t� d �S r� )r rw r� r r r r �_next_side_effect�
s z$mock_open.._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..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 i |��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 sp d| _ t| �D ]\}zt| |�}W n ty6 Y qY n0 t|t�sDqt|j�|�t�rXq|j | u rt
|� qd S r8 ) r� r� r& r; r$ r r r_ r� r� r )r5 r� r� r r r r % s
r c @ s e Zd 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 A s
z_AsyncIterator.__init__c � s( z t | j�W S ty Y n0 t�d S r/ )r� r r� r� r� r r r r� G s
z_AsyncIterator.__anext__N)rN rQ r� r9 r� r r r r r� = s r� )F)F)NFNNN)FFNN)Nr� )}�__all__r� rd r r' r� r} �builtinsr �typesr r r Z
unittest.utilr � functoolsr r r� rt 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< r9 r� r; r� r? r rP rR rS r� r� r� r r� r� r� r� ZmultipleZstopallrb 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[ r& r r r r� r1 r� r6 r r r
r r r� r r r r � s0
1 h4 < �
�
2�
N �� ���
� 8+ :
�
�
N