Your IP : 216.73.216.213
�
g fg �
@ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Td d l m Z y d d l Z Wn e k
rz d Z Yn XGd d � d e j
� Z Gd d � d e j
� Z Gd d
� d
e j
� Z
e j e d � Gd d
� d
e j
� � Z Gd d � d e � Z Gd d � d e j
� Z Gd d � d e j
� Z Gd d � d e j
� Z Gd d � d e j
� Z e d k r�e j � n d S)z9Unit tests for contextlib.py, and other context managers.� N)�*)�supportc @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z e
j d d � � Z e
j d d � � Z
d d � Z d S)�ContextManagerTestCasec s} g � t � f d d � � } | � �7 } | j � d g � | j | d � � j | � Wd QX| j � d d d g � d S)Nc 3 s# � j d � d V� j d � d S)N� �* i� )�append� )�stater �9/opt/alt/python34/lib64/python3.4/test/test_contextlib.py�woohoo s
z@ContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor r i� )�contextmanager� assertEqualr )�selfr �xr )r r
�test_contextmanager_plain s z0ContextManagerTestCase.test_contextmanager_plainc s� g � t � f d d � � } | j t � �P | � �@ } | j � d g � | j | d � � j | � t � � Wd QXWd QX| j � d d d g � d S)Nc 3 s+ � j d � z d VWd � j d � Xd S)Nr r i� )r r )r r r
r s
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor r i� )r � assertRaises�ZeroDivisionErrorr
r )r r r r )r r
�test_contextmanager_finally s
z2ContextManagerTestCase.test_contextmanager_finallyc C sK t d d � � } | � } | j � | j | j t t d � d � � d S)Nc s s d Vd S)Nr r r r r
�whee0 s zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r � __enter__� assertFalse�__exit__� TypeError)r r �ctxr r r
� test_contextmanager_no_reraise/ s
z5ContextManagerTestCase.test_contextmanager_no_reraisec C sK t d d � � } | � } | j � | j t | j t t d � d � d S)Nc s s y d VWn
d VYn Xd S)Nr r r r r
�whoo9 s zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoor )r r r � RuntimeErrorr r )r r r r r r
�*test_contextmanager_trap_yield_after_throw8 s
zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc s� g � t � � f d d � � } | � �C } � j � d g � � j | d � � j | � t d � � Wd QX� j � d d d g � d S)Nc 3 so � j d � y d VWnR t k
rj } z2 � j | j d � � j � d d d g � WYd d } ~ Xn Xd S)Nr r r i� )r r �argsr
)�e)r r r r
r G s
zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor r i� )r r
r r )r r r r )r r r
�test_contextmanager_exceptE s
z1ContextManagerTestCase.test_contextmanager_exceptc C s1 d d � } t | d d � d d � � � } | S)Nc s � f d d � } | S)Nc s1 x* � j � D] \ } } t | | | � q
W| S)N)�items�setattr)�func�k�v)�kwr r
�decorateX s zXContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )r' r( r )r' r
�attribsW s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribsr �barc S s d S)zWhee!Nr )Zspamr r r
�baz] s zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz)r )r r) r+ r r r
� _create_contextmanager_attribsV s z5ContextManagerTestCase._create_contextmanager_attribsc C s6 | j � } | j | j d � | j | j d � d S)Nr+ r* )r, r
�__name__r )r r+ r r r
�test_contextmanager_attribsc s z2ContextManagerTestCase.test_contextmanager_attribsc C s# | j � } | j | j d � d S)NzWhee!)r, r
�__doc__)r r+ r r r
� test_contextmanager_doc_attribh s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s) | j � d � } | j | j d � d S)NzWhee!)r, r
r/ )r r+ r r r
�*test_instance_docstring_given_cm_docstringm s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc
C sP t d d � � } | d d d d d d d d
� � } | j | d � Wd QXd S) Nc s s | | | | f Vd S)Nr )r r$ r �kwdsr r r
r t s z4ContextManagerTestCase.test_keywords.<locals>.woohoor � r$ � r �! r2 �, )r3 r4 r5 r6 )r r
)r r �targetr r r
�
test_keywordsr s $z$ContextManagerTestCase.test_keywordsN)r- �
__module__� __qualname__r r r r r! r, r. r �requires_docstringsr0 r1 r8 r r r r
r s
r c @ s= e Z d Z e j d d � � Z d d � Z d d � Z d S)�ClosingTestCasec C s, t j } t d � } | j | j | � d S)N)�closingr/ r
)r � cm_docstring�objr r r
�test_instance_docs} s z"ClosingTestCase.test_instance_docsc
st g � G� f d d � d � } | � } | j � g � t | � � } | j | | � Wd QX| j � d g � d S)Nc s" e Z d Z � f d d � Z d S)z'ClosingTestCase.test_closing.<locals>.Cc s � j d � d S)Nr )r )r )r r r
�close� s z-ClosingTestCase.test_closing.<locals>.C.closeN)r- r9 r: rA r )r r r
�C� s rB r )r
r= )r rB r �yr )r r
� test_closing� s z ClosingTestCase.test_closingc s� g � G� f d d � d � } | � } | j � g � | j t � �2 t | � � } | j | | � d d Wd QXWd QX| j � d g � d S)Nc s" e Z d Z � f d d � Z d S)z-ClosingTestCase.test_closing_error.<locals>.Cc s � j d � d S)Nr )r )r )r r r
rA � s z3ClosingTestCase.test_closing_error.<locals>.C.closeN)r- r9 r: rA r )r r r
rB � s rB r r )r
r r r= )r rB r rC r )r r
�test_closing_error� s z"ClosingTestCase.test_closing_errorN)r- r9 r: r r; r@ rD rE r r r r
r< { s r< c @ s e Z d Z d d � Z d S)�FileContextTestCasec C s� t j � } z� d } t | d � �$ } | j | j � | j d � Wd QX| j | j � d } | j t � �K t | d � �5 } | j | j � | j | j
� d � d d Wd QXWd QX| j | j � Wd t j | � Xd S)N�wzBooh
�rr r )
�tempfileZmktemp�openr �closed�write�
assertTruer r r
�readr �unlink)r Ztfn�fr r r
� testWithOpen� s z FileContextTestCase.testWithOpenN)r- r9 r: rQ r r r r
rF � s rF z!Threading required for this test.c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�LockContextTestCasec C s� | j | � � | � | j | � � Wd QX| j | � � | j t � �* | � | j | � � d d Wd QXWd QX| j | � � d S)Nr r )r rM r r )r �lock�lockedr r r
� boilerPlate� s z LockContextTestCase.boilerPlatec C s# t j � } | j | | j � d S)N)� threadingZLockrU rT )r rS r r r
� testWithLock� s z LockContextTestCase.testWithLockc C s# t j � } | j | | j � d S)N)rV �RLockrU � _is_owned)r rS r r r
�
testWithRLock� s z!LockContextTestCase.testWithRLockc s2 t j � � � f d d � } | j � | � d S)Nc s
� j � S)N)rY r )rS r r
rT � s z5LockContextTestCase.testWithCondition.<locals>.locked)rV Z ConditionrU )r rT r )rS r
�testWithCondition� s z%LockContextTestCase.testWithConditionc s2 t j � � � f d d � } | j � | � d S)Nc s% � j d � r � j � d Sd Sd S)NFT)�acquire�releaser )rS r r
rT � s
z5LockContextTestCase.testWithSemaphore.<locals>.locked)rV Z SemaphorerU )r rT r )rS r
�testWithSemaphore� s z%LockContextTestCase.testWithSemaphorec s2 t j � � � f d d � } | j � | � d S)Nc s% � j d � r � j � d Sd Sd S)NFT)r\ r] r )rS r r
rT � s
z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)rV ZBoundedSemaphorerU )r rT r )rS r
�testWithBoundedSemaphore� s z,LockContextTestCase.testWithBoundedSemaphoreN) r- r9 r: rU rW rZ r[ r^ r_ r r r r
rR � s
rR c @ s@ e Z d Z d Z d Z d Z d Z d d � Z d d � Z d S)� mycontextz9Example decoration-compatible context manager for testingFNc C s
d | _ | S)NT)�started)r r r r
r � s zmycontext.__enter__c G s | | _ | j S)N)�exc�catch)r rb r r r
r � s zmycontext.__exit__) r- r9 r: r/ ra rb rc r r r r r r
r` � s r` c @ s� e Z d Z e j d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d � Z d S)�TestContextDecoratorc C s) t j } t � } | j | j | � d S)N)r` r/ r
)r r> r? r r r
r@ � s z'TestContextDecorator.test_instance_docsc C sO t � } | �' } | j | | � | j | j � Wd QX| j | j d � d S)N)NNN)r` �assertIsrM ra r
rb )r �context�resultr r r
�test_contextdecorator� s
z*TestContextDecorator.test_contextdecoratorc C s� t � } | j t d � � | � t d � � Wd QXWd QX| j | j � | j | j d t � t � } d | _ | � t d � � Wd QX| j | j � | j | j d t � d S)Nr r T)r` �assertRaisesRegex� NameError�assertIsNotNonerb re rc )r rf r r r
�$test_contextdecorator_with_exception s z9TestContextDecorator.test_contextdecorator_with_exceptionc sB t � � � � � f d d � � } | � � j � j d � d S)Nc s$ � j � j � � j � j � d S)N)� assertIsNonerb rM ra r )rf r r r
�test s z1TestContextDecorator.test_decorator.<locals>.test)NNN)r` r
rb )r rn r )rf r r
�test_decorator s z#TestContextDecorator.test_decoratorc so t � � � � � f d d � � } � j t d � � | � Wd QX� j � j � � j � j d t � d S)Nc s0 � j � j � � j � j � t d � � d S)Nr )rm rb rM ra rj r )rf r r r
rn ! s z@TestContextDecorator.test_decorator_with_exception.<locals>.testr r )r` ri rj rk rb re )r rn r )rf r r
� test_decorator_with_exception s
z2TestContextDecorator.test_decorator_with_exceptionc s t � � G� f d d � d t � } | � } | j d d � | j | j d � | j | j d � | j | j d � | � } | j d d d � | j | j d � | j | j d � | j | j d � | � } | j d d d d � | j | j d � | j | j d � d S)Nc s% e Z d Z � d d d � � Z d S)z9TestContextDecorator.test_decorating_method.<locals>.TestNc S s | | _ | | _ | | _ d S)N)�a�b�c)r rq rr rs r r r
�method2 s z@TestContextDecorator.test_decorating_method.<locals>.Test.method)r- r9 r: rt r )rf r r
�Test0 s ru r � rq rr rs )r` �objectrt r
rq rr rs )r ru rn r )rf r
�test_decorating_method- s z+TestContextDecorator.test_decorating_methodc C s@ Gd d � d t � } | j t � � | � � Wd QXWd QXd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z7TestContextDecorator.test_typo_enter.<locals>.mycontextc S s d S)Nr )r r r r
� __unter__M s zATestContextDecorator.test_typo_enter.<locals>.mycontext.__unter__c W s d S)Nr )r rb r r r
r O s z@TestContextDecorator.test_typo_enter.<locals>.mycontext.__exit__N)r- r9 r: ry r r r r r
r` L s r` )�ContextDecoratorr �AttributeError)r r` r r r
�test_typo_enterK s
z$TestContextDecorator.test_typo_enterc C s@ Gd d � d t � } | j t � � | � � Wd QXWd QXd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z6TestContextDecorator.test_typo_exit.<locals>.mycontextc S s d S)Nr )r r r r
r Y s z@TestContextDecorator.test_typo_exit.<locals>.mycontext.__enter__c W s d S)Nr )r rb r r r
�__uxit__[ s z?TestContextDecorator.test_typo_exit.<locals>.mycontext.__uxit__N)r- r9 r: r r} r r r r
r` X s r` )rz r r{ )r r` r r r
�test_typo_exitW s
z#TestContextDecorator.test_typo_exitc sq Gd d � d t � } Gd d � d | t � } | � � � � � f d d � � } | � � j � j d � d S)Nc @ s4 e Z d Z d Z d Z d d � Z d d � Z d S)zHTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontextFNc S s
d | _ | S)NT)ra )r r r r
r h s zRTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__enter__c W s
| | _ d S)N)rb )r rb r r r
r l s zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r- r9 r: ra rb r r r r r r
� somecontextd s r c @ s e Z d Z d S)zFTestContextDecorator.test_contextdecorator_as_mixin.<locals>.mycontextN)r- r9 r: r r r r
r` o s r` c s$ � j � j � � j � j � d S)N)rm rb rM ra r )rf r r r
rn s s zATestContextDecorator.test_contextdecorator_as_mixin.<locals>.test)NNN)rw rz r
rb )r r r` rn r )rf r r
� test_contextdecorator_as_mixinc s z3TestContextDecorator.test_contextdecorator_as_mixinc s� t � f d d � � } g � | d � � � f d d � � } | d � � j � d d d g � g � | d � � j � d d d g � d S) Nc 3 s# � j | � d V� j d � d S)Ni� )r )rC )r r r
r | s
zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor c s$ � j � d g � � j | � d S)Nr )r
r )r )r r r r
rn � s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.testZ somethingi� zsomething else)r r
)r r rn r )r r r
� test_contextmanager_as_decorator{ s !
z5TestContextDecorator.test_contextmanager_as_decoratorN)r- r9 r: r r; r@ rh rl ro rp rx r| r~ r� r� r r r r
rd � s rd c @ s� e Z d Z e j d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% S)&�
TestExitStackc C s) t j } t � } | j | j | � d S)N)� ExitStackr/ r
)r r> r? r r r
r@ � s z TestExitStack.test_instance_docsc C s t � � Wd QXd S)N)r� )r r r r
�test_no_resources� s
z TestExitStack.test_no_resourcesc s~ f i f d i f d i f f t d d � f d t d d � f d t d d � f g } g � � f d d � } t � �� } x� t | � D]� \ } } | r� | r� | j | | | � } nE | r� | j | | � } n* | r� | j | | � } n | j | � } | j | | � q� WxP | j D]E } | j | j | � | j | j | j � | j | j
| j
� qWWd QX| j � | � d S)
Nr rv Zexamplec s � j | | f � d S)zTest metadata propagationN)r )r r2 )rg r r
�_exit� s z*TestExitStack.test_callback.<locals>._exit)r )r rv )r )r rv ) �dictr� �reversed�callbackre �_exit_callbacks� __wrapped__ZassertNotEqualr- rm r/ r
)r Zexpectedr� �stackr r2 rP �wrapperr )rg r
�
test_callback� s. zTestExitStack.test_callbackc
sc t � � � f d d � } d d � } � f d d � } Gd d � d t � } t � �} | j | � � j | j d | � | | � } | j | � � j | j d j | � | j | � � j | j d
| � | | � } | j | � � j | j d j | � | j | � � j | j d | � | j | � � j | j d | � d d
Wd QXd S)Nc s � j | � � d S)N)re )�exc_typerb �exc_tb)�
exc_raisedr r r
� _expect_exc� s z,TestExitStack.test_push.<locals>._expect_excc W s d S)NTr )� exc_detailsr r r
�
_suppress_exc� s z.TestExitStack.test_push.<locals>._suppress_excc s+ � j | � � j | � � j | � d S)N)rm )r� rb r� )r r r
�
_expect_ok� s
z+TestExitStack.test_push.<locals>._expect_okc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z'TestExitStack.test_push.<locals>.ExitCMc S s
| | _ d S)N)� check_exc)r r� r r r
�__init__� s z0TestExitStack.test_push.<locals>.ExitCM.__init__c S s | j d � d S)NzShould not be called!)�fail)r r r r
r � s z1TestExitStack.test_push.<locals>.ExitCM.__enter__c W s | j | � d S)N)r� )r r� r r r
r � s z0TestExitStack.test_push.<locals>.ExitCM.__exit__N)r- r9 r: r� r r r r r r
�ExitCM� s r� r r ���r� r� r� r� r� )r rw r� �pushre r� �__self__)r r� r� r� r� r� �cmr )r� r r
� test_push� s*
zTestExitStack.test_pushc s� G� f d d � d t � } g � | � } t � �c } | j � f d d � � } | j | � | j | � | j | j d j | � � j d � Wd QX| j � d d d d g � d S)
Nc s4 e Z d Z � f d d � Z � f d d � Z d S)z0TestExitStack.test_enter_context.<locals>.TestCMc s � j d � d S)Nr )r )r )rg r r
r � s z:TestExitStack.test_enter_context.<locals>.TestCM.__enter__c s � j d � d S)N� )r )r r� )rg r r
r � s z9TestExitStack.test_enter_context.<locals>.TestCM.__exit__N)r- r9 r: r r r )rg r r
�TestCM� s r� c s � j d � d S)N� )r r )rg r r
r� � s z/TestExitStack.test_enter_context.<locals>._exitr rv r� r� r� )
rw r� r� rk �
enter_contextre r� r� r r
)r r� r� r� r� r )rg r
�test_enter_context� s
z TestExitStack.test_enter_contextc sq g � t � �F } | j � f d d � � } | j | � | j � � j d � Wd QX| j � d d g � d S)Nc s � j d � d S)Nr )r r )rg r r
r� � s z'TestExitStack.test_close.<locals>._exitrv r )r� r� rk rA r r
)r r� r� r )rg r
�
test_close� s
zTestExitStack.test_closec s� g � t � �H } | j � f d d � � } | j | � | j � } � j d � Wd QX� j d � | j � | j � d d d g � d S)Nc s � j d � d S)Nr� )r r )rg r r
r� � s z)TestExitStack.test_pop_all.<locals>._exitr rv r� )r� r� rk �pop_allr rA r
)r r� r� � new_stackr )rg r
� test_pop_all� s
zTestExitStack.test_pop_allc C sG | j t � �2 t � �" } | j d d � � d d Wd QXWd QXd S)Nc W s d S)NFr )rb r r r
�<lambda> s z/TestExitStack.test_exit_raise.<locals>.<lambda>r r )r r r� r� )r r� r r r
�test_exit_raise s z TestExitStack.test_exit_raisec
C s1 t � �" } | j d d � � d d Wd QXd S)Nc W s d S)NTr )rb r r r
r� s z2TestExitStack.test_exit_suppress.<locals>.<lambda>r r )r� r� )r r� r r r
�test_exit_suppress s z TestExitStack.test_exit_suppressc $ C s9 Gd d � d � } Gd d � d � } Gd d � d � } yX | t � �F | t t � �0 | � � | t � �
d d Wd QXWd QXWd QXWd QXWnd t k
r� } zD | j | j t � | j | j j t � | j | j j j � WYd d } ~ Xn X| j d � | j d } | j | t � | j | j t � d S)
Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zFTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcc S s
| | _ d S)N)rb )r rb r r r
r� s zOTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__init__c S s | S)Nr )r r r r
r s zPTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__enter__c W s
| j � d S)N)rb )r r� r r r
r s zOTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__N)r- r9 r: r� r r r r r r
�RaiseExc s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zQTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s | | _ | | _ d S)N)�outer�inner)r r� r� r r r
r� s zZTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__init__c S s | S)Nr )r r r r
r " s z[TestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__enter__c W s% y
| j � Wn | j � Yn Xd S)N)r� r� )r r� r r r
r $ s
zZTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__N)r- r9 r: r� r r r r r r
�RaiseExcWithContext s r� c @ s( e Z d Z d d � Z d d � Z d S)zITestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExcc S s | S)Nr )r r r r
r + s zSTestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__enter__c W s | t | � _ d S)NT)�type�
saved_details)r r� r r r
r - s zRTestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__N)r- r9 r: r r r r r r
� SuppressExc* s r� r r z0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr{ �
ValueError�assertIsInstance� __context__rm r� r� r )r r� r� r� rb � inner_excr r r
�&test_exit_exception_chaining_reference s"
$)
z4TestExitStack.test_exit_exception_chaining_referencec s4 d d � } d � � f d d � } yk t � �\ } | j | t � | j | t � | j | t � | j | � | j | t � d d Wd QXWnd t k
r� } zD | j | j t � | j | j j t � | j | j j j � WYd d } ~ Xn X| j
d � � d } | j | t � | j | j t � d S)Nc S s
| � d S)Nr )rb r r r
� raise_excE s z=TestExitStack.test_exit_exception_chaining.<locals>.raise_excc s
| � d S)NTr )r� )r� r r
� suppress_excI s z@TestExitStack.test_exit_exception_chaining.<locals>.suppress_excr r z0Expected IndexError, but no exception was raised) r� r� r� r� r{ r� r� r� r� rm r� r )r r� r� r� rb r� r )r� r
� test_exit_exception_chainingC s&
)
z*TestExitStack.test_exit_exception_chainingc C s" d d � } d d � } y9 t � �* } | j d d � � | j | t � Wd QXWn5 t k
r� } z | j | t � WYd d } ~ Xn X| j d � yC t � �4 } | j | t � | j | � | j | t � Wd QXWn5 t k
r} z | j | t � WYd d } ~ Xn X| j d � d S) Nc S s
| � d S)Nr )rb r r r
r� d s zDTestExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W s d S)NTr )r� r r r
r� g s zGTestExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S s d S)Nr r r r r
r� l s zCTestExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>z0Expected IndexError, but no exception was raisedz.Expected KeyError, but no exception was raised)r� r� r� � Exceptionr� r� r� r� )r r� r� r� rb r r r
�#test_exit_exception_non_suppressingb s" #
#z1TestExitStack.test_exit_exception_non_suppressingc C s. t d d � � } t d � } t d � } t d � } t d � } yU t � �F } | j | | � � | j | | � � | j | | � � | � Wd QXWn� t k
r)} zp | j | | � | j | j | � | j | j j | � | j | j j j | � | j | j j j j � WYd d } ~ Xn Xd S)Nc
s s z d VWd | � Xd S)Nr )rb r r r
�gets_the_context_right s zVTestExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr rv r� r� )r r� r� r� re r� rm )r r� �exc1�exc2�exc3�exc4r� rb r r r
�(test_exit_exception_with_correct_context} s$ z6TestExitStack.test_exit_exception_with_correct_contextc C s@ d d � } t d � } t d � } t d � } t d � } t d � } yB t � �3 } | j | | | � | j | | | � | � Wd QXWn� t k
r;} z� | j | | � | j | j | � | j | j j | � | j | j j j | � | j | j j j j | � | j | j j j j j � WYd d } ~ Xn Xd S)Nc
S s z
| � Wd | � Xd S)Nr )r� Z outer_excr r r
� raise_nested� s
zMTestExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr rv r� r� � )r� r� r� re r� rm ) r r� r� r� r� r� Zexc5r� rb r r r
�)test_exit_exception_with_existing_context� s( z7TestExitStack.test_exit_exception_with_existing_contextc C sp d d � } y+ t � � } | j | � d d Wd QXWn2 t k
rk } z | j d � WYd d } ~ Xn Xd S)Nc W s d S)NTr )r� r r r
r� � s z@TestExitStack.test_body_exception_suppress.<locals>.suppress_excr r z%Expected no exception, got IndexError)r� r� r� r� )r r� r� rb r r r
� test_body_exception_suppress� s
z*TestExitStack.test_body_exception_suppressc
C sO t � �@ } | j d d � � | j d d � � | j d d � � Wd QXd S)Nc W s d S)NTr )rb r r r
r� � s zETestExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s d d S)Nr r r )rb r r r
r� � s c W s i d S)Nr r )rb r r r
r� � s )r� r� )r r� r r r
�%test_exit_exception_chaining_suppress� s z3TestExitStack.test_exit_exception_chaining_suppressc
C s: t � �+ } x! t d � D] } | j t � q WWd QXd S)Ni' )r� �ranger� �int)r r� �ir r r
�test_excessive_nesting� s z$TestExitStack.test_excessive_nestingc C sr Gd d � d t � } | � } t � | _ t � } | j t | j | � | j | � | j | j d | � d S)Nc @ s e Z d Z d S)z3TestExitStack.test_instance_bypass.<locals>.ExampleN)r- r9 r: r r r r
�Example� s r� r r� ) rw r r� r r{ r� r� re r� )r r� r� r� r r r
�test_instance_bypass� s
z"TestExitStack.test_instance_bypassN)r- r9 r: r r; r@ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r� � s$ "
0
r� c @ sa e Z d Z e j d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z
d
S)�TestRedirectStdoutc C s, t j } t d � } | j | j | � d S)N)�redirect_stdoutr/ r
)r r> r? r r r
r@ � s z%TestRedirectStdout.test_instance_docsc C s* t j } t d � | j t j | � d S)N)�sys�stdoutr� re )r � orig_stdoutr r r
�test_no_redirect_in_init� s
z+TestRedirectStdout.test_no_redirect_in_initc C sq t j � } d } t j } t | � � t | � Wd QX| j t j | � | j � j � } | j | | � d S)Nz<Consider an API like help(), which prints directly to stdout)
�io�StringIOr� r� r� �printre �getvalue�stripr
)r rP �msgr� �sr r r
�test_redirect_to_string_io� s
z-TestRedirectStdout.test_redirect_to_string_ioc
C s5 t j � } t | � � } | j | | � Wd QXd S)N)r� r� r� re )r rP � enter_resultr r r
�test_enter_result_is_target� s z.TestRedirectStdout.test_enter_result_is_targetc C s� t j � } t | � } t j } | � t d d d �Wd QX| � t d � Wd QX| j t j | � | j � } | j | d � d S)N�Hello�end� zWorld!z
Hello World!
) r� r� r� r� r� r� re r� r
)r rP �
write_to_fr� r� r r r
�test_cm_is_reusable� s z&TestRedirectStdout.test_cm_is_reusablec C s� t j � } t | � } t j } | �, t d d d �| � t d � Wd QXWd QX| j t j | � | j � } | j | d � d S)Nr� r� r� zWorld!z
Hello World!
) r� r� r� r� r� r� re r� r
)r rP r� r� r� r r r
�test_cm_is_reentrant s z'TestRedirectStdout.test_cm_is_reentrantN) r- r9 r: r r; r@ r� r� r� r� r� r r r r
r� � s
r� c @ s� e Z d Z e j d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d S)� TestSuppressc C s) t j } t � } | j | j | � d S)N)�suppressr/ r
)r r> r? r r r
r@ s z TestSuppress.test_instance_docsc C s& t t � � } | j | � Wd QXd S)N)r� r� rm )r r� r r r
�test_no_result_from_enter s z&TestSuppress.test_no_result_from_enterc C s0 t t � � | j t d d � d � Wd QXd S)Nrv r� � )r� r� r
�pow)r r r r
�test_no_exception s
z TestSuppress.test_no_exceptionc C s! t t � � t d � Wd QXd S)Nr� )r� r �len)r r r r
�test_exact_exception s
z!TestSuppress.test_exact_exceptionc C s t t � �
d d Wd QXd S)Nr� �2 )r� � LookupError)r r r r
�test_exception_hierarchy$ s
z%TestSuppress.test_exception_hierarchyc C s5 | j t � � t t � �
d d Wd QXWd QXd S)Nr r )r r r� r )r r r r
�test_other_exception( s
z!TestSuppress.test_other_exceptionc C s2 | j t � � t � �
d d Wd QXWd QXd S)Nr r )r r r� )r r r r
� test_no_args- s
zTestSuppress.test_no_argsc C sB t t t � �
d d Wd QXt t t � � t d � Wd QXd S)Nr r r� )r� r r r� )r r r r
� test_multiple_exception_args2 s z)TestSuppress.test_multiple_exception_argsc C ss t t � } | � Wd QX| � t d � Wd QX| �* | � t d � Wd QXd } d d Wd QX| j | � d S)Nr� Tr r )r� r� r� rM )r Zignore_exceptionsZouter_continuedr r r
r� 8 s z!TestSuppress.test_cm_is_reentrantN)r- r9 r: r r; r@ r� r� r� r� r� r� r� r� r r r r
r� s r� �__main__)r/ r� r� rI Zunittest�
contextlibrn r rV � ImportErrorZTestCaser r< rF Z
skipUnlessrR rz r` rd r� r� r� r- �mainr r r r
�<module> s.
l! 0�� J66