File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_monkey.cpython-38.pyc
U
W[ � @ sP d Z ddlmZmZ ddlmZ ddlmZ G dd� d�ZG dd� dej �Z
d S )
z%
Tests for L{twisted.python.monkey}.
� )�division�absolute_import)�unittest)�
MonkeyPatcherc @ s e Zd Zdd� ZdS )�TestObjc C s d| _ d| _d| _d S )Nz foo valuez bar valuez baz value)�foo�bar�baz��self� r �:/usr/lib/python3/dist-packages/twisted/test/test_monkey.py�__init__ s zTestObj.__init__N)�__name__�
__module__� __qualname__r r r r r
r s r c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�MonkeyPatcherTestsz;
Tests for L{MonkeyPatcher} monkey-patching class.
c C s t � | _t � | _t� | _d S �N)r �
testObject�originalObjectr �
monkeyPatcherr
r r r
�setUp s zMonkeyPatcherTests.setUpc C sJ | j �� | �| jj| jj� | �| jj| jj� | �| jj| jj� dS )zL
A monkey patcher without patches shouldn't change a thing.
N)r �patch� assertEqualr r r r r r
r r r
�
test_empty! s
z MonkeyPatcherTests.test_emptyc C sZ t | jddf| jddf�}|�� | �d| jj� | �d| jj� | �| jj| jj� dS )z}
Constructing a L{MonkeyPatcher} with patches should add all of the
given patches to the patch list.
r �hahar ZheheN)r r r r r r r r )r Zpatcherr r r
�test_constructWithPatches. s
�z,MonkeyPatcherTests.test_constructWithPatchesc C s0 | j �| jdd� | j �� | �| jjd� dS )zf
Patching an attribute that exists sets it to the value defined in the
patch.
r r N)r �addPatchr r r r r
r r r
�test_patchExisting; s
z%MonkeyPatcherTests.test_patchExistingc C s& | j �| jdd� | �t| j j� dS )zT
Patching a non-existing attribute fails with an C{AttributeError}.
Znowherezblow up pleaseN)r r r � assertRaises�AttributeErrorr r
r r r
�test_patchNonExistingE s �z(MonkeyPatcherTests.test_patchNonExistingc C s` | j �| jdd� | j �| jdd� | j �� | �| jjd� | j �� | �| jj| jj� dS )z|
Adding a patch for an object and attribute that already have a patch
overrides the existing patch.
r �blahZBLAHN)r r r r r r �restorer r
r r r
�test_patchAlreadyPatchedN s
z+MonkeyPatcherTests.test_patchAlreadyPatchedc C s\ | j �| jdd� | j �� | j �� | �| jj| jj� | j �� | �| jj| jj� dS )zH
Restoring an already-restored monkey patch is a no-op.
r r" N)r r r r r# r r r r
r r r
�test_restoreTwiceIsANoOp[ s
z+MonkeyPatcherTests.test_restoreTwiceIsANoOpc sD g � d
� fdd� }| j j|dddd�}| �d|� | �d g� � dS ) z�
runWithPatches should run the given callable, passing in all arguments
and keyword arguments, and return the return value of the callable.
Nc s � � | ||f� dS )Nr )�append)�a�b�c��logr r
�fn s z;MonkeyPatcherTests.test_runWithPatchesDecoration.<locals>.f� � �
)r) r )r- r. r/ )N)r �runWithPatchesr �r r, �resultr r* r
� test_runWithPatchesDecorationg s
z0MonkeyPatcherTests.test_runWithPatchesDecorationc sn � fdd�}� j �� jdd� � j �|�}� �d� jj� jjf|� � j �|�}� �d� jj� jjf|� dS )z�
We should be able to call the same function with runWithPatches more
than once. All patches should apply for each call.
c s � j j� j j� j jfS r )r r r r r r
r r
r, | s �z9MonkeyPatcherTests.test_repeatedRunWithPatches.<locals>.fr r N)r r r r0 r r r r r1 r r
r
�test_repeatedRunWithPatchesw s � �z.MonkeyPatcherTests.test_repeatedRunWithPatchesc C sN | j �| jdd� | �| jj| jj� | j �dd� � | �| jj| jj� dS )zo
C{runWithPatches} should restore the original values after the function
has executed.
r r c S s d S r r r r r r
�<lambda>� � z@MonkeyPatcherTests.test_runWithPatchesRestores.<locals>.<lambda>N)r r r r r r r0 r
r r r
�test_runWithPatchesRestores� s z.MonkeyPatcherTests.test_runWithPatchesRestoresc sn � fdd�}� j �� jdd� � j �� jdd� � �t� j j|� � �� jj� jj� � �� jj � jj � dS )zv
Test runWithPatches restores the original values even when the function
raises an exception.
c s, � � � jjd� � � � jjd� td��d S )Nr �blahblahzSomething went wrong!)r r r r � RuntimeErrorr r
r r
�_� s zDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.<locals>._r r r r8 N)
r r r r r9 r0 r r r r )r r: r r
r
�&test_runWithPatchesRestoresOnException� s z9MonkeyPatcherTests.test_runWithPatchesRestoresOnExceptionN)r r r �__doc__r r r r r! r$ r% r3 r4 r7 r; r r r r
r s
r N) r< Z
__future__r r Z
twisted.trialr Ztwisted.python.monkeyr r ZSynchronousTestCaser r r r r
�<module> s