EVOLUTION-NINJA
Edit File: testmagicmethods.cpython-38.pyc
U �i�_�? � @ sf d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z G dd� dej �Z e dkrbe� � dS )� N)� AsyncMock�Mock� MagicMock�_magicsc @ s. e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z e � d5e j kd6�d7d8� �Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGS )H�TestMockingMagicMethodsc C sF t � }| �t|d�� t � |_| �t|d�� |`| �t|d�� d S )N� __getitem__)r � assertFalse�hasattrr � assertTrue��self�mock� r �F/opt/python38/lib/python3.8/unittest/test/testmock/testmagicmethods.py�test_deleting_magic_methods s z3TestMockingMagicMethods.test_deleting_magic_methodsc sH t � � � `| �t� fdd�� t � � � d � `| �t� fdd�� d S )Nc s � d S �N�foor r �r r r �<lambda> � z<TestMockingMagicMethods.test_magicmock_del.<locals>.<lambda>r c s � d S r r r r r r r ! r )r r � assertRaises� TypeError�r r r r �test_magicmock_del s z*TestMockingMagicMethods.test_magicmock_delc C sd t � }dd� }||_| �|j|� | �|d |df� | �|�d�|df� ||_| �|j|� d S )Nc S s | dfS )N�fishr �r �namer r r �f& s z=TestMockingMagicMethods.test_magic_method_wrapping.<locals>.fr r )r r Z assertIsNot� assertEqualZassertIs)r r r r r r �test_magic_method_wrapping$ s z2TestMockingMagicMethods.test_magic_method_wrappingc sD t � }t � � t tg �d�|_| �t|�g � | �t� fdd�� d S )N�� return_valuec s t � �S �N)�listr �Zmock2r r r 8 r zSTestMockingMagicMethods.test_magic_methods_isolated_between_mocks.<locals>.<lambda>)r �iter�__iter__r r# r r )r Zmock1r r$ r �)test_magic_methods_isolated_between_mocks2 s zATestMockingMagicMethods.test_magic_methods_isolated_between_mocksc C s< t � }| �t|�dt|� � dd� |_| �t|�d� d S )Nz<Mock id='%s'>c S s dS r r ��sr r r r > r z3TestMockingMagicMethods.test_repr.<locals>.<lambda>r )r r �repr�id�__repr__r r r r � test_repr; s z!TestMockingMagicMethods.test_reprc C s: t � }| �t|�t�|�� dd� |_| �t|�d� d S )Nc S s dS r r r( r r r r E r z2TestMockingMagicMethods.test_str.<locals>.<lambda>r )r r �str�object�__str__r r r r �test_strB s z TestMockingMagicMethods.test_strc s� t � �| �t�fdd�� �fdd�}�fdd�}| �t|� | �t|� i � � fdd�}� fd d �}� fd d �}|�_|�_|�_| �t�fd d�� d�d< | �� ddi� | ��d d� �d= | �� i � d S )Nc s � d S r r r r r r r L r z;TestMockingMagicMethods.test_dict_methods.<locals>.<lambda>c s � d= d S r r r r r r �_delM s z7TestMockingMagicMethods.test_dict_methods.<locals>._delc s d� d< d S )N� r r r r r r �_setO s z7TestMockingMagicMethods.test_dict_methods.<locals>._setc s � | S r" r �r) r ��_dictr r �getitemU s z:TestMockingMagicMethods.test_dict_methods.<locals>.getitemc s |� |< d S r" r )r) r �valuer6 r r �setitemW s z:TestMockingMagicMethods.test_dict_methods.<locals>.setitemc s � |= d S r" r r5 r6 r r �delitemY s z:TestMockingMagicMethods.test_dict_methods.<locals>.delitemc s � d S r r r r r r r ` r �barr )r r r � __setitem__r � __delitem__�KeyErrorr )r r2 r4 r8 r: r; r )r7 r r �test_dict_methodsI s& z)TestMockingMagicMethods.test_dict_methodsc s� t � }� d� _| �t� fdd�� � fdd�}|� _| �� d � � | �� jd� � `dd� }| �t|� � |� _� d 7 � | �� |� | �� jd � | �t� fd d�� |� _| �d � � � | �� jd � d S )Nr c s � d S �Nr3 r r r r r r l r z6TestMockingMagicMethods.test_numeric.<locals>.<lambda>c s � j |7 _ | S r" �r9 �r �otherr r r �addn s z1TestMockingMagicMethods.test_numeric.<locals>.addr3 c S s | d7 } d S rA r r r r r �iaddv s z2TestMockingMagicMethods.test_numeric.<locals>.iadd� � c s d� S rA r r r r r r ~ r � � )r r9 r r �__add__r �__iadd__�__radd__)r �originalrE rF r r r � test_numerich s$ z$TestMockingMagicMethods.test_numericc s� t � }� d� _| �t� fdd�� � fdd�}|� _| �� d � � | �� jd� � `dd � }| �t|� � |� _� d � | �� |� | �� jd� | �t� fd d�� |� _| �d � � � | �� jd � d S )N� c s � d S )N� r r r r r r � r z7TestMockingMagicMethods.test_division.<locals>.<lambda>c s � j | _ | S r" rB rC r r r �truediv� s z6TestMockingMagicMethods.test_division.<locals>.truedivrQ rJ c S s | d } d S )N� r r r r r �itruediv� s z7TestMockingMagicMethods.test_division.<locals>.itruediv� c s d� S )NrU r r r r r r � r g �?rS )r r9 r r � __truediv__r � __itruediv__� __rtruediv__)r rN rR rT r r r � test_division� s$ z%TestMockingMagicMethods.test_divisionc C s> t � }| �t|�t �|�� dd� }||_| �t|�d� d S )Nc S s dS rA r r( r r r �_hash� s z0TestMockingMagicMethods.test_hash.<locals>._hashr3 )r r �hash�__hash__)r r rZ r r r � test_hash� s z!TestMockingMagicMethods.test_hashc C s0 t � }| �t|�� dd� |_| �t|�� d S )Nc S s dS )NFr r( r r r r � r z6TestMockingMagicMethods.test_nonzero.<locals>.<lambda>)r r �bool�__bool__r �r �mr r r � test_nonzero� s z$TestMockingMagicMethods.test_nonzeroc C s" t � }dd� }| |_ |_ |_|_| �|dk � | �|dk� | �|dk� | �|dk� | �tdd� � | �tdd� � | �tdd� � | �tdd� � | �td d� � | �td d� � | �td d� � | �td d� � | �td d� � | �tdd� � | �tdd� � | �tdd� � d S )Nc S s dS )NTr �r) �or r r �comp� s z5TestMockingMagicMethods.test_comparison.<locals>.compr3 c S s t � t� k S r" �r r/ r r r r r � r z9TestMockingMagicMethods.test_comparison.<locals>.<lambda>c S s t � t� k S r" �r/ r r r r r r � r c S s t � t � k S r" �r r r r r r � r c S s t � t� kS r" rf r r r r r � r c S s t � t� kS r" rg r r r r r � r c S s t � t � kS r" rh r r r r r � r c S s t � t� kS r" rf r r r r r � r c S s t � t� kS r" rg r r r r r � r c S s t � t � kS r" rh r r r r r � r c S s t � t� kS r" rf r r r r r � r c S s t � t� kS r" rg r r r r r � r c S s t � t � kS r" rh r r r r r � r )r �__lt__�__gt__�__le__�__ge__r r r )r r re r r r �test_comparison� s& z'TestMockingMagicMethods.test_comparisonc C s$ t � t� fD ]�}| �||kd� | �||kt� | �||kd� | �||kt� | �|t� kd� | �|t� kd� dd� }||_| �|dk� | �|dk� dd� }||_ | �|dk� | �|dk� q t� }d|j_ | �|dkt� | �|dkd� d|j _ | �|dkt� | �|dkd� d S ) NTFc S s |dkS rA r rC r r r �eq� s z1TestMockingMagicMethods.test_equality.<locals>.eqr3 rS c S s |dkS rA r rC r r r �ne� s z1TestMockingMagicMethods.test_equality.<locals>.ne) r r r �assertIsInstancer^ r/ �__eq__r r �__ne__r! )r r rn ro r r r � test_equality� s, z%TestMockingMagicMethods.test_equalityc s� t � � | �tt� � | �tt� � | �t� fdd�� dd� � _| �t� �d� dd� � _| �d� � | � d� � dd� � _ | �t � �t d�� d S ) Nc s d� kS r r r r r r r � r z@TestMockingMagicMethods.test_len_contains_iter.<locals>.<lambda>c S s dS )NrG r r( r r r r � r rG c S s |dkS rA r rc r r r r � r r3 c S s t d�S )N� foobarbaz)r% r( r r r r � r rt ) r r r �lenr% �__len__r � __contains__ZassertIn� assertNotInr& r# r r r r �test_len_contains_iter� s z.TestMockingMagicMethods.test_len_contains_iterc C s� t � }tdddg�|j_| �t|�dddg� dt|d�_| �t|d�� | �t |�� t D ]}| � t||�� q\| �t|d�� d S )N� rQ r3 Fr_ Z __nonzero__Z __imaginary__) r r% r& r! r r# �getattrr r r^ r r )r r �entryr r r �test_magicmock� s z&TestMockingMagicMethods.test_magicmockc C sr t � }| �|t� kt� | �|t� kt� | �|t� kd� | �|t� kd� | �||kd� | �||kd� d S )NFT)r rp r/ r^ r r r r r �test_magic_mock_equality s z0TestMockingMagicMethods.test_magic_mock_equalityc C sX t � }| �t|�d� | �t|�d� | �t|�d� | �t� |� | �t|�d� | �t|�g � | �t |�t� |�� | �t |�t� |�� | � t|�� | �t|�|�� � | �t�|�|�� � | �t�|�|�� � | �t�|�|�� � | � t�|j�� | � t�|j�� | � |jt � | � |jt � | �t |�d� | �t |�d� d S �Nrz y �?g �?r Z0o1Z0x1) r r �int�complex�floatrx r/ ru r# r[ r\ r. r0 r r^ �round� __round__�math�trunc� __trunc__�floor� __floor__�ceil�__ceil__�asyncio�iscoroutinefunction� __aexit__� __aenter__rp �oct�hexr r r r �test_asyncmock_defaults s( z/TestMockingMagicMethods.test_asyncmock_defaultsc C sX t � }| �t|�d� | �t|�d� | �t|�d� | �t� |� | �t|�d� | �t|�g � | �t |�t� |�� | �t |�t� |�� | � t|�� | �t|�|�� � | �t�|�|�� � | �t�|�|�� � | �t�|�|�� � | � t�|j�� | � t�|j�� | � |jt � | � |jt � | �t |�d� | �t |�d� d S r ) r r r� r� r� rx r/ ru r# r[ r\ r. r0 r r^ r� r� r� r� r� r� r� r� r� r� r� r� r� rp r r� r� r r r r �test_magicmock_defaults, s( z/TestMockingMagicMethods.test_magicmock_defaultsc C s6 t � }|�� }|�� | �t�|�|� |j�� d S r" )r � __fspath__� reset_mockr �os�fspathZassert_called_once)r r Z expected_pathr r r �test_magic_methods_fspathG s z1TestMockingMagicMethods.test_magic_methods_fspathc s� G dd� dt �}t|d�� | �t� fdd�� ttg �d�� _| �t� �g � G dd� dt �}t|d�� | �t� fd d�� � fd d �}| �t|� t|d�� | �t� �g � | �t|� d S ) Nc @ s e Zd Zdd� ZdS )zETestMockingMagicMethods.test_magic_methods_and_spec.<locals>.Iterablec S s d S r" r r r r r r& R r zNTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.Iterable.__iter__N��__name__� __module__� __qualname__r& r r r r �IterableQ s r� )�specc s � j S r" �r& r r r r r U r zETestMockingMagicMethods.test_magic_methods_and_spec.<locals>.<lambda>r c @ s e Zd ZdS )zHTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.NonIterableN�r� r� r� r r r r � NonIterableZ s r� c s � j S r" r� r r r r r ] r c s t tg �d�� _d S �Nr �r r% �__int__r r r r �set_int_ s zDTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.set_int� r/ r r �AttributeErrorr% r& r r# r �r r� r� r� r r r �test_magic_methods_and_specP s z3TestMockingMagicMethods.test_magic_methods_and_specc s� G dd� dt �}t|d�� | �t� fdd�� ttg �d�� _| �t� �g � G dd� dt �}t|d�� | �t� fd d�� � fd d �}| �t|� t|d�� | �t� �g � | �t|� d S ) Nc @ s e Zd Zdd� ZdS )zITestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.Iterablec S s d S r" r r r r r r& j r zRTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.Iterable.__iter__Nr� r r r r r� i s r� )Zspec_setc s � j S r" r� r r r r r m r zITestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.<lambda>r c @ s e Zd ZdS )zLTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.NonIterableNr� r r r r r� r s r� c s � j S r" r� r r r r r u r c s t tg �d�� _d S r� r� r r r r r� w s zHTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.set_intr� r� r r r � test_magic_methods_and_spec_seth s z7TestMockingMagicMethods.test_magic_methods_and_spec_setc s$ t � � � fdd�}| �td|� d S )Nc s dd� � _ d S )Nc S s d S r" r r r r r r � r zdTestMockingMagicMethods.test_setting_unsupported_magic_method.<locals>.set_setattr.<locals>.<lambda>)� __setattr__r r r r � set_setattr� s zRTestMockingMagicMethods.test_setting_unsupported_magic_method.<locals>.set_setattrz9Attempting to set unsupported magic method '__setattr__'.)r ZassertRaisesRegexr� )r r� r r r �%test_setting_unsupported_magic_method� s �z=TestMockingMagicMethods.test_setting_unsupported_magic_methodc C s> t � }|j}dd� }| �||�t � |� }| �||�t � d S )Nc S s t | �jd S )Nrz )�type�__mro__)�objr r r � _get_type� s zKTestMockingMagicMethods.test_attributes_and_return_value.<locals>._get_type)r r r )r r �attrr� Zreturnedr r r � test_attributes_and_return_value� s z8TestMockingMagicMethods.test_attributes_and_return_valuec C s@ t � }| �|jt � d|d d j_| �|d d d d� d S )Nr3 rz rQ )r rp r r! r r r r r �"test_magic_methods_are_magic_mocks� s z:TestMockingMagicMethods.test_magic_methods_are_magic_mocksc C s6 t � }t|� | �|jj� |�� | �|jj� d S r" )r r. r r0 Zcalledr� r r r r r � test_magic_method_reset_mock� s z4TestMockingMagicMethods.test_magic_method_reset_mockc C s6 t � t� fD ]$}dd� }||_| �t|�dg� q d S )Nc S s dgS r r r r r r �_dir� s z.TestMockingMagicMethods.test_dir.<locals>._dirr )r r �__dir__r �dir)r r r� r r r �test_dir� s z TestMockingMagicMethods.test_dirZPyPyz This fails differently on pypyc C s" t � }dgj|_| �tt|� d S rA )r r& r r r% r` r r r �test_bound_methods� s z*TestMockingMagicMethods.test_bound_methodsc C s( G dd� dt �}|� }| �|j|� d S )Nc @ s e Zd ZdS )z;TestMockingMagicMethods.test_magic_method_type.<locals>.FooNr� r r r r �Foo� s r� )r rp r� )r r� r r r r �test_magic_method_type� s z.TestMockingMagicMethods.test_magic_method_typec C s& t � }dt|�j_| �t|�d� d S r )r r� r0 r! r r. r` r r r �test_descriptor_from_class� s z2TestMockingMagicMethods.test_descriptor_from_classc C s| t � }dddg|j_| �t|�dddg� | �t|�dddg� tdddg�|j_| �t|�dddg� | �t|�g � d S )Nrz rQ r3 rS � rG )r r& r! r r# r% r` r r r �"test_iterable_as_iter_return_value� s z:TestMockingMagicMethods.test_iterable_as_iter_return_valuec C sf t � }| �|d t � d|j_d|j_d|j_| �|d d� | �d| d� |d }| �|d� d S )Nrz �* i� � )r rp � __matmul__r! � __rmatmul__� __imatmul__r r` r r r � test_matmul� s z#TestMockingMagicMethods.test_matmulc C s� t � }| �td|�t � d|j_| �t|d�d� t � }td|�}| �|t � |�d�}| �|t � t|d�}| �|t � |�d�}| �|t � d S )Nr� )rQ rz rQ )r rp �divmod� __divmod__r! r � __rdivmod__)r ra r Z foo_directr< Z bar_directr r r �test_divmod_and_rdivmod� s z/TestMockingMagicMethods.test_divmod_and_rdivmodc C s" t f ddi�}| �t|�d� d S )N�__str__.return_value�12)r r r. r` r r r � test_magic_in_initialization� s z4TestMockingMagicMethods.test_magic_in_initializationc C sX t f ddi�}d|j_| �t|�d� t f ddi�}|jf ddi� | �t|�d� d S )Nr� r� Z13Z14)r r0 r! r r. Zconfigure_mockr` r r r �)test_changing_magic_set_in_initialization� s zATestMockingMagicMethods.test_changing_magic_set_in_initializationN))r� r� r� r r r r' r- r1 r@ rO rY r] rb rm rs ry r} r~ r� r� r� r� r� r� r� r� r� r� �unittestZskipIf�sys�versionr� r� r� r� r� r� r� r� r r r r r sF r �__main__)r� r� r� r� r� Z unittest.mockr r r r ZTestCaser r� �mainr r r r �<module> s w