U  W[ã@sPdZddlmZmZddlmZddlmZGdd„dƒZGdd„dej ƒZ d S) z% Tests for L{twisted.python.monkey}. é)ÚdivisionÚabsolute_import)Úunittest)Ú MonkeyPatcherc@seZdZdd„ZdS)ÚTestObjcCsd|_d|_d|_dS)Nz foo valuez bar valuez baz value)ÚfooÚbarÚbaz©Úself©r ú:/usr/lib/python3/dist-packages/twisted/test/test_monkey.pyÚ__init__szTestObj.__init__N)Ú__name__Ú __module__Ú __qualname__rr r r r rsrc@sheZdZdZdd„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. cCstƒ|_tƒ|_tƒ|_dS©N)rÚ testObjectÚoriginalObjectrÚ monkeyPatcherr r r r ÚsetUpszMonkeyPatcherTests.setUpcCsJ|j ¡| |jj|jj¡| |jj|jj¡| |jj|jj¡dS)zL A monkey patcher without patches shouldn't change a thing. N)rÚpatchÚ assertEqualrrrrr r r r r Ú test_empty!s zMonkeyPatcherTests.test_emptycCsZt|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ÚhaharZheheN)rrrrrrrr )r Zpatcherr r r Útest_constructWithPatches.s  ÿz,MonkeyPatcherTests.test_constructWithPatchescCs0|j |jdd¡|j ¡| |jjd¡dS)zf Patching an attribute that exists sets it to the value defined in the patch. rrN)rÚaddPatchrrrrr r r r Útest_patchExisting;s z%MonkeyPatcherTests.test_patchExistingcCs&|j |jdd¡| t|jj¡dS)zT Patching a non-existing attribute fails with an C{AttributeError}. Znowherezblow up pleaseN)rrrÚ assertRaisesÚAttributeErrorrr r r r Útest_patchNonExistingEs ÿz(MonkeyPatcherTests.test_patchNonExistingcCs`|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)rrrrrrÚrestorerr r r r Útest_patchAlreadyPatchedNs   z+MonkeyPatcherTests.test_patchAlreadyPatchedcCs\|j |jdd¡|j ¡|j ¡| |jj|jj¡|j ¡| |jj|jj¡dS)zH Restoring an already-restored monkey patch is a no-op. rr"N)rrrrr#rrrr r r r Útest_restoreTwiceIsANoOp[s    z+MonkeyPatcherTests.test_restoreTwiceIsANoOpcsDg‰d ‡fdd„ }|jj|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. Ncsˆ |||f¡dS)Nr)Úappend)ÚaÚbÚc©Úlogr r Úfnsz;MonkeyPatcherTests.test_runWithPatchesDecoration..fééé )r)r)r-r.r/)N)rÚrunWithPatchesr©r r,Úresultr r*r Útest_runWithPatchesDecorationgs  z0MonkeyPatcherTests.test_runWithPatchesDecorationcsn‡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. csˆjjˆjjˆjjfSr)rrrr r r r r r,|s ÿz9MonkeyPatcherTests.test_repeatedRunWithPatches..frrN)rrrr0rrrr r1r r r Útest_repeatedRunWithPatchesws  ÿ þz.MonkeyPatcherTests.test_repeatedRunWithPatchescCsN|j |jdd¡| |jj|jj¡|j dd„¡| |jj|jj¡dS)zo C{runWithPatches} should restore the original values after the function has executed. rrcSsdSrr r r r r Ú‘óz@MonkeyPatcherTests.test_runWithPatchesRestores..N)rrrrrrr0r r r r Útest_runWithPatchesRestoresŠsz.MonkeyPatcherTests.test_runWithPatchesRestorescsn‡fdd„}ˆj ˆjdd¡ˆj ˆjdd¡ˆ tˆjj|¡ˆ ˆjjˆjj¡ˆ ˆjj ˆjj ¡dS)zv Test runWithPatches restores the original values even when the function raises an exception. cs,ˆ ˆjjd¡ˆ ˆjjd¡tdƒ‚dS)NrÚblahblahzSomething went wrong!)rrrrÚ RuntimeErrorr r r r Ú_šszDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.._rrrr8N) rrrrr9r0rrrr)r r:r r r Ú&test_runWithPatchesRestoresOnException•s  z9MonkeyPatcherTests.test_runWithPatchesRestoresOnExceptionN)rrrÚ__doc__rrrrr!r$r%r3r4r7r;r r r r rs       rN) r<Z __future__rrZ twisted.trialrZtwisted.python.monkeyrrZSynchronousTestCaserr r r r Ús