Sindbad~EG File Manager
a
��g�/ � @ 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 �;/usr/local/lib/python3.9/unittest/test/testmock/testwith.pyr
� 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 sR t dt tj�� | �ttjd� W d � n1 s60 Y | �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 s� | � t��V tdt tj��& | �ttjd� t� �W d � n1 sH0 Y W d � n1 sf0 Y | �ttj� d S r ) � assertRaisesr
r r r r r r r r r r r � test_with_statement_exception s
Bz&WithTest.test_with_statement_exceptionc C s^ t dt ��0}| �t|d� | �t|t�d� W d � n1 sB0 Y | �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 sd G dd� dt �}|j}t� |d�� | �|j|d� W d � n1 sH0 Y | �|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��� tdt ��X}tdt ��,}| �t|d� | �t|d� W d � n1 sX0 Y W d � n1 sv0 Y W d � n1 s�0 Y | �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 s` t dt tj��.}| �t|d� | �|tjd� W d � n1 sD0 Y | �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 sp t � }t � |_t � |_d|j_|� }| �||jj� W d � n1 sH0 Y |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 sx t � }| �t��< |� dd W d � n1 s40 Y W d � n1 sR0 Y |j�� | �|jj� d S )Nr" � )r r � TypeErrorr. r1 r r/ Zcalled)r r2 r r r �$test_context_manager_with_magic_mockO s D
z-WithTest.test_context_manager_with_magic_mockc C s� t dt tj��f}| �t|d� t dt �� }| �t|d� W d � n1 sP0 Y | �t|d� W d � n1 s|0 Y | �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 s� t dt ��X}| �t|d� t dt �� }| �t|d� W d � n1 sL0 Y W d � n1 sj0 Y | �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 d � n1 s:0 Y | �|i � | �t��X t �|ddi��( | �|ddi� td��W d � n1 s�0 Y W d � n1 s�0 Y | �|i � d S )N�a�bZKonrad)r �dictr r � NameError)r r" r r r �test_dict_context_managerr s . Dz"WithTest.test_dict_context_managerc C s� G dd� d�}|� }t j|ddd��l}t j|ddd��}|�� W d � n1 sT0 Y | �|jd� | �|jd� |�� W d � n1 s�0 Y | �|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� r 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- r4 r7 r8 r9 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 sZ t � }tdt |dd��$}| �||� td� W d � n1 sB0 Y |�d� d S �N�%s.openT�Zcreater" )r r r �assertIs�open�assert_called_once_with�r r2 Zpatchedr r r �test_mock_open� s
&zTestMockOpen.test_mock_openc C s� t � }|j}tdt |dd��@ td��}|�� W d � n1 sF0 Y W d � n1 sd0 Y td�t� �� t� �� t� �d d d �g}| � |j
|� | � ||� d S rE ) r r0 r r rI �readr r. r/ r �
mock_callsrH )r r2 �handler? �expected_callsr r r � test_mock_open_context_manager� s
D�z+TestMockOpen.test_mock_open_context_managerc C s� t � }tdt |dd��p td��}|�� W d � n1 s@0 Y td��}|�� W d � n1 sp0 Y W d � n1 s�0 Y 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 r2 r? rP r r r �-test_mock_open_context_manager_multiple_times� s
&
D�z:TestMockOpen.test_mock_open_context_manager_multiple_timesc C sb t � }t|� tdt |dd��$}| �||� td� W d � n1 sJ0 Y |�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 s\ t dd�}tdt |dd�� td�}|�� }W d � n1 sB0 Y | �|d� d S )Nr" �Z read_datarF TrG rR �r r r rI rM r �r r2 �h�resultr r r �test_read_data� s
&zTestMockOpen.test_read_datac C s� t dd�}tdt |dd��0 td�}|�� }|�� }|�� }W d � n1 sR0 Y | �|d� | �|d� | �|d � | �|�� d
� t d d�}tdt |dd�� td�}|�� }W d � n1 s�0 Y | �|d � | �|�� d
� d S ) N� foo
bar
baz
rU rF TrG rR �foo
�bar
�baz
� r" �r r r rI �readliner )r r2 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 d � n1 sH0 Y | �|d d
� | �|d d � | �|d
d� | �|�� d� | �t�� t|� W d � n1 s�0 Y d S )Nr[ rU rF TrG rR c S s g | ]}|�qS r r ��.0�lr r r �
<listcomp>� r z6TestMockOpen.test_dunder_iter_data.<locals>.<listcomp>r r\ rA r] � r^ r_ ) r r r rI r ra r �
StopIteration�next)r r2 rX �linesr r r �test_dunder_iter_data� s
, z"TestMockOpen.test_dunder_iter_datac C s� t dd�}tdt |dd��6 td�}t|�}t|�}dd� |D �}W d � n1 sX0 Y | �|d � | �|d
� | �|d d � | �|�� d
� d S )Nr[ rU rF TrG rR c S s g | ]}|�qS r r rf r r r ri � r z/TestMockOpen.test_next_data.<locals>.<listcomp>r\ r] r r^ r_ )r r r rI rl r ra )r r2 rX rb rc rm r r r �test_next_data� s
, zTestMockOpen.test_next_datac C s� t dd�}tdt |dd�� td�}|�� }W d � n1 sB0 Y | �|g d�� t dd�}tdt |dd�� td�}|�� }W d � n1 s�0 Y | �|g d �� d S )
Nr[ rU rF TrG rR )r\ r] r^ � foo
bar
baz)r\ r] Zbaz�r r r rI � readlinesr rW r r r �test_readlines_data� s
&
&z TestMockOpen.test_readlines_datac C s~ t dd�}tdt |dd��B tdd��}|�� }W d � n1 sF0 Y W d � n1 sd0 Y | �|d� d S )N� �rU rF TrG �abc�rbrV )r r2 r? rY r r r �test_read_bytes s
Dz TestMockOpen.test_read_bytesc C s� t dd�}tdt |dd��R tdd��(}|�� }|�� }|�� }W d � n1 sV0 Y W d � n1 st0 Y | �|d� | �|d � | �|d
� d S ) N� abc
def
ghi
rU rF TrG ru rv � abc
� def
� ghi
r` )r r3 r? rb rc rd r r r �test_readline_bytes s
D z TestMockOpen.test_readline_bytesc C s� t dd�}tdt |dd��B tdd��}|�� }W d � n1 sF0 Y W d � n1 sd0 Y | �|g d�� d S ) Nrx rU rF TrG ru rv )ry rz r{ rq )r r3 r? rY r r r �test_readlines_bytes s
Dz!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_datar2 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 � n1 sJ0 Y | �|d� | �|dd g� t dd�}tdt |dd��( td�}|�� }|�� }W d � n1 s�0 Y | �|d� | �|d
� d S ) Nr[ rU rF TrG rR r\ r] r^ zbar
baz
)r r r rI ra rr r rM )r r2 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 r0 ra rr r )r r2 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 r| r} 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
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists