\drbdZddlmZddlmZGddZGddejZdS) z% Tests for L{twisted.python.monkey}. ) MonkeyPatcher)unittestceZdZdZdS)TestObjc0d|_d|_d|_dS)Nz foo valuez bar valuez baz value)foobarbazselfs 8lib/python3.11/site-packages/twisted/test/test_monkey.py__init__zTestObj.__init__sN)__name__ __module__ __qualname__rrr rr s#rrc`eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdS)MonkeyPatcherTestsz; Tests for L{MonkeyPatcher} monkey-patching class. cxt|_t|_t|_dSN)r testObjectoriginalObjectr monkeyPatcherr s r setUpzMonkeyPatcherTests.setUps+!))%ii*__rc4|j||jj|jj||jj|jj||jj|jjdS)zL A monkey patcher without patches shouldn't change a thing. N)rpatch assertEqualrrrr r r s r test_emptyzMonkeyPatcherTests.test_emptys   """ ,0$/2EFFF ,0$/2EFFF ,0$/2EFFFFFrcBt|jddf|jddf}||d|jj|d|jj||jj|jjdS)z} Constructing a L{MonkeyPatcher} with patches should add all of the given patches to the patch list. rhahar heheN)rrrrrr rr )r patchers r test_constructWithPatchesz,MonkeyPatcherTests.test_constructWithPatches*s  _eV ,tv.N     !4555 !4555 ,0$/2EFFFFFrc|j|jdd|j||jjddS)zf Patching an attribute that exists sets it to the value defined in the patch. rr!N)raddPatchrrrrr s r test_patchExistingz%MonkeyPatcherTests.test_patchExisting7sW ##DOUFCCC   """ ,f55555rc|j|jdd|t|jjdS)zT Patching a non-existing attribute fails with an C{AttributeError}. nowherezblow up pleaseN)rr&r assertRaisesAttributeErrorrr s r test_patchNonExistingz(MonkeyPatcherTests.test_patchNonExisting@sD ##DOY@PQQQ .$*<*BCCCCCrc|j|jdd|j|jdd|j||jjd|j||jj|jjdS)z| Adding a patch for an object and attribute that already have a patch overrides the existing patch. rblahBLAHN)rr&rrrrrestorerr s r test_patchAlreadyPatchedz+MonkeyPatcherTests.test_patchAlreadyPatchedGs ##DOUFCCC ##DOUFCCC   """ ,f555 ""$$$ ,d.A.EFFFFFrc|j|jdd|j|j||jj|jj|j||jj|jjdS)zH Restoring an already-restored monkey patch is a no-op. rr.N)rr&rrr0rrrr s r test_restoreTwiceIsANoOpz+MonkeyPatcherTests.test_restoreTwiceIsANoOpSs ##DOUFCCC   """ ""$$$ ,d.A.EFFF ""$$$ ,d.A.EFFFFFrcgd fd }|j|ddd}|d||dgdS) z runWithPatches should run the given callable, passing in all arguments and keyword arguments, and return the return value of the callable. Nc8|||fdS)Nr)append)abclogs r fz;MonkeyPatcherTests.test_runWithPatchesDecoration..fes JJ1ay ! ! !5r )r9r)r<r=r>r)rrunWithPatchesr)r r;resultr:s @r test_runWithPatchesDecorationz0MonkeyPatcherTests.test_runWithPatchesDecoration^s}       #221ab2AA ''' *s+++++rcpfd}jjddj|}djjjjf|j|}djjjjf|dS)z We should be able to call the same function with runWithPatches more than once. All patches should apply for each call. cJjjjjjjfSr)rrr r r sr r;z9MonkeyPatcherTests.test_repeatedRunWithPatches..fssO')QR Rrrr!N)rr&rr?rrr r )r r;r@s` r test_repeatedRunWithPatchesz.MonkeyPatcherTests.test_repeatedRunWithPatchesms  S S S S S ##DOUFCCC#22155  T(,d.A.E F   #22155  T(,d.A.E F     rc&|j|jdd||jj|jj|jd||jj|jjdS)zo C{runWithPatches} should restore the original values after the function has executed. rr!cdSrrrrr z@MonkeyPatcherTests.test_runWithPatchesRestores..s$rN)rr&rrrrr?r s r test_runWithPatchesRestoresz.MonkeyPatcherTests.test_runWithPatchesRestoress ##DOUFCCC ,0$/2EFFF )),,777 ,0$/2EFFFFFrcfd}jjddjjddtjj|jjjjjj jj dS)zv Test runWithPatches restores the original values even when the function raises an exception. cjjdjjdt d)Nr!blahblahzSomething went wrong!)rrrr RuntimeErrorr sr _zDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.._sH   T_0& 9 9 9   T_0* = = =677 7rrr!r rKN) rr&rr*rLr?rrrr )r rMs` r &test_runWithPatchesRestoresOnExceptionz9MonkeyPatcherTests.test_runWithPatchesRestoresOnExceptions  8 8 8 8 8 ##DOUFCCC ##DOUJGGG ,(:(I1MMM ,d.A.EFFF ,d.A.EFFFFFrc|j|jdd|j5||jjddddn #1swxYwY||jj|jjdS)z L{MonkeyPatcher} is a context manager that applies its patches on entry and restore original values on exit. rz patched valueN)rr&rrrrr s r test_contextManagerz&MonkeyPatcherTests.test_contextManagers ##DOUOLLL   C C   T_0/ B B B C C C C C C C C C C C C C C C ,d.A.EFFFFFs!AAAc|t5|j5td#1swxYwY ddddS#1swxYwYdS)zh Exceptions propagate through the L{MonkeyPatcher} context-manager exit method. somethingN)r*rLrr s r 'test_contextManagerPropagatesExceptionsz:MonkeyPatcherTests.test_contextManagerPropagatesExceptionss   | , , 0 0# 0 0";/// 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s'A 37 A 7 A  AAN)rrr__doc__rrr$r'r,r1r3rArDrHrNrPrSrrr rrs--- G G G G G G666DDD G G G G G G , , ,   &GGGGGG$GGG00000rrN)rTtwisted.python.monkeyr twisted.trialrrSynchronousTestCaserrrr rXs 0/////""""""Y0Y0Y0Y0Y05Y0Y0Y0Y0Y0r