Your IP : 216.73.217.32
a
XC?h�� � @ 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/python39/lib64/python3.9/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 sD G dd� dt �}| �t�� |� W d � n1 s60 Y 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 sr g � t � fdd��}|� �4}| �� dg� | �|d� � �|� W d � n1 sT0 Y | �� g d�� 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- r. r/ )�contextmanager� assertEqualr1 �r
r4 �xr r2 r �test_contextmanager_plain: s (z0ContextManagerTestCase.test_contextmanager_plainc s� g � t � fdd��}| �t��` |� �:}| �� dg� | �|d� � �|� t� �W d � n1 sf0 Y W d � n1 s�0 Y | �� g d�� d S )Nc 3 s0 � � d� zdV W � � d� n � � d� 0 d S r, r0 r r2 r r r4 I s
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor- r. r5 )r6 r �ZeroDivisionErrorr7 r1 r8 r r2 r �test_contextmanager_finallyG s
Bz2ContextManagerTestCase.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)r6 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 n0 d S r r r r r r �whoob s
zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoor> )r6 r r � RuntimeErrorr r )r
rA r? r r r �*test_contextmanager_trap_yield_after_throwa s
�zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc s| g �t � �fdd��}|� �<}� ��dg� � �|d� ��|� td��W d � n1 s^0 Y � ��g d�� d S )Nc
3 s^ �� d� z
dV W nD tyX } z,�� | jd � � ��g d�� W Y d } ~ n
d } ~ 0 0 d S )Nr- r. r r5 )r1 r; r r7 )�e�r
r3 r r r4 p s
zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor- r. r/ r5 )r6 r7 r1 r; r8 r rE r �test_contextmanager_exceptn s
&z1ContextManagerTestCase.test_contextmanager_exceptc C s� t dd� �}G dd� dt�}td�|d�fD ]�}| jt|�d��� z.|� � |�W d � n1 sd0 Y W n0 ty� } z| �||� W Y d }~nd }~0 0 | �|� d�� W d � q,1 s�0 Y q,d S )Nc s s
d V d S r r r r r r r4 � s zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohooc @ s e Zd ZdS )zYContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.StopIterationSubclassNr r r r r �StopIterationSubclass� s rG �spam)�typez was suppressed)r6 �
StopIteration�subTestrI � Exceptionr �fail)r
r4 rG �stop_exc�exr r r �#test_contextmanager_except_stopiter s
&"z:ContextManagerTestCase.test_contextmanager_except_stopiterc
C s� d}i }t |||� |d }td�}z.|� � |�W d � n1 sF0 Y W n0 ty� } z| �||� W Y d }~nd }~0 0 | �d� d S )Nztfrom __future__ import generator_stop
from contextlib import contextmanager
@contextmanager
def woohoo():
yield
r4 rH z StopIteration was suppressed)�execrJ rL r rM )r
�code�localsr4 rN rO r r r �!test_contextmanager_except_pep479� s &"z8ContextManagerTestCase.test_contextmanager_except_pep479c
C s t dd� �}z.|� � t�W d � n1 s.0 Y W nT ty� } z<| �t|�t� | �|jd d� | �|j t� W Y d }~n
d }~0 0 z2|� � t
d��W d � n1 s�0 Y W nT t�y } z:| �t|�t
� | �|jd d� | � |j � W Y d }~n
d }~0 0 d S )Nc
s s> z
d V W n. t y8 } ztd�| �W Y d } ~ n
d } ~ 0 0 d S )N�issue29692:Chained)rL rB ��excr r r �test_issue29692� s
zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r rU zissue29692:Unchained) r6 r; rL r rI rB r7 r �assertIsInstance� __cause__rJ � assertIsNone)r
rX rO 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 )rc rd r rb r �attribs� s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)r> c S s dS )�Whee!Nr )rH r r r �baz� s zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz�r6 )r
re rh r r r � _create_contextmanager_attribs� s
z5ContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nrh rf )rj r7 r r> �r
rh r r r �test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nrg �rj r7 �__doc__rk r r r � test_contextmanager_doc_attrib� s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s | � � d �}| �|jd� d S rm rn rk r r r �*test_instance_docstring_given_cm_docstring� s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc C sL t dd� �}|ddddd�� }| �|d� W d � n1 s>0 Y d S ) Nc s s | |||fV d S r r �r
r_ r �kwdsr r r r4 � s z4ContextManagerTestCase.test_keywords.<locals>.woohoo� � �! �, rr )rt ru rv rw �r6 r7 )r
r4 �targetr r r �
test_keywords� s
z$ContextManagerTestCase.test_keywordsc sR G dd� d�}t � fdd��}||� |� d�� W d � n1 sD0 Y d S )Nc @ s e Zd ZdS )z0ContextManagerTestCase.test_nokeepref.<locals>.ANr r r r r �A� s r{ c 3 s6 t �| �} t �|�}� �| � � � �|� � d V d S r )�weakref�refr[ ��a�br r r r4 � s
z5ContextManagerTestCase.test_nokeepref.<locals>.woohoo�r� ri )r
r{ r4 r r r �test_nokeepref� s
z%ContextManagerTestCase.test_nokeeprefc C s� t dd� �}| �t�� |� W d � n1 s20 Y | �t�� |dd� W d � n1 sf0 Y | �t�� |dd� W d � n1 s�0 Y d S )Nc s s
d V d S r r r~ r r r r4 � s z8ContextManagerTestCase.test_param_errors.<locals>.woohoo� � r� )r6 r r �r
r4 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- )r7 )Zbefore)�depthr
r r r4 � s
z5ContextManagerTestCase.test_recursive.<locals>.woohooc s � dk r�� d S )N�
r r )r� � recursiver r r� s z8ContextManagerTestCase.test_recursive.<locals>.recursiverx r� r )r� r� r
r �test_recursive� s z%ContextManagerTestCase.test_recursiveN)r r r r: r< r@ rC rF rP rT r\ rj rl r �requires_docstringsrp rq rz 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 )�closingro r7 �r
Z cm_docstring�objr r r �test_instance_docs s z"ClosingTestCase.test_instance_docsc sn g � G � fdd�d�}|� }| � � g � t|�� }| � ||� W d � n1 sR0 Y | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z'ClosingTestCase.test_closing.<locals>.Cc s � � d� d S r� r0 r r2 r r �close s z-ClosingTestCase.test_closing.<locals>.C.closeN�r r r r� r r2 r r �C s r� r- )r7 r� �r
r� r9 �yr r2 r � test_closing s
*z ClosingTestCase.test_closingc s� g � G � fdd�d�}|� }| � � g � | �t��L t|��$}| � ||� dd W d � n1 sf0 Y W d � n1 s�0 Y | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z-ClosingTestCase.test_closing_error.<locals>.Cc s � � d� d S r� r0 r r2 r r r� ' s z3ClosingTestCase.test_closing_error.<locals>.C.closeNr� r r2 r r r� & s r� r- r )r7 r r; r� r� r r2 r �test_closing_error$ s
Dz"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 sL G dd� d�}|� }t |�� }| �||� W d � n1 s>0 Y d S )Nc @ s e Zd ZdS )z/NullcontextTestCase.test_nullcontext.<locals>.CNr r r r r r� 4 s r� )� nullcontextr )r
r� �cZc_inr r r �test_nullcontext3 s
z$NullcontextTestCase.test_nullcontextN)r r r r� r r r r r� 2 s r� c @ s e Zd Zdd� ZdS )�FileContextTestCasec C s t �� }z�d }t|d��&}| �|j� |�d� W d � n1 sD0 Y | �|j� d }| �t��^ t|d��4}| �|j� | � |�
� d� dd W d � n1 s�0 Y W d � n1 s�0 Y | �|j� W t � |� n t � |� 0 d S )N�wzBooh
�rr- r )
�tempfileZmktemp�openr) �closed�writer' r r; r7 �readr �unlink)r
Ztfn�fr r r � testWithOpen= s ( Dz FileContextTestCase.testWithOpenN)r r r r� r r r r r� ; s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�LockContextTestCasec C s� | � |� � |� | �|� � W d � n1 s20 Y | � |� � | �t��H |�$ | �|� � dd W d � n1 s�0 Y W d � n1 s�0 Y | � |� � d S �Nr- r )r) r' r r; )r
�lock�lockedr r r � boilerPlateQ s * Dz LockContextTestCase.boilerPlatec C s t �� }| �||j� d S r )� threadingZLockr� r� �r
r� r r r � testWithLock\ 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� f s z5LockContextTestCase.testWithCondition.<locals>.locked)r� Z Conditionr� �r
r� r r� r �testWithConditiond s z%LockContextTestCase.testWithConditionc s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S �NFT��acquire�releaser r� r r r� l s
z5LockContextTestCase.testWithSemaphore.<locals>.locked)r� Z Semaphorer� r� r r� r �testWithSemaphorej s z%LockContextTestCase.testWithSemaphorec s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S r� r� r r� r r r� v s
z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)r� ZBoundedSemaphorer� r� r r� r �testWithBoundedSemaphoret s z,LockContextTestCase.testWithBoundedSemaphoreN) r r r r� r� r� r� r� r� r r r r r� O 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 )rW �catch�r
rW r r r r � s zmycontext.__exit__) r r r ro r� rW 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� ro r7 r� r r r r� � s z'TestContextDecorator.test_instance_docsc C sT t � }|�(}| �||� | �|j� W d � n1 s80 Y | �|jd� d S )N�NNN)r� r r' r� r7 rW )r
�context�resultr r r �test_contextdecorator� s
*z*TestContextDecorator.test_contextdecoratorc C s� t � }| �td��< |� td��W d � n1 s60 Y W d � n1 sT0 Y | �|j� | �|jd t� t � }d|_|� td��W d � n1 s�0 Y | �|j� | �|jd t� d S )Nr> r T)r� �assertRaisesRegex� NameError�assertIsNotNonerW r r� )r
r� r r r �$test_contextdecorator_with_exception� s D & z9TestContextDecorator.test_contextdecorator_with_exceptionc s0 t � � � � �fdd��}|� ��� jd� d S )Nc s �� � j� ��� j� d S r �r[ rW r' r� r �r� r
r r �test� s z1TestContextDecorator.test_decorator.<locals>.testr� )r� r7 rW �r
r� r r� r �test_decorator� s
z#TestContextDecorator.test_decoratorc sl t � � � � �fdd��}��td�� |� W d � n1 s@0 Y ��� j� ��� jd t� d S )Nc s$ �� � j� ��� j� td��d S )Nr> )r[ rW r' r� r� r r� r r r� � s z@TestContextDecorator.test_decorator_with_exception.<locals>.testr> r )r� r� r� r� rW 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� r7 r r� r� )r
r� r� r r� r �test_decorating_method� s z+TestContextDecorator.test_decorating_methodc C sd G dd� dt �}| �t��6 |� � W d � n1 s80 Y W d � n1 sV0 Y 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 sd G dd� dt �}| �t��6 |� � W d � n1 s80 Y W d � n1 sV0 Y 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 rV r� r r r r
s zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r r r r� rW 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� r7 rW )r
r� r� r� r r� r � test_contextdecorator_as_mixin s z3TestContextDecorator.test_contextdecorator_as_mixinc sb t �fdd��}g �|d�� �fdd��}|d� � ��g d�� g �|d� � ��g d �� d S )
Nc 3 s � � | � d V � � d� d S )Nr/ r0 )r� r2 r r r4 s
zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor- c s � � �dg� ��| � d S r� )r7 r1 )r9 rE r r r� ! s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.test� something)r- r� r/ �something else)r- r� r/ rx )r
r4 r� r rE 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'� Zd(d)� ZdS )*�TestBaseExitStackNc C s" | j j}| � � }| �|j|� d S r )�
exit_stackro r7 r� r r r r� 1 s z$TestBaseExitStack.test_instance_docsc C s, | � � � W d � n1 s 0 Y d S r )r� r r r r �test_no_resources8 s
z#TestBaseExitStack.test_no_resourcesc s8 di fdi fdi fdt dd�fdt dd�fdt dd�fdt ddd�fg}g � � fd d
�}| �� ��}t|�D ]p\}}|r�|r�|j|g|�R i |��}n:|r�|j|g|�R � }n"|r�|j|fi |��}n
|�|�}| �||� qj|jD ]>}| �|d j|� | �|d j|j� | � |d j
|j
� q�W d � n1 �s80 Y | � � |� g � | �� ��}| � t
�� |jdd � W d � n1 �s�0 Y | � t
�� | jjd d � W d � n1 �s�0 Y | � t
�� |j|dd
� W d � n1 �s�0 Y W d � n1 �s 0 Y | � � g � d S )Nr )r- )r- r� r- )Zexampler� � )r
�callbackc s � � | |f� dS )zTest metadata propagationNr0 )r rs �r� r r �_exitG s z.TestBaseExitStack.test_callback.<locals>._exit)�argr� )r� r� )�dictr� �reversedr� r �_exit_callbacks� __wrapped__ZassertNotEqualr r[ ro r7 r r )r
Zexpectedr� �stackr rs r� �wrapperr r� r �
test_callback< sD �
6
, . Nz 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 d � n1 �s20 Y d S ) Nc s �� | � � d S r )r �r rW Zexc_tb�Z
exc_raisedr
r r � _expect_excg s z0TestBaseExitStack.test_push.<locals>._expect_excc W s dS r� r �� exc_detailsr r r �
_suppress_exci s z2TestBaseExitStack.test_push.<locals>._suppress_excc s" � � | � � � |� � � |� d S r )r[ r� r r r �
_expect_okk 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__p s z4TestBaseExitStack.test_push.<locals>.ExitCM.__init__c S s | � d� d S )NzShould not be called!)rM r r r r r r s z5TestBaseExitStack.test_push.<locals>.ExitCM.__enter__c W s | j |� d S r r� �r
r� r r r r t s z4TestBaseExitStack.test_push.<locals>.ExitCM.__exit__N�r r r r� r r r r r r �ExitCMo s r� ���r- r )r; r� r� �pushr r� �__self__)r
r� r� r� r� r� �cmr r� r � test_pushe s*
zTestBaseExitStack.test_pushc s� G � fdd�dt �}g � |� }| �� �X}|j� fdd��}| �|� |�|� | �|jd d j|� � �d� W d � n1 s�0 Y | � � g d�� d S ) Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z4TestBaseExitStack.test_enter_context.<locals>.TestCMc s � � d� d S r� r0 r r� r r r � s z>TestBaseExitStack.test_enter_context.<locals>.TestCM.__enter__c s � � d� d S �Nr� r0 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� r0 r r� r r r� � s z3TestBaseExitStack.test_enter_context.<locals>._exitr r- r� )r- r� r� r� )
r� r� r� r� �
enter_contextr r� r r1 r7 )r
r r r� r� r r� r �test_enter_context� s
(z$TestBaseExitStack.test_enter_contextc sn g � | � � �>}|j� fdd��}| �|� |�� � �d� W d � n1 sP0 Y | �� ddg� d S )Nc s � � d� d S r� r0 r r� r r r� � s z+TestBaseExitStack.test_close.<locals>._exitr� r- )r� r� r� r� r1 r7 )r
r� r� r r� r �
test_close� s
(z TestBaseExitStack.test_closec s� g � | � � �>}|j� fdd��}| �|� |�� }� �d� W d � n1 sP0 Y � �d� |�� | �� g d�� d S )Nc s � � d� d S r r0 r r� r r r� � s z-TestBaseExitStack.test_pop_all.<locals>._exitr- r� )r- r� r� )r� r� r� �pop_allr1 r� r7 )r
r� r� � new_stackr r� r � test_pop_all� s
(
z TestBaseExitStack.test_pop_allc C sl | � t��N | �� �&}|�dd� � dd W d � n1 s@0 Y W d � n1 s^0 Y d S )Nc W s dS )NFr rV r r r �<lambda>� � z3TestBaseExitStack.test_exit_raise.<locals>.<lambda>r- r )r r; r� r �r
r� r r r �test_exit_raise� s
z!TestBaseExitStack.test_exit_raisec C sB | � � �&}|�dd� � dd W d � n1 s40 Y d S )Nc W s dS r� r rV 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 ��� |tt��f |� �@ |t�� dd W d � n1 sp0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W nT t �y* } z:| �|jt� | �|jjt� | �|jjj� W Y d }~nd }~0 0 | �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 rV 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 rV 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 n0 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� )rI �
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� �
ValueErrorrY � __context__r[ rM r r; )r
r r r rW � inner_excr r r �&test_exit_exception_chaining_reference� s"
�&
z8TestBaseExitStack.test_exit_exception_chaining_referencec
s
dd� }d � � fdd�}zn| � � �R}|�|t� |�|t� |�|t� |�|� |�|t� dd W d � n1 sz0 Y W nR ty� } z:| �|jt� | �|jjt� | � |jjj� W Y d }~nd }~0 0 | �
d� � d }| �|t� | �|jt � d S )Nc S s | �d S r r rV 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 rY r r[ rM r; )r
r r" r� rW r r r! r � test_exit_exception_chaining� s&
*&
z.TestBaseExitStack.test_exit_exception_chainingc s� G dd� dt �� t� fdd���t��fdd��}�|fD ]�}��� �| z0|� � t� �W d � n1 sl0 Y W n0 � y� } z��|j� W Y d }~nd }~0 0 ��d� W d � q:1 s�0 Y q:d S )Nc @ s e Zd ZdS )zPTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.MyExceptionNr r r r r � MyException s r$ c 3 s@ z
d V W n0 t y: � � } z| �W d | _nd | _0 Y n0 d S r )�
BaseExceptionr rV )r$ r r �my_cm s
zJTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.my_cmc 3 s> �� � �"} | �� � � | V W d � n1 s00 Y d S r )r� r )r� )r& r
r r �my_cm_with_exit_stack s
zZTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.my_cm_with_exit_stackr )rL r6 rK r r[ r rM )r
r' r rW r )r$ r& r
r �)test_exit_exception_explicit_none_context s
("z;TestBaseExitStack.test_exit_exception_explicit_none_contextc
C s" dd� }dd� }zF| � � �*}|�dd� � |�|t� W d � n1 sJ0 Y W n0 ty� } z| �|t� W Y d }~nd }~0 0 | �d� zN| � � �2}|�|t� |�|� |�|t� W d � n1 s�0 Y W n2 t�y } z| �|t� W Y d }~nd }~0 0 | �d� d S ) Nc S s | �d S r r rV r r r r 0 s zHTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W s dS r� r r� r r r r" 3 s zKTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S s d S r r r r r r r
8 r zGTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>r z.Expected KeyError, but no exception was raised)r� r� r rL rY rM r r )r
r r" r� rW r r r �#test_exit_exception_non_suppressing. s"
."
."z5TestBaseExitStack.test_exit_exception_non_suppressingc
C s� t dd� �}td�}td�}td�}td�}zZ| �� �>}|�||�� |�||�� |�||�� |�W d � n1 sz0 Y W nr ty� } zZ| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj� W Y d }~n
d }~0 0 d S )Nc s s zd V W | �n| �0 d S r r rV r r r �gets_the_context_rightK s zZTestBaseExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr- r� r� r� )r6 rL r� r r r r[ )r
r* �exc1�exc2�exc3�exc4r� rW r r r �(test_exit_exception_with_correct_contextI s(
&
�z:TestBaseExitStack.test_exit_exception_with_correct_contextc
C s dd� }t d�}t d�}t d�}t d�}t d�}zL| �� �0}|�|||� |�|||� |�W d � n1 sp0 Y W n� t �y } zp| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj|� | �|jjjjj� W Y d }~n
d }~0 0 d S )Nc S s z | �W |�n|�0 d S r r )r Z outer_excr r r � raise_nestedk s zQTestBaseExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr- r� r� r� r� )rL r� r� r r r[ ) r
r0 r+ r, r- r. Zexc5r� rW r r r �)test_exit_exception_with_existing_contexth s,
& � �z;TestBaseExitStack.test_exit_exception_with_existing_contextc
C sz dd� }z>| � � �"}|�|� dd W d � n1 s:0 Y W n. tyt } z| �d� W Y d }~n
d }~0 0 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 rM )r
r" r� rW r r r � test_body_exception_suppress� s
*z.TestBaseExitStack.test_body_exception_suppressc C sV | � � �:}|�dd� � |�dd� � |�dd� � W d � n1 sH0 Y d S )Nc W s dS r� r rV r r r r
� r zITestBaseExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s dd S r� r rV r r r r
� r c W s i d S r� r rV r r r r
� r r r r r r �%test_exit_exception_chaining_suppress� s
z7TestBaseExitStack.test_exit_exception_chaining_suppressc C sD | � � �(}td�D ]}|�t� qW d � n1 s60 Y d S )Ni' )r� �ranger� �int)r
r� �ir r r �test_excessive_nesting� 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 �Example� r r8 r r- ) r� r r� r r� r r r r� )r
r8 r r� r r r �test_instance_bypass� s
z&TestBaseExitStack.test_instance_bypassc s� G dd� dt �� G dd� dt�}t� fdd��}tdd� �}| �� ��X}| �� �0}|�|� � |�|� � |d ��W d � n1 s�0 Y W d � n1 s�0 Y |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� r r: c @ s e Zd ZdS )zLTestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueRuntimeErrorNr r r r r �UniqueRuntimeError� r r; c
3 s> z
dV W n. t y8 } z� d�| �W Y d } ~ n
d } ~ 0 0 d S )Nr- z
new exception�rL rV �r: r r �second� s
z@TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.secondc
s s8 z
dV W n( t y2 } z| �W Y d } ~ n
d } ~ 0 0 d S r� r< rV r r r �first� s
z?TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) rL rB r6 r r� r Z exceptionrY r r[ rZ r )r
r; r> r? Zerr_ctxZes_ctxrW r r= r � test_dont_reraise_RuntimeError� s"
D 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/ r1 r2 r3 r7 r9 r@ r r r r r� . s,
)"
0 "
r� c @ s e Zd ZeZdS )�
TestExitStackN)r r r � ExitStackr� r r r r rA � s rA 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_streamro r7 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_streamrD r )r
� orig_stdoutr r r �test_no_redirect_in_init� s
z+TestRedirectStream.test_no_redirect_in_initc C s� t �� }d}tt| j�}| �|��$ t|tt| j�d� W d � n1 sL0 Y | �tt| j�|� |�� � � }| �
||� d S )Nz<Consider an API like help(), which prints directly to stdout��file) �io�StringIOrE rF rG rD �printr �getvalue�stripr7 )r
r� �msgrH �sr r r �test_redirect_to_string_io� s 2 z-TestRedirectStream.test_redirect_to_string_ioc C sB t �� }| �|�� }| �||� W d � n1 s40 Y d S r )rL rM rD 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 d � n1 sN0 Y |�$ tdtt| j�d� W d � n1 s�0 Y | �tt| j�|� |�� }| � |d� d S �N�Hello� )�endrK zWorld!rJ z
Hello World!
�
rL rM rD rE rF rG rN r rO r7 �r
r� Z
write_to_frH rR r r r �test_cm_is_reusable� s
42z&TestRedirectStream.test_cm_is_reusablec C s� t �� }| �|�}tt| j�}|�^ tddtt| j�d� |�$ tdtt| j�d� W d � n1 sh0 Y W d � n1 s�0 Y | �tt| j�|� |�� }| � |d� d S rV rZ r[ r r r �test_cm_is_reentrant� s
Pz'TestRedirectStream.test_cm_is_reentrant)
r r r rD rG r r� r� rI rS rU r\ r] r r r r rC � s
rC c @ s e Zd ZeZdZdS )�TestRedirectStdout�stdoutN)r r r �redirect_stdoutrD rG r r r r r^ s r^ c @ s e Zd ZeZdZdS )�TestRedirectStderr�stderrN)r r r �redirect_stderrrD rG r r r r ra s ra 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 )�suppressro r7 r� r r r r� s z TestSuppress.test_instance_docsc C s6 t t��}| �|� W d � n1 s(0 Y d S r )re r r[ )r
rT r r r �test_no_result_from_enter s
z&TestSuppress.test_no_result_from_enterc C s> t t��" | �tdd�d� W d � n1 s00 Y d S )Nr� r� � )re r r7 �powr r r r �test_no_exception s
z TestSuppress.test_no_exceptionc C s4 t t�� td� W d � n1 s&0 Y d S )Nr� )re r �lenr r r r �test_exact_exception s
z!TestSuppress.test_exact_exceptionc C s4 t t�� dd W d � n1 s&0 Y d S )NrW �2 )re � LookupErrorr r r r �test_exception_hierarchy$ s
z%TestSuppress.test_exception_hierarchyc C s^ | � t��@ tt�� dd W d � n1 s20 Y W d � n1 sP0 Y d S r� )r r; re r r r r r �test_other_exception( s
z!TestSuppress.test_other_exceptionc C s\ | � t��> t� � dd W d � n1 s00 Y W d � n1 sN0 Y d S r� )r r; re r r r r � test_no_args- s zTestSuppress.test_no_argsc C sh t tt�� dd W d � n1 s(0 Y t tt�� td� W d � n1 sZ0 Y d S )Nr- r r� )re r; r rj r r r r � test_multiple_exception_args2 s & z)TestSuppress.test_multiple_exception_argsc C s� t t�}|� W d � n1 s"0 Y |� td� W d � n1 sN0 Y |�H |� td� W d � n1 s�0 Y d}dd W d � n1 s�0 Y | �|� d S )Nr� Tr- r )re rL rj r' )r
Zignore_exceptionsZouter_continuedr r r r] 8 s &&&z!TestSuppress.test_cm_is_reentrantN)r r r r r� r� rf ri rk rn ro rp rq r] r r r r rd s
rd �__main__) ro rL rF r� r� Zunittest�
contextlibr� r r| ZTestCaser r+ r� r� r� r� r� r� r� r� rA rC r^ ra rd r �mainr r r r �<module> s: + Y" 0 ! 96