File: //usr/local/lib/python3.7/unittest/test/testmock/__pycache__/testmagicmethods.cpython-37.pyc
B
"��gL8 � @ sJ d dl Z d dlZd dlmZmZmZ G dd� de j�ZedkrFe � � dS )� N)�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� Ze� d1e j kd2�d3d4� �Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCS )D�TestMockingMagicMethodsc C sF t � }| �t|d�� t � |_| �t|d�� |`| �t|d�� d S )N� __getitem__)r � assertFalse�hasattrr �
assertTrue)�self�mock� r �C/usr/local/lib/python3.7/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 )Nr 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 )�mock2r r
r 5 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_mocks/ 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 )Nr r )�sr r r
r ; r z3TestMockingMagicMethods.test_repr.<locals>.<lambda>r )r r �repr�id�__repr__)r
r r r r
� test_repr8 s
z!TestMockingMagicMethods.test_reprc C s: t � }| �t|�t�|�� dd� |_| �t|�d� d S )Nc S s dS )Nr r )r r r r
r B r z2TestMockingMagicMethods.test_str.<locals>.<lambda>r )r r �str�object�__str__)r
r r r r
�test_str? 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 )Nr r r )r r r
r I r z;TestMockingMagicMethods.test_dict_methods.<locals>.<lambda>c s
� d= d S )Nr r r )r r r
�_delJ s z7TestMockingMagicMethods.test_dict_methods.<locals>._delc s d� d< d S )N� r r r )r r r
�_setL s z7TestMockingMagicMethods.test_dict_methods.<locals>._setc s � | S )Nr )r r )�_dictr r
�getitemR s z:TestMockingMagicMethods.test_dict_methods.<locals>.getitemc s |� |< d S )Nr )r r �value)r, r r
�setitemT s z:TestMockingMagicMethods.test_dict_methods.<locals>.setitemc s
� |= d S )Nr )r r )r, r r
�delitemV s z:TestMockingMagicMethods.test_dict_methods.<locals>.delitemc s � d S )Nr r r )r r r
r ] r �barr )r r r � __setitem__r � __delitem__�KeyErrorr )r
r) r+ r- r/ r0 r )r, r r
�test_dict_methodsF 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 )Nr* r r )r r r
r i r z6TestMockingMagicMethods.test_numeric.<locals>.<lambda>c s � j |7 _ | S )N)r. )r
�other)r r r
�addk s z1TestMockingMagicMethods.test_numeric.<locals>.addr* c S s | d7 } d S )Nr* r )r r r r
�iadds s z2TestMockingMagicMethods.test_numeric.<locals>.iadd� � c s d� S )Nr* r r )r r r
r { r � � )r r. r r �__add__r �__iadd__�__radd__)r
�originalr7 r8 r )r r
� test_numerice 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 )N)r. )r
r6 )r r r
�truediv� s z6TestMockingMagicMethods.test_division.<locals>.truedivrC r< c S s | d } d S )N� r )r r r r
�itruediv� s z7TestMockingMagicMethods.test_division.<locals>.itruediv� c s d� S )NrG r r )r r r
r � r g �?rE )r r. r r � __truediv__r � __itruediv__� __rtruediv__)r
r@ rD rF r )r r
�
test_division� s$
z%TestMockingMagicMethods.test_divisionc C s> t � }| �t|�t �|�� dd� }||_| �t|�d� d S )Nc S s dS )Nr* r )r r r r
�_hash� s z0TestMockingMagicMethods.test_hash.<locals>._hashr* )r r �hash�__hash__)r
r rL 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>.compr* c S s t � t� k S )N)r r& r r r r
r � r z9TestMockingMagicMethods.test_comparison.<locals>.<lambda>c S s t � t� k S )N)r& r r r r r
r � r c S s t � t � k S )N)r r r r r
r � r c S s t � t� kS )N)r r& r r r r
r � r c S s t � t� kS )N)r& r r r r r
r � r c S s t � t � kS )N)r r r r r
r � r c S s t � t� kS )N)r r& r r r r
r � r c S s t � t� kS )N)r& r r r r r
r � r c S s t � t � kS )N)r r r r r
r � r c S s t � t� kS )N)r r& r r r r
r � r c S s t � t� kS )N)r& r r r r r
r � r c S s t � t � kS )N)r r r r r
r � r )r �__lt__�__gt__�__le__�__ge__r r r )r
r rU r r r
�test_comparison� s& z'TestMockingMagicMethods.test_comparisonc C s( x�t � t� fD ]�}| �||kd� | �||kt� | �||kd� | �||kt� | �|t� kd� | �|t� kd� dd� }||_| �|dk� | �|dk� dd� }||_ | �|dk� | �|dk� qW t� }d|j_
| �|dkt� | �|dkd� d|j _
| �|dkt� | �|dkd� d S ) NTFc S s |dkS )Nr* r )r
r6 r r r
�eq� s z1TestMockingMagicMethods.test_equality.<locals>.eqr* rE c S s |dkS )Nr* r )r
r6 r r r
�ne� s z1TestMockingMagicMethods.test_equality.<locals>.ne) r r r �assertIsInstancerP r& �__eq__r r �__ne__r )r
r r[ r\ 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 )Nr r r )r r r
r � r z@TestMockingMagicMethods.test_len_contains_iter.<locals>.<lambda>c S s dS )Nr9 r )r r r r
r � r r9 c S s |dkS )Nr* r )r rT r r r
r � r r* c S s t d�S )N� foobarbaz)r )r r r r
r � r ra ) 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 |�� x t
D ]}| � t||�� q^W | �t|d�� d S )N� rC r* FrQ Z __nonzero__Z
__imaginary__) r r r r r r �getattrr r rP 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 r] r& rP r )r
r r r r
�test_magic_mock_equality s z0TestMockingMagicMethods.test_magic_mock_equalityc C s� t � }| �t|�d� | �t|�d� | �t|�d� | �t� |� | �t|�d� | �t|�g � | �t |�t�
|�� | �t |�t� |�� | �
t|�� | �t|�d� | �t|�d� d S )Nrg y �?g �?r Z0o1Z0x1)r r �int�complex�floatre r& rb r rM rN r% r' r rP �oct�hex)r
r r r r
�test_magicmock_defaults s z/TestMockingMagicMethods.test_magicmock_defaultsc 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 )Nr )r
r r r
r % s zNTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.Iterable.__iter__N)�__name__�
__module__� __qualname__r r r r r
�Iterable$ s ru )�specc s � j S )N)r r )r r r
r ) r zETestMockingMagicMethods.test_magic_methods_and_spec.<locals>.<lambda>)r c @ s e Zd ZdS )zHTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.NonIterableN)rr rs rt r r r r
� NonIterable. s rw c s � j S )N)r r )r r r
r 1 r c s t tg �d�� _d S )N)r )r r �__int__r )r r r
�set_int3 s zDTestMockingMagicMethods.test_magic_methods_and_spec.<locals>.set_int) r& r r �AttributeErrorr r r r r )r
ru rw ry r )r r
�test_magic_methods_and_spec# 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 )Nr )r
r r r
r > s zRTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.Iterable.__iter__N)rr rs rt r r r r r
ru = s ru )Zspec_setc s � j S )N)r r )r r r
r B r zITestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.<lambda>)r c @ s e Zd ZdS )zLTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.NonIterableN)rr rs rt r r r r
rw G s rw c s � j S )N)r r )r r r
r J r c s t tg �d�� _d S )N)r )r r rx r )r r r
ry L s zHTestMockingMagicMethods.test_magic_methods_and_spec_set.<locals>.set_int) r& r r rz r r r r r )r
ru rw ry r )r r
� test_magic_methods_and_spec_set< s
z7TestMockingMagicMethods.test_magic_methods_and_spec_setc s$ t � � � fdd�}| �td|� d S )Nc s dd� � _ d S )Nc S s d S )Nr )r
r r r r
r X r zdTestMockingMagicMethods.test_setting_unsupported_magic_method.<locals>.set_setattr.<locals>.<lambda>)� __setattr__r )r r r
� set_setattrW s zRTestMockingMagicMethods.test_setting_unsupported_magic_method.<locals>.set_setattrz9Attempting to set unsupported magic method '__setattr__'.)r ZassertRaisesRegexrz )r
r~ r )r r
�%test_setting_unsupported_magic_methodU s
z=TestMockingMagicMethods.test_setting_unsupported_magic_methodc C s> t � }|j}dd� }| �||�t � |� }| �||�t � d S )Nc S s t | �jd S )Nrg )�type�__mro__)�objr r r
� _get_typeb 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 )Nr* rg rC )r r] r r r )r
r r r r
�"test_magic_methods_are_magic_mocksl s z:TestMockingMagicMethods.test_magic_methods_are_magic_mocksc C s6 t � }t|� | �|jj� |�� | �|jj� d S )N)r r% r r' ZcalledZ
reset_mockr )r
r r r r
� test_magic_method_reset_mockt s
z4TestMockingMagicMethods.test_magic_method_reset_mockc C s: x4t � t� fD ]$}dd� }||_| �t|�dg� qW d S )Nc S s dgS )Nr 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 )Nr* )r r r r r )r
rR 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>.FooN)rr rs rt r r r r
�Foo� s r� )r r] rx )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 )Nr )r r� r' r r r% )r
rR 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 )Nrg rC r* rE � r9 )r r r r r r )r
rR 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 )Nrg �* i� � )r r] �
__matmul__r � __rmatmul__� __imatmul__r )r
rR 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� )rC rg rC )r r] �divmod�
__divmod__r r � __rdivmod__)r
rR r Z
foo_directr1 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 )Nz__str__.return_value�12)r r r% )r
rR 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 )Nz__str__.return_valuer� Z13Z14)r r' r r r% Zconfigure_mock)r
rR r r r
�)test_changing_magic_set_in_initialization� s zATestMockingMagicMethods.test_changing_magic_set_in_initializationN)'rr rs rt r r r r r$ r( r5 rA rK rO rS rZ r` rf rj rk rq r{ r| r r� r� r� r� �unittestZskipIf�sys�versionr� r� r� r� r� r� r� r� r r r r
r s@
r �__main__)
r� r� Z
unittest.mockr r r ZTestCaser rr �mainr r r r
�<module> s O