File: //opt/python38/lib/python3.8/unittest/test/testmock/__pycache__/testwith.cpython-38.opt-1.pyc
U
�i�_�/ � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZm Z m
Z
m Z e j Z
e jZG dd� de�ZG dd� de j�ZG d d
� d
e j�Zed kr�e �� dS ) � N)�catch_warnings)� is_instance)� MagicMock�Mock�patch�sentinel� mock_open�callc @ s e Zd ZdS )�SampleExceptionN)�__name__�
__module__� __qualname__� r r �>/opt/python38/lib/python3.8/unittest/test/testmock/testwith.pyr
s r
c @ 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 5 Q R X | �ttj� d S �N� %s.something� unpatched)r r r �
Something2� assertEqual� something� Something��selfr r r �test_with_statement s z WithTest.test_with_statementc
C sZ | � t��8 tdt tj�� | �ttjd� t� �W 5 Q R X W 5 Q R X | �ttj� d S r ) � assertRaisesr
r r r r r r r 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 5 Q R X | �ttj� d S )Nr r 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 5 Q R X | �|j|� d S )Nc @ s e Zd ZdZdS )z6WithTest.test_patch_object_with_statement.<locals>.Foo�fooN)r r r
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 5 Q R X W 5 Q R X W 5 Q R X | �ttj� | �ttj� d S )NT)�recordr �%s.something_elser ) r r r r r �something_elser r �
SomethingElse�r r Zmock_something_elser r r �test_with_statement_nested1 s �"z#WithTest.test_with_statement_nestedc C sL t dt tj��$}| �t|d� | �|tjd� W 5 Q R X | �ttj� d S )Nr r z wrong patch�r r r ZPatchedr 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 5 Q R X |j�� |j�d d d � d S )NF)r � __enter__�__exit__� return_valuer �assert_called_with)r �mock�mr r r �testContextManagerMockingC s
z"WithTest.testContextManagerMockingc
C sP t � }| �t�� |� dd W 5 Q R X W 5 Q R X |j�� | �|jj� d S )Nr! � )r r � TypeErrorr- r0 r r. Zcalled)r r1 r r r �$test_context_manager_with_magic_mockO s
z-WithTest.test_context_manager_with_magic_mockc
C sr t dt tj��H}| �t|d� t dt ��}| �t|d� W 5 Q R X | �t|d� W 5 Q R X | �ttjd� d S )Nr r z restored with wrong instancez not restoredr+ )r r Z
mock_againr r r �"test_with_statement_same_attributeY s �z+WithTest.test_with_statement_same_attributec
C sl t dt ��:}| �t|d� t dt ��}| �t|d� W 5 Q R X W 5 Q R X | �ttj� | �ttj� d S )Nr r r& )r r r r r' r r r( r) r r r � test_with_statement_imbricatedf s �z'WithTest.test_with_statement_imbricatedc
C s� i }t �|ddi�� | �|ddi� W 5 Q R X | �|i � | �t��: t �|ddi�� | �|ddi� td��W 5 Q R X W 5 Q R X | �|i � d S )N�a�bZKonrad)r �dictr r � NameError)r r! r r r �test_dict_context_managerr s z"WithTest.test_dict_context_managerc
C s� G dd� d�}|� }t j|ddd��N}t j|ddd��}|�� W 5 Q R X | �|jd� | �|jd� |�� W 5 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� � 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, r3 r6 r7 r8 r= rC 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 5 Q R X |�d� d S �N�%s.openT�Zcreater! )r r r �assertIs�open�assert_called_once_with�r r1 Zpatchedr r r �test_mock_open� s
zTestMockOpen.test_mock_openc
C s� t � }|j}tdt |dd��" td��}|�� W 5 Q R X W 5 Q R X td�t� �� t� �� t� �d d d �g}| � |j
|� | � ||� d S rE ) r r/ r r rI �readr r- r. r �
mock_callsrH )r r1 �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 5 Q R X td��}|�� W 5 Q R X W 5 Q R X td�t� �� t� �� t� �d d d �td�t� �� t� �� t� �d d d �g}| �|j |� d S )NrF TrG r! �bar)
r r r rI rM r r- r. r rN )r r1 r> rP 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 5 Q R X |�d� d S rE )r r r r rH rI rJ rK r r r �test_explicit_mock� s z TestMockOpen.test_explicit_mockc C sH t dd�}tdt |dd�� td�}|�� }W 5 Q R X | �|d� d S )Nr! �Z read_datarF TrG rR �r r r rI rM r �r r1 �h�resultr r r �test_read_data� s
zTestMockOpen.test_read_datac C s� t dd�}tdt |dd��& td�}|�� }|�� }|�� }W 5 Q R X | �|d� | �|d� | �|d � | �|�� d
� t d d�}tdt |dd�� td�}|�� }W 5 Q R X | �|d � | �|�� d
� d S ) N� foo
bar
baz
rU rF TrG rR �foo
�bar
�baz
� r! �r r r rI �readliner )r r1 rX �line1�line2�line3rY r r r �test_readline_data� s
z TestMockOpen.test_readline_datac C s� t dd�}tdt |dd�� td�}dd� |D �}W 5 Q R X | �|d d
� | �|d d � | �|d
d� | �|�� d� | �t�� t|� W 5 Q R X d S )Nr[ rU rF TrG rR c S s g | ]}|�qS r r ��.0�lr r r �
<listcomp>� s z6TestMockOpen.test_dunder_iter_data.<locals>.<listcomp>r r\ rA r] � r^ r_ ) r r r rI r ra r �
StopIteration�next)r r1 rX �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 5 Q R X | �|d � | �|d
� | �|d d � | �|�� d
� d S )Nr[ rU rF TrG rR c S s g | ]}|�qS r r rf r r r ri � s z/TestMockOpen.test_next_data.<locals>.<listcomp>r\ r] r r^ r_ )r r r rI rl r ra )r r1 rX rb rc rm r r r �test_next_data� s
zTestMockOpen.test_next_datac C s� t dd�}tdt |dd�� td�}|�� }W 5 Q R X | �|ddd g� t d
d�}tdt |dd�� td�}|�� }W 5 Q R X | �|ddd g� d S ) Nr[ rU rF TrG rR r\ r] r^ � foo
bar
bazZbaz�r r r rI � readlinesr rW r r r �test_readlines_data� s
z TestMockOpen.test_readlines_datac
C sV t dd�}tdt |dd��$ tdd��}|�� }W 5 Q R X W 5 Q R X | �|d� d S )N� �rU rF TrG �abc�rbrV )r r1 r> rY r r r �test_read_bytes s
z TestMockOpen.test_read_bytesc
C s~ t dd�}tdt |dd��4 tdd�� }|�� }|�� }|�� }W 5 Q R X W 5 Q R X | �|d� | �|d � | �|d
� d S � Ns abc
def
ghi
rU rF TrG ru rv s abc
s def
s ghi
r` )r r2 r> rb rc rd r r r �test_readline_bytes s
z TestMockOpen.test_readline_bytesc
C s\ t dd�}tdt |dd��$ tdd��}|�� }W 5 Q R X W 5 Q R X | �|dd d
g� d S rx rq )r r2 r> rY 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 )Nrp rU �
)r r rM )r Z some_datar1 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 5 Q R X | �|d� | �|dd g� t dd�}tdt |dd�� td�}|�� }|�� }W 5 Q R X | �|d� | �|d
� d S ) Nr[ rU rF TrG rR r\ r] r^ zbar
baz
)r r r rI ra rr r rM )r r1 rX rb �restr r r �test_interleaved_reads5 s
z#TestMockOpen.test_interleaved_readsc C s� t dd�}|� }d|j_d|j_dg|j_| �|�� d� | �|�� d� | �|�� dg� | �|�� d� | �|�� d� d S )Nr! rU rR )r rM r/ ra rr r )r r1 rO r r r � test_overriding_return_valuesI s
z*TestMockOpen.test_overriding_return_valuesN)r r r
rL rQ rS rT rZ re rn ro rs rw ry rz r| r~ r r r r r rD � s
rD �__main__)Zunittest�warningsr Z unittest.test.testmock.supportr Z
unittest.mockr r r r r r r r r( r' � Exceptionr
ZTestCaser rD r �mainr r r r �<module> s ~ M