관리-도구
편집 파일: mock.cpython-310.opt-2.pyc
o 6��fs� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z m Z m Z ddl mZ ddlmZmZ ddlmZ G dd � d e�Zd d � ee�D �Zd ZeZd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z d�dd�Z!d d � Z"d d � 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/Z0e-j1Z2h d0�Z3d1d2� Z4G d3d4� d4e5�Z6d5d6� Z7G d7d8� d8e*�Z8G d9d:� d:e*�Z9G d;d<� d<e9�Z:e�;e:j<�Z=G d=d>� d>e5�Z>d?d@� Z?G dAdB� dBe9�Z@G dCdD� dDe@e:�ZAdEdF� ZBdGdH� ZCdIdJ� ZDG dKdL� dLe*�ZEdMdN� ZFe.dddddfddO�dPdQ�ZG d�dRdS�ZHe.dddddfddO�dTdU�ZIG dVdW� dWe*�ZJdXdY� ZKdZd[� ZLeGeI_*eJeI_MeHeI_NeLeI_Od\eI_Pd]ZQd^ZRd_�Sd`da� eR�T� D ��ZUd_�Sdbda� eR�T� D ��ZVh dc�ZWddde� ZXdfd � d_�SeQeReUeVg��T� D �ZYh dg�ZZdhhZ[eZe[B Z\eYeWB Z]e]e\B Z^h di�Z_djdk� dldk� dmdk� dndk� do�Z`eaeaeaeadpddddqdrd dpdds� Zbdtdu� Zcdvdw� Zddxdy� Zedzd{� Zfecedeeefd|�Zgd}d~� ZhG dd�� d�e9�ZiG d�d�� d�eie:�ZjG d�d�� d�ei�ZkG d�d�� d�eieA�ZlG d�d�� d�e9�ZmG d�d�� d�e9�ZnG d�d�� d�enekeA�ZoG d�d�� d�e*�Zpep� Zqd�d�� ZrG d�d�� d�es�Ztetdd��Zu d�ddO�d�d��Zvd�d�� ZwG d�d�� d�e*�Zxeyev�eyeqjz�fZ{da|d�d�� Z}d�d�d��Z~G d�d�� d�eA�Zd�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�partial)�RLockc @ � e Zd ZdS )�InvalidSpecErrorN��__name__� __module__� __qualname__� r r �4/opt/alt/python310/lib64/python3.10/unittest/mock.pyr ( � r c C s h | ] }|� d �s|�qS ��_�� startswith)�.0�namer r r � <setcomp>, � r( Tc C s< t | �r t| t�s dS t| d�rt| d�} t| �p t�| �S )NF�__func__)�_is_instance_mock� isinstancer �hasattr�getattrr �inspectZ isawaitable��objr r r � _is_async_obj4 s r2 c C s t | dd �r t| �S dS )N�__code__F)r. r ��funcr r r �_is_async_func< s r6 c C s t t| �t�S �N)� issubclass�typer r0 r r r r+ C s r+ c C s t | t�pt | t�ot| t�S r7 )r, � BaseExceptionr9 r8 r0 r r r � _is_exceptionI s �r; c C s t | t�r t| d�r | jS | S �N�mock)r, � FunctionTypesr- r= r0 r r r � _extract_mockP s r? c C s� t | t�r|s| j} d}nt | t�s#z| j} W n ty" Y d S w |r+t| d �}n| }z| t�|�fW S t y? Y d S w �NT) r, r9 �__init__r> �__call__�AttributeErrorr r/ � signature� ValueError)r5 Z as_instanceZeat_selfZsig_funcr r r �_get_signature_objectY s$ � �rF Fc sN t | ||�� � d u r d S � \} � � fdd�}t| |� |t|�_� t|�_d S )Nc s � j |i |�� d S r7 ��bind��self�args�kwargs��sigr r �checksig{ � z"_check_signature.<locals>.checksig)rF �_copy_func_detailsr9 �_mock_check_sig� __signature__)r5 r= � skipfirst�instancerO r rM r �_check_signaturev s rV c C s6 dD ]}z t ||t| |�� W q ty Y qw d S )N)r �__doc__�__text_signature__r � __defaults__�__kwdefaults__)�setattrr. rC )r5 �funcopy� attributer r r rQ � s ��rQ c C s@ t | t�rdS t | tttf�rt| j�S t| dd �d ur dS dS )NTrB F)r, r9 � staticmethod� classmethodr � _callabler* r. r0 r r r r` � s r` c C s t | �ttfv S r7 )r9 �list�tupler0 r r r �_is_list� s rc c C sH t | t�st| dd �d uS | f| j D ] }|j�d�d ur! dS qdS )NrB TF)r, r9 r. �__mro__�__dict__�get)r1 �baser r r �_instance_callable� s �rh c s� t |t�}t|||�}|d u r| S |\}� � fdd�}t||� |j}|�� s)d}|| d�}d| } t| |� || } t| | � � | S )Nc s � j | i |�� d S r7 rG �rK rL rM r r rO � rP z _set_signature.<locals>.checksigr\ )Z _checksig_r= zYdef %s(*args, **kwargs): _checksig_(*args, **kwargs) return mock(*args, **kwargs))r, r9 rF rQ r � isidentifier�exec� _setup_func) r= �originalrU rT �resultr5 rO r' �context�srcr\ r rM r �_set_signature� s$ � rq 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 � � j | i |��S r7 )�assert_called_withri �r= r r rs � � z'_setup_func.<locals>.assert_called_withc rr r7 )� assert_calledri rt r r rv � ru z"_setup_func.<locals>.assert_calledc rr r7 )�assert_not_calledri rt r r rw � ru z&_setup_func.<locals>.assert_not_calledc rr r7 )�assert_called_onceri rt r r rx � ru z'_setup_func.<locals>.assert_called_oncec rr r7 )�assert_called_once_withri rt r r ry � ru z,_setup_func.<locals>.assert_called_once_withc rr r7 )�assert_has_callsri rt r r rz � ru z%_setup_func.<locals>.assert_has_callsc rr r7 )�assert_any_callri rt r r r{ � ru z$_setup_func.<locals>.assert_any_callc sB t � � _t � � _��� � j} t| �r | �ur | �� d S d S d S r7 )� _CallList� method_calls� mock_calls� reset_mock� return_valuer+ )�ret�r\ r= r r r � s �z _setup_func.<locals>.reset_mockFr )r= �called� call_count� call_argsr| �call_args_listr} r~ r� � side_effect�_mock_childrenrs ry rz r{ r rv rw rx rS �_mock_delegate) r\ r= rN rs rv rw rx ry rz r{ r r r� r rl � s8 rl c sJ t jj� _d� _d � _t� � _� fdd�}dD ] }t� |t||�� qd S )Nr c s t � j| �|i |��S r7 )r. r= )�attrrK rL rt r r �wrapper � 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) �asyncio� coroutines� _is_coroutine� await_count� await_argsr| �await_args_listr[ r )r= r� r] r rt r �_setup_async_mock� s �r� c C s d| dd� | kS )N�__%s__� ���r �r' r r r � _is_magic rP r� c @ �& e Zd Z dd� Zdd� Zdd� ZdS )�_SentinelObjectc C s || _ d S r7 r� �rJ r' r r r rA � z_SentinelObject.__init__c C � d| j S �Nz sentinel.%sr� �rJ r r r �__repr__ r� z_SentinelObject.__repr__c C r� r� r� r� r r r � __reduce__! r� z_SentinelObject.__reduce__N)r r r rA r� r� r r r r r� s r� c @ r� )� _Sentinelc C s i | _ d S r7 )� _sentinelsr� r r r rA ' r� z_Sentinel.__init__c C s |dkrt �| j�|t|��S )N� __bases__)rC r� � setdefaultr� r� r r r � __getattr__* s z_Sentinel.__getattr__c C � dS )Nr r r� r r r r� 0 � z_Sentinel.__reduce__N)r r r rA r� r� r r r r r� % s r� > � _mock_name�_mock_new_parentr� �_mock_return_valuer� �_mock_new_name� _mock_parent�_mock_side_effectc C s8 t �| � d| }| |fdd�}| |fdd�}t||�S )NZ_mock_c S s" | j }|d u r t| |�S t||�S r7 )r� r. )rJ r' � _the_namerN r r r �_getE s z"_delegating_property.<locals>._getc S s, | j }|d u r|| j|< d S t|||� d S r7 )r� re r[ )rJ �valuer' r� rN r r r �_setJ s z"_delegating_property.<locals>._set)�_allowed_names�add�property)r' r� r� r� r r r �_delegating_propertyB s r� c @ � e Zd Zdd� Zdd� ZdS )r| c C sl t |t�s t�| |�S t|�}t| �}||krdS td|| d �D ]}| ||| � }||kr3 dS q"dS )NFr � T)r, ra � __contains__�len�range)rJ r� Z len_valueZlen_self�iZsub_listr r r r� W s �z_CallList.__contains__c C s t �t| ��S r7 )�pprintZpformatra r� r r r r� e � z_CallList.__repr__N)r r r r� r� r r r r r| U s r| c C s� t |�}t|�s dS |js|js|jd us|jd ur dS | }|d ur/||u r(dS |j}|d us"|r7| |_||_|r?| |_||_dS )NFT)r? r+ r� r� r� r� )�parentr� r' �new_name�_parentr r r �_check_and_set_parenti s( �r� c @ r� )� _MockIterc C s t |�| _d S r7 )�iterr1 )rJ r1 r r r rA � r� z_MockIter.__init__c C s t | j�S r7 )�nextr1 r� r r r �__next__� r� z_MockIter.__next__N)r r r rA r� r r r r r� � s r� c @ s e Zd ZeZdZdd� ZdS )�BaseNc O � d S r7 r rI r r r rA � r� z Base.__init__)r r r r r� r� rA r r r r r� � s r� c @ sr e Zd Z e� Zdd� Z dKdd�Zdd � ZdLd d �Z dMd d �Z dd� Z dd� Z dZ e e e e �Ze dd� �Zed�Zed�Zed�Zed�Zed�Zdd� Zd d � Ze ee�ZdNddd �d d �Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)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)dLdBdC�Z*dDdE� Z+dFdG� Z,dPdIdJ�Z-dS )Qr c O s| | f}t | t�s*tj| g|�R i |��j}|�d|�d��}|d ur*t|�r*t| f}t| j|d| j i�}t t | �� |�}|S )N�spec_set�specrW ) r8 �AsyncMockMixin� _MOCK_SIGZ bind_partialZ argumentsrf r2 r9 r rW � _safe_superr �__new__)�clsrK �kw�basesZ bound_argsZspec_arg�newrU r r r r� � s zNonCallableMock.__new__N� Fc K s� |d u r|}| j } || d< || d< || d< || d< d| d< |d ur%|}d}| d u r-|d u} | �||| | � i | d< || d < d | d < d| d < d | d < d | d< t� | d< t� | d< t� | d< | | d< | rj| jdi | �� 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_callsr} � _mock_unsafer )re �_mock_add_specr| �configure_mockr� r rA )rJ r� r r' r� r� � _spec_state� _new_name� _new_parent�_spec_as_instance� _eat_self�unsaferL re r r r rA � s< �zNonCallableMock.__init__c C s2 t |�}d |_d |_d|_d |_t| ||� d S �Nr� )r? r� r� r� r� r[ )rJ r= r] Z inner_mockr r r � attach_mock� s zNonCallableMock.attach_mockc C s | � ||� d S r7 )r� �rJ 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�r*|}nt|�}t|||�} | o9| d }t |�}| j} || d< || d< || d< || d< || d< d S )Nr� � _spec_class� _spec_set�_spec_signature� _mock_methods� _spec_asyncs) �dirr r. �appendrc r, r9 rF re ) rJ r� r� r� r� r� r� r� r� �resre r r r r� � s, � � z NonCallableMock._mock_add_specc C s8 | j }| jd ur | jj}|tu r| j| dd�}|| _|S )N�()�r� r� )r� r� r� r �_get_child_mock)rJ r� r r r Z__get_return_value s �z"NonCallableMock.__get_return_valuec C s. | j d ur || j _d S || _t| |d d� d S )Nr� )r� r� r� r� )rJ 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 r t| �S | j S r7 )r� r9 r� r r r � __class__' s zNonCallableMock.__class__r� r� r� r� r~ c C sN | j }|d u r | jS |j}|d ur%t|�s%t|t�s%t|�s%t|�}||_|S r7 )r� r� r� �callabler, r� r; )rJ � delegatedZsfr r r Z__get_side_effect4 s ��z!NonCallableMock.__get_side_effectc C s* t |�}| j}|d u r|| _d S ||_d S r7 )� _try_iterr� r� r� )rJ r� r� r r r Z__set_side_effect? s z!NonCallableMock.__set_side_effect�r� r� c C s� |d u rg }t | �|v rd S |�t | �� d| _d | _d| _t� | _t� | _t� | _|r0t | _ |r5d | _ | j � � D ]}t|t�sE|tu rFq:|j|||d� q:| j }t|�ra|| urc|�|� d S d S d S )NFr r� )�idr� r� r� r� r| r~ r� r} r r� r� r� �valuesr, � _SpecState�_deletedr r+ )rJ Zvisitedr� r� �childr� r r r r J s0 �zNonCallableMock.reset_mockc K sZ t |�� dd� d�D ] \}}|�d�}|�� }| }|D ]}t||�}q t|||� q d S )Nc S s | d � d�S )Nr �.)�count)�entryr r r �<lambda>u s z0NonCallableMock.configure_mock.<locals>.<lambda>)�keyr� )�sorted�items�split�popr. r[ )rJ rL �arg�valrK �finalr1 r r r r r� h s � �z NonCallableMock.configure_mockc C s� |dv rt |��| jd ur || jvs|tv r t d| ��nt|�r%t |��| js?| jr0|| jvr?|�d�r?t |�d|�d���tj�� | j� |�}|t u rQt |��|d u rrd }| j d urbt | j |�}| j | |||| d�}|| j|< nJt|t�r�zt|j|j|j|j|j�}W n ty� | jd p�| }td|�d |�d | �d |j�d � ��w || j|< W d � |S W d � |S W d � |S 1 s�w Y |S ) N> r� r� z Mock object has no attribute %r)�assertZassretZasertZaseertZassrtz6 is not a valid assertion. Use a spec for the mock if z is meant to be an attribute.)r� r' r r� r� r� �Cannot autospec attr � from target �, as it has already been mocked out. [target=�, attr=�])rC r� � _all_magicsr� r� r% r �_lockr� rf r� r� r. r� r, r� r r� r� rU r� r' r re )rJ r' rn r � target_namer r r r� ~ sr � �� � � ����� � � � � � � zNonCallableMock.__getattr__c C s� | j g}| j}| }d}|dgkrd}|d ur0|}|�|j | � d}|j dkr)d}|j}|d ustt|��}|jp:d}t|�dkrK|d dvrK|d7 }||d< d�|�S )Nr� r� r� r= r� )r� z().r )r� r� r� ra �reversedr� r� �join)rJ Z _name_listr� Zlast�dotZ_firstr r r �_extract_mock_name� s* � z"NonCallableMock._extract_mock_namec C s^ | � � }d}|dvrd| }d}| jd ur"d}| jr d}|| jj }dt| �j||t| �f S )Nr� )r= zmock.z name=%rz spec=%rz spec_set=%rz<%s%s%s id='%s'>)r r� r� r r9 r� )rJ r' Z name_stringZ spec_stringr r r r� � s �zNonCallableMock.__repr__c C sx t st�| �S | jp g }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 � <listcomp>� s �z+NonCallableMock.__dir__.<locals>.<listcomp>c S s g | ] }|� d �s|�qS r" r$ �r& �er r r r � r) c S s" g | ] }|� d �r t|�r|�qS r" )r% r� r r r r r � s �) r �object�__dir__r� r� r9 ra re r� r r �set)rJ ZextrasZ from_typeZ from_dictZfrom_child_mocksr r r r � s �zNonCallableMock.__dir__c sJ |t v r t��||�S �jr#�jd ur#|�jvr#|�jvr#td| ��|tv r/d| }t|��|tv rq�jd urC|�jvrCtd| ��t |�s\t t ��|t ||�� |� � �fdd�}n*t �|d |� t t ��||� |�j|< n|dkrz|�_d S t �|||�r�|�j|< �jr�t�|�s���� � 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 r7 r �rK r� �rm rJ r r r � z-NonCallableMock.__setattr__.<locals>.<lambda>r� r� z Cannot set )r� r � __setattr__r� r� re rC �_unsupported_magicsr r+ r[ r9 � _get_methodr� r� r� r� r- r )rJ r' r� �msg� mock_namer r r r! � s8 zNonCallableMock.__setattr__c C s� |t v r|t| �jv rtt| �|� || jvrd S | j�|t�}|| jv r.tt| �� |� n|t u r6t |��|tur>| j|= t | j|< d S r7 ) r r9 re �delattrr� rf �_missingr� r � __delattr__r� rC )rJ r' r1 r r r r( s zNonCallableMock.__delattr__c C s | j pd}t|||�S r< )r� �_format_call_signature�rJ rK rL 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� )rJ rK rL �action�message�expected_stringr� Z actual_stringr r r � _format_mock_failure_message. s z,NonCallableMock._format_mock_failure_messagec C sl |s| j S d }|�dd��d�}| j}|D ] }|�|�}|d u s&t|t�r) |S t|�}|j}|j }q|S )Nr� r� r� )r� �replacer r� rf r, r� r? )rJ r' rN �names�childrenr� r r r � _get_call_signature_from_name6 s �z-NonCallableMock._get_call_signature_from_namec C s� t |t�rt|�dkr| �|d �}n| j}|d urVt|�dkr(d}|\}}n|\}}}z|j|i |��}t||j|j�W S t yU } z |� d �W Y d }~S d }~ww |S )Nr� r r� ) r, rb r� r3 r� rH r rK rL � TypeError�with_traceback)rJ �_callrN r' rK rL Z bound_callr r r r � _call_matcherW s" ��z NonCallableMock._call_matcherc C s2 | j dkrd| jp d| j | �� f }t|��d S )Nr z9Expected '%s' to not have been called. Called %s times.%sr= �r� r� � _calls_repr�AssertionError�rJ r$ r r r rw r � ���z!NonCallableMock.assert_not_calledc C s& | j dkrd| jp d }t|��d S )Nr z"Expected '%s' to have been called.r= )r� r� r: r; r r r rv | s ��z NonCallableMock.assert_calledc C s2 | j dksd| jp d| j | �� f }t|��d S )Nr� z:Expected '%s' to have been called once. Called %s times.%sr= r8 r; r r r rx � r< z"NonCallableMock.assert_called_oncec s� �j d u r��� ��}d}d||f }t|��� ��fdd�}��t� �fdd��}���j �}||krDt|t�r<|nd }t|� �|�d S )Nz not called.z0expected call not found. Expected: %s Actual: %sc s �� � ��} | S r7 �r/ �r$ �rK rL rJ r r �_error_message� s z:NonCallableMock.assert_called_with.<locals>._error_messageT��two)r� r+ r: r7 �_Callr, � Exception)rJ rK rL �expected�actualZ error_messager@ �causer r? r rs � s � �z"NonCallableMock.assert_called_withc O s> | j dksd| jp d| j | �� f }t|��| j|i |��S )Nr� z3Expected '%s' to be called once. Called %s times.%sr= )r� r� r9 r: rs �rJ rK rL r$ r r r ry � s ��z'NonCallableMock.assert_called_once_withc s � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|sN||vrL|d u r-d}n d�dd� |D ��}t|� d t|�� � jd d ��d �� ��|�d S t|�}g }|D ]}z|�|� W qV t ym |� |� Y qVw |rtd � j pvdt |�|f �|�d S )Nc � g | ]}� � |��qS r �r7 �r& �cr� r r r � r z4NonCallableMock.assert_has_calls.<locals>.<listcomp>c s � � | ] }t |t�r|V qd S r7 �r, rD r r r r � <genexpr>� � � z3NonCallableMock.assert_has_calls.<locals>.<genexpr>c 3 � � | ]}� � |�V qd S r7 rJ rK r� r r rO � � � zCalls not found.z+Error processing expected calls. Errors: {}c S � g | ] }t |t�r |nd �qS r7 rN r r r r r � � �� Expected: ZActual)�prefixr� z@%r does not contain all of %r in its call list, found %r insteadr= ) r� r| r~ �formatr: r9 �rstripra �removerE r� r� rb ) rJ �calls� any_orderrE rG Z all_calls�problem� not_found�kallr r� r rz � sX ������� ������z NonCallableMock.assert_has_callsc �l � � t||fdd��}t|t�r|nd }� fdd�� jD �}|s'|t|�vr4� �||�}td| �|�d S )NTrA c rI r rJ rK r� r r r � r z3NonCallableMock.assert_any_call.<locals>.<listcomp>z%s call not found)r7 rC r, rD r� � _AnyComparerr+ r: �rJ rK rL rE rG rF r. r r� r r{ � s ���z NonCallableMock.assert_any_callc K s� | j rd|v rd|d � �nd}| �� | }t|��|�d�}|| jd v r.tdi |��S t| �}t|t�r>|t v r>t}n1t|t �rU|t v sO| j rR|| j v rRt}n t}nt|t �sjt|t�rbt}n t|t�rit}n|jd }|di |��S )Nr' r� r� r� r� r� r )r� r rC rf re r r9 r8 r �_async_method_magicsr� �_all_sync_magicsr� � CallableMixinr r r rd )rJ r� r] r% r� �_type�klassr r r r� � s4 � � z NonCallableMock._get_child_mock�Callsc C s$ | j sdS d|� dt| j �� d�S )Nr� � z: r� )r~ r )rJ rV r r r r9 s zNonCallableMock._calls_repr) NNNNNNr� NFNF�F)FFr7 )r )rg ).r r r r r r� rA r� r� r� Z"_NonCallableMock__get_return_valueZ"_NonCallableMock__set_return_valueZ"_NonCallableMock__return_value_docr� r� r� r� r� r� r� r� r~ Z!_NonCallableMock__get_side_effectZ!_NonCallableMock__set_side_effectr� r r� r� r r� r r! r( r+ r/ r3 r7 rw rv rx rs ry rz r{ r� r9 r r r r r � sd �- � � 0' ! -&r c @ s e Zd Z dd� ZdS )r` c C s, | D ]}t dd� t||�D ��r dS qdS )Nc S s g | ]\}}||k�qS r r )r& rE rF r r r r + s ��z-_AnyComparer.__contains__.<locals>.<listcomp>TF)�all�zip)rJ �itemr6 r r r r� ( s ��z_AnyComparer.__contains__N)r r r r� r r r r r` # s r` c C sF | d u r| S t | �r | S t| �r| S zt| �W S ty" | Y S w r7 )r; r` r� r4 r0 r r r r� 3 s �r� c @ sR e Zd Zddedddddddf dd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )rd Nr� c K s: || j d< tt| �j||||||| | fi | �� || _d S )Nr� )re r� rd rA r� ) rJ r� r� r� r r' r� r� r� r� r� rL r r r rA D s �� zCallableMixin.__init__c O r� r7 r rI r r r rR P s z CallableMixin._mock_check_sigc O s0 | j |i |�� | j|i |�� | j|i |��S r7 )rR �_increment_mock_call� _mock_callrI r r r rB U s zCallableMixin.__call__c O s | j |i |��S r7 )�_execute_mock_callrI r r r rn ] ru zCallableMixin._mock_callc O s d| _ | jd7 _t||fdd�}|| _| j�|� | jd u}| j}| j}|dk}| j �td||f�� | j }|d ur�|rW|j �t|||f�� |jd u}|rW|jd | }t|||f�} |j �| � |jrz|rld} nd} |jdk}|j| | }|j }|d us<d S d S )NTr� rA r� r� r� ) r� r� rC r� r� r� r� r� r� r~ r� r} ) rJ rK rL r6 Zdo_method_callsZmethod_call_nameZmock_call_nameZ is_a_callr� Zthis_mock_callr r r r rm ` s6 �z"CallableMixin._increment_mock_callc O s� | j }|d ur)t|�r |�t|�s t|�}t|�r|�n||i |��}|tur)|S | jtur1| jS | jd ur>| j|i |��S | jS r7 )r� r; r` r� r r� r� r� )rJ rK rL �effectrn r r r ro � s" � z CallableMixin._execute_mock_call) r r r r rA rR rB rn rm ro r r r r rd B s � .rd c @ r )r Nr r r r r r � r! r c C s2 zt | |�W S ty t|� t | |� Y S w r7 )r. rC � __import__)�thing�comp� import_pathr r r � _dot_lookup� s �ru c C sB | � d�}|�d�}t|�}|D ]}|d| 7 }t|||�}q|S )Nr� r z.%s)r r rq ru )�targetZ componentsrt rr rs r r r � _importer� s rw c C s( d}|D ] }|| v rt |�d���qd S )N)Z autospectZ auto_specZset_specz5 might be a typo; use unsafe=True if this is intended)� RuntimeError)Zkwargs_to_checkZtyposZtypor r r �_check_spec_arg_typos� s ���ry c @ s� e Zd ZdZg Zdd�dd�Zdd� Zdd � Zd d � Ze j d d � �Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� ZdS ) �_patchNF�r� c C s� |d ur|t ur td��|d urtd��| st| � t|�r)td|�d|�d���t|�r8td|�d|�d���|| _|| _|| _|| _|| _ || _ d| _ || _ || _ | | _g | _d S )Nz,Cannot use 'new' and 'new_callable' togetherz1Cannot use 'autospec' and 'new_callable' togetherzCannot spec attr z0 as the spec has already been mocked out. [spec=r z? as the spec_set target has already been mocked out. [spec_set=F)r rE ry r+ r �getterr] r� � new_callabler� �createZ has_localr� �autospecrL �additional_patchers) rJ r| r] r� r� r~ r� r r} rL r� r r r rA sD ������ 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 - � �z _patch.copy.<locals>.<listcomp>) rz r| r] r� r� r~ r� r r} rL �attribute_namer� )rJ �patcherr r r r� & s ��z _patch.copyc C �2 t |t�r | �|�S t�|�r| �|�S | �|�S r7 �r, r9 �decorate_classr/ r �decorate_async_callable�decorate_callable�rJ r5 r r r rB 3 � z_patch.__call__c C sN t |�D ] }|�tj�s qt||�}t|d�sq| �� }t||||�� q|S �NrB )r� r% r � TEST_PREFIXr. r- r� r[ )rJ rf r� � attr_valuer� r r r r� ; s z_patch.decorate_classc c s� � g }t �� �4}|jD ] }|�|�}|jd ur |�|� q |jtu r'|�|� q |t |�7 }||fV W d � d S 1 s>w Y d S r7 ) � contextlib� ExitStack� patchings� enter_contextr� �updater� r r� rb )rJ �patchedrK �keywargs� extra_args� exit_stack�patchingr r r r �decoration_helperI s � � "�z_patch.decoration_helperc �> t � d�r � j��� � S t� �� ��fdd����g�_�S )Nr� c sD �� �| |��\}}� |i |��W d � S 1 sw Y d S r7 �r� �rK r� ZnewargsZ newkeywargs�r5 r� rJ r r r� ^ s � $�z)_patch.decorate_callable.<locals>.patched�r- r� r� r r� r r� r r� X � z_patch.decorate_callablec r� )Nr� c � sL ��� �| |��\}}� |i |��I d H W d � S 1 s w Y d S r7 r� r� r� r r r� o s ��$�z/_patch.decorate_async_callable.<locals>.patchedr� r� r r� r r� i r� z _patch.decorate_async_callablec C s� | � � }| j}t}d}z|j| }W n ttfy# t||t�}Y nw d}|tv r2t|t �r2d| _ | j sA|tu rAtd||f ��||fS )NFTz!%s does not have the attribute %r) r| r] r re rC �KeyErrorr. � _builtinsr, r r~ )rJ rv r' rm �localr r r � get_originalz s" � �z_patch.get_originalc C s� | j | j| j}}}| j| j}}| j}| �� | _|du r!d }|du r'd }|du r-d }|d ur9|d ur9td��|d usA|d urI|dvrItd��| � � \}}|t u �r/|d u �r/d} |du rj|}|du ri|}d }n|d urw|du rv|}d }n|du r}|}|d us�|d ur�|t u r�td��t |t �r�d} |d u r�t |�r�t} nt} i } |d ur�|} n)|d us�|d ur�|} |d ur�|} t| �r�d| v} nt| � } t | �r�t} n| r�t} |d ur�|| d< |d ur�|| d < t | t �r�t| t�r�| jr�| j| d < | �|� | di | ��}| �r.t|��r.|} |d u�r|} t| ��s t| ��s t} | �d � | d|d d �| ��|_nl|d u�r�|t u�r=td ��|t u �rFtd��t|�}|du �rQ|}t| j��rgt d| j�d| j�d|�d���t|��r�t | jd| j�}t d| j�d|�d| j�d|�d� ��t |f|| jd�|��}n|�r�td��|}|| _ || _ t!�"� | _#z9t$| j| j|� | j%d u�r�i }| j t u �r�||| j%< | j&D ]}| j#�'|�}|j t u �r�|�|� �q�|W S |W S | j(t)�*� � �s� Y 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=TruerB r� r� r' r� r� zBautospec creates the mock for you. Can't specify autospec and new.z%Can't use 'autospec' with create=Truer z: as the patch target has already been mocked out. [target=r r r r r )r� �_namez.Can't pass kwargs to a mock we aren't creatingr )+r� r� r� r rL r} r| rv r4 r� r r, r9 r2 r r rc r� r r8 r r] r� r+ rh r r� �boolr r. r � temp_original�is_localr� r� � _exit_stackr[ r� r� r� �__exit__�sys�exc_info)rJ r� r� r� r rL r} rm r� Zinherit�Klass�_kwargsZ this_specZ not_callabler Znew_attrr� r� r r r r � __enter__� s �� �� � �� � ��� ������ ��z_patch.__enter__c G s� | j r| jturt| j| j| j� n t| j| j� | js2t| j| j�r)| jdv r2t| j| j| j� | `| ` | `| j }| ` |j |� S )N)rW r rY �__annotations__rZ ) r� r� r r[ rv r] r& r~ r- r� r� )rJ r� r� r r r r� $ s z_patch.__exit__c C s | � � }| j�| � |S r7 )r� �_active_patchesr� �rJ rn r r r �start9 � z _patch.startc C s6 z| j �| � W n ty Y d S w | �d d d �S r7 )r� rY rE r� r� r r r �stop@ � �z _patch.stop)r r r r� r� rA r� rB r� r� �contextmanagerr� r� r� r� r� r� r� r� r r r r rz � s$ �% rz c sL z � � dd�\� }W n tttfy td� ����w � fdd�}||fS )Nr� r� z,Need a valid target to patch. You supplied: c � t � �S r7 �rw r �rv r r r R � z _get_target.<locals>.<lambda>)�rsplitr4 rE rC )rv r] r| r r� r � _get_targetL s �� r� r{ c sD t � �tu rt� �d���� fdd�} t| |||||||| |d� S )Nz3 must be the actual object to be patched, not a strc � � S r7 r r r� r r r l � z _patch_object.<locals>.<lambda>r{ )r9 �strr4 rz ) rv r] r� r� r~ r� r r} r� rL r| r r� r � _patch_objectV s � �r� c s� t � �tu r� fdd�}n� fdd�}|std��t|�� �}|d \} } t|| | |||||i � } | | _|dd � D ]\} } t|| | |||||i � } | | _| j�| � q;| S )Nc r� r7 r� r r� r r r � r� z!_patch_multiple.<locals>.<lambda>c r� r7 r r r� r r r � r� z=Must supply at least one keyword argument with patch.multipler r� ) r9 r� rE ra r rz r� r� r� ) rv r� r~ r� r r} rL r| r r] r� r� Z this_patcherr r� r �_patch_multiples s. � � �r� c K s* t | �\} } t| | ||||||||d� S )Nr{ )r� rz ) rv r� r� r~ r� r r} r� rL r| r] r r r r � s G �r c @ sh e Zd Z ddd�Zdd� Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z dS )� _patch_dictr Fc K s, || _ t|�| _| j�|� || _d | _d S r7 )�in_dict�dictr� r� �clear� _original)rJ r� r� r� rL r r r rA s z_patch_dict.__init__c C r� r7 r� )rJ �fr r r rB r� z_patch_dict.__call__c � t � �� �fdd��}|S )Nc s, �� � z � | i |��W ��� S ��� w r7 �r� � _unpatch_dictr �r� rJ r r �_inner% s z-_patch_dict.decorate_callable.<locals>._inner�r �rJ r� r� r r� r r� $ � z _patch_dict.decorate_callablec r� )Nc � s4 ��� � z� | i |��I d H W ��� S ��� w r7 r� r r� r r r� 1 s �z3_patch_dict.decorate_async_callable.<locals>._innerr� r� r r� r r� 0 r� z#_patch_dict.decorate_async_callablec C sX t |�D ]%}t||�}|�tj�r)t|d�r)t| j| j| j �}||�}t |||� q|S r� ) r� r. r% r r� r- r� r� r� r� r[ )rJ rf r� r� Z decoratorZ decoratedr r r r� < s � �z_patch_dict.decorate_classc C s | � � | jS r7 )r� r� r� r r r r� G s z_patch_dict.__enter__c C s� | j }t| jt�rt| j�| _| j}| j}z|�� }W n ty1 i }|D ]}|| ||< q&Y nw || _|r;t |� z|� |� W d S tyX |D ]}|| ||< qLY d S w r7 ) r� r, r� r� rw r� r� rC r� � _clear_dictr� )rJ r� r� r� rm r r r r r� M s. �� ��z_patch_dict._patch_dictc C sP | j }| j}t|� z|�|� W d S ty' |D ]}|| ||< qY d S w r7 )r� r� r� r� rC )rJ r� rm r r r r r� i s ��z_patch_dict._unpatch_dictc G s | j d ur | �� dS �NF)r� r� )rJ rK r r r r� v s z_patch_dict.__exit__c C s | � � }tj�| � |S r7 )r� rz r� r� r� r r r r� } r� z_patch_dict.startc C s6 zt j�| � W n ty Y d S w | �d d d �S r7 )rz r� rY rE r� r� r r r r� � r� z_patch_dict.stopN)r F)r r r rA rB r� r� r� r� r� r� r� r� r� r r r r r� � s r� c C s< z| � � W d S ty t| �}|D ]}| |= qY d S w r7 )r� rC ra )r� �keysr r r r r� � s ��r� c C s t tj�D ]} | �� qd S r7 )r rz r� r� )r r r r �_patch_stopall� 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 � � | ]}d | V qdS )zi%sNr �r& �nr r r rO � � � rO c c r� )zr%sNr r� r r r rO � r� > � __reversed__r� Z__getinitargs__� __format__�__getnewargs_ex__�__set__� __delete__�__get__� __setstate__r� � __setformat__�__subclasses__� __getstate__r � __getformat__� __reduce_ex__� __missing__�__getnewargs__c s � fdd�}| |_ |S )Nc s � | g|�R i |��S r7 r �rJ rK r� r4 r r �method� r� z_get_method.<locals>.method)r )r' r5 r� r r4 r r# � s r# c C s h | ]}d | �qS )r� r )r& r� r r r r( � r� > � __aexit__� __aenter__� __anext__� __aiter__> r� r! �__instancecheck__rA � __prepare__r� �__del__�__subclasscheck__c C � t �| �S r7 )r �__hash__r� r r r r � � r c C r� r7 )r �__str__r� r r r r � r� c C r� r7 )r � __sizeof__r� r r r r � r� c C s" t | �j� d| �� � dt| �� �S )N�/)r9 r r r� r� r r r r � s " )r� r� r� � __fspath__r� y �?g �?) �__lt__�__gt__�__le__�__ge__�__int__r� �__len__r� � __complex__� __float__�__bool__� __index__r� c � � fdd�}|S )Nc s$ � j j}|tur |S � | u rdS tS r@ )�__eq__r� r �NotImplemented)�other�ret_valr� r r r � s z_get_eq.<locals>.__eq__r )rJ r r r� r �_get_eq� � r c r )Nc s � j jturtS � | u rdS tS r� )�__ne__r� r r )r r� r r r s z_get_ne.<locals>.__ne__r )rJ r r r� r �_get_ne s r c r )Nc s � j j} | tu r tg �S t| �S r7 )�__iter__r� r r� �r r� r r r s z_get_iter.<locals>.__iter__r )rJ r r r� r � _get_iter r r c r )Nc s( � j j} | tu rttg ��S tt| ��S r7 )r� r� r �_AsyncIteratorr� r r� r r r� s z"_get_async_iter.<locals>.__aiter__r )rJ r� r r� r �_get_async_iter s r )r r r r� c C sf t �|t�}|tur||_d S t�|�}|d ur!|| �}||_d S t�|�}|d ur1|| �|_d S d S r7 )�_return_valuesrf r r� �_calculate_return_value�_side_effect_methodsr� )r= r� r' ZfixedZreturn_calculatorr� Z side_effectorr r r �_set_return_value* s �r c @ r� )� MagicMixinc O �* | � � tt| �j|i |�� | � � d S r7 )�_mock_set_magicsr� r rA r� r r r rA = � zMagicMixin.__init__c C s� t tB }|}t| dd �d ur,|�| j�}t� }|| }|D ]}|t| �jv r+t| |� q |tt| �j� }t| �}|D ] }t ||t || �� q;d S )Nr� ) �_magicsrb r. � intersectionr� r r9 re r& r[ � MagicProxy)rJ Z orig_magicsZ these_magicsZ remove_magicsr re r r r r C s ��zMagicMixin._mock_set_magicsN)r r r rA r r r r r r < s r c @ � e Zd Z ddd�ZdS )r Fc C � | � ||� | �� d S r7 �r� r r� r r r r� ] � z"NonCallableMagicMock.mock_add_specNri �r r r r� r r r r r [ s r c @ s e Zd Zdd� ZdS )�AsyncMagicMixinc O r r7 )r r� r" rA r� r r r rA h r zAsyncMagicMixin.__init__N�r r r rA r r r r r" g s r" c @ r )r Fc C r r7 r r� r r r r� x r zMagicMock.mock_add_specNri r! r r r r r m s r c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) r c C s || _ || _d S r7 �r' r� )rJ r' r� r r r rA � s zMagicProxy.__init__c C s8 | j }| j}|j|||d�}t|||� t|||� |S )N)r' r� r� )r' r� r� r[ r )rJ r r� �mr r r � create_mock� s � zMagicProxy.create_mockNc C s | � � S r7 )r& )rJ r1 re r r r r� � s zMagicProxy.__get__r7 )r r r rA 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|i |�� tjj| jd< d| jd< d | jd< t� | jd< ttd�}t j |_ || jd< d| jd < t � | jd < i | jd < d | jd < d S ) Nr� r Z_mock_await_countZ_mock_await_argsZ_mock_await_args_list�r� r3 r r rY rZ r� ) �superrA r� r� r� re r| r r r/ Z CO_COROUTINE�co_flagsrb )rJ rK rL � code_mock�r� r r rA � s zAsyncMockMixin.__init__c � s �t ||fdd�}| jd7 _|| _| j�|� | j}|d ur]t|�r&|�t|�sAzt|�}W n t y9 t �w t|�r@|�nt |�rP||i |��I d H }n||i |��}|t ur]|S | j t ure| jS | jd ur�t | j�rz| j|i |��I d H S | j|i |��S | jS )NTrA r� )rC r� r� r� r� r� r; r` r� � StopIteration�StopAsyncIterationr r r� r� r� )rJ rK rL r6 rp rn r r r ro � s< � �� z!AsyncMockMixin._execute_mock_callc C s* | j dkrd| jp d� d�}t|��d S )Nr � Expected r= z to have been awaited.�r� r� r: r; r r r r� � s �z AsyncMockMixin.assert_awaitedc C s2 | j dksd| jp d� d| j � d�}t|��d S �Nr� r. r= z$ to have been awaited once. Awaited � times.r/ r; r r r r� � � ��z"AsyncMockMixin.assert_awaited_oncec s� �j d u r��� ��}td|� d���� ��fdd�}��t� �fdd��}���j �}||kr@t|t�r8|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_messageTrA )r� r+ r: r7 rC r, rD )rJ rK rL rE r@ rF rG r r? r r� � s �z"AsyncMockMixin.assert_awaited_withc O s> | j dksd| jp d� d| j � d�}t|��| j|i |��S r0 )r� r� r: r� rH r r r r� � s �z'AsyncMockMixin.assert_awaited_once_withc r_ )NTrA c rI r rJ rK r� r r r r z3AsyncMockMixin.assert_any_await.<locals>.<listcomp>z%s await not found)r7 rC r, rD r� r` r+ r: ra r r� r r� s ���z AsyncMockMixin.assert_any_awaitFc s� � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|sI||vrG|d u r-d}n d�dd� |D ��}t|� d t|�� d � j� ��|�d S t|�}g }|D ]}z|�|� W qQ tyh |�|� Y qQw |rutd t |�f �|�d S ) Nc rI r rJ rK r� r r r r z4AsyncMockMixin.assert_has_awaits.<locals>.<listcomp>c s rM r7 rN r r r r rO rP z3AsyncMockMixin.assert_has_awaits.<locals>.<genexpr>c 3 rQ r7 rJ rK r� r r rO rR zAwaits not found.z,Error processing expected awaits. Errors: {}c S rS r7 rN r r r r r ' rT rU z Actual: z %r not all found in await list) r� r| r� rW r: ra rY rE r� rb ) rJ rZ r[ rE rG Z all_awaitsr\ r] r^ r r� r r� sP ������� � ���z AsyncMockMixin.assert_has_awaitsc C s2 | j dkrd| jp d� d| j � d�}t|��d S )Nr r. r= z# to not have been awaited. Awaited r1 r/ r; r r r r� = r2 z!AsyncMockMixin.assert_not_awaitedc s, t � j|i |�� d| _d | _t� | _d S �Nr )r( r r� r� r| r� rI r+ r r r F s zAsyncMockMixin.reset_mockri )r r r r� r� r� r� rA ro r� r� r� r� r� r� r� r � __classcell__r r r+ r r� � s ( , r� c @ r )r Nr r r r r r P r! r c @ r� )�_ANYc C r� r@ r �rJ r r r r r ~ r� z _ANY.__eq__c C r� r� r r7 r r r r � r� z _ANY.__ne__c C r� )Nz<ANY>r r� r r r r� � r� z _ANY.__repr__N)r r r r r r� r r r r r6 { s r6 c C s` d| }d}d� dd� |D ��}d� dd� |�� D ��}|r |}|r,|r(|d7 }||7 }|| S )Nz%s(%%s)r� z, c S s g | ]}t |��qS r )�repr)r& r 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' rK rL r- Zformatted_argsZ args_stringZ kwargs_stringr r r r) � s �r) c @ s� e Zd Z d dd�Z d dd �Zd d � ZejZd d � Zdd� Z dd� Z dd� Z e dd� �Z e dd� �Zdd� Zdd� ZdS ) rC r r� NFTc C s� d}i }t |�}|dkr|\}}}n9|dkr2|\} } t| t�r,| }t| t�r)| }n"| }n | | }}n|dkrK|\}t|t�rA|}n t|t�rI|}n|}|rUt�| ||f�S t�| |||f�S )Nr � r� r� )r� r, r� rb r� ) r� r� r' r� rB � from_kallrK rL �_len�first�secondr r r r� � s. z _Call.__new__c C s || _ || _|| _d S r7 )r� r� �_mock_from_kall)rJ r� r' r� rB r: r r r rA � s z_Call.__init__c C sR zt |�}W n ty t Y S w d}t | �dkr | \}}n| \}}}t| dd �r8t|dd �r8| j|jkr8dS d}|dkrDdi }}nU|dkrN|\}}}nK|dkrq|\} t| t�r_| }i }n:t| t�rl| }di }}n-d}| }n(|dkr�|\} } t| t�r�| }t| t�r�| i }}nd| }}n| | }}ndS |r�||kr�dS ||f||fkS ) Nr� r� r� Fr r r9 r� )r� r4 r r. r� r, rb r� ) rJ 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 rtd||fdd�S | j d }t| j ||f|| d�S )Nr� r� r� r$ �r� rC r* r r r rB s z_Call.__call__c C s2 | j d u r t|dd�S d| j |f }t|| dd�S )NF)r' r: z%s.%s)r' r� r: r? )rJ r� r' r r r r� s z_Call.__getattr__c C s |t jv rt�t �| |�S r7 )rb re rC �__getattribute__)rJ r� r r r r@ s z_Call.__getattribute__c C s. t | �dkr| \}}||fS | \}}}||fS )Nr� )r� r* r r r �_get_call_arguments$ s �z_Call._get_call_argumentsc C � | � � d S r4 �rA r� r r r rK , � z _Call.argsc C rB )Nr� rC r� r r r rL 0 rD z _Call.kwargsc C s| | j s| jpd}|�d�rd| }|S t| �dkr d}| \}}n| \}}}|s*d}n|�d�s4d| }nd| }t|||�S )Nr r� zcall%sr� zcall.%s)r> r� r% r� r) )rJ r' rK rL r r r r� 4 s z_Call.__repr__c C s< g }| }|d ur|j r|�|� |j}|d us tt|��S r7 )r> r� r� r| r )rJ Zvalsrr r r r � call_listI s � z_Call.call_list)r r� NFT)r NNFT)r r r r� rA r r r rB r� r@ rA r� rK rL r� rE r r r r rC � s( � �5 rC )r: c K sj t | �r t| �} t| t�}t| �rtd| �d���t| �}d| i} |r)d| i} n| d u r/i } | r7|r7d| d< |s=t|� | �|� t} t � | �rLi } n |rW|rTt d��t } nt | �s^t} n |rh|rht| �sht} | �d|�}|} |d u rvd } | d||| |d �| ��} t| t�r�t| | �} |r�t| � nt| | ||� |d ur�|s�| |j|< |r�|s�d |vr�t| |dd | d �| _t| �D ]w} t| �r�q�zt| | �}W n ty� Y q�w d|i}|r�d|i}t|t�s�t ||| | |�}|| j| < n6| }t| t�r�| j }t | | |�}||d< t |��r t }nt}|d|| | |d�|��}|| j| < t|||d� t|t��r2t | | |� q�| S )Nz'Cannot autospec a Mock object. [object=r r� r� Tr� zJInstance can not be True when create_autospec is mocking an async functionr' r� )r� r� r� r' r� r� )rU r� r� r� )r� r' r� r� )rT r )!rc r9 r, r+ r r6 ry r� r r/ Zisdatadescriptorrx r r` r rh r r> rq r� rV r� r r� r� r� r. rC r� r= � _must_skipr r[ )r� r� rU r� r� r� rL �is_typeZ is_async_funcr� r� r� r= r rm r� r� rT Z child_klassr r r r Y s� � �� � � � �� �r c C sx t | t�s|t| di �v rdS | j} | jD ]#}|j�|t�}|tu r$qt |tt f�r. dS t |t �r7| S dS |S )Nre F) r, r9 r. r� rd re rf r r^ r_ r> )r� r rG rf rn r r r rF � s rF c @ s e Zd Z ddd�ZdS )r� FNc C s( || _ || _|| _|| _|| _|| _d S r7 )r� �idsr� r� rU r' )rJ r� r� r� r' rH rU r r r rA s z_SpecState.__init__)FNNNFr# r r r r r� s �r� c C s t | t�r t�| �S t�| �S r7 )r, �bytes�io�BytesIO�StringIO)� read_datar r r � _to_stream s rN r� c s( t ��}|d g���fdd�}��fdd�}� �fdd����fdd�� ��fd d �}td u rEd d l}ttt|j���tt|j����a| d u rOt d t d �} t td����j _ d �j _ d �j_ d �j_ d �j_ |�j_�� �d< �d �j_|�j_� �j_|�j_����fdd�}|| _�| _ | S )Nc �( �j jd ur �j jS � d j | i |��S r4 )� readlinesr� ri ��_state�handler r �_readlines_side_effect. � z)mock_open.<locals>._readlines_side_effectc rO r4 )�readr� ri rQ r r �_read_side_effect3 rU z$mock_open.<locals>._read_side_effectc ? s( � � � E d H �d j | i |��V q�NTr )�readlineri )�_iter_side_effectrR r r �_readline_side_effect8 s � �z(mock_open.<locals>._readline_side_effectc 3 s4 � �j jd ur �j jV q� d D ]} | V qd S rX )rY r� )�linerQ r r rZ = s � � �z$mock_open.<locals>._iter_side_effectc s �j jd ur �j jS t� d �S r4 )rY r� r� r rQ r r �_next_side_effectD s z$mock_open.<locals>._next_side_effectr �open)r' r� )r� r� c s6 t ���d< �jj�d kr� � �d< �d �j_tS )Nr r� )rN rY r� r ri )r[ rR rS rM r r � reset_data` s z mock_open.<locals>.reset_data)rN � file_spec�_iora r r� � TextIOWrapper�unionrK r r^ r� r� �writerV rY rP r� r r� )r= rM Z _read_datarT rW r] ra r_ r )rZ r[ rR rS rM r r s: " r c @ s( e Zd Z dd� Zddd�Zdd� ZdS ) r c K s t di |��S )Nr )r )rJ rL r r r r� v r� z PropertyMock._get_child_mockNc C s | � S r7 r )rJ r1 Zobj_typer r r r� y s zPropertyMock.__get__c C s | |� d S r7 r )rJ r1 r r r r r� { s zPropertyMock.__set__r7 )r r r r� r� r� r r r r r m s r c C sn d| _ t| �D ],}zt| |�}W n ty Y qw t|t�s!qt|j�|�t�r+q|j | u r4t |� qd S r@ ) r� r� r. rC r, r r� rf r� r� r )r= r� r% r r r r s � ��r c @ s e Zd Z dd� Zdd� ZdS )r c C s& || _ ttd�}tj|_|| jd< d S )Nr' r3 )�iteratorr r r/ ZCO_ITERABLE_COROUTINEr) re )rJ re r* r r r rA � s z_AsyncIterator.__init__c � s$ �zt | j�W S ty Y t�w r7 )r� re r, r- r� r r r r� � s � �z_AsyncIterator.__anext__N)r r r rA r� r r r r r � s r ri )NFNNN)FFNNr� )��__all__r� r� rJ r/ r� r� �builtinsr �typesr r r Z unittest.utilr � functoolsr r Z threadingr rD r r� r� r r( r� r2 r6 r+ r; r? rF rV rQ r` rc rh rq rl r� r� r r� r� r r �MISSINGr' ZDELETEDr� r� r� ra r| r� r� r� r rD rA r� r` r� rd r ru rw ry rz r� r� r� r r� r� r� r� ZmultipleZstopallr� Z magic_methodsZnumericsr r Zinplace�rightZ _non_defaultsr# r rb Z_sync_async_magicsZ _async_magicsrc r r" r r r r r r r r r r r r" r r r� r r6 r r) rb rC r r rF r� r9 r r> r` rN r r r r r r r r �<module> sD 1 h4 R � � �2� �R �� ��� � <+ :� � � N