← Back
Editing: 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
Save File
Cancel