o ¯bã@s@dZddlmZddlmZGdd„dƒZGdd„dejƒZdS) z% Tests for L{twisted.python.monkey}. é)Ú MonkeyPatcher)Úunittestc@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__s zTestObj.__init__N)Ú__name__Ú __module__Ú __qualname__r r r r r r s rc@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Ú monkeyPatcherrr r r ÚsetUps zMonkeyPatcherTests.setUpcCsJ|j ¡| |jj|jj¡| |jj|jj¡| |jj|jj¡dS)zL A monkey patcher without patches shouldn't change a thing. N)rÚpatchÚ assertEqualrrrrrrr r r Ú test_emptys 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ÚhaharÚheheN)rrrrrrrr)r Úpatcherr 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ÚaddPatchrrrrrr r r Útest_patchExisting7s z%MonkeyPatcherTests.test_patchExistingcCs&|j |jdd¡| t|jj¡dS)zT Patching a non-existing attribute fails with an C{AttributeError}. Únowherezblow up pleaseN)rrrÚ assertRaisesÚAttributeErrorrrr r r Útest_patchNonExisting@sz(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ÚblahÚBLAHN)rrrrrrÚrestorerrr r r Útest_patchAlreadyPatchedGs   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%rrrrr r r Útest_restoreTwiceIsANoOpSs    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 Úfesz;MonkeyPatcherTests.test_runWithPatchesDecoration..fééé )r+r)r/r0r1r)rÚrunWithPatchesr©r r.Úresultr r,r Útest_runWithPatchesDecoration^s  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)rrrrr rr r r.ssz9MonkeyPatcherTests.test_repeatedRunWithPatches..frrN)rrrr2rrrrr3r rr Útest_repeatedRunWithPatchesms  ÿ ÿ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 Ú‡sz@MonkeyPatcherTests.test_runWithPatchesRestores..N)rrrrrrr2rr 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ƒ‚)NrÚblahblahzSomething went wrong!)rrrrÚ RuntimeErrorr rr r Ú_szDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.._rrrr9N) rrrr r:r2rrrr)r r;r rr Ú&test_runWithPatchesRestoresOnExceptionŠs z9MonkeyPatcherTests.test_runWithPatchesRestoresOnExceptionN)r rrÚ__doc__rrrrr"r&r'r5r6r8r<r r r r rs      rN)r=Útwisted.python.monkeyrÚ twisted.trialrrÚSynchronousTestCaserr r r r Ús