CasperSecurity
o
�b� � @ s@ d Z ddlmZ ddlmZ G dd� d�ZG dd� dej�ZdS ) z%
Tests for L{twisted.python.monkey}.
� )�
MonkeyPatcher)�unittestc @ 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 �heheN)r r r r r r r r )r �patcherr
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_patchExisting7 s
z%MonkeyPatcherTests.test_patchExistingc C s& | j �| jdd� | �t| j j� dS )zT
Patching a non-existing attribute fails with an C{AttributeError}.
�nowherezblow up pleaseN)r r r � assertRaises�AttributeErrorr r r
r
r �test_patchNonExisting@ 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 �blah�BLAHN)r r r r r r �restorer r r
r
r �test_patchAlreadyPatchedG 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_restoreTwiceIsANoOpS 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 �fe s z;MonkeyPatcherTests.test_runWithPatchesDecoration.<locals>.f� � �
)r+ r )r/ r0 r1 r )r �runWithPatchesr �r r. �resultr
r, r � test_runWithPatchesDecoration^ 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 s z9MonkeyPatcherTests.test_repeatedRunWithPatches.<locals>.fr r N)r r r r2 r r r r r3 r
r r �test_repeatedRunWithPatchesm 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>� s z@MonkeyPatcherTests.test_runWithPatchesRestores.<locals>.<lambda>N)r r r r r r r2 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��)Nr �blahblahzSomething went wrong!)r r r r � RuntimeErrorr
r r
r �_� s zDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.<locals>._r r r r9 N)
r r r r r: r2 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' r5 r6 r8 r< r
r
r
r r s
r N)r= �twisted.python.monkeyr �
twisted.trialr r �SynchronousTestCaser r
r
r
r �<module> s