← Back
Editing: mock.cpython-37.opt-2.pyc
B YI~d�C � @ s\ d Z dZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z dd� ee�D �ZefZdejkr�ddlZeejjfZd ZeZd d � Zd d � Zdd� Zdd� Z d�dd�Z dd� Z dd� Z dd� Z dd � Z!d�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(� Z.d)d*d+d,d-d.d/d0hZ/d1d2� Z0G d3d4� d4e1�Z2d5d6� Z3G d7d8� d8e%�Z4G d9d:� d:e%�Z5G d;d<� d<e5�Z6d=d>� Z7G d?d@� d@e5�Z8G dAdB� dBe8e6�Z9dCdD� Z:dEdF� Z;dGdH� Z<G dIdJ� dJe%�Z=dKdL� Z>e)dddddfdMdN�Z?d�dOdP�Z@e)dddddfdQdR�ZAG dSdT� dTe%�ZBdUdV� ZCdWdX� ZDe?eA_%eBeA_Ee@eA_FeDeA_GdYeA_HdZZId[ZJd\�Kd]d^� eJ�L� D ��ZMd\�Kd_d^� eJ�L� D ��ZNd`dadbdcdddedfdgdhdidjdkdldmdndodpdqhZOdrds� ZPdtd� d\�KeIeJeMeNg��L� D �ZQeQeOB ZRdudvdwdxdydzd{d|hZSd}d~� dd~� d�d~� d��ZTeUeUeUeUd�dddd�d�d d�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYeWeXeYd��ZZd�d�� Z[G d�d�� d�e%�Z\G d�d�� d�e\e6�Z]G d�d�� d�e\e9�Z^G d�d�� d�e%�Z_G d�d�� d�e%�Z`e`� Zad�d�� ZbG d�d�� d�ec�Zdeddd��Zed�d�d��Zfd�d�� Zgd�d�� ZhG d�d�� d�e%�Ziejef�ejeajk�fZldamd�d�� Znd�d�d��ZoG d�d�� d�e9�Zpd�d�� ZqdS )�)�Mock� MagicMock�patch�sentinel�DEFAULT�ANY�call�create_autospec� FILTER_DIR�NonCallableMock�NonCallableMagicMock� mock_open� PropertyMock�sealz1.0� N)� ModuleType� MethodType)�wraps�partialc C s h | ]}|� d �s|�qS )�_)� startswith)�.0�name� r �2/opt/alt/python37/lib64/python3.7/unittest/mock.py� <setcomp>& s r �javaTc C s t t| �t�S )N)� issubclass�typer )�objr r r �_is_instance_mock5 s r c C s t | t�p t | t�o t| t�S )N)� isinstance�BaseExceptionsr r )r r r r � _is_exception; s r" c C s" t | t�rt| d�r| jS | S d S )N�mock)r � FunctionTypes�hasattrr# )r r r r � _extract_mockB s r& c C s� t | t�r4|s4y | j} W n tk r, d S X d}n*t | t�s^y | j} W n tk r\ d S X |rnt| d �}n| }y| t�|�fS t k r� d S X d S )NT) r r �__init__�AttributeErrorr$ �__call__r �inspectZ signature� ValueError)�funcZ as_instanceZeat_selfZsig_funcr r r �_get_signature_objectK s$ r- Fc sN t | ||�� � d krd S � \} � � fdd�}t| |� |t|�_� t|�_d S )Nc s � j ||� d S )N)�bind)� _mock_self�args�kwargs)�sigr r �checksigp s z"_check_signature.<locals>.checksig)r- �_copy_func_detailsr �_mock_check_sig� __signature__)r, r# � skipfirst�instancer3 r )r2 r �_check_signaturek s r9 c C s> x8dD ]0}yt ||t| |�� W q tk r4 Y qX qW d S )N)�__name__�__doc__�__text_signature__� __module__� __defaults__�__kwdefaults__)�setattr�getattrr( )r, �funcopy� attributer r r r4 w s r4 c C s@ t | t�rdS t | tttf�r(t| j�S t| dd �d k r<dS dS )NTr) F)r r � staticmethod� classmethodr � _callable�__func__rA )r r r r rF � s rF c C s t | �ttfkS )N)r �list�tuple)r r r r �_is_list� s rJ c C sH t | t�st| dd �d k S x(| f| j D ]}|j�d�d k r(dS q(W dS )Nr) TF)r r rA �__mro__�__dict__�get)r �baser r r �_instance_callable� s rO c s� t |�s d S t|t�}t|||�}|d kr.| S |\}� � fdd�}t||� |j}|�� s^d}|| d�}d| } t| |� || } t| | � � | S )Nc s � j | |� d S )N)r. )r0 r1 )r2 r r r3 � s z _set_signature.<locals>.checksigrB )Z _checksig_r# zYdef %s(*args, **kwargs): _checksig_(*args, **kwargs) return mock(*args, **kwargs)) rF r r r- r4 r: � isidentifier�exec� _setup_func) r# �originalr8 r7 �resultr, r3 r �context�srcrB r )r2 r �_set_signature� s$ rW c s� �� _ t��sd 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 )N)�assert_called_with)r0 r1 )r# r r rX � s z'_setup_func.<locals>.assert_called_withc s � j | |�S )N)� assert_called)r0 r1 )r# r r rY � s z"_setup_func.<locals>.assert_calledc s � j | |�S )N)�assert_not_called)r0 r1 )r# r r rZ � s z&_setup_func.<locals>.assert_not_calledc s � j | |�S )N)�assert_called_once)r0 r1 )r# r r r[ � s z'_setup_func.<locals>.assert_called_oncec s � j | |�S )N)�assert_called_once_with)r0 r1 )r# r r r\ � s z,_setup_func.<locals>.assert_called_once_withc s � j | |�S )N)�assert_has_calls)r0 r1 )r# r r r] � s z%_setup_func.<locals>.assert_has_callsc s � j | |�S )N)�assert_any_call)r0 r1 )r# r r r^ � s z$_setup_func.<locals>.assert_any_callc s: t � � _t � � _��� � j} t| �r6| �k r6| �� d S )N)� _CallList� method_calls� mock_calls� reset_mock� return_valuer )�ret)rB r# r r rb � s z _setup_func.<locals>.reset_mockFr )r# r �called� call_count� call_argsr_ �call_args_listr` ra rc � side_effect�_mock_childrenrX r\ r] r^ rb rY rZ r[ r6 �_mock_delegate) rB r# r2 rX rY rZ r[ r\ r] r^ rb r )rB r# r rR � s< rR c C s d| dd� | kS )Nz__%s__� ���r )r r r r � _is_magic� s rn c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_SentinelObjectc C s || _ d S )N)r )�selfr r r r r' � s z_SentinelObject.__init__c C s d| j S )Nz sentinel.%s)r )rp r r r �__repr__� s z_SentinelObject.__repr__c C s d| j S )Nz sentinel.%s)r )rp r r r � __reduce__ s z_SentinelObject.__reduce__N)r: r= � __qualname__r' rq rr r r r r ro � s ro c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _Sentinelc C s i | _ d S )N)� _sentinels)rp r r r r' s z_Sentinel.__init__c C s |dkr t �| j�|t|��S )N� __bases__)r( ru � setdefaultro )rp r r r r � __getattr__ s z_Sentinel.__getattr__c C s dS )Nr r )rp r r r rr s z_Sentinel.__reduce__N)r: r= rs r' rx rr r r r r rt s rt c C s$ t | �ttttfkr t | �| �S | S )N)r �dictrH rI �set)�valuer r r �_copy s r| rc �_mock_return_valueri �_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 )N)rk rA )rp r � _the_namer2 r r r �_get+ s z"_delegating_property.<locals>._getc S s* | j }|d kr|| j|< n t|||� d S )N)rk rL r@ )rp r{ r r� r2 r r r �_set0 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 )r_ c C sn t |t�st�| |�S t|�}t| �}||kr2dS x6td|| d �D ] }| ||| � }||krFdS qFW dS )NFr � T)r rH � __contains__�len�range)rp r{ Z len_valueZlen_self�iZsub_listr r r r� = s z_CallList.__contains__c C s t �t| ��S )N)�pprintZpformatrH )rp r r r rq K s z_CallList.__repr__N)r: r= rs r� rq r r r r r_ ; s r_ c C s� t |�}t|�sdS |js4|js4|jd k s4|jd k r8dS | }x |d k rZ||krRdS |j}q>W |rl| |_||_|r|| |_||_dS )NFT)r& r r� r� r r� )�parentr{ r �new_name�_parentr r r �_check_and_set_parentO s& r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _MockIterc C s t |�| _d S )N)�iterr )rp r r r r r' k s z_MockIter.__init__c C s | S )Nr )rp r r r �__iter__m s z_MockIter.__iter__c C s t | j�S )N)�nextr )rp r r r �__next__o s z_MockIter.__next__N)r: r= rs r' r� r� r r r r r� j s r� c @ s e Zd ZeZdZdd� ZdS )�BaseNc O s d S )Nr )rp r0 r1 r r r r' u s z Base.__init__)r: r= rs r r} r~ r' r r r r r� r s r� c @ sT e Zd Zdd� ZdGdd�Zdd � ZdHd d �ZdId d �Zdd� Zdd� Z dZ e ee e �Z e dd� �Z ed�Zed�Zed�Zed�Zed�Zdd� Zd d � Ze ee�ZdJddd �d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;d<� Z%d=d>� Z&d?d@� Z'dKdAdB�Z(dCdD� Z)dEdF� Z*dS )Lr c O s$ t | j| fd| ji�}t�|�}|S )Nr; )r r: r; �object�__new__)�clsr0 �kw�newr8 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_sealedTrj � _mock_wrapsrk Z _mock_calledZ_mock_call_argsr Z_mock_call_countZ_mock_call_args_listZ_mock_mock_callsr` � _mock_unsafe)rL �_mock_add_specr_ �configure_mock� _safe_superr r' )rp �specr r �spec_setr� � _spec_state� _new_name� _new_parent�_spec_as_instance� _eat_selfZunsafer1 rL r r r r' � s: zNonCallableMock.__init__c C s0 t |�}d |_d |_d|_d |_t| ||� d S )Nr� )r& r r� r� r� r@ )rp r# rC Z inner_mockr r r � attach_mock� s zNonCallableMock.attach_mockc C s | � ||� d S )N)r� )rp r� r� r r r � mock_add_spec� s z NonCallableMock.mock_add_specc C sz d }d }|d k rPt |�sPt|t�r(|}nt|�}t|||�}|oF|d }t|�}| j}||d< ||d< ||d< ||d< d S )Nr� � _spec_class� _spec_set�_spec_signature� _mock_methods)rJ r r � _get_classr- �dirrL ) rp r� r� r� r� r� r� ZresrL r r r r� � s z NonCallableMock._mock_add_specc C s8 | j }| jd k r| jj}|tkr4| j| dd�}|| _|S )Nz())r� r� )r} rk rc r �_get_child_mock)rp rd 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 )Nz())rk rc r} r� )rp 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 )N)r� r )rp r r r � __class__� s zNonCallableMock.__class__re rf rg rh ra c C sN | j }|d kr| jS |j}|d k rJt|�sJt|t�sJt|�sJt|�}||_|S )N)rk r~ ri �callabler r� r" )rp � delegatedZsfr r r Z__get_side_effect s z!NonCallableMock.__get_side_effectc C s( t |�}| j}|d kr || _n||_d S )N)� _try_iterrk r~ ri )rp r{ r� r r r Z__set_side_effect s z!NonCallableMock.__set_side_effect)rc ri c C s� |d kr g }t | �|kr d S |�t | �� d| _d | _d| _t� | _t� | _t� | _|r^t | _ |rhd | _ x0| j � � D ]"}t|t�st|tkr�qt|�|� qtW | j }t|�r�|| k r�|�|� d S )NFr )�id�appendre rg rf r_ ra rh r` r r} r~ rj �valuesr � _SpecState�_deletedrb r )rp Zvisitedrc ri �childrd r r r rb s, zNonCallableMock.reset_mockc K s` xZt |�� dd� d�D ]B\}}|�d�}|�� }| }x|D ]}t||�}q:W t|||� qW d S )Nc S s | d � d�S )Nr �.)�count)�entryr r r �<lambda>I � z0NonCallableMock.configure_mock.<locals>.<lambda>)�keyr� )�sorted�items�split�poprA r@ )rp r1 �arg�valr0 �finalr r� r r r r� < s z NonCallableMock.configure_mockc C s� |dkrt |��n:| jd k r<|| jks.|tkrLt d| ��nt|�rLt |��| jsd|�d�rdt |��| 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)�assertZassret)r� r r r� r� )r( r� � _all_magicsrn r� r rj rM r� r� rA r� r r� r r� r� r8 r� r )rp r rT r r r r rx R s6 zNonCallableMock.__getattr__c C s� | j g}| j}| }d}|dgkr$d}t� }xT|d k r~|}|�|j | � d}|j dkrZd}|j}t|�|krnP |�t|�� q,W tt|��}|jp�d}t |�dkr�|d dkr�|d7 }||d< d� |�S )Nr� z()r� r# r� )z()z().r ) r� r� rz r� r� r� rH �reversedr� r� �join)rp Z _name_listr� Zlast�dot�seenZ_firstr r r �_extract_mock_namex s0 z"NonCallableMock._extract_mock_namec C s^ | � � }d}|dkr d| }d}| jd k rDd}| jr8d}|| 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: r r� )rp r Z name_stringZ spec_stringr r r rq � 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 � <listcomp>� 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 rn )r r� r r r r� � s ) r r� �__dir__r� r� r rH rL rj r� r� rz )rp �extrasZ 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 )Nr )r0 r� )rS rp r r r� � r� z-NonCallableMock.__setattr__.<locals>.<lambda>r� r� z Cannot set )r� r� � __setattr__r� r� rL r( �_unsupported_magicsr� r r@ r � _get_methodr� rj r� r� r% r� )rp r r{ �msg� mock_namer )rS rp r r� � s8 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 )N) r� r rL �delattrrj rM �_missingr� r � __delattr__r� r( )rp r r r r r r� � s zNonCallableMock.__delattr__c C s | j pd}t|||�S )Nr# )r� �_format_call_signature)rp r0 r1 r r r r �_format_mock_call_signature� s z+NonCallableMock._format_mock_call_signaturec C sD d}| � ||�}| j}t|�dkr.|dd � }| j |� }|||f S )Nz!Expected call: %s Actual call: %s� r� )r� rg r� )rp r0 r1 �message�expected_stringrg 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}x@|D ]8}|�|�}|d ksLt|t�rPP q,t|�}|j}|j }q,W |S )Nz()r� r� )r� �replacer� rj rM r r� r& )rp r r2 �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 |\}}}y||j||�fS tk r� } z |�d �S d }~X Y q�X n|S d S )Nrl r r� )r rI r� r� r� r. � TypeError�with_traceback)rp �_callr2 r r0 r1 r� r r r � _call_matcher( s z NonCallableMock._call_matcherc C s. | }|j dkr*d|jpd|j f }t|��d S )Nr z7Expected '%s' to not have been called. Called %s times.r# )rf r� �AssertionError)r/ rp r� r r r rZ B s z!NonCallableMock.assert_not_calledc C s( | }|j dkr$d|j pd}t|��d S )Nr z"Expected '%s' to have been called.r# )rf r� r� )r/ rp r� r r r rY K s z NonCallableMock.assert_calledc C s. | }|j dks*d|jpd|j f }t|��d S )Nr� z8Expected '%s' to have been called once. Called %s times.r# )rf r� r� )r/ rp r� r r r r[ T s z"NonCallableMock.assert_called_oncec s| | ��j d kr(��� ��}td|f ��� ��fdd�}��� �f�}���j �}||krxt|t�rh|nd }t|� �|�d S )Nz Expected call: %s Not calledc s �� � ��} | S )N)r� )r� )r0 r1 rp r r �_error_messageg s z:NonCallableMock.assert_called_with.<locals>._error_message)rg r� r� r� r � Exception)r/ r0 r1 �expectedr� �actual�causer )r0 r1 rp r rX ] s z"NonCallableMock.assert_called_withc O s6 | }|j dks*d|jpd|j f }t|��|j||�S )Nr� z1Expected '%s' to be called once. Called %s times.r# )rf r� r� rX )r/ r0 r1 rp r� r r r r\ q s z'NonCallableMock.assert_called_once_withc s� � fdd�|D �}t dd� |D �d �}t� fdd�� jD ��}|s�||kr�|d krXd}nd�dd� |D ��}td |t|�� jf �|�d S t|�}g }x:|D ]2}y|�|� W q� tk r� |�|� Y q�X q�W |r�td t |�f �|�d S ) Nc s g | ]}� � |��qS r )r� )r �c)rp r r r� � s z4NonCallableMock.assert_has_calls.<locals>.<listcomp>c s s | ]}t |t�r|V qd S )N)r r� )r r� r r r � <genexpr>� s z3NonCallableMock.assert_has_calls.<locals>.<genexpr>c 3 s | ]}� � |�V qd S )N)r� )r r� )rp r r r� � s zCalls not found.z+Error processing expected calls. Errors: {}c S s g | ]}t |t�r|nd �qS )N)r r� )r r� r r r r� � s z%s Expected: %r Actual: %rz %r not all found in call list) r� r_ ra �formatr� rH �remover+ r� rI ) rp ZcallsZ any_orderr� r� Z all_callsZproblemZ not_foundZkallr )rp r r] | s4 z NonCallableMock.assert_has_callsc sZ � � ||f�}� fdd�� jD �}||krVt|t�r8|nd }� �||�}td| �|�d S )Nc s g | ]}� � |��qS r )r� )r r� )rp r r r� � s z3NonCallableMock.assert_any_call.<locals>.<listcomp>z%s call not found)r� rh r r� r� r� )rp r0 r1 r� r� r� r� r )rp r r^ � s z NonCallableMock.assert_any_callc K sx t | �}t|t�s2t|t�r"t}q<t|t�r<t}n |jd }| jrnd|krVd|d nd}| � � | }t |��|f |�S )Nr� r r� z()) r r � CallableMixinr r r r rK r� r� r( )rp r� �_type�klassrC r� r r r r� � s z NonCallableMock._get_child_mock) NNNNNNr� NFNF)F)FF)N)F)+r: r= rs r� r' r� r� r� Z"_NonCallableMock__get_return_valueZ"_NonCallableMock__set_return_valueZ"_NonCallableMock__return_value_docr� rc r� r� re rf rg rh ra Z!_NonCallableMock__get_side_effectZ!_NonCallableMock__set_side_effectri rb r� rx r� rq r� r� r� r� r� r� r� rZ rY r[ rX r\ r] r^ r� r r r r r z sT * & ' ! *r c C sF | d kr | S t | �r| S t| �r$| S yt| �S tk r@ | S X d S )N)r" rF r� r� )r r r r r� � s r� c @ sB e Zd Zddedddddddf dd�Zdd� Zdd� Zd d � ZdS ) r� Nr� c K s6 || j d< tt| �j||||||| | f| � || _d S )Nr} )rL r� r� r' ri ) rp r� ri rc r r r� r� r� r� r� r1 r r r r' � s zCallableMixin.__init__c O s d S )Nr )rp r0 r1 r r r r5 � s z CallableMixin._mock_check_sigc O s | j ||� | j||�S )N)r5 � _mock_call)r/ r0 r1 r r r r) � s zCallableMixin.__call__c O s� | }d|_ | jd7 _t||fdd�}||_|j�|� t� }|jd k }|j}|j }|dk} |j �td||f�� |j } x�| d k �r*|r�| j �t|||f�� | jd k }|r�| jd | }t|||f�} | j �| � | j �r| r�d} nd} | j dk} | j | | }| j } t | �} | |k�r P |�| � q|W |j}|d k �r�t|��rL|�n,t|��snt|�}t|��rx|�n |||�}|tk �r�|S |jtk �r�|jS |jd k �r�|j||�S |jS )NTr� )�twoz()r� r� )re rf �_Callrg rh r� rz r r� r� ra r� r` r� r� ri r" rF r� r r} rc r� )r/ r0 r1 rp r� r� Zdo_method_callsZmethod_call_nameZmock_call_nameZ is_a_callr� Zthis_mock_callr� Z_new_parent_idZeffectrT r r r r� � s` zCallableMixin._mock_call)r: r= rs r r' r5 r) r� r r r r r� � s r� c @ s e Zd ZdS )r N)r: r= rs r r r r r I s 1r c C s2 y t | |�S tk r, t|� t | |�S X d S )N)rA r( � __import__)�thing�comp� import_pathr r r � _dot_lookup~ s r c C sF | � d�}|�d�}t|�}x$|D ] }|d| 7 }t|||�}q"W |S )Nr� r z.%s)r� r� r� r )�targetZ componentsr r� r� r r r � _importer� s r c C s t | d�S )N�is_local)r% )�patcherr r r � _is_started� s r c @ sd e Zd ZdZg Zdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� 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+ �getterrC r� � new_callabler� �createZ has_localr� �autospecr1 �additional_patchers) rp r rC r� r� r r� r r r1 r r r r' � 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>) r r rC r� r� r r� r r r1 �attribute_namer )rp r r r r r � s z _patch.copyc C s t |t�r| �|�S | �|�S )N)r r �decorate_class�decorate_callable)rp r, r r r r) � s z_patch.__call__c C sR xLt |�D ]@}|�tj�s q t||�}t|d�s2q | �� }t||||�� q W |S )Nr) )r� r r � TEST_PREFIXrA r% r r@ )rp r� �attr� attr_valuer r r r r � s z_patch.decorate_classc s< t � d�r� j�| � � S t� �� �fdd���| g�_�S )N� patchingsc sr g }t �� �\}xB�jD ]8}|�|�}|jd k r:|�|� q|jtkr|�|� qW | t |�7 } � | |�S Q R X d S )N) � contextlib� ExitStackr � enter_contextr �updater� r r� rI )r0 Zkeywargs� extra_args� exit_stack�patchingr� )r, �patchedr r r � s z)_patch.decorate_callable.<locals>.patched)r% r r� r )rp r, r )r, r r r � s z_patch.decorate_callablec C s� | � � }| j}t}d}y|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) r rC r rL r( �KeyErrorrA � _builtinsr r r )rp r r rS �localr r r � get_original� s z_patch.get_originalc C sn | 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�rR|d k�rRd} |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} t } i } |d k �rF|} nN|d k �sZ|d k �r�|} |d k �rl|} t| ��r�d| k} n t| � } | �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 �r|} t| ��s0t| ��s0t} | �d � | f |d d �| ��|_nl|d k �r�|t k �rntd ��|t k�r�td��t|�}|dk�r�|}t|f|| jd�|��}n|�r�td��|}|| _|| _ t � � | _ yrt | j| j|� | j!d k �rHi }| j t k�r||| j!< x2| j"D ](}| j �#|�}|j t k�r|�|� �qW |S |S | j$t%�&� � �sd� 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 z())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� r r1 r r r r� r! r r r r rJ r� r r r rC r r rO r� rc �boolr � temp_originalr r r � _exit_stackr@ r r r �__exit__�sys�exc_info)rp r� r� r� r r1 r rS r Zinherit�Klass�_kwargsZ this_specZ not_callableZnew_attrr r r� r r r � __enter__ s� z_patch.__enter__c G s� t | �std��| jr4| jtk r4t| j| j| j� n>t| j| j� | j srt | j| j�r`| jdkrrt| j| j| j� | `| `| `| j }| ` |j |� S )Nz stop called on unstarted patcher)r; r= r> �__annotations__r? ) r � RuntimeErrorr r$ r r@ r rC r� r r% r% r& )rp r( r r r r r& � s z_patch.__exit__c C s | � � }| j�| � |S )N)r+ �_active_patchesr� )rp rT r r r �start� s z _patch.startc C s4 y| j �| � W n tk r$ Y nX | �d d d �S )N)r. r� r+ r& )rp r r r �stop� s z _patch.stop)r: r= rs r r. r' r r) r r r! r+ r& r/ r0 r r r r r � s r c sP y� � 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 )N)r r )r r r r� � r� z _get_target.<locals>.<lambda>)�rsplitr� r+ )r rC r r )r r � _get_target� s r2 c s> t � �tkrt� �d���� fdd�} t| ||||||||� S )Nz3 must be the actual object to be patched, not a strc s � S )Nr r )r r r r� � r� z _patch_object.<locals>.<lambda>)r �strr� r ) r rC r� r� r r� r r r1 r r )r r � _patch_object� s r4 c s� t � �tkr� fdd�}n � fdd�}|s2td��t|�� �}|d \} } t|| | |||||i � } | | _xB|dd � D ]2\} } t|| | |||||i � } | | _| j�| � qvW | S )Nc s t � �S )N)r r )r r r r� � r� z!_patch_multiple.<locals>.<lambda>c s � S )Nr r )r r r r� � r� z=Must supply at least one keyword argument with patch.multipler r� ) r r3 r+ rH r� r r r r� ) r r� r r� r r r1 r r� rC r� r Z this_patcherr )r r �_patch_multiple� s& r5 c K s$ t | �\}} t|| |||||||� S )N)r2 r ) r r� r� r r� r r r1 r rC r r r r s E 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 )N)�in_dictry r� r �clear� _original)rp r7 r� r8 r1 r r r r' t s z_patch_dict.__init__c s. t � t�r��� �S t� �� �fdd��}|S )Nc s" �� � z � | |�S ��� X d S )N)r6 � _unpatch_dict)r0 r� )�frp r r �_inner� s z$_patch_dict.__call__.<locals>._inner)r r r r )rp r; r< r )r; rp r r) } s z_patch_dict.__call__c C s\ xVt |�D ]J}t||�}|�tj�r t|d�r t| j| j| j �}||�}t |||� q W |S )Nr) ) r� rA r r r r% r6 r7 r� r8 r@ )rp r� r r Z decoratorZ decoratedr r r r � s z_patch_dict.decorate_classc C s | � � d S )N)r6 )rp r r r r+ � s z_patch_dict.__enter__c C s� | j }t| jt�r t| j�| _| j}| j}y |�� }W n2 tk rh i }x|D ]}|| ||< qPW Y nX || _|r|t |� y|� |� W n. tk r� x|D ]}|| ||< q�W Y nX d S )N) r� r r7 r3 r r8 r r( r9 � _clear_dictr )rp r� r7 r8 rS r� r r r r6 � s& z_patch_dict._patch_dictc C sV | j }| j}t|� y|�|� W n. tk rP x|D ]}|| ||< q8W Y nX d S )N)r7 r9 r= r r( )rp r7 rS r� r r r r: � s z_patch_dict._unpatch_dictc G s | � � dS )NF)r: )rp r0 r r r r& � s z_patch_dict.__exit__N)r F) r: r= rs r' r) r r+ r6 r: r& r/ r0 r r r r r6 W s r6 c C sB y | � � W n0 tk r< t| �}x|D ] }| |= q*W Y nX d S )N)r8 r( rH )r7 �keysr� r r r r= � s r= c C s xt tj�D ] } | �� q W d S )N)r� r r. r0 )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 trunc floor ceil bool next 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� � s r� c c s | ]}d | V qdS )zr%sNr )r rA r r r r� � s �__get__�__set__� __delete__� __reversed__� __missing__rr � __reduce_ex__Z__getinitargs__�__getnewargs__� __getstate__� __setstate__� __getformat__Z __setformat__rq r� �__subclasses__� __format__�__getnewargs_ex__c s � fdd�}| |_ |S )Nc s � | f|�|�S )Nr )rp r0 r� )r, r r �method s z_get_method.<locals>.method)r: )r r, rO r )r, r r� s r� c C s h | ] }d | �qS )z__%s__r )r rO r r r r s rx r� r' r� � __prepare__�__instancecheck__�__subclasscheck__�__del__c C s t �| �S )N)r� �__hash__)rp r r r r� r� r� c C s t �| �S )N)r� �__str__)rp r r r r� r� c C s t �| �S )N)r� � __sizeof__)rp r r r r� r� )rT rU rV r� y �?g �?) �__lt__�__gt__�__le__�__ge__�__int__r� �__len__r& � __complex__� __float__�__bool__� __index__c s � fdd�}|S )Nc s$ � j j}|tk r|S � | kr dS tS )NT)�__eq__r} r �NotImplemented)�other�ret_val)rp r r ra 0 s z_get_eq.<locals>.__eq__r )rp ra r )rp r �_get_eq/ s re c s � fdd�}|S )Nc s � j jtk rtS � | kr dS tS )NF)�__ne__r} r rb )rc )rp r r rf : s z_get_ne.<locals>.__ne__r )rp rf r )rp r �_get_ne9 s rg c s � fdd�}|S )Nc s � j j} | tkrtg �S t| �S )N)r� r} r r� )rd )rp r r r� C s z_get_iter.<locals>.__iter__r )rp r� r )rp r � _get_iterB s rh )ra rf r� c C s� t �|t�}|tk r ||_d S t�|�}|d k rdy || �}W n tk rX t|�}Y nX ||_d S t�|�}|d k r�|| �|_d S )N)�_return_valuesrM r rc �_calculate_return_valuer( �_side_effect_methodsri )r# rO r ZfixedZreturn_calculatorrc Z side_effectorr r r �_set_return_valueT s rl c @ s e Zd Zdd� Zdd� ZdS )� MagicMixinc O s& | � � tt| �j||� | � � d S )N)�_mock_set_magicsr� rm r' )rp r0 r� r r r r' l s zMagicMixin.__init__c C s� t }t| dd �d k rTt �| j�}t� }t | }x$|D ] }|t| �jkr4t| |� q4W |tt| �j� }t| �}x |D ]}t||t || �� qtW d S )Nr� ) �_magicsrA � intersectionr� rz r rL r� r@ � MagicProxy)rp Z these_magicsZ remove_magicsr� r� r r r rn r s zMagicMixin._mock_set_magicsN)r: r= rs r' rn r r r r rm k s rm c @ s e Zd Zddd�ZdS )r Fc C s | � ||� | �� d S )N)r� rn )rp r� r� r r r r� � s z"NonCallableMagicMock.mock_add_specN)F)r: r= rs r� r r r r r � s r c @ s e Zd Zddd�ZdS )r Fc C s | � ||� | �� d S )N)r� rn )rp r� r� r r r r� � s zMagicMock.mock_add_specN)F)r: r= rs r� r r r r r � s r c @ s. e Zd Zdd� Zdd� Zdd� Zd dd �ZdS ) rq c C s || _ || _d S )N)r r� )rp r r� r r r r' � s zMagicProxy.__init__c O s | � � }|||�S )N)� create_mock)rp r0 r1 �mr r r r) � s zMagicProxy.__call__c C s8 | j }| j}|j|||d�}t|||� t|||� |S )N)r r� r� )r r� r� r@ rl )rp r� r� rs r r r rr � s zMagicProxy.create_mockNc C s | � � S )N)rr )rp r r� r r r rB � s zMagicProxy.__get__)N)r: r= rs r' r) rr rB r r r r rq � s rq c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_ANYc C s dS )NTr )rp rc r r r ra � s z _ANY.__eq__c C s dS )NFr )rp rc r r r rf � s z _ANY.__ne__c C s dS )Nz<ANY>r )rp r r r rq � s z _ANY.__repr__N)r: r= rs ra rf rq r r r r rt � s rt c C sd d| }d}d� dd� |D ��}d� dd� t|�� �D ��}|rD|}|r\|rT|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� )r r0 r1 r� Zformatted_args� args_stringZ kwargs_stringr r r r� � s r� c @ s^ e Zd Zddd�Zddd �Zd d � ZejZd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dS )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� rl r� )r� r r3 rI r� ) r� r{ r r� r� � from_kallr0 r1 �_len�first�secondr r r r� � s. z _Call.__new__c C s || _ || _|| _d S )N)r� r �_mock_from_kall)rp r{ r r� r� rw r r r r' s z_Call.__init__c C sf |t kr dS y t|�}W n tk r, dS X d}t| �dkrH| \}}n | \}}}t| dd �rzt|dd �rz| j|jkrzdS d}|dkr�di }}n�|dkr�|\}}}n�|d kr�|\} t| t�r�| }i }n"t| t�r�| }di }}nd}| }nV|dk�r>|\} } t| t��r2| }t| t��r&| i }}n d| }}n | | }}ndS |�rV||k�rVdS ||f||fkS ) NTFr� rl r r r r� r� )r r� r� rA r r rI r3 ) rp rc Z len_otherZ self_nameZ self_argsZ self_kwargsZ other_nameZ other_argsZ other_kwargsr{ ry rz r r r ra sP z _Call.__eq__c O s< | j d kr td||fdd�S | j d }t| j ||f|| d�S )Nr� z())r )r r� )r� r� )rp r0 r1 r r r r r) Y s z_Call.__call__c C s2 | j d krt|dd�S d| j |f }t|| dd�S )NF)r rw z%s.%s)r r� rw )r� r� )rp r r r r r rx a s z_Call.__getattr__c O s | � d�||�S )Nr� )rx )rp r0 r1 r r r r� h s z _Call.countc O s | � d�||�S )N�index)rx )rp r0 r1 r r r r| k s z _Call.indexc C s| | j s&| jpd}|�d�r"d| }|S t| �dkr@d}| \}}n0| \}}}|sTd}n |�d�shd| }nd| }t|||�S )Nr z()zcall%srl zcall.%s)r{ r� r r� r� )rp r r0 r1 r r r rq n s z_Call.__repr__c C s8 g }| }x"|d k r*|j r"|�|� |j}q W tt|��S )N)r{ r� r r_ r� )rp Zvalsr� r r r � call_list� s z_Call.call_list)r r� NFT)r NNFT)r: r= rs r� r' ra r� rf r) rx r� r| rq r} r r r r r� � s 7r� )rw c K s$ t | �rt| �} t| t�}d| i}|r0d| i}n | d kr<i }|rL|rLd|d< |�|� t}t�| �rji }n"t| �sxt}n|r�|r�t | �s�t}|� d|�}|} |d kr�d} |f ||| |d�|��} t| t �r�t | | �} nt | | ||� |d k r�|s�| |j|< |�r$|�s$d|k�r$t| |dd | d �| _x�t| �D ]�} t| ��r@�q.yt| | �} W n tk �rh �w.Y nX d| i}|�r�d| i}t| t ��s�t| || | |�} | | j| < nZ| }t| t ��r�| j}t| | |�}||d < tf || | |d �|��} | | j| < t | | |d � t| t ��r.t| | | � �q.W | S )Nr� r� Tr� r r� )r� r� r� r rc z())r8 r" r� r� )r� r r� r� )r7 )rJ r r r r r* ZisdatadescriptorrF r rO r� r$ rW r9 rj r rc r� rn rA r( r� r# � _must_skipr@ )r� r� r8 r� r" r1 �is_typer* r) r� r# r� rS r� r� r7 r r r r � st r c C st t | t�s$|t| di �kr dS | j} xJ| jD ]@}|j�|t�}|tkrHq,t |tt f�rZdS t |t �rh|S dS q,W |S )NrL F) r r rA r� rK rL rM r rD rE r$ )r� r� r r� rT r r r r~ s r~ c C s$ y| j S tk r t| �S X d S )N)r� r( r )r r r r r� ) s r� c @ s e Zd Zddd�ZdS )r� FNc C s( || _ || _|| _|| _|| _|| _d S )N)r� �idsr� r� r8 r )rp r� r� r� r r� r8 r r r r' 3 s z_SpecState.__init__)FNNNF)r: r= rs r' r r r r r� 1 s r� c C s" t | t�rt�| �S t�| �S d S )N)r �bytes�io�BytesIO�StringIO)� read_datar r r � _to_streamH 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 )Nr )� readlinesrc )r0 r1 )�_state�handler r �_readlines_side_effect^ s z)mock_open.<locals>._readlines_side_effectc s$ �j jd k r�j jS � d j | |�S )Nr )�readrc )r0 r1 )r� r� r r �_read_side_effectc s z$mock_open.<locals>._read_side_effectc ? s( � � E d H x�d j | |�V qW d S )Nr )�readline)r0 r1 )�_iter_side_effectr� r r �_readline_side_effecth s z(mock_open.<locals>._readline_side_effectc 3 s8 �j jd k r x�j jV qW x� d D ] } | V q&W d S )Nr )r� rc )�line)r� r� r r r� m s z$mock_open.<locals>._iter_side_effectc s �j jd k r�j jS t� d �S )Nr )r� rc r� r )r� r� r r �_next_side_effectt 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� ri r )r0 r1 )r� r� r� r� r r � reset_data� s z mock_open.<locals>.reset_data)r� � file_spec�_iorH rz r� � TextIOWrapper�unionr� r r� r+ rc �writer� r� r� ri r� r� )r# r� Z _read_datar� r� r� r� r� r )r� r� r� r� r� r r O s8 " r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c K s t f |�S )N)r )rp r1 r r r r� � s z PropertyMock._get_child_mockc C s | � S )Nr )rp r Zobj_typer r r rB � s zPropertyMock.__get__c C s | |� d S )Nr )rp r r� r r r rC � s zPropertyMock.__set__N)r: r= rs r� rB rC r r r r r � s r c C s` d| _ xTt| �D ]H}yt| |�}W n tk r8 wY nX t|t�sFq|j| krt|� qW d S )NT)r� r� rA r( r r r� r )r# r rs r r r r � s r )F)F)NFNNN)FFNN)Nr� )r�__all__� __version__r� r* r� r' �builtinsr �typesr r � functoolsr r r� r � BaseExceptionr! �platformr �langZ Throwabler �superr� r r" r& r- r9 r4 rF rJ rO rW rR rn r� ro rt r r �MISSINGr� ZDELETEDr� r| r� r� rH r_ r� r� r� r r� r� r r r r r r2 r4 r5 r r6 r= r? ry ZmultipleZstopallr Z magic_methodsZnumericsr� r� Zinplace�rightZ _non_defaultsr� ro r� r� rj rb ri re rg rh rk rl rm r r rq rt r r� rI r� r r r~ r� r� r ra r$ r� r� r r r r r r r �<module> s 5 Zi5 1Jv , w N
Save File
Cancel