shell bypass 403
o
a b�g � @ sN d Z dZdZddlZddlZddlmZ ddlmZm Z m
Z
G dd� de�Z d d
� Z G d d � d e
�Ze d
d� �ZG dd� de
�ZG dd� de�ZG dd� dej�ZG dd� de
�ZG dd� deje�ZG dd� deje�ZG dd � d eej�ZG d d � d ej�ZG d d � d ej�ZG d!d"� d"ej�ZG d#d$� d$ej�Ze d%kr�e� � dS dS )&z7Unit tests for the with statement specified in PEP 343.z
Mike Blandzmbland at acm dot org� N)�deque)�_GeneratorContextManager�contextmanager� nullcontextc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�MockContextManagerc s" t � j|� d| _d| _d | _d S �NF)�super�__init__� enter_called� exit_called� exit_args��self�args�� __class__� �7/usr/local/python-3.10/lib/python3.10/test/test_with.pyr s
zMockContextManager.__init__c C � d| _ t�| �S �NT)r
r � __enter__�r r r r r �
z MockContextManager.__enter__c C s" d| _ |||f| _t�| |||�S r )r r r �__exit__�r �type�value� tracebackr r r r s
�zMockContextManager.__exit__)�__name__�
__module__� __qualname__r r r �
__classcell__r r r r r
s r c s � fdd�}|S )Nc s t � | |�S �N)r )r �kwds��funcr r �helper s z#mock_contextmanager.<locals>.helperr )r% r& r r$ r �mock_contextmanager s r' c @ � e Zd Zdd� ZdS )� MockResourcec C s d| _ d| _d S r )�yielded�stoppedr r r r r & r zMockResource.__init__N�r r r r r r r r r) % � r) c c s* � t � } z d| _| V W d| _d S d| _w r )r) r* r+ )�mockr r r � mock_contextmanager_generator+ s �r/ c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�Nestedc G s || _ d | _d S r" )�managers�entered�r r2 r r r r 7 r zNested.__init__c C sh | j d ur td��t� | _ g }z| jD ]}|�|�� � | j �|� qW |S | jt� � � s1� Y |S )NzContext is not reentrant)
r3 � RuntimeErrorr r2 �appendr �
appendleftr �sys�exc_info)r �vars�mgrr r r r ; s
���zNested.__enter__c G s` |}| j D ]}z |j|� rd}W q t�� }Y qd | _ ||ur.|d |d ��|d ��d S )N�NNNr � � )r3 r r8 r9 �with_traceback)r r9 �exr; r r r r I s
� �zNested.__exit__N�r r r r r r r r r r r1 5 s r1 c @ r0 )�
MockNestedc G s( t j| g|�R � d| _d| _d | _d S r )r1 r r
r r r4 r r r r Z s
zMockNested.__init__c C r r )r
r1 r r r r r r ` r zMockNested.__enter__c G s d| _ || _tj| g|�R � S r )r r r1 r �r r9 r r r r d s zMockNested.__exit__NrA r r r r rB Y s rB c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�FailureTestCasec C � dd� }| � t|� d S )Nc S s* t � W d � d S 1 sw Y d S r" ��foor r r r �fooNotDeclaredl s *z5FailureTestCase.testNameError.<locals>.fooNotDeclared)� assertRaises� NameError)r rH r r r �
testNameErrork s z FailureTestCase.testNameErrorc �. G dd� dt �� � fdd�}| �td|� d S )Nc @ r( )z<FailureTestCase.testEnterAttributeError1.<locals>.LacksEnterc S � d S r" r r r r r r r � zEFailureTestCase.testEnterAttributeError1.<locals>.LacksEnter.__exit__N)r r r r r r r r �
LacksEnterq r- rO c �0 � � } | � W d � d S 1 sw Y d S r" r rF �rO r r �
fooLacksEnteru � *z?FailureTestCase.testEnterAttributeError1.<locals>.fooLacksEnterr ��object�assertRaisesRegex�AttributeError)r rR r rQ r �testEnterAttributeError1p � z(FailureTestCase.testEnterAttributeError1c rL )Nc @ � e Zd ZdS )zCFailureTestCase.testEnterAttributeError2.<locals>.LacksEnterAndExitN�r r r r r r r �LacksEnterAndExit{ s r\ c rP r" r rF �r\ r r �fooLacksEnterAndExit~ rS zFFailureTestCase.testEnterAttributeError2.<locals>.fooLacksEnterAndExitr rT )r r^ r r] r �testEnterAttributeError2z s z(FailureTestCase.testEnterAttributeError2c rL )Nc @ r( )z9FailureTestCase.testExitAttributeError.<locals>.LacksExitc S rM r" r r r r r r � rN zCFailureTestCase.testExitAttributeError.<locals>.LacksExit.__enter__N)r r r r r r r r � LacksExit� r- r` c rP r" r rF �r` r r � fooLacksExit� rS z<FailureTestCase.testExitAttributeError.<locals>.fooLacksExitr rT )r rb r ra r �testExitAttributeError� rY z&FailureTestCase.testExitAttributeErrorc C s dd� }| � t||� d S )Nc S s t | dd� d S )N� Zsingle)�compile)�sr r r �shouldRaiseSyntaxError� s zGFailureTestCase.assertRaisesSyntaxError.<locals>.shouldRaiseSyntaxError)rI � SyntaxError)r Zcodestrrg r r r �assertRaisesSyntaxError� s z'FailureTestCase.assertRaisesSyntaxErrorc C � | � d� | � d� d S )Nzwith mock as None:
passzwith mock as (None):
pass�ri r r r r �testAssignmentToNoneError� �
�z)FailureTestCase.testAssignmentToNoneErrorc C rj )Nzwith mock as None,:
passz with mock as (None,):
passrk r r r r �,testAssignmentToTupleOnlyContainingNoneError� rm z<FailureTestCase.testAssignmentToTupleOnlyContainingNoneErrorc C s | � d� d S )Nz%with mock as (foo, None, bar):
passrk r r r r �(testAssignmentToTupleContainingNoneError� s �z8FailureTestCase.testAssignmentToTupleContainingNoneErrorc s< G dd� dt �� � �fdd�}��t|� ���jd � d S )Nc @ � e Zd Zdd� Zdd� ZdS )z4FailureTestCase.testEnterThrows.<locals>.EnterThrowsc S � t d��)Nz Enter threw�r5 r r r r r � � z>FailureTestCase.testEnterThrows.<locals>.EnterThrows.__enter__c W rM r" r r
r r r r � rN z=FailureTestCase.testEnterThrows.<locals>.EnterThrows.__exit__N�r r r r r r r r r � EnterThrows� � ru c s8 � � } d �_ | �
�_ W d � d S 1 sw Y d S r" rF )�ct�ru r r r � shouldThrow� s
"�z4FailureTestCase.testEnterThrows.<locals>.shouldThrow)rU rI r5 � assertEqualrG �r ry r rx r �testEnterThrows� s z FailureTestCase.testEnterThrowsc �, G dd� dt �� � fdd�}| �t|� d S )Nc @ rp )z2FailureTestCase.testExitThrows.<locals>.ExitThrowsc S rM r" r r r r r r � rN z<FailureTestCase.testExitThrows.<locals>.ExitThrows.__enter__c W rq )N�* rr r
r r r r � rs z;FailureTestCase.testExitThrows.<locals>.ExitThrows.__exit__Nrt r r r r �
ExitThrows� rv r c s, � � � W d � d S 1 sw Y d S r" r r �r r r ry � � "�z3FailureTestCase.testExitThrows.<locals>.shouldThrow)rU rI r5 r{ r r� r �testExitThrows� s z FailureTestCase.testExitThrowsN)
r r r rK rX r_ rc ri rl rn ro r| r� r r r r rD j s
rD c @ sX e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
dd� Z dS )� ContextmanagerAssertionMixinc C s t d�| _d S )Nztest exception)r5 �TEST_EXCEPTIONr r r r �setUp� s z"ContextmanagerAssertionMixin.setUpc C s* | � |j� | �|j� | �|jd � d S r" )�
assertTruer
� assertFalser rz r �r � mock_managerr r r � assertInWithManagerInvariants� � z:ContextmanagerAssertionMixin.assertInWithManagerInvariantsc C s* | � |j� | � |j� | �|j|� d S r" )r� r
r rz r )r r� r r r r � assertAfterWithManagerInvariants� r� z=ContextmanagerAssertionMixin.assertAfterWithManagerInvariantsc C s | � |d� d S )Nr< )r� r� r r r �'assertAfterWithManagerInvariantsNoError� s �zDContextmanagerAssertionMixin.assertAfterWithManagerInvariantsNoErrorc C s | � |j� | �|j� d S r" )r� r* r� r+ �r Zmock_generatorr r r � assertInWithGeneratorInvariants� � z<ContextmanagerAssertionMixin.assertInWithGeneratorInvariantsc C � | � |j� | � |j� d S r" �r� r* r+ r� r r r �)assertAfterWithGeneratorInvariantsNoError� r� zFContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsNoErrorc C s | j �r" )r� r r r r �raiseTestException� � z/ContextmanagerAssertionMixin.raiseTestExceptionNc C sx | � |j� | � |j� |d u r | �|jd | j� t| j�}| �|jd |� | �|jd |� | �|jd d � d S )Nr= r r> ) r� r
r rz r r� r ZassertIsInstanceZ assertIsNot)r r� �exc_typer r r �)assertAfterWithManagerInvariantsWithError� s
zFContextmanagerAssertionMixin.assertAfterWithManagerInvariantsWithErrorc C r� r" r� r� r r r �+assertAfterWithGeneratorInvariantsWithError� r� zHContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsWithErrorr" ) r r r r� r� r� r� r� r� r� r� r� r r r r r� � s
� r� c @ �D e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�NonexceptionalTestCasec C s, t � � W d � d S 1 sw Y d S r" �r/ r r r r �testInlineGeneratorSyntax� r� z0NonexceptionalTestCase.testInlineGeneratorSyntaxc C s8 t � }|� W d � n1 sw Y | �|� d S r" )r/ r� )r r. r r r �testUnboundGenerator� s
�z+NonexceptionalTestCase.testUnboundGeneratorc C s> t � �
}| �|� W d � n1 sw Y | �|� d S r" �r/ r� r� �r rG r r r � testInlineGeneratorBoundSyntax� s �z5NonexceptionalTestCase.testInlineGeneratorBoundSyntaxc C sB d }t � �
}| �|� W d � n1 sw Y | �|� d S r" r� r� r r r �*testInlineGeneratorBoundToExistingVariable� s
�zANonexceptionalTestCase.testInlineGeneratorBoundToExistingVariablec C sD t � �| _| �| j� W d � n1 sw Y | �| j� d S r" )r/ rG r� r� r r r r �(testInlineGeneratorBoundToDottedVariable s
�z?NonexceptionalTestCase.testInlineGeneratorBoundToDottedVariablec C sV t � }|�}| �|� | �|� W d � n1 sw Y | �|� | �|� d S r" )r/ r� r� r� r� )r r. rG r r r �testBoundGenerator s
�
z)NonexceptionalTestCase.testBoundGeneratorc C s� t � }|�E}t � }|� }| �|� | �|� | �|� | �|� W d � n1 s*w Y | �|� | �|� | �|� | �|� W d � n1 sMw Y | �|� | �|� d S r" )r/ r� r� r� r� )r �mock_arG �mock_b�barr r r �testNestedSingleStatements s
�
�
z1NonexceptionalTestCase.testNestedSingleStatementsN)
r r r r� r� r� r� r� r� r� r r r r r� � s r� c @ r� )� NestedNonexceptionalTestCasec C s0 t t� �� W d � d S 1 sw Y d S r" �r1 r/ r r r r �"testSingleArgInlineGeneratorSyntax# s "�z?NestedNonexceptionalTestCase.testSingleArgInlineGeneratorSyntaxc C �F t � }t|��
}| �|� W d � n1 sw Y | �|� d S r" �r/ r1 r� r� �r �mrG r r r � testSingleArgBoundToNonTuple' �
�z9NestedNonexceptionalTestCase.testSingleArgBoundToNonTuplec C r� r" r� r� r r r �2testSingleArgBoundToSingleElementParenthesizedList/ r� zONestedNonexceptionalTestCase.testSingleArgBoundToSingleElementParenthesizedListc C rE )Nc S s4 t t� �� \} }W d � d S 1 sw Y d S r" r� )rG r� r r r �shouldThrowValueError8 s "�ziNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleError.<locals>.shouldThrowValueError)rI �
ValueError)r r� r r r �-testSingleArgBoundToMultipleElementTupleError7 s zJNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleErrorc C s^ t � }t|�}|� | �|� | �|� W d � n1 s w Y | �|� | �|� d S r" �r/ rB r� r� )r r' � mock_nestedr r r �testSingleArgUnbound= s
�
z1NestedNonexceptionalTestCase.testSingleArgUnboundc C s� t � }t � }t � }t|||�}|� | �|� | �|� | �|� | �|� W d � n1 s0w Y | �|� | �|� | �|� | �|� d S r" r� )r r� �n�or� r r r �testMultipleArgUnboundF s
�
z3NestedNonexceptionalTestCase.testMultipleArgUnboundc C s� t t� t� t� �}|� \}}}| �|� | �|� | �|� | �|� W d � n1 s-w Y | �|� | �|� | �|� | �|� d S r" )rB r/ r� r� r� r� )r r� r� r� r� r r r �testMultipleArgBoundU s �
�
z1NestedNonexceptionalTestCase.testMultipleArgBoundN)
r r r r� r� r� r� r� r� r� r r r r r� ! s r� c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�ExceptionalTestCasec s: t � � � �fdd�}��t|� ��� � ���j� d S )Nc sJ � ��_ ��� � ���j � ��� W d � d S 1 s w Y d S r" )�resourcer� r� r� r ��cmr r r ry f s
"�z;ExceptionalTestCase.testSingleResource.<locals>.shouldThrow)r/ rI r5 r� r� r� r{ r r� r �testSingleResourced s
z&ExceptionalTestCase.testSingleResourcec s0 t � � � �fdd�}��t|� ��� t� d S )Nc s4 � ��_ dd W d � d S 1 sw Y d S �Nr= r )r� r r� r r ry q s
"�z@ExceptionalTestCase.testExceptionNormalized.<locals>.shouldThrow)r/ rI �ZeroDivisionErrorr� r{ r r� r �testExceptionNormalizedo s z+ExceptionalTestCase.testExceptionNormalizedc sX t � � t � �� ��fdd�}��t|� ��� � ���� ���j� ���j� d S )Nc s� � �?�_ ��#�_��� � ���� ���j � ���j� ��� W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S r" �rG r� r� r� r� r �r� r� r r r ry | s
��"�zCExceptionalTestCase.testNestedSingleStatements.<locals>.shouldThrow)r/ rI r5 r� r� rG r� r{ r r� r r� y �
z.ExceptionalTestCase.testNestedSingleStatementsc sn t � � t � �t� ���� ���fdd�}��t|� ��� � ���� ���� ���j� ���j� d S )Nc sp ��,\�_ �_��� � ���� ���� ���j � ���j� ��� W d � d S 1 s1w Y d S r" )�
resource_a�
resource_br� r� r� r �Zcm_aZcm_br� r r r ry � s
"�zOExceptionalTestCase.testMultipleResourcesInSingleStatement.<locals>.shouldThrow)r/ rB rI r5 r� r� r� r� r{ r r� r �&testMultipleResourcesInSingleStatement� s
z:ExceptionalTestCase.testMultipleResourcesInSingleStatementc s| t � � t � �d �_� ��fdd�}��t|� ��� � ���j� ���jd � ���j � ���j
� ���j d � d S )Nc s� � �4�_ ��� � ���j � ��� �� �_W d � n1 s!w Y W d � d S W d � d S 1 s9w Y d S r" )rG r� r� r� r� r r� r r ry � s
��"�zPExceptionalTestCase.testNestedExceptionBeforeInnerStatement.<locals>.shouldThrow) r/ r� rI r5 r� r� rG rz r� r
r r r{ r r� r �'testNestedExceptionBeforeInnerStatement� s
z;ExceptionalTestCase.testNestedExceptionBeforeInnerStatementc sX t � � t � �� ��fdd�}��t|� ��� � ���� ���j� ���j� d S )Nc s� � �7�_ �� �_��� � ���� ���j � ���j� W d � n1 s(w Y ��� W d � d S 1 s<w Y d S r" r� r r� r r ry � s
�
"�zOExceptionalTestCase.testNestedExceptionAfterInnerStatement.<locals>.shouldThrow) r/ rI r5 r� r� r� rG r� r� r{ r r� r �&testNestedExceptionAfterInnerStatement� r� z:ExceptionalTestCase.testNestedExceptionAfterInnerStatementc sP t dd� �� � fdd�}| �td�� |� W d � d S 1 s!w Y d S )Nc s � � d V d S r" r r r r r r� � � �
z8ExceptionalTestCase.testRaisedStopIteration1.<locals>.cmc �$ � � � t d��1 s w Y d S �N� from with��
StopIterationr �r� r r ry � � �zAExceptionalTestCase.testRaisedStopIteration1.<locals>.shouldThrowr� )r rV r� r{ r r� r �testRaisedStopIteration1� s
"�z,ExceptionalTestCase.testRaisedStopIteration1c sT G dd� dt �� � fdd�}| �td�� |� W d � d S 1 s#w Y d S )Nc @ rp )z8ExceptionalTestCase.testRaisedStopIteration2.<locals>.cmc S rM r" r r r r r r � rN zBExceptionalTestCase.testRaisedStopIteration2.<locals>.cm.__enter__c S rM r" r r r r r r � rN zAExceptionalTestCase.testRaisedStopIteration2.<locals>.cm.__exit__Nrt r r r r r� � rv r� c r� r� r� r r� r r ry � r� zAExceptionalTestCase.testRaisedStopIteration2.<locals>.shouldThrowr� )rU rV r� r{ r r� r �testRaisedStopIteration2� s
"�z,ExceptionalTestCase.testRaisedStopIteration2c sN t dd� �� � fdd�}| �t�� |� W d � d S 1 s w Y d S )Nc s r� r" r r r r r r� � r� z8ExceptionalTestCase.testRaisedStopIteration3.<locals>.cmc s( � � � t tg ���1 s
w Y d S r" )�next�iterr r� r r ry � s �zAExceptionalTestCase.testRaisedStopIteration3.<locals>.shouldThrow)r rI r� r{ r r� r �testRaisedStopIteration3� s
"�z,ExceptionalTestCase.testRaisedStopIteration3c s( t dd� �� � fdd�}| �t|� d S )Nc s r� r" r r r r r r� � r� z8ExceptionalTestCase.testRaisedGeneratorExit1.<locals>.cmc r� r� ��
GeneratorExitr r� r r ry � r� zAExceptionalTestCase.testRaisedGeneratorExit1.<locals>.shouldThrow)r rI r� r{ r r� r �testRaisedGeneratorExit1� s
z,ExceptionalTestCase.testRaisedGeneratorExit1c r} )Nc @ rp )z8ExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cmc S rM r" r r r r r r � rN zBExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cm.__enter__c S rM r" r r r r r r � rN zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cm.__exit__Nrt r r r r r� � rv r� c r� r� r� r r� r r ry r� zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.shouldThrow)rU rI r� r{ r r� r �testRaisedGeneratorExit2� s z,ExceptionalTestCase.testRaisedGeneratorExit2c s\ G dd� dt �� � �fdd�}|� � �fdd�}��t|� � �fdd�}��t|� d S ) Nc @ r0 )z0ExceptionalTestCase.testErrorsInBool.<locals>.cmc s G � fdd�d�}|� | _ d S )Nc s e Zd Z� fdd�ZdS )zGExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__.<locals>.Boolc s � � S r" r r ��bool_conversionr r �__bool__ r� zPExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__.<locals>.Bool.__bool__N)r r r r� r r� r r �Bool
s r� �Z exit_result)r r� r� r r� r r s z9ExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__c S � dS )N� r r r r r r rN z:ExceptionalTestCase.testErrorsInBool.<locals>.cm.__enter__c S s | j S r" r� �r �a�b�cr r r r r� z9ExceptionalTestCase.testErrorsInBool.<locals>.cm.__exit__NrA r r r r r� s r� c �< � dd� �� �� d� W d � d S 1 sw Y d S )Nc S r� r r r r r r �<lambda> � zJExceptionalTestCase.testErrorsInBool.<locals>.trueAsBool.<locals>.<lambda>�Should NOT see this��failr r� r r �
trueAsBool � "�z8ExceptionalTestCase.testErrorsInBool.<locals>.trueAsBoolc r� )Nc S r� r r r r r r r� r� zKExceptionalTestCase.testErrorsInBool.<locals>.falseAsBool.<locals>.<lambda>z Should raiser� r r� r r � falseAsBool r� z9ExceptionalTestCase.testErrorsInBool.<locals>.falseAsBoolc r� )Nc S s dd S r� r r r r r r� ! s zJExceptionalTestCase.testErrorsInBool.<locals>.failAsBool.<locals>.<lambda>r� r� r r� r r �
failAsBool r� z8ExceptionalTestCase.testErrorsInBool.<locals>.failAsBool)rU rI �AssertionErrorr� )r r� r� r� r r� r �testErrorsInBool s z$ExceptionalTestCase.testErrorsInBoolN)r r r r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� c s
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �NonLocalFlowControlTestCasec C sX d} |d7 }t � �
|d7 } W d � q$1 sw Y |d7 }q| �|d� d S )Nr Tr= �
�d � �r/ rz �r �counterr r r �
testWithBreak( s ��z)NonLocalFlowControlTestCase.testWithBreakc C sb d} |d7 }|dkr q)t � �
|d7 } W d � q1 s w Y |d7 }q| �|d� d S )Nr Tr= r> r� r� � r� r� r r r �testWithContinue2 s ��z,NonLocalFlowControlTestCase.testWithContinuec C s dd� }| � |� d� d S )Nc S sJ d} | d7 } t � � | d7 } | W d � S 1 sw Y | d7 } q)Nr Tr= r� r� r� )r� r r r rG ? s ��z7NonLocalFlowControlTestCase.testWithReturn.<locals>.foor� )rz r� r r r �testWithReturn> s z*NonLocalFlowControlTestCase.testWithReturnc C s& dd� }t |� �}| �|ddg� d S )Nc s s: � t � � dV dV W d � d S 1 sw Y d S )Nr� �
r� r r r r �genJ s
�"�z6NonLocalFlowControlTestCase.testWithYield.<locals>.genr� r )�listrz )r r �xr r r �
testWithYieldI s
z)NonLocalFlowControlTestCase.testWithYieldc C sl d}z |d7 }t � � |d7 }t�1 sw Y |d7 }W n ty. | �|d� Y d S w | �d� d S )Nr r= r� r� r� zDidn't raise RuntimeError)r/ r5 rz r� r� r r r �
testWithRaiseQ s � �z)NonLocalFlowControlTestCase.testWithRaiseN)r r r r� r� r r r r r r r r� & s
r� c @ r0 )�AssignmentTargetTestCasec C sj dg d�i}t � �"|d d< | �t|�� �dg� | �|d d jt� W d � n1 s.w Y t � �&t|�� �d d< | �t|�� �dg� | �|d d jt� W d � n1 s_w Y t � � |d< t|�� �}|�� | �|ddg� W d � n1 s�w Y G dd� d�}|� }t � �|_| �t |d�d� W d � d S 1 s�w Y d S ) Nr= �r r= r> r r> c @ rZ )z;AssignmentTargetTestCase.testSingleComplexTarget.<locals>.CNr[ r r r r �Cm � r rG T)
r/ rz r �keysr r) �values�sortrG �hasattr)r �targetsr r �blahr r r �testSingleComplexTargeta s& � � �
"�z0AssignmentTargetTestCase.testSingleComplexTargetc C s� G dd� d�}dg d�i}|� �!\|d d< |d d< |d d< | � |dg d�i� W d � n1 s4w Y |� �-\t|�� �d d< t|�� �d d< t|�� �d d< | � |dg d�i� W d � n1 slw Y |� �\|d< |d< |d < | � |ddd d�� W d � n1 s�w Y G d
d � d �}|� }|� �$\|_|_|_| � |jd� | � |jd� | � |jd � W d � d S 1 s�w Y d S ) Nc @ rp )z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.Cc S r� )N�r= r> r� r r r r r r t r� zHAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__enter__c S rM r" r �r �t�v�tbr r r r u r� zGAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__exit__Nrt r r r r r s � r r= r r r> r )r� r> r= r� c @ rZ )z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.BNr[ r r r r �B} r
r )rz r r ZoneZtwoZthree)r r r r r r r r �testMultipleComplexTargetsr s$ & �> � �"�z3AssignmentTargetTestCase.testMultipleComplexTargetsc C sd t tdd���!�^}}}| �|d� | �|ddg� | �|d� W d � d S 1 s+w Y d S )Nr= � r> r� � )r �rangerz r� r r r �testWithExtendedTargets� s
"�z0AssignmentTargetTestCase.testWithExtendedTargetsN)r r r r r r r r r r r _ s r c @ rp )� ExitSwallowsExceptionTestCasec C sf G dd� d�}z |� � dd W d � W d S 1 s w Y W d S t y2 | �d� Y d S w )Nc @ rp )zSExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallowc S rM r" r r r r r r � r� z]ExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallow.__enter__c S r� r r r r r r r � r� z\ExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallow.__exit__Nrt r r r r �AfricanSwallow� r r r= r z,ZeroDivisionError should have been swallowed�r� r� )r r r r r � testExitTrueSwallowsException� s
&� �z;ExitSwallowsExceptionTestCase.testExitTrueSwallowsExceptionc C sd G dd� d�}z|� � dd W d � n1 sw Y W n
t y* Y d S w | �d� d S )Nc @ rp )zZExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallowc S rM r" r r r r r r � r� zdExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallow.__enter__c S r� r r r r r r r � r� zcExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallow.__exit__Nrt r r r r �EuropeanSwallow� r r" r= r z)ZeroDivisionError should have been raisedr )r r" r r r �#testExitFalseDoesntSwallowException� s
�� �zAExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowExceptionN)r r r r! r# r r r r r � s
r c @ st e Zd ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )�
NestedWithc @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
zNestedWith.DummyNFc C s( |d u r| }|| _ || _d| _d| _d S r )r �gobbler
r )r r r% r r r r � s
zNestedWith.Dummy.__init__c C s d| _ | jS r )r
r r r r r r � s zNestedWith.Dummy.__enter__c G s d| _ || _| jr dS d S r )r r9 r% rC r r r r � s
�zNestedWith.Dummy.__exit__r rA r r r r �Dummy� s
r& c @ r( )zNestedWith.InitRaisesc C � t � �r" rr r r r r r � � z NestedWith.InitRaises.__init__Nr, r r r r �
InitRaises� r- r) c @ rp )zNestedWith.EnterRaisesc C r' r" rr r r r r r � r( z NestedWith.EnterRaises.__enter__c G rM r" r rC r r r r � r� z NestedWith.EnterRaises.__exit__Nrt r r r r � EnterRaises� r r* c @ rp )zNestedWith.ExitRaisesc C rM r" r r r r r r � r� z NestedWith.ExitRaises.__enter__c G r' r" rr rC r r r r � r( z NestedWith.ExitRaises.__exit__Nrt r r r r �
ExitRaises� r r+ c C s� | � � �(}| � � �}| �|j� | �|j� W d � n1 s w Y W d � n1 s/w Y | �|j� | �|j� d S r" )r& r� r
r )r r� r� r r r �testNoExceptions� s �� zNestedWith.testNoExceptionsc C sz z*| � � � }| �� � W d � n1 sw Y W d � n1 s$w Y W n Y | �|j� | �|j� d S r" )r& r) r� r
r �r r� r r r �testExceptionInExprList� s �� � z"NestedWith.testExceptionInExprListc C s� z/| � � �!}| �� �
| �d� W d � n1 sw Y W d � n1 s)w Y W n ty8 Y nw | �d� | �|j� | �|j� d S )Nzbody of bad with executedzRuntimeError not reraised)r& r* r� r5 r� r
r r- r r r �testExceptionInEnter� s �� � �
z NestedWith.testExceptionInEnterc C s� d}| j dd�� }| �� �
d}W d � n1 sw Y W d � n1 s)w Y | �|j� | �|j� | �|� | �|jd d � d S )NFT)r% r )r& r+ r� r
r ZassertNotEqualr9 )r Z
body_executedr� r r r �testExceptionInExit� s ��
z NestedWith.testExceptionInExitc C s� | j dd��C\}}| j dd��"\}}| �d|� | �d|� | �d|� | �d|� W d � n1 s4w Y W d � d S W d � d S 1 sLw Y d S )N)r= r> )r )r� � r= r> r� r1 )r& rz )r Za1Za2�b1�b2r r r �testEnterReturnsTuple� s
� P�z NestedWith.testEnterReturnsTupleN)
r r r rU r& r) r* r+ r, r. r/ r0 r4 r r r r r$ � s r$ �__main__) �__doc__�
__author__Z __email__r8 Zunittest� collectionsr �
contextlibr r r r r' rU r) r/ r1 rB ZTestCaserD r� r� r� r� r� r r r$ r �mainr r r r �<module> s:
$S/5�B D9,L �