Your IP : 216.73.216.85
U
i�f*� � @ sV d Z ddlZddlZddlZddlZddlZddlT ddlmZ ddl Z G dd� dej
�Z G dd� dej
�Z G d d
� d
ej
�Z
G d d � d ej
�ZG d
d� dej
�ZG dd� dej
�ZG dd� de�ZG dd� dej
�ZG dd� d�ZG dd� deej
�ZG dd� d�ZG dd � d eej
�ZG d d � d eej
�ZG d d � d ej
�Zed!k�rRe�� dS )"z9Unit tests for contextlib.py, and other context managers.� N)�*)�supportc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestAbstractContextManagerc C s* G dd� dt �}|� }| �|�� |� d S )Nc s e Zd Z� fdd�Z� ZS )z;TestAbstractContextManager.test_enter.<locals>.DefaultEnterc s t � j|� d S �N��super�__exit__��self�args�� __class__� �9/opt/alt/python38/lib64/python3.8/test/test_contextlib.pyr s zDTestAbstractContextManager.test_enter.<locals>.DefaultEnter.__exit__��__name__�
__module__� __qualname__r �
__classcell__r r r r � DefaultEnter s r )�AbstractContextManager�assertIs� __enter__)r
r Zmanagerr r r �
test_enter s z%TestAbstractContextManager.test_enterc C s0 G dd� dt �}| �t�� |� W 5 Q R X d S )Nc @ s e Zd ZdS )zETestAbstractContextManager.test_exit_is_abstract.<locals>.MissingExitN�r r r r r r r � MissingExit s r )r � assertRaises� TypeError)r
r r r r �test_exit_is_abstract s z0TestAbstractContextManager.test_exit_is_abstractc C s� G dd� d�}| � t|t�� G dd� dt�}| � t|t�� G dd� d|�}| �t|t�� G dd� d|�}| �t|t�� d S ) Nc @ s e Zd Zdd� Zdd� ZdS )zRTestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratchc S s | S r r �r
r r r r s z\TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__enter__c S s d S r r )r
�exc_type� exc_value� tracebackr r r r " s z[TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__exit__N�r r r r r r r r r �ManagerFromScratch s r$ c s e Zd Z� fdd�Z� ZS )zLTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnterc s t � j|� d S r r r r r r r ( s zUTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnter.__exit__r r r r r r ' s r c @ s e Zd ZdZdS )zGTestAbstractContextManager.test_structural_subclassing.<locals>.NoEnterN)r r r r r r r r �NoEnter- s r% c @ s e Zd ZdZdS )zFTestAbstractContextManager.test_structural_subclassing.<locals>.NoExitN)r r r r r r r r �NoExit2 s r& )�
assertTrue�
issubclassr � assertFalse)r
r$ r r% r& r r r �test_structural_subclassing s z6TestAbstractContextManager.test_structural_subclassingN)r r r r r r* r r r r r
s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
jdd� �Ze
jdd� �Zdd� Zdd � Zd d � Zd d � Zd!S )"�ContextManagerTestCasec s` g � t � fdd��}|� �*}| �� dg� | �|d� � �|� W 5 Q R X | �� dddg� d S )Nc 3 s � � d� dV � � d� d S )N� �* �� ��appendr ��stater r �woohoo< s
z@ContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor, r- r. )�contextmanager� assertEqualr0 �r
r3 �xr r1 r �test_contextmanager_plain: s z0ContextManagerTestCase.test_contextmanager_plainc
s| g � t � fdd��}| �t��B |� �0}| �� dg� | �|d� � �|� t� �W 5 Q R X W 5 Q R X | �� dddg� d S )Nc 3 s& � � d� z
dV W 5 � � d� X d S )Nr, r. r- r/ r r1 r r r3 I s
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor, r- r. )r4 r �ZeroDivisionErrorr5 r0 r6 r r1 r �test_contextmanager_finallyG s
z2ContextManagerTestCase.test_contextmanager_finallyc C s6 t dd� �}|� }|�� | �|�ttd�d �� d S )Nc s s
d V d S r r r r r r �wheeY s zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r4 r r) r r )r
r; �ctxr r r � test_contextmanager_no_reraiseX s
z5ContextManagerTestCase.test_contextmanager_no_reraisec C s6 t dd� �}|� }|�� | �t|jttd�d � d S )Nc s s" z
d V W n d V Y nX d S r r r r r r �whoob s
zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoor< )r4 r r � RuntimeErrorr r )r
r? r= r r r �*test_contextmanager_trap_yield_after_throwa s
�zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc sj g �t � �fdd��}|� �2}� ��dg� � �|d� ��|� td��W 5 Q R X � ��dddg� d S )Nc
3 s^ �� d� z
dV W nD tk
rX } z&�� | jd � � ��dddg� W 5 d } ~ X Y nX d S )Nr, r- r r. )r0 r9 r r5 )�e�r
r2 r r r3 p s
zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor, r- r. )r4 r5 r0 r9 r6 r rC r �test_contextmanager_exceptn s
z1ContextManagerTestCase.test_contextmanager_exceptc
C s� t d�}tdd� �}z2| �td�� |� �
|�W 5 Q R X W 5 Q R X W n. tk
rt } z| �||� W 5 d }~X Y n X | �d� d S )N�spamc s s
d V d S r r r r r r r3 � s zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohoo�
StopIteration� StopIteration was suppressed)rF r4 ZassertWarnsRegex�DeprecationWarning� Exceptionr �fail)r
�stop_excr3 �exr r r �#test_contextmanager_except_stopiter s
� z:ContextManagerTestCase.test_contextmanager_except_stopiterc
C s| d}i }t |||� |d }td�}z|� �
|�W 5 Q R X W n. tk
rl } z| �||� W 5 d }~X Y n X | �d� d S )Nztfrom __future__ import generator_stop
from contextlib import contextmanager
@contextmanager
def woohoo():
yield
r3 rE rG )�execrF rI r rJ )r
�code�localsr3 rK rL r r r �!test_contextmanager_except_pep479� s z8ContextManagerTestCase.test_contextmanager_except_pep479c
C s� t dd� �}z|� �
t�W 5 Q R X W nR tk
rx } z4| �t|�t� | �|jd d� | �|j t� W 5 d }~X Y nX z |� � t
d��W 5 Q R X W nP tk
r� } z2| �t|�t
� | �|jd d� | � |j � W 5 d }~X Y nX d S )Nc
s s< z
d V W n, t k
r6 } ztd�| �W 5 d } ~ X Y nX d S )N�issue29692:Chained)rI r@ ��excr r r �test_issue29692� s
zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r rR zissue29692:Unchained) r4 r9 rI r �typer@ r5 r �assertIsInstance� __cause__rF � assertIsNone)r
rU rL r r r �?test_contextmanager_do_not_unchain_non_stopiteration_exceptions� s
zVContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptionsc C s" dd� }t |dd�dd� ��}|S )Nc s � fdd�}|S )Nc s" � � � D ]\}}t| ||� q| S r )�items�setattr)�func�k�v��kwr r �decorate� s zXContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )ra rb r r` r �attribs� s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)r< c S s dS )�Whee!Nr )rE r r r �baz� s zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz�r4 )r
rc rf r r r � _create_contextmanager_attribs� s
z5ContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nrf rd )rh r5 r r< �r
rf r r r �test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nre �rh r5 �__doc__ri r r r � test_contextmanager_doc_attrib� s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s | � � d �}| �|jd� d S rk rl ri r r r �*test_instance_docstring_given_cm_docstring� s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc C s8 t dd� �}|ddddd��}| �|d� W 5 Q R X d S ) Nc s s | |||fV d S r r �r
r] r �kwdsr r r r3 � s z4ContextManagerTestCase.test_keywords.<locals>.woohoo� � �! �, rp )rr rs rt ru �r4 r5 )r
r3 �targetr r r �
test_keywords� s
z$ContextManagerTestCase.test_keywordsc s> G dd� d�}t � fdd��}||� |� d�� W 5 Q R X d S )Nc @ s e Zd ZdS )z0ContextManagerTestCase.test_nokeepref.<locals>.ANr r r r r �A� s ry c 3 s6 t �| �} t �|�}� �| � � � �|� � d V d S r )�weakref�refrY ��a�br r r r3 � s
z5ContextManagerTestCase.test_nokeepref.<locals>.woohoo�r~ rg )r
ry r3 r r r �test_nokeepref� s
z%ContextManagerTestCase.test_nokeeprefc C sl t dd� �}| �t�� |� W 5 Q R X | �t�� |dd� W 5 Q R X | �t�� |dd� W 5 Q R X d S )Nc s s
d V d S r r r| r r r r3 � s z8ContextManagerTestCase.test_param_errors.<locals>.woohoo� � r )r4 r r �r
r3 r r r �test_param_errors� s
z(ContextManagerTestCase.test_param_errorsc s@ d� t � �fdd��}|� � �fdd����� ��� d� d S )Nr c 3 s* � } � d7 � d V � d8 � �� � | � d S �Nr, )r5 )Zbefore)�depthr
r r r3 � s
z5ContextManagerTestCase.test_recursive.<locals>.woohooc s � dk r�� d S )N�
r r )r� � recursiver r r� s z8ContextManagerTestCase.test_recursive.<locals>.recursiverv r� r )r� r� r
r �test_recursive� s z%ContextManagerTestCase.test_recursiveN)r r r r8 r: r> rA rD rM rQ rZ rh rj r �requires_docstringsrn ro rx r� r� r� r r r r r+ 8 s$
r+ c @ s* e Zd Zejdd� �Zdd� Zdd� ZdS )�ClosingTestCasec C s t j}t d �}| �|j|� d S r )�closingrm r5 �r
Z cm_docstring�objr r r �test_instance_docs s z"ClosingTestCase.test_instance_docsc sZ g � G � fdd�d�}|� }| � � g � t|��}| � ||� W 5 Q R X | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z'ClosingTestCase.test_closing.<locals>.Cc s � � d� d S r� r/ r r1 r r �close s z-ClosingTestCase.test_closing.<locals>.C.closeN�r r r r� r r1 r r �C s r� r, )r5 r� �r
r� r7 �yr r1 r � test_closing s
z ClosingTestCase.test_closingc
sx g � G � fdd�d�}|� }| � � g � | �t��. t|��}| � ||� dd W 5 Q R X W 5 Q R X | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z-ClosingTestCase.test_closing_error.<locals>.Cc s � � d� d S r� r/ r r1 r r r� $ s z3ClosingTestCase.test_closing_error.<locals>.C.closeNr� r r1 r r r� # s r� r, r )r5 r r9 r� r� r r1 r �test_closing_error! s
z"ClosingTestCase.test_closing_errorN)r r r r r� r� r� r� r r r r r�
s
r� c @ s e Zd Zdd� ZdS )�NullcontextTestCasec C s8 G dd� d�}|� }t |��}| �||� W 5 Q R X d S )Nc @ s e Zd ZdS )z/NullcontextTestCase.test_nullcontext.<locals>.CNr r r r r r� 1 s r� )� nullcontextr )r
r� �cZc_inr r r �test_nullcontext0 s
z$NullcontextTestCase.test_nullcontextN)r r r r� r r r r r� / s r� c @ s e Zd Zdd� ZdS )�FileContextTestCasec
C s� t �� }z�d }t|d�� }| �|j� |�d� W 5 Q R X | �|j� d }| � t
��@ t|d��*}| �|j� | � |� � d� dd W 5 Q R X W 5 Q R X | �|j� W 5 t�|� X d S )N�wzBooh
�rr, r )
�tempfileZmktempr �unlink�openr) �closed�writer' r r9 r5 �read)r
Ztfn�fr r r � testWithOpen: s z FileContextTestCase.testWithOpenN)r r r r� r r r r r� 8 s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�LockContextTestCasec
C s~ | � |� � |� | �|� � W 5 Q R X | � |� � | �t��* |� | �|� � dd W 5 Q R X W 5 Q R X | � |� � d S �Nr, r )r) r' r r9 )r
�lock�lockedr r r � boilerPlateN s z LockContextTestCase.boilerPlatec C s t �� }| �||j� d S r )� threadingZLockr� r� �r
r� r r r � testWithLockY s z LockContextTestCase.testWithLockc C s t �� }| �||j� d S r )r� �RLockr� � _is_ownedr� r r r �
testWithRLock] s z!LockContextTestCase.testWithRLockc s$ t �� � � fdd�}| �� |� d S )Nc s � � � S r )r� r �r� r r r� c s z5LockContextTestCase.testWithCondition.<locals>.locked)r� Z Conditionr� �r
r� r r� r �testWithConditiona s z%LockContextTestCase.testWithConditionc s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S �NFT��acquire�releaser r� r r r� i s
z5LockContextTestCase.testWithSemaphore.<locals>.locked)r� Z Semaphorer� r� r r� r �testWithSemaphoreg s z%LockContextTestCase.testWithSemaphorec s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S r� r� r r� r r r� s s
z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)r� ZBoundedSemaphorer� r� r r� r �testWithBoundedSemaphoreq s z,LockContextTestCase.testWithBoundedSemaphoreN) r r r r� r� r� r� r� r� r r r r r� L s
r� c @ s, e Zd ZdZdZdZdZdd� Zdd� ZdS )� mycontextz9Example decoration-compatible context manager for testingFNc C s
d| _ | S �NT��startedr r r r r � s zmycontext.__enter__c G s || _ | jS r )rT �catch�r
rT r r r r � s zmycontext.__exit__) r r r rm r� rT r� r r r r r r r� | s r� c @ sb e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� ZdS )�TestContextDecoratorc C s t j}t � }| �|j|� d S r )r� rm r5 r� r r r r� � s z'TestContextDecorator.test_instance_docsc C s@ t � }|� }| �||� | �|j� W 5 Q R X | �|jd� d S )N�NNN)r� r r' r� r5 rT )r
�context�resultr r r �test_contextdecorator� s
z*TestContextDecorator.test_contextdecoratorc
C s� t � }| �td�� |� td��W 5 Q R X W 5 Q R X | �|j� | �|jd t� t � }d|_|� td��W 5 Q R X | �|j� | �|jd t� d S )Nr< r T)r� �assertRaisesRegex� NameError�assertIsNotNonerT r r� )r
r� r r r �$test_contextdecorator_with_exception� s z9TestContextDecorator.test_contextdecorator_with_exceptionc s0 t � � � � �fdd��}|� ��� jd� d S )Nc s �� � j� ��� j� d S r �rY rT r' r� r �r� r
r r �test� s z1TestContextDecorator.test_decorator.<locals>.testr� )r� r5 rT �r
r� r r� r �test_decorator� s
z#TestContextDecorator.test_decoratorc sX t � � � � �fdd��}��td�� |� W 5 Q R X ��� j� ��� jd t� d S )Nc s$ �� � j� ��� j� td��d S )Nr< )rY rT r' r� r� r r� r r r� � s z@TestContextDecorator.test_decorator_with_exception.<locals>.testr< r )r� r� r� r� rT r r� r r� r � test_decorator_with_exception� s z2TestContextDecorator.test_decorator_with_exceptionc s� t � � G � fdd�dt�}|� }|�dd� | �|jd� | �|jd� | �|jd � |� }|�ddd� | �|jd� | �|jd� | �|jd� |� }|jddd� | �|jd� | �|jd� d S ) Nc s e Zd Z� ddd��ZdS )z9TestContextDecorator.test_decorating_method.<locals>.TestNc S s || _ || _|| _d S r )r} r~ r� )r
r} r~ r� r r r �method� s z@TestContextDecorator.test_decorating_method.<locals>.Test.method)N)r r r r� r �r� r r �Test� s r� r, � r} r~ r� r| )r� �objectr� r5 r} r~ r� )r
r� r� r r� r �test_decorating_method� s z+TestContextDecorator.test_decorating_methodc
C s< G dd� dt �}| �t�� |� � W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z7TestContextDecorator.test_typo_enter.<locals>.mycontextc S s d S r r r r r r � __unter__� s zATestContextDecorator.test_typo_enter.<locals>.mycontext.__unter__c W s d S r r r� r r r r � s z@TestContextDecorator.test_typo_enter.<locals>.mycontext.__exit__N)r r r r� r r r r r r� � s r� ��ContextDecoratorr �AttributeError�r
r� r r r �test_typo_enter� s z$TestContextDecorator.test_typo_enterc
C s< G dd� dt �}| �t�� |� � W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z6TestContextDecorator.test_typo_exit.<locals>.mycontextc S s d S r r r r r r r � s z@TestContextDecorator.test_typo_exit.<locals>.mycontext.__enter__c W s d S r r r� r r r �__uxit__� s z?TestContextDecorator.test_typo_exit.<locals>.mycontext.__uxit__N)r r r r r� r r r r r� � s r� r� r� r r r �test_typo_exit� s z#TestContextDecorator.test_typo_exitc sR G dd� dt �}G dd� d|t�}|� � � � �fdd��}|� ��� jd� d S )Nc @ s$ e Zd ZdZdZdd� Zdd� ZdS )zHTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontextFNc S s
d| _ | S r� r� r r r r r s zRTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__enter__c W s
|| _ d S r rS r� r r r r s zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r r r r� rT r r r r r r � somecontext� s r� c @ s e Zd ZdS )zFTestContextDecorator.test_contextdecorator_as_mixin.<locals>.mycontextNr r r r r r�
s r� c s �� � j� ��� j� d S r r� r r� r r r� s zATestContextDecorator.test_contextdecorator_as_mixin.<locals>.testr� )r� r� r5 rT )r
r� r� r� r r� r � test_contextdecorator_as_mixin� s z3TestContextDecorator.test_contextdecorator_as_mixinc sf t �fdd��}g �|d�� �fdd��}|d� � ��dddg� g �|d� � ��dddg� d S ) Nc 3 s � � | � d V � � d� d S )Nr. r/ )r� r1 r r r3 s
zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor, c s � � �dg� ��| � d S r� )r5 r0 )r7 rC r r r� s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.testZ somethingr. zsomething elserv )r
r3 r� r rC r � test_contextmanager_as_decorator s z5TestContextDecorator.test_contextmanager_as_decoratorN)r r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s� e Zd ZdZejdd� �Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZdS )(�TestBaseExitStackNc C s" | j j}| � � }| �|j|� d S r )�
exit_stackrm r5 r� r r r r� . s z$TestBaseExitStack.test_instance_docsc C s | � � � W 5 Q R X d S r )r� r r r r �test_no_resources5 s
z#TestBaseExitStack.test_no_resourcesc s� di fdi fdi fdt dd�fdt dd�fdt dd�fdt ddd�fg}g � � fd d
�}| �� ��}t|�D ]d\}}|r�|r�|j|f|�|�}n4|r�|j|f|�� }n |r�|j|f|�}n
|�|�}| �||� qj|jD ]>}| �|d j|� | �|d j|j� | � |d j
|j
� q�W 5 Q R X | � � |� g � | �� �p}| � t
�� |jdd � W 5 Q R X | � t
�� | jjd d � W 5 Q R X | �t�� |j|dd
� W 5 Q R X W 5 Q R X | � � dddifg� d S )Nr )r, )r, r� r, )Zexampler� � )r
�callbackc s � � | |f� dS )zTest metadata propagationNr/ )r rq �r� r r �_exitD s z.TestBaseExitStack.test_callback.<locals>._exit)�argr� )r� r� r� )�dictr� �reversedr� r �_exit_callbacks� __wrapped__ZassertNotEqualr rY rm r5 r r Z assertWarnsrH )r
Zexpectedr� �stackr rq r� �wrapperr r� r �
test_callback9 sD �
"z TestBaseExitStack.test_callbackc s* t � � �fdd�}dd� }�fdd�}G dd� dt�}��� ��}|�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� |�|� ��|jd d
|� d
d W 5 Q R X d S ) Nc s �� | � � d S r )r �r rT Zexc_tb�Z
exc_raisedr
r r � _expect_excd s z0TestBaseExitStack.test_push.<locals>._expect_excc W s dS r� r �� exc_detailsr r r �
_suppress_excf s z2TestBaseExitStack.test_push.<locals>._suppress_excc s" � � | � � � |� � � |� d S r )rY r� r r r �
_expect_okh s
z/TestBaseExitStack.test_push.<locals>._expect_okc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z+TestBaseExitStack.test_push.<locals>.ExitCMc S s
|| _ d S r �� check_exc)r
r� r r r �__init__m s z4TestBaseExitStack.test_push.<locals>.ExitCM.__init__c S s | � d� d S )NzShould not be called!)rJ r r r r r o s z5TestBaseExitStack.test_push.<locals>.ExitCM.__enter__c W s | j |� d S r r� �r
r� r r r r q s z4TestBaseExitStack.test_push.<locals>.ExitCM.__exit__N�r r r r� r r r r r r �ExitCMl s r� ���r, r )r9 r� r� �pushr r� �__self__)r
r� r� r� r� r� �cmr r� r � test_pushb s*
zTestBaseExitStack.test_pushc s� G � fdd�dt �}g � |� }| �� �N}|j� fdd��}| �|� |�|� | �|jd d j|� � �d� W 5 Q R X | � � dddd g� d S )
Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z4TestBaseExitStack.test_enter_context.<locals>.TestCMc s � � d� d S r� r/ r r� r r r � s z>TestBaseExitStack.test_enter_context.<locals>.TestCM.__enter__c s � � d� d S �Nr� r/ r� r� r r r � s z=TestBaseExitStack.test_enter_context.<locals>.TestCM.__exit__Nr# r r� r r �TestCM� s r c s � � d� d S )Nr� r/ r r� r r r� � s z3TestBaseExitStack.test_enter_context.<locals>._exitr� r, r� r� r� )
r� r� r� r� �
enter_contextr r� r� r0 r5 )r
r r� r� r� r r� r �test_enter_context� s
z$TestBaseExitStack.test_enter_contextc sZ g � | � � �4}|j� fdd��}| �|� |�� � �d� W 5 Q R X | �� ddg� d S )Nc s � � d� d S r� r/ r r� r r r� � s z+TestBaseExitStack.test_close.<locals>._exitr� r, )r� r� r� r� r0 r5 )r
r� r� r r� r �
test_close� s
z TestBaseExitStack.test_closec sn g � | � � �4}|j� fdd��}| �|� |�� }� �d� W 5 Q R X � �d� |�� | �� dddg� d S )Nc s � � d� d S r r/ r r� r r r� � s z-TestBaseExitStack.test_pop_all.<locals>._exitr, r� r� )r� r� r� �pop_allr0 r� r5 )r
r� r� � new_stackr r� r � test_pop_all� s
z TestBaseExitStack.test_pop_allc
C sD | � t��0 | �� � }|�dd� � dd W 5 Q R X W 5 Q R X d S )Nc W s dS )NFr rS r r r �<lambda>� � z3TestBaseExitStack.test_exit_raise.<locals>.<lambda>r, r )r r9 r� r� �r
r� r r r �test_exit_raise� s
z!TestBaseExitStack.test_exit_raisec C s. | � � � }|�dd� � dd W 5 Q R X d S )Nc W s dS r� r rS r r r r � r
z6TestBaseExitStack.test_exit_suppress.<locals>.<lambda>r, r �r� r� r r r r �test_exit_suppress� s
z$TestBaseExitStack.test_exit_suppressc C s
G dd� d�}G dd� d�}G dd� d�}z\|t ��J |tt��4 |� �" |t�� dd W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W nP t k
r� } z2| �|jt� | �|jjt� | �|jjj� W 5 d }~X Y n X | �d � |jd }| �|t� | �|jt � d S )
Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zJTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcc S s
|| _ d S r rS r� r r r r� � s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__init__c S s | S r r r r r r r � s zTTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__enter__c W s
| j �d S r rS r� r r r r � s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__Nr� r r r r �RaiseExc� s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zUTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s || _ || _d S r )�outer�inner)r
r r r r r r� � s z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__init__c S s | S r r r r r r r � s z_TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__enter__c W s" z
| j �W n | j�Y nX d S r )r r r� r r r r � s
z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__Nr� r r r r �RaiseExcWithContext� s r c @ s e Zd Zdd� Zdd� ZdS )zMTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExcc S s | S r r r r r r r � s zWTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__enter__c W s |t | �_dS r� )rV �
saved_detailsr� r r r r � s
zVTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__Nr# r r r r � SuppressExc� s r r, r �0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr� �
ValueErrorrW � __context__rY rJ r r9 )r
r r r rT � inner_excr r r �&test_exit_exception_chaining_reference� s"
4"
z8TestBaseExitStack.test_exit_exception_chaining_referencec
s� dd� }d � � fdd�}zZ| � � �H}|�|t� |�|t� |�|t� |�|� |�|t� dd W 5 Q R X W nP tk
r� } z2| �|jt� | �|jjt� | � |jjj� W 5 d }~X Y n X | �
d� � d }| �|t� | �|jt � d S )Nc S s | �d S r r rS r r r � raise_exc� s zATestBaseExitStack.test_exit_exception_chaining.<locals>.raise_excc s | � dS r� r r� �r r r � suppress_exc� s zDTestBaseExitStack.test_exit_exception_chaining.<locals>.suppress_excr, r r ) r� r� r r r� r� r rW r rY rJ r9 )r
r r r� rT r r r r � test_exit_exception_chaining� s&
"
z.TestBaseExitStack.test_exit_exception_chainingc
C s� dd� }dd� }z2| � � � }|�dd� � |�|t� W 5 Q R X W n. tk
rp } z| �|t� W 5 d }~X Y n X | �d� z:| � � �(}|�|t� |�|� |�|t� W 5 Q R X W n. tk
r� } z| �|t� W 5 d }~X Y n X | �d� d S ) Nc S s | �d S r r rS r r r r s zHTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W s dS r� r r� r r r r s zKTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S s d S r r r r r r r r
zGTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>r z.Expected KeyError, but no exception was raised)r� r� r rI rW rJ r r� )r
r r r� rT r r r �#test_exit_exception_non_suppressing s"
z5TestBaseExitStack.test_exit_exception_non_suppressingc
C s� t dd� �}td�}td�}td�}td�}zF| �� �4}|�||�� |�||�� |�||�� |�W 5 Q R X W np tk
r� } zR| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj� W 5 d }~X Y nX d S )Nc s s z
d V W 5 | �X d S r r rS r r r �gets_the_context_right& s
zZTestBaseExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr, r� r� r� )r4 rI r� r r r rY )r
r! �exc1�exc2�exc3�exc4r� rT r r r �(test_exit_exception_with_correct_context$ s(
�z:TestBaseExitStack.test_exit_exception_with_correct_contextc
C s� dd� }t d�}t d�}t d�}t d�}t d�}z8| �� �&}|�|||� |�|||� |�W 5 Q R X W n� t k
r� } zh| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj|� | �|jjjjj� W 5 d }~X Y nX d S )Nc S s z| �W 5 |�X d S r r )r Z outer_excr r r � raise_nestedF s zQTestBaseExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr, r� r� r� r� )rI r� r� r r rY ) r
r' r" r# r$ r% Zexc5r� rT r r r �)test_exit_exception_with_existing_contextC s.
� �z;TestBaseExitStack.test_exit_exception_with_existing_contextc
C sd dd� }z*| � � �}|�|� dd W 5 Q R X W n, tk
r^ } z| �d� W 5 d }~X Y nX d S )Nc W s dS r� r r� r r r r ` s zDTestBaseExitStack.test_body_exception_suppress.<locals>.suppress_excr, r z%Expected no exception, got IndexError)r� r� r rJ )r
r r� rT r r r � test_body_exception_suppress_ s
z.TestBaseExitStack.test_body_exception_suppressc C sB | � � �0}|�dd� � |�dd� � |�dd� � W 5 Q R X d S )Nc W s dS r� r rS r r r r k r
zITestBaseExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s dd S r� r rS r r r r l r
c W s i d S r� r rS r r r r m r
r
r r r r �%test_exit_exception_chaining_suppressi s
z7TestBaseExitStack.test_exit_exception_chaining_suppressc C s0 | � � � }td�D ]}|�t� qW 5 Q R X d S )Ni' )r� �ranger� �int)r
r� �ir r r �test_excessive_nestingo s
z(TestBaseExitStack.test_excessive_nestingc C sZ G dd� dt �}|� }t � |_| �� }| �t|j|� |�|� | �|jd d |� d S )Nc @ s e Zd ZdS )z7TestBaseExitStack.test_instance_bypass.<locals>.ExampleNr r r r r �Examplev s r/ r� r, ) r� r r� r r� r r� r r� )r
r/ r� r� r r r �test_instance_bypassu s
z&TestBaseExitStack.test_instance_bypassc
s� G dd� dt �� G dd� dt�}t� fdd��}tdd� �}| �� ��:}| �� �&}|�|� � |�|� � |d ��W 5 Q R X W 5 Q R X |j}| �|� � | �|j|� | � |jj� | � |jj
� | � |j
|j� d S )
Nc @ s e Zd ZdS )zITestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueExceptionNr r r r r �UniqueException� s r1 c @ s e Zd ZdS )zLTestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueRuntimeErrorNr r r r r �UniqueRuntimeError� s r2 c
3 s< z
dV W n, t k
r6 } z� d�| �W 5 d } ~ X Y nX d S )Nr, z
new exception�rI rS �r1 r r �second� s
z@TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.secondc
s s6 z
dV W n& t k
r0 } z| �W 5 d } ~ X Y nX d S r� r3 rS r r r �first� s
z?TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) rI r@ r4 r r� r Z exceptionrW r rY rX r )r
r2 r5 r6 Zerr_ctxZes_ctxrT r r4 r � test_dont_reraise_RuntimeError~ s"
z0TestBaseExitStack.test_dont_reraise_RuntimeError)r r r r� r r� r� r� r� r r r r r r r r r r& r( r) r* r. r0 r7 r r r r r� + s*
)"
0
r� c @ s e Zd ZeZdS )�
TestExitStackN)r r r � ExitStackr� r r r r r8 � s r8 c @ sJ e Zd ZdZdZejdd� �Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dS )�TestRedirectStreamNc C s$ | j j}| � d �}| �|j|� d S r )�redirect_streamrm r5 r� r r r r� � s
z%TestRedirectStream.test_instance_docsc C s. t t| j�}| �d � | �t t| j�|� d S r )�getattr�sys� orig_streamr; r )r
� orig_stdoutr r r �test_no_redirect_in_init� s
z+TestRedirectStream.test_no_redirect_in_initc C sr t �� }d}tt| j�}| �|�� t|tt| j�d� W 5 Q R X | �tt| j�|� |�� � � }| �
||� d S )Nz<Consider an API like help(), which prints directly to stdout��file) �io�StringIOr< r= r> r; �printr �getvalue�stripr5 )r
r� �msgr? �sr r r �test_redirect_to_string_io� s z-TestRedirectStream.test_redirect_to_string_ioc C s. t �� }| �|��}| �||� W 5 Q R X d S r )rC rD r; r )r
r� � enter_resultr r r �test_enter_result_is_target� s z.TestRedirectStream.test_enter_result_is_targetc C s� t �� }| �|�}tt| j�}|� tddtt| j�d� W 5 Q R X |� tdtt| j�d� W 5 Q R X | �tt| j�|� |�� }| � |d� d S �N�Hello� )�endrB zWorld!rA z
Hello World!
�
rC rD r; r< r= r> rE r rF r5 �r
r� Z
write_to_fr? rI r r r �test_cm_is_reusable� s
z&TestRedirectStream.test_cm_is_reusablec
C s� t �� }| �|�}tt| j�}|�@ tddtt| j�d� |� tdtt| j�d� W 5 Q R X W 5 Q R X | �tt| j�|� |�� }| � |d� d S rM rQ rR r r r �test_cm_is_reentrant� s
(z'TestRedirectStream.test_cm_is_reentrant)
r r r r; r> r r� r� r@ rJ rL rS rT r r r r r: � s
r: c @ s e Zd ZeZdZdS )�TestRedirectStdout�stdoutN)r r r �redirect_stdoutr; r> r r r r rU � s rU c @ s e Zd ZeZdZdS )�TestRedirectStderr�stderrN)r r r �redirect_stderrr; r> r r r r rX � s rX c @ sZ e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dS )� TestSuppressc C s t j}t � }| �|j|� d S r )�suppressrm r5 r� r r r r� � s z TestSuppress.test_instance_docsc C s" t t��}| �|� W 5 Q R X d S r )r\ r rY )r
rK r r r �test_no_result_from_enter� s
z&TestSuppress.test_no_result_from_enterc C s* t t�� | �tdd�d� W 5 Q R X d S )Nr� r� � )r\ r r5 �powr r r r �test_no_exception� s
z TestSuppress.test_no_exceptionc C s t t�� td� W 5 Q R X d S )Nr� )r\ r �lenr r r r �test_exact_exception� s
z!TestSuppress.test_exact_exceptionc C s t t�� dd W 5 Q R X d S )NrN �2 )r\ � LookupErrorr r r r �test_exception_hierarchy� s
z%TestSuppress.test_exception_hierarchyc
C s6 | � t��" tt�� dd W 5 Q R X W 5 Q R X d S r� )r r9 r\ r r r r r �test_other_exception s
z!TestSuppress.test_other_exceptionc
C s4 | � t�� t� � dd W 5 Q R X W 5 Q R X d S r� )r r9 r\ r r r r � test_no_args s zTestSuppress.test_no_argsc C s@ t tt�� dd W 5 Q R X t tt�� td� W 5 Q R X d S )Nr, r r� )r\ r9 r ra r r r r � test_multiple_exception_args
s z)TestSuppress.test_multiple_exception_argsc
C sr t t�}|� W 5 Q R X |� td� W 5 Q R X |�* |� td� W 5 Q R X d}dd W 5 Q R X | �|� d S )Nr� Tr, r )r\ rI ra r' )r
Zignore_exceptionsZouter_continuedr r r rT s
z!TestSuppress.test_cm_is_reentrantN)r r r r r� r� r] r` rb re rf rg rh rT r r r r r[ � s
r[ �__main__) rm rC r= r� r� Zunittest�
contextlibr� r rz ZTestCaser r+ r� r� r� r� r� r� r� r� r8 r: rU rX r[ r �mainr r r r �<module> s8 + V" 0 ! x96