U “j@sXdZddlmZddlmZddlmZddlmZGdddZ Gdd d ej Z d S) z% Tests for L{twisted.python.monkey}. ) annotations)NoReturn) MonkeyPatcher)unittestc@seZdZddddZdS)TestObjNonereturncCsd|_d|_d|_dS)Nz foo valuez bar valuez baz value)foobarbazselfr.f )r-r )r2r3r4)N)rrunWithPatchesrrr1resultrr/rtest_runWithPatchesDecoration`s  z0MonkeyPatcherTests.test_runWithPatchesDecorationcstddfdd }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. ztuple[str, str, str]rcsjjjjjjfSr)rr r r rr rrr1usz9MonkeyPatcherTests.test_repeatedRunWithPatches..fr rN)rr rr5rrr r r6rr rtest_repeatedRunWithPatchesos  z.MonkeyPatcherTests.test_repeatedRunWithPatchescCsN|j|jdd||jj|jj|jdd||jj|jjdS)zo C{runWithPatches} should restore the original values after the function has executed. r rcSsdSrrrrrrz@MonkeyPatcherTests.test_runWithPatchesRestores..N)rr rrrr r5r rrrtest_runWithPatchesRestoressz.MonkeyPatcherTests.test_runWithPatchesRestorescstddfdd }jjddjjddtjj|jjjjjj jj d S) zv Test runWithPatches restores the original values even when the function raises an exception. rrcs,jjdjjdtddS)NrblahblahzSomething went wrong!)rrr r RuntimeErrorrr rr_szDMonkeyPatcherTests.test_runWithPatchesRestoresOnException.._r rr r=N) rr rr"r>r5rr rr )rr?rr r&test_runWithPatchesRestoresOnExceptions z9MonkeyPatcherTests.test_runWithPatchesRestoresOnExceptionc CsL|j|jdd|j||jjdW5QRX||jj|jjdS)z L{MonkeyPatcher} is a context manager that applies its patches on entry and restore original values on exit. r z patched valueN)rr rrr rr rrrtest_contextManagersz&MonkeyPatcherTests.test_contextManagerc Cs4|t |jtdW5QRXW5QRXdS)zh Exceptions propagate through the L{MonkeyPatcher} context-manager exit method. Z somethingN)r"r>rr rrr'test_contextManagerPropagatesExceptionss z:MonkeyPatcherTests.test_contextManagerPropagatesExceptionsN)rrr__doc__rrrr!r$r'r(r8r9r<r@rArBrrrrrs       rN) rC __future__rZtyping_extensionsrZtwisted.python.monkeyrZ twisted.trialrrZSynchronousTestCaserrrrrs