Your IP : 216.73.216.213
3
� f�( � @ 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 j� dS )
� N)�catch_warnings)� is_instance)� MagicMock�Mock�patch�sentinel� mock_open�callc @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�WithTestc C s> t dt tj�� | jttjd� W d Q R X | jttj� d S )Nz %s.something� unpatched)r �__name__r �
Something2� assertEqual� something� Something)�self� r �D/opt/alt/python36/lib64/python3.6/unittest/test/testmock/testwith.py�test_with_statement s z WithTest.test_with_statementc C sj y8t dt tj�� | jttjd� td��W d Q R X W n tk
rL Y n X | jd� | jttj� 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 ��&}| jt|d� | jt|t�d� W d Q R X | jttj� 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}tj |d�� | j|j|d� W d Q R X | j|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 ��"}| jt|d� | jt|d� W d Q R X W d Q R X W d Q R X | jttj� | jttj� 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��$}| jt|d� | j|tjd� W d Q R X | jttj� 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_|�}| j||jj� W d Q R X |jj� |jjd 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 � }| jt�� |� dd W d Q R X W d Q R X |jj� | 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}| jt|d� t dt ��}| jt|d� W d Q R X | jt|d� W d Q R X | jttjd� 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 ��:}| jt|d� t dt ��}| jt|d� W d Q R X W d Q R X | jttj� | jttj� 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 j|ddi�� | j|ddi� W d Q R X | j|i � | jt��: t j|ddi�� | j|ddi� td��W d Q R X W d Q R X | j|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_managerN)r r r r r r r r% r' r. r2 r3 r4 r9 r r r r r
s
r
c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) � TestMockOpenc
C sF t � }tdt |dd��}| j||� td� W d Q R X |jd� 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��}|j� W d Q R X W d Q R X td�t� j� t� j� t� jd d d �g}| j |j
|� | j ||� d S )Nz%s.openT)r; r ) r r* r r r= �readr r( r) r �
mock_callsr< )r r, �handle�f�expected_callsr r r � test_mock_open_context_manager� s
z+TestMockOpen.test_mock_open_context_managerc C s� t � }tdt |dd��> td��}|j� W d Q R X td��}|j� W d Q R X W d Q R X td�t� j� t� j� t� jd d d �td�t� j� t� j� t� jd d d �g}| j|j |� d S )Nz%s.openT)r; r �bar)
r r r r= rA r r( r) r rB )r r, rD rE 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��}| j||� td� W d Q R X |jd� d S )Nz%s.openT)r; r )r r r r r< r= r> )r r, r? r r r �test_explicit_mock� s z TestMockOpen.test_explicit_mockc
C sH t dd�}tdt |dd�� td�}|j� }W d Q R X | j|d� d S )Nr )� read_dataz%s.openT)r; rG )r r r r= rA r )r r, �h�resultr r r �test_read_data� s
zTestMockOpen.test_read_datac C s� t dd�}tdt |dd��& td�}|j� }|j� }|j� }W d Q R X | j|d� | j|d� | j|d � t d
d�}tdt |dd�� td�}|j� }W d Q R X | j|d
� d S ) Nz foo
bar
baz
)rJ z%s.openT)r; rG zfoo
zbar
zbaz
r )r r r r= �readliner )r r, rK �line1�line2�line3rL r r r �test_readline_data� s
z TestMockOpen.test_readline_datac C s� t dd�}tdt |dd�� td�}|j� }W d Q R X | j|ddd g� t d
d�}tdt |dd�� td�}|j� }W d Q R X | j|ddd g� d S ) Nz foo
bar
baz
)rJ z%s.openT)r; rG zfoo
zbar
zbaz
z foo
bar
bazZbaz)r r r r= � readlinesr )r r, rK rL r r r �test_readlines_data� s
z TestMockOpen.test_readlines_datac C sV t dd�}tdt |dd��$ tdd��}|j� }W d Q R X W d Q R X | j|d� d S )N� �)rJ z%s.openT)r; �abc�rb)r r r r= rA r )r r, rD rL r r r �test_read_bytes� s
z TestMockOpen.test_read_bytesc C s~ t dd�}tdt |dd��4 tdd�� }|j� }|j� }|j� }W d Q R X W d Q R X | j|d� | j|d � | j|d
� d S ) Ns abc
def
ghi
)rJ z%s.openT)r; rV rW s abc
s def
s ghi
)r r r r= rN r )r r- rD rO rP rQ r r r �test_readline_bytes� s
z TestMockOpen.test_readline_bytesc C s\ t dd�}tdt |dd��$ tdd��}|j� }W d Q R X W d Q R X | j|dd d
g� d S ) Ns abc
def
ghi
)rJ z%s.openT)r; rV rW s abc
s def
s ghi
)r r r r= rS r )r r- rD rL r r r �test_readlines_bytes� s
z!TestMockOpen.test_readlines_bytesc C s& d}t |d�}| j|� jd�|� d S )Nz foo
bar
baz)rJ �
)r r rA )r Z some_datar, 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�}|j� }|j� }W d Q R X | j|d� | j|dd g� t dd�}tdt |dd�� td�}|j� }|j� }W d Q R X | j|d� | j|d
� d S ) Nz foo
bar
baz
)rJ z%s.openT)r; rG zfoo
zbar
zbaz
zbar
baz
)r r r r= rN rS r rA )r r, rK rO �restr r r �test_interleaved_reads s
z#TestMockOpen.test_interleaved_readsc C s� t dd�}|� }d|j_d|j_dg|j_| j|j� d� | j|j� d� | j|j� dg� | j|j� d� | j|j� d� d S )Nr )rJ rG )r rA r* rN rS r )r r, rC r r r � test_overriding_return_values s
z*TestMockOpen.test_overriding_return_valuesN)r r r r@ rF rH rI rM rR rT rX rY rZ r\ r^ r_ 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 u *