Your IP : 216.73.216.213
B
� f0 � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZm Z m
Z
m Z e j Z
e jZG dd� de j�ZG dd� de j�Zed kr�e �� dS )
� N)�catch_warnings)� is_instance)� MagicMock�Mock�patch�sentinel� mock_open�callc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�WithTestc C s> t dt tj�� | �ttjd� W d Q R X | �ttj� d S )Nz %s.something� unpatched)r �__name__r �
Something2� assertEqual� something� Something)�self� r �D/opt/alt/python37/lib64/python3.7/unittest/test/testmock/testwith.py�test_with_statement s z WithTest.test_with_statementc C sj y8t dt tj�� | �ttjd� td��W d Q R X W n tk
rL Y n X | �d� | �ttj� d S )Nz %s.somethingr �powzpatch swallowed exception) r r r r
r r � ExceptionZfailr )r r r r � test_with_statement_exception s
z&WithTest.test_with_statement_exceptionc C sJ t dt ��&}| �t|d� | �t|t�d� W d Q R X | �ttj� d S )Nz %s.somethingr zpatching wrong type) r r r r �
assertTruer r r r )r �mock_somethingr r r �test_with_statement_as" s
z WithTest.test_with_statement_asc C sP G dd� dt �}|j}t� |d�� | �|j|d� W d Q R X | �|j|� d S )Nc @ s e Zd ZdZdS )z6WithTest.test_patch_object_with_statement.<locals>.Foo�fooN)r �
__module__� __qualname__r r r r r �Foo+ s r r r )�objectr r ZassertNotEqualr )r r Zoriginalr r r � test_patch_object_with_statement* s
z)WithTest.test_patch_object_with_statementc C s� t dd��R tdt ��:}tdt ��"}| �t|d� | �t|d� W d Q R X W d Q R X W d Q R X | �ttj� | �ttj� d S )NT)�recordz %s.somethingz%s.something_elser ) r r r r r �something_elser r �
SomethingElse)r r �mock_something_elser r r �test_with_statement_nested3 s $z#WithTest.test_with_statement_nestedc C sL t dt tj��$}| �t|d� | �|tjd� W d Q R X | �ttj� d S )Nz %s.somethingr z wrong patch)r r r �Patchedr r r )r r r r r � test_with_statement_specified> s z&WithTest.test_with_statement_specifiedc C s\ t � }t � |_t � |_d|j_|�}| �||jj� W d Q R X |j�� |j�d d d � d S )NF)r � __enter__�__exit__� return_valuer �assert_called_with)r �mock�mr r r �testContextManagerMockingE s
z"WithTest.testContextManagerMockingc
C sP t � }| �t�� |� dd W d Q R X W d Q R X |j�� | �|jj� d S )Nr � )r � assertRaises� TypeErrorr( r+ r r) Zcalled)r r, r r r �$test_context_manager_with_magic_mockQ s
z-WithTest.test_context_manager_with_magic_mockc
C sr t dt tj��H}| �t|d� t dt ��}| �t|d� W d Q R X | �t|d� W d Q R X | �ttjd� d S )Nz %s.somethingr z restored with wrong instancez not restored)r r r r& r r r )r r Z
mock_againr r r �"test_with_statement_same_attribute[ s z+WithTest.test_with_statement_same_attributec
C sl t dt ��:}| �t|d� t dt ��}| �t|d� W d Q R X W d Q R X | �ttj� | �ttj� d S )Nz %s.somethingr z%s.something_else)r r r r r" r r r# )r r r$ r r r � test_with_statement_imbricatedh s z'WithTest.test_with_statement_imbricatedc
C s� i }t �|ddi�� | �|ddi� W d Q R X | �|i � | �t��: t �|ddi�� | �|ddi� td��W d Q R X W d Q R X | �|i � d S )N�a�bZKonrad)r �dictr r0 � NameError)r r r r r �test_dict_context_managert s z"WithTest.test_dict_context_managerc
C s� G dd� d�}|� }t j|ddd��N}t j|ddd��}|�� W d Q R X | �|jd� | �|jd� |�� W d Q R X | �|jd� d S )Nc @ s e Zd Zdd� ZdS )z5WithTest.test_double_patch_instance_method.<locals>.Cc S s d S )Nr )r r r r �f� s z7WithTest.test_double_patch_instance_method.<locals>.C.fN)r r r r: r r r r �C� s r; r: T)Zautospec� r )r r r: r Z
call_count)r r; �cZpatch1Zpatch2r r r �!test_double_patch_instance_method� s z*WithTest.test_double_patch_instance_methodN)r r r r r r r r% r' r. r2 r3 r4 r9 r> r r r r r
s
r
c @ 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 S ) � TestMockOpenc C sF t � }tdt |dd��}| �||� td� W d Q R X |�d� d S )Nz%s.openT)�creater )r r r �assertIs�open�assert_called_once_with)r r, �patchedr r r �test_mock_open� s
zTestMockOpen.test_mock_openc
C s� t � }|j}tdt |dd��" td��}|�� W d Q R X W d Q R X td�t� �� t� �� t� �d d d �g}| � |j
|� | � ||� d S )Nz%s.openT)r@ r ) r r* r r rB �readr r( r) r �
mock_callsrA )r r, �handler: �expected_callsr r r � test_mock_open_context_manager� s
z+TestMockOpen.test_mock_open_context_managerc C s� t � }tdt |dd��> td��}|�� W d Q R X td��}|�� W d Q R X W d Q R X td�t� �� t� �� t� �d d d �td�t� �� t� �� t� �d d d �g}| �|j |� d S )Nz%s.openT)r@ r �bar)
r r r rB rF r r( r) r rG )r r, r: rI r r r �-test_mock_open_context_manager_multiple_times� s
z:TestMockOpen.test_mock_open_context_manager_multiple_timesc C sN t � }t|� tdt |dd��}| �||� td� W d Q R X |�d� d S )Nz%s.openT)r@ r )r r r r rA rB rC )r r, rD r r r �test_explicit_mock� s z TestMockOpen.test_explicit_mockc C sH t dd�}tdt |dd�� td�}|�� }W d Q R X | �|d� d S )Nr )� read_dataz%s.openT)r@ rK )r r r rB rF r )r r, �h�resultr r r �test_read_data� s
zTestMockOpen.test_read_datac C s� t dd�}tdt |dd��& td�}|�� }|�� }|�� }W d Q R X | �|d� | �|d� | �|d � | �|�� d
� t d d�}tdt |dd�� td�}|�� }W d Q R X | �|d � | �|�� d
� d S ) Nz foo
bar
baz
)rN z%s.openT)r@ rK zfoo
zbar
zbaz
� r )r r r rB �readliner )r r, rO �line1�line2�line3rP r r r �test_readline_data� s
z TestMockOpen.test_readline_datac C s� t dd�}tdt |dd�� td�}dd� |D �}W d Q R X | �|d d
� | �|d d � | �|d
d� | �|�� d� | �t�� t|� W d Q R X d S )Nz foo
bar
baz
)rN z%s.openT)r@ rK c S s g | ]}|�qS r r )�.0�lr r r �
<listcomp>� s z6TestMockOpen.test_dunder_iter_data.<locals>.<listcomp>r zfoo
r< zbar
� zbaz
rR ) r r r rB r rS r0 �
StopIteration�next)r r, rO �linesr r r �test_dunder_iter_data� s
z"TestMockOpen.test_dunder_iter_datac C s� t dd�}tdt |dd��, td�}t|�}t|�}dd� |D �}W d Q R X | �|d � | �|d
� | �|d d � | �|�� d
� d S )Nz foo
bar
baz
)rN z%s.openT)r@ rK c S s g | ]}|�qS r r )rX rY r r r rZ � s z/TestMockOpen.test_next_data.<locals>.<listcomp>zfoo
zbar
r zbaz
rR )r r r rB r] r rS )r r, rO rT rU r^ r r r �test_next_data� s
zTestMockOpen.test_next_datac C s� t dd�}tdt |dd�� td�}|�� }W d Q R X | �|ddd g� t d
d�}tdt |dd�� td�}|�� }W d Q R X | �|ddd g� d S ) Nz foo
bar
baz
)rN z%s.openT)r@ rK zfoo
zbar
zbaz
z foo
bar
bazZbaz)r r r rB � readlinesr )r r, rO rP r r r �test_readlines_data� s
z TestMockOpen.test_readlines_datac
C sV t dd�}tdt |dd��$ tdd��}|�� }W d Q R X W d Q R X | �|d� d S )N� �)rN z%s.openT)r@ �abc�rb)r r r rB rF r )r r, r: rP r r r �test_read_bytes s
z TestMockOpen.test_read_bytesc
C s~ t dd�}tdt |dd��4 tdd�� }|�� }|�� }|�� }W d Q R X W d Q R X | �|d� | �|d � | �|d
� d S ) Ns abc
def
ghi
)rN z%s.openT)r@ rd re s abc
s def
s ghi
)r r r rB rS r )r r- r: rT rU rV r r r �test_readline_bytes s
z TestMockOpen.test_readline_bytesc
C s\ t dd�}tdt |dd��$ tdd��}|�� }W d Q R X W d Q R X | �|dd d
g� d S ) Ns abc
def
ghi
)rN z%s.openT)r@ rd re s abc
s def
s ghi
)r r r rB ra r )r r- r: rP r r r �test_readlines_bytes# s
z!TestMockOpen.test_readlines_bytesc C s� d}t |d�}| �|� �d�|d d� � | �|� �d�|d d� � |� }| �|�d�|d d� � | �|�d�|dd � � d S )Nz foo
bar
baz)rN �
)r r rF )r Z some_datar, r: r r r �!test_mock_open_read_with_argument+ s
z.TestMockOpen.test_mock_open_read_with_argumentc C s� t dd�}tdt |dd�� td�}|�� }|�� }W d Q R X | �|d� | �|dd g� t dd�}tdt |dd�� td�}|�� }|�� }W d Q R X | �|d� | �|d
� d S ) Nz foo
bar
baz
)rN z%s.openT)r@ rK zfoo
zbar
zbaz
zbar
baz
)r r r rB rS ra r rF )r r, rO rT �restr r r �test_interleaved_reads8 s
z#TestMockOpen.test_interleaved_readsc C s� t dd�}|� }d|j_d|j_dg|j_| �|�� d� | �|�� d� | �|�� dg� | �|�� d� | �|�� d� d S )Nr )rN rK )r rF r* rS ra r )r r, rH r r r � test_overriding_return_valuesL s
z*TestMockOpen.test_overriding_return_valuesN)r r r rE rJ rL rM rQ rW r_ r` rb rf rg rh rj rl rm r r r r r? � s
r? �__main__)Zunittest�warningsr Z unittest.test.testmock.supportr Z
unittest.mockr r r r r r r r r# r" ZTestCaser
r? r �mainr r r r �<module> s M