Dashboard
PHP:
8.3.33
OS:
Linux
User:
letter7
/
/
usr
/
lib
/
mysqlsh
/
lib
/
python3.8
/
unittest
/
test
/
testmock
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: testwith.cpython-38.pyc
U �:�e�/ � @ 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 �D/usr/local/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
Save
Cancel