Your IP : 216.73.216.196
o
0�i�� � @ sl d Z ddlZddlZddlZddlZddlZddlZddlT ddlm Z ddl
m Z ddl Z G dd� dej
�ZG dd � d ej
�ZG d
d � d ej
�ZG d d
� d
ej
�ZG dd� dej
�ZG dd� dej
�ZG dd� de�ZG dd� dej
�ZG dd� d�ZG dd� deej
�ZG dd� d�ZG d d � d eej
�ZG d d � d eej
�ZG d d!� d!ej
�Z e d"kr�e� � dS dS )#z9Unit tests for contextlib.py, and other context managers.� N)�*)�support)� os_helperc @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�TestAbstractContextManagerc C s* G dd� dt �}|� }| �|�� |� d S )Nc � e Zd Z� fdd�Z� ZS )z;TestAbstractContextManager.test_enter.<locals>.DefaultEnterc � t � j|� d S �N��super�__exit__��self�args�� __class__� �;/opt/alt/python310/lib64/python3.10/test/test_contextlib.pyr � zDTestAbstractContextManager.test_enter.<locals>.DefaultEnter.__exit__��__name__�
__module__� __qualname__r �
__classcell__r r r r � DefaultEnter � r )�AbstractContextManager�assertIs� __enter__)r r Zmanagerr r r �
test_enter s z%TestAbstractContextManager.test_enterc C sF G dd� dt �}| �t�� |� W d � d S 1 s w Y d S )Nc @ � e Zd ZdS )zETestAbstractContextManager.test_exit_is_abstract.<locals>.MissingExitN�r r r r r r r � MissingExit � 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 @ � e Zd Zdd� Zdd� ZdS )zRTestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratchc S � | S r r �r r r r r " � z\TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__enter__c S � d S r r )r �exc_type� exc_value� tracebackr r r r $ r* z[TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__exit__N�r r r r r r r r r �ManagerFromScratch! � r0 c r )zLTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnterc r r r
r
r r r r * r zUTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnter.__exit__r r r r r r ) r r c @ � e Zd ZdZdS )zGTestAbstractContextManager.test_structural_subclassing.<locals>.NoEnterN)r r r r r r r r �NoEnter/ � r3 c @ r2 )zFTestAbstractContextManager.test_structural_subclassing.<locals>.NoExitN)r r r r r r r r �NoExit4 r4 r5 )�
assertTrue�
issubclassr � assertFalse)r r0 r r3 r5 r r r �test_structural_subclassing s z6TestAbstractContextManager.test_structural_subclassingN)r r r r r& r9 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 dd� Z
ejdd� �Zejdd� �Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�ContextManagerTestCasec sr g � t � fdd��}|� �}| �� dg� | �|d� � �|� W d � n1 s*w Y | �� g d�� d S )Nc 3 s � � � d� dV � � d� d S �N� �* �� ��appendr ��stater r �woohoo> � �
z@ContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor< r= �r< r= r> )�contextmanager� assertEqualr@ �r rC �xr rA r �test_contextmanager_plain< s �z0ContextManagerTestCase.test_contextmanager_plainc s� g � t � fdd��}| �t��) |� �}| �� dg� | �|d� � �|� t� �1 s,w Y W d � n1 s;w Y | �� g d�� d S )Nc 3 s0 � � � d� z dV W � � d� d S � � d� w r; r? r rA r r rC K s
�
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor< r= rE )rF r$ �ZeroDivisionErrorrG r@ rH r rA r �test_contextmanager_finallyI s
� �z2ContextManagerTestCase.test_contextmanager_finallyc C s@ t dd� �}z|� � dd W d � n1 sw Y W n ty7 } z t�|j�}W Y d }~nd }~ww | �t|�d� | �|d jd� | �|d jd� G dd� dt �}z|� � |d ��1 sfw Y W n |y� } z t�|j�}W Y d }~nd }~ww | �t|�d� | �|d jd� | �|d jd
� G d d � d t
�}t
d
�|d
�fD ]n}| j t |�d��[ z|� � |�1 s�w Y W n t |�y� } z| �
||� t�|j�}W Y d }~n
d }~ww | �|� d�� | �t|�d� | �|d jd� | �|d jd� W d � n 1 �sw Y q�d S )Nc s � � d V d S r r r r r r �f[ � �
z?ContextManagerTestCase.test_contextmanager_traceback.<locals>.fr< r � test_contextmanager_traceback�1/0c @ r )zRContextManagerTestCase.test_contextmanager_traceback.<locals>.RuntimeErrorSubclassNr! r r r r �RuntimeErrorSubclassj r# rR r= z raise RuntimeErrorSubclass(42)c @ r )zSContextManagerTestCase.test_contextmanager_traceback.<locals>.StopIterationSubclassNr! r r r r �StopIterationSubclassw r# rS �spam��type� was suppressedzraise stop_exc)rF rK r. �
extract_tb�
__traceback__rG �len�name�line� RuntimeError�
StopIteration�subTestrV r �fail)r rN �eZframesrR rS �stop_excr r r rP Z s^
��������� �� ���z4ContextManagerTestCase.test_contextmanager_tracebackc C s6 t dd� �}|� }|�� | �|�ttd�d �� d S )Nc s rM r r r r r r �whee� rO zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)rF r r8 r r% )r rc �ctxr r r � test_contextmanager_no_reraise� s
z5ContextManagerTestCase.test_contextmanager_no_reraisec C s6 t dd� �}|� }|�� | �t|jttd�d � d S )Nc s s" � zd V W d S d V Y d S r r r r r r �whoo� s
� zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoord )rF r r$ r] r r% )r rg re r r r �*test_contextmanager_trap_yield_after_throw� s
�zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc sn g �t � �fdd��}|� �}� ��dg� � �|d� ��|� td��1 s(w Y � ��g d�� d S )Nc
3 s` � �� d� zdV W d S ty/ } z�� | jd � � ��g d�� W Y d } ~ d S d } ~ ww )Nr< r= r rE )r@ rK r rG )ra �r rB r r rC � s �
��zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor< r= r> rE )rF rG r@ rK rH r ri r �test_contextmanager_except� s
�z1ContextManagerTestCase.test_contextmanager_exceptc C s� t dd� �}G dd� dt�}td�|d�fD ]K}| jt|�d��9 z|� � |�1 s+w Y W n tyI } z | �||� W Y d }~n
d }~ww | �|� d�� W d � n1 s\w Y qd S )Nc s rM r r r r r r rC � rO zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohooc @ r )zYContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.StopIterationSubclassNr! r r r r rS � r# rS rT rU rW )rF r^ r_ rV � Exceptionr r` )r rC rS rb �exr r r �#test_contextmanager_except_stopiter� s$
���� ���z:ContextManagerTestCase.test_contextmanager_except_stopiterc
C s� d}i }t |||� |d }td�}z|� � |�1 s w Y W n ty; } z
| �||� W Y d }~d S d }~ww | �d� d S )Nztfrom __future__ import generator_stop
from contextlib import contextmanager
@contextmanager
def woohoo():
yield
rC rT z StopIteration was suppressed)�execr^ rk r r` )r �code�localsrC rb rl r r r �!test_contextmanager_except_pep479� s ���z8ContextManagerTestCase.test_contextmanager_except_pep479c
C s� t dd� �}z|� � t�1 sw Y W n* ty@ } z | �t|�t� | �|jd d� | �|j t� W Y d }~nd }~ww z|� � t
d��1 sMw Y W d S ty~ } z | �t|�t
� | �|jd d� | � |j � W Y d }~d S d }~ww )Nc
s s2 � zd V W d S t y } ztd�| �d } ~ ww )N�issue29692:Chained)rk r] ��excr r r �test_issue29692� � �
��zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r rr zissue29692:Unchained) rF rK rk r rV r] rG r �assertIsInstance� __cause__r^ � 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 )r� r� r r� r �attribs� s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)rd c S � dS )�Whee!Nr )rT r r r �baz� � zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz�rF )r r� r� r r r � _create_contextmanager_attribs� s
z5ContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nr� r� )r� rG r rd �r r� r r r �test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nr� �r� rG �__doc__r� r r r � test_contextmanager_doc_attrib s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s | � � d �}| �|jd� d S r� r� r� r r r �*test_instance_docstring_given_cm_docstring s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc C sN t dd� �}|ddddd��}| �|d� W d � d S 1 s w Y d S ) Nc s s � | |||fV d S r r �r r} r �kwdsr r r rC s �z4ContextManagerTestCase.test_keywords.<locals>.woohoo� � �! �, r� )r� r� r� r� �rF rG )r rC �targetr r r �
test_keywords
s
"�z$ContextManagerTestCase.test_keywordsc sT G dd� d�}t � fdd��}||� |� d�� W d � d S 1 s#w Y d S )Nc @ r )z0ContextManagerTestCase.test_nokeepref.<locals>.ANr! r r r r �A r# r� c 3 s@ � t �| �} t �|�}t�� � �| � � � �|� � d V d S r )�weakref�refr Z
gc_collectry ��a�br) r r rC s �
z5ContextManagerTestCase.test_nokeepref.<locals>.woohoo�r� r� )r r� rC r r) r �test_nokeepref s "�z%ContextManagerTestCase.test_nokeeprefc C s� t dd� �}| �t�� |� W d � n1 sw Y | �t��
|dd� W d � n1 s3w Y | �t�� |dd� W d � d S 1 sNw Y d S )Nc s rM r r r� r r r rC ' rO z8ContextManagerTestCase.test_param_errors.<locals>.woohoo� � r� )rF r$ r% �r rC 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< )rG )Zbefore)�depthr r r rC 4 s �z5ContextManagerTestCase.test_recursive.<locals>.woohooc s � dk r �� d S d S )N�
r r )r� � recursiver r r� = s
�z8ContextManagerTestCase.test_recursive.<locals>.recursiver� r� r )r� r� r r �test_recursive2 s z%ContextManagerTestCase.test_recursiveN)r r r rJ rL rP rf rh rj rm rq rz r� r� r �requires_docstringsr� r� r� r� r� r� r r r r r: : s(
2
r: c @ s* e Zd Zejdd� �Zdd� Zdd� ZdS )�ClosingTestCasec C s t j}t d �}| �|j|� d S r )�closingr� rG �r Z cm_docstring�objr r r �test_instance_docsH s z"ClosingTestCase.test_instance_docsc sn g � G � fdd�d�}|� }| � � g � t|��}| � ||� W d � n1 s)w Y | � � dg� d S )Nc � e Zd Z� fdd�ZdS )z'ClosingTestCase.test_closing.<locals>.Cc � � � d� d S r� r? r) rA r r �closeR � z-ClosingTestCase.test_closing.<locals>.C.closeN�r r r r� r rA r r �CQ � r� r< )rG r� �r r� rI �yr rA r � test_closingO s
�z ClosingTestCase.test_closingc s� g � G � fdd�d�}|� }| � � g � | �t��& t|��}| � ||� dd W d � n1 s3w Y W d � n1 sBw Y | � � dg� d S )Nc r� )z-ClosingTestCase.test_closing_error.<locals>.Cc r� r� r? r) rA r r r� ] r� z3ClosingTestCase.test_closing_error.<locals>.C.closeNr� r rA r r r� \ r� r� r< r )rG r$ rK r� r� r rA r �test_closing_errorZ s
�� �z"ClosingTestCase.test_closing_errorN)r r r r r� r� r� r� r r r r r� F s
r� c @ � e Zd Zdd� ZdS )�NullcontextTestCasec C sN G dd� d�}|� }t |��}| �||� W d � d S 1 s w Y d S )Nc @ r )z/NullcontextTestCase.test_nullcontext.<locals>.CNr! r r r r r� j r# r� )� nullcontextr )r r� �cZc_inr r r �test_nullcontexti s
"�z$NullcontextTestCase.test_nullcontextN)r r r r� r r r r r� h s r� c @ r� )�FileContextTestCasec C s t �� }zxd }t|ddd��}| �|j� |�d� W d � n1 s$w Y | �|j� d }| �t��1 t|ddd��}| �|j� | � |�
� d� dd W d � n1 s[w Y W d � n1 sjw Y | �|j� W t � |� d S t � |� w )N�wzutf-8)�encodingzBooh
�rr< r )
�tempfileZmktemp�openr8 �closed�writer6 r$ rK rG �readr �unlink)r ZtfnrN r r r � testWithOpens s& �
�� �z FileContextTestCase.testWithOpenN)r r r r� r r r r r� q s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�LockContextTestCasec C s� | � |� � |� | �|� � W d � n1 sw Y | � |� � | �t��$ |� | �|� � dd W d � n1 sAw Y W d � n1 sPw Y | � |� � d S �Nr< r )r8 r6 r$ rK )r �lock�lockedr r r � boilerPlate� s �
�� �z LockContextTestCase.boilerPlatec C � t �� }| �||j� d S r )� threadingZLockr� r� �r r� r r r � testWithLock� � z LockContextTestCase.testWithLockc C r� r )r� �RLockr� � _is_ownedr� r r r �
testWithRLock� r� z!LockContextTestCase.testWithRLockc �$ t �� � � fdd�}| �� |� d S )Nc s � � � S r )r� r �r� r r r� � s z5LockContextTestCase.testWithCondition.<locals>.locked)r� Z Conditionr� �r r� r r� r �testWithCondition� s z%LockContextTestCase.testWithConditionc r� )Nc � � � d�r � �� dS dS �NFT��acquire�releaser r� r r r� � �
z5LockContextTestCase.testWithSemaphore.<locals>.locked)r� Z Semaphorer� r� r r� r �testWithSemaphore� � z%LockContextTestCase.testWithSemaphorec r� )Nc r� r� r� r r� r r r� � r� z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)r� ZBoundedSemaphorer� r� r r� r �testWithBoundedSemaphore� r� z,LockContextTestCase.testWithBoundedSemaphoreN) r r r r� r� r� r� r� r� r r r r r� � s
r� c @ s, e Zd ZdZdZdZdZdd� Zdd� ZdS )� mycontextz9Example decoration-compatible context manager for testingFNc C �
d| _ | S �NT��startedr) r r r r � � zmycontext.__enter__c G s || _ | jS r )rt �catch�r rt r r r r � s zmycontext.__exit__) r r r r� 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 � t j}t � }| �|j|� d S r )r� r� rG r� r r r r� � � z'TestContextDecorator.test_instance_docsc C sT t � }|�}| �||� | �|j� W d � n1 s w Y | �|jd� d S )N�NNN)r� r r6 r� rG rt )r �context�resultr r r �test_contextdecorator� s �z*TestContextDecorator.test_contextdecoratorc C s� t � }| �td�� |� td��1 sw Y W d � n1 s#w Y | �|j� | �|jd t� t � }d|_|� td��1 sGw Y | �|j� | �|jd t� d S )Nrd 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 � �� � j� ��� j� d S r �ry rt r6 r� r �r� r r r �test� � z1TestContextDecorator.test_decorator.<locals>.testr� )r� rG rt �r r� r r� r �test_decorator� s
z#TestContextDecorator.test_decoratorc sl t � � � � �fdd��}��td�� |� W d � n1 s w Y ��� j� ��� jd t� d S )Nc s �� � j� ��� j� td��)Nrd )ry rt r6 r� r� r r� r r r� � s z@TestContextDecorator.test_decorator_with_exception.<locals>.testrd 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.methodr )r r r r r �r� r r �Test s r r< � r� r� r� r� )r� �objectr rG r� r� r� )r r r� r r r �test_decorating_method s z+TestContextDecorator.test_decorating_methodc C �v G dd� dt �}| �t��$ |� � W d � n1 s w Y W d � d S W d � d S 1 s4w Y d S )Nc @ r' )z7TestContextDecorator.test_typo_enter.<locals>.mycontextc S r+ r r r) r r r � __unter__! r* zATestContextDecorator.test_typo_enter.<locals>.mycontext.__unter__c W r+ r r r� r r r r # r* z@TestContextDecorator.test_typo_enter.<locals>.mycontext.__exit__N)r r r r r r r r r r� r1 r� ��ContextDecoratorr$ �AttributeError�r r� r r r �test_typo_enter � ��"�z$TestContextDecorator.test_typo_enterc C r )Nc @ r' )z6TestContextDecorator.test_typo_exit.<locals>.mycontextc S r+ r r r) r r r r - r* z@TestContextDecorator.test_typo_exit.<locals>.mycontext.__enter__c W r+ r r r� r r r �__uxit__/ r* z?TestContextDecorator.test_typo_exit.<locals>.mycontext.__uxit__N)r r r r r r r r r r� , r1 r� r
r
r r r �test_typo_exit+ r 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 r� r� r� r) r r r r < r� zRTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__enter__c W �
|| _ d S r rs r� r r r r @ �
zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r r r r� rt r r r r r r � somecontext8 s
r c @ r )zFTestContextDecorator.test_contextdecorator_as_mixin.<locals>.mycontextNr! r r r r r� C r# r� c r� r r� r r� r r r� G r� zATestContextDecorator.test_contextdecorator_as_mixin.<locals>.testr� )r r rG rt )r r r� r� r r� r � test_contextdecorator_as_mixin7 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> r? )r� rA r r rC P rD zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor< c s � � �dg� ��| � d S r� )rG r@ )rI ri r r r� W s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.test� something)r< r r> �something else)r< r r> r� )r rC r� r ri r � test_contextmanager_as_decoratorO 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)� Zd*d+� ZdS ),�TestBaseExitStackNc C s" | j j}| � � }| �|j|� d S r )�
exit_stackr� rG r� r r r r� g s z$TestBaseExitStack.test_instance_docsc C s. | � � � W d � d S 1 sw Y d S r )r r) r r r �test_no_resourcesn s
"�z#TestBaseExitStack.test_no_resourcesc s0 di fdi fdi fdt dd�fdt dd�fdt dd�fdt ddd�fg}g � � fd d
�}| �� �h}t|�D ]8\}}|rJ|rJ|j|g|�R i |��}n |rV|j|g|�R � }n|rb|j|fi |��}n|�|�}| �||� q5|jD ] }| �|d j|� | �|d j|j� | � |d j
|j
� qqW d � n1 s�w Y | � � |� g � | �� �[}| � t
�� |jdd � W d � n1 s�w Y | � t
�� | jjd d � W d � n1 s�w Y | � t
�� |j|dd
� W d � n1 s�w Y W d � n 1 �s w Y | � � g � d S )Nr )r< )r< r r< )Zexampler� � )r �callbackc s � � | |f� dS )zTest metadata propagationNr? )r r� �r� r r �_exit} s z.TestBaseExitStack.test_callback.<locals>._exit)�argr )r r )�dictr �reversedr r �_exit_callbacks� __wrapped__ZassertNotEqualr ry r� rG r$ r% )r �expectedr �stackr r� rN �wrapperr r r �
test_callbackr sR �
� �
� � �� �z TestBaseExitStack.test_callbackc s@ t � � �fdd�}dd� }�fdd�}G dd� dt�}��� �w}|�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� |�|� ��|jd d
|� d
d W d � d S 1 s�w Y d S ) Nc s �� | � � d S r )r �r, rt Zexc_tb�Z
exc_raisedr r r � _expect_exc� r z0TestBaseExitStack.test_push.<locals>._expect_excc W r� r� r �� exc_detailsr r r �
_suppress_exc� r* z2TestBaseExitStack.test_push.<locals>._suppress_excc s" � � | � � � |� � � |� d S r )ry r) r) r r �
_expect_ok� s
z/TestBaseExitStack.test_push.<locals>._expect_okc @ r )z+TestBaseExitStack.test_push.<locals>.ExitCMc S r r �� check_exc)r r1 r r r �__init__� r z4TestBaseExitStack.test_push.<locals>.ExitCM.__init__c S s | � d� d S )NzShould not be called!)r` r) r r r r � r� z5TestBaseExitStack.test_push.<locals>.ExitCM.__enter__c W s | j |� d S r r0 �r r- r r r r � r� z4TestBaseExitStack.test_push.<locals>.ExitCM.__exit__N�r r r r2 r r r r r r �ExitCM� � r5 ���r< r )rK r r �pushr r# �__self__)r r+ r. r/ r5 r& �cmr r* r � test_push� s,
"�zTestBaseExitStack.test_pushc s� G � fdd�dt �}g � |� }| �� �,}|j� fdd��}| �|� |�|� | �|jd d j|� � �d� W d � n1 sBw Y | � � g d�� d S ) Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z4TestBaseExitStack.test_enter_context.<locals>.TestCMc r� r� r? r) r r r r � r� z>TestBaseExitStack.test_enter_context.<locals>.TestCM.__enter__c r� �Nr� r? r3 r r r r � r� z=TestBaseExitStack.test_enter_context.<locals>.TestCM.__exit__Nr/ r r r r �TestCM� s r= c r� )Nr r? r r r r r � � z3TestBaseExitStack.test_enter_context.<locals>._exitr7 r< r )r< r r� r )
r r r r� �
enter_contextr r# r9 r@ rG )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 s(w Y | �� ddg� d S )Nc r� r� r? r r r r r � r> z+TestBaseExitStack.test_close.<locals>._exitr r< )r r r� r� r@ rG )r r& r r r r �
test_close� s
�z TestBaseExitStack.test_closec s� g � | � � � }|j� fdd��}| �|� |�� }� �d� W d � n1 s(w Y � �d� |�� | �� g d�� d S )Nc r� r< r? r r r r r � r> z-TestBaseExitStack.test_pop_all.<locals>._exitr< r )r< r r� )r r r� �pop_allr@ r� rG )r r& r � new_stackr r r � test_pop_all� s
�
z TestBaseExitStack.test_pop_allc C s~ | � t��0 | �� �}|�dd� � dd W d � n1 s w Y W d � d S W d � d S 1 s8w Y d S )Nc W r� )NFr rs r r r �<lambda>� r� z3TestBaseExitStack.test_exit_raise.<locals>.<lambda>r< r )r$ rK r r8 �r r& r r r �test_exit_raise� s
��"�z!TestBaseExitStack.test_exit_raisec C sD | � � �}|�dd� � dd W d � d S 1 sw Y d S )Nc W r� r� r rs r r r rE � r� z6TestBaseExitStack.test_exit_suppress.<locals>.<lambda>r< r �r r8 rF r r r �test_exit_suppress� s
"�z$TestBaseExitStack.test_exit_suppressc
C s� dd� }z | � � �}|�|t� dd W d � n1 s w Y W n ty8 } z|}W Y d }~nd }~ww | �|t� t�|j�}dg| j ddg }| �dd � |D �|� | �|j t
� t�|j j�}| �d
d � |D �d g� d S ) Nc S � | �r r rs r r r � raise_exc� r* zBTestBaseExitStack.test_exit_exception_traceback.<locals>.raise_excr< r )� test_exit_exception_tracebackz with self.exit_stack() as stack:)�
_exit_wrapperzcallback(*args, **kwds))rK z raise excc S � g | ]}|j |jf�qS r �r[ r\ ��.0rN r r r �
<listcomp> � zCTestBaseExitStack.test_exit_exception_traceback.<locals>.<listcomp>c S rN r rO rP r r r rR rS )rL rQ ) r r �
ValueErrorrw r. rX rY � callback_error_internal_framesrG � __context__rK )r rK r& ra rt Z ve_framesr% Z
zde_framesr r r rL � s:
���� ������z/TestBaseExitStack.test_exit_exception_tracebackc C s\ G dd� d�}G dd� d�}G dd� d�}zV|t ��H |tt��3 |� � |t�� dd W d � n1 s8w Y W d � n1 sGw Y W d � n1 sVw Y W d � n1 sew Y W n) t y� } z | �|jt� | �|jjt� | �|jjj� W Y d }~n
d }~ww | �d � |jd }| �|t� | �|jt � d S )
Nc @ r )zJTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcc S r r rs r� r r r r2 r zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__init__c S r( r r r) r r r r r* zTTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__enter__c W s | j �r rs r3 r r r r s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__Nr4 r r r r �RaiseExc r6 rW c @ r )zUTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s || _ || _d S r )�outer�inner)r rX rY r r r r2 s
z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__init__c S r( r r r) r r r r " r* z_TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__enter__c W s z| j � | j�r )rY rX r3 r r r r $ s z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__Nr4 r r r r �RaiseExcWithContext s rZ c @ r' )zMTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExcc S r( r r r) r r r r + r* zWTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__enter__c W s |t | �_dS r� )rV �
saved_detailsr3 r r r r - s
zVTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__Nr/ r r r r � SuppressExc* r1 r\ r< r �0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr rT rw rV ry r` r[ rK )r rW rZ r\ rt � inner_excr r r �&test_exit_exception_chaining_reference s6
�� �� �� �� ��
z8TestBaseExitStack.test_exit_exception_chaining_referencec
s
dd� }d � � fdd�}z7| � � �)}|�|t� |�|t� |�|t� |�|� |�|t� dd W d � n1 s=w Y W n) tyl } z | �|jt� | �|jjt� | � |jjj� W Y d }~n
d }~ww | �
d� � d }| �|t� | �|jt � d S )Nc S rJ r r rs r r r rK E r* zATestBaseExitStack.test_exit_exception_chaining.<locals>.raise_excc s | � dS r� r r, �r[ r r � suppress_excI s zDTestBaseExitStack.test_exit_exception_chaining.<locals>.suppress_excr< r r] ) r r r^ r_ r r8 rT rw rV ry r` rK )r rK rc r& rt r` r rb r � test_exit_exception_chainingC s.
�� ��
z.TestBaseExitStack.test_exit_exception_chainingc s� G dd� dt �� t� fdd���t��fdd��}�|fD ]E}��� �7 z|� � t� �1 s/w Y W n � yM } z ��|j� W Y d }~n
d }~ww ��d� W d � n1 s]w Y q d S )Nc @ r )zPTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.MyExceptionNr! r r r r � MyExceptionf r# re c 3 s2 � zd V W d S t y � � } z| �d | _w w r )�
BaseExceptionrV rs )re r r �my_cmi s � �zJTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.my_cmc 3 sB � �� � �} | �� � � | V W d � d S 1 sw Y d S r )r r? )r& )rg r r r �my_cm_with_exit_stackt s
�
"�zZTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.my_cm_with_exit_stackr] )rk rF r_ r^ ry rV r` )r rh r: rt r )re rg r r �)test_exit_exception_explicit_none_contextb s(
���
� ���z;TestBaseExitStack.test_exit_exception_explicit_none_contextc
C s" dd� }dd� }z#| � � �}|�dd� � |�|t� W d � n1 s%w Y W n tyC } z | �|t� W Y d }~n
d }~ww | �d� z'| � � �}|�|t� |�|� |�|t� W d � n1 sjw Y W n ty� } z
| �|t� W Y d }~d S d }~ww | �d� d S ) Nc S rJ r r rs r r r rK � r* zHTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W r� r� r r, r r r rc � r* zKTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S r+ r r r r r r rE � r� zGTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>r] z.Expected KeyError, but no exception was raised)r r r^ rk rw r` r_ r8 )r rK rc r& rt r r r �#test_exit_exception_non_suppressing� s2
����
����z5TestBaseExitStack.test_exit_exception_non_suppressingc
C s� t dd� �}td�}td�}td�}td�}z'| �� �}|�||�� |�||�� |�||�� |�1 s6w Y W d S tyw } z.| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj� W Y d }~d S d }~ww )Nc s s � zd V W | �| �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 )rF rk r r? r rV ry )r rk �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�}z | �� �}|�|||� |�|||� |�1 s1w Y W d S t y} } z9| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj|� | �|jjjjj� W Y d }~d S d }~ww )Nc S s
z| �|�r r )r` Z outer_excr r r � raise_nested� s zQTestBaseExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr< r r� r r� )rk r r r rV ry ) r rq rl rm rn ro Zexc5r& rt r r r �)test_exit_exception_with_existing_context� s2
� � ���z;TestBaseExitStack.test_exit_exception_with_existing_contextc
C s~ dd� }z"| � � �}|�|� dd W d � W d S 1 s w Y W d S ty> } z | �d� W Y d }~d S d }~ww )Nc W r� r� r r, r r r rc � r* zDTestBaseExitStack.test_body_exception_suppress.<locals>.suppress_excr< r z%Expected no exception, got IndexError)r r8 r^ r` )r rc r& rt r r r � test_body_exception_suppress� s
&���z.TestBaseExitStack.test_body_exception_suppressc C sX | � � � }|�dd� � |�dd� � |�dd� � W d � d S 1 s%w Y d S )Nc W r� r� r rs r r r rE � r� zITestBaseExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s dd S r� r rs r r r rE � � c W s i d S r� r rs r r r rE � rt rH rF r r r �%test_exit_exception_chaining_suppress� s
"�z7TestBaseExitStack.test_exit_exception_chaining_suppressc C sF | � � �}td�D ]}|�t� q W d � d S 1 s w 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 @ r )z7TestBaseExitStack.test_instance_bypass.<locals>.ExampleNr! r r r r �Example� � rz r7 r< ) r r r r$ r r? r8 r r# )r rz r: r& r r r �test_instance_bypass� s
z&TestBaseExitStack.test_instance_bypassc s� G dd� dt �� G dd� dt�}t� fdd��}tdd� �}| �� ��%}| �� �}|�|� � |�|� � |d ��1 s<w Y W d � n1 sKw Y |j}| �|� � | �|j|� | � |jj� | � |jj
� | � |j
|j� d S )
Nc @ r )zITestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueExceptionNr! r r r r �UniqueException� r{ r} c @ r )zLTestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueRuntimeErrorNr! r r r r �UniqueRuntimeError� r{ r~ c
3 s2 � zdV W d S t y } z� d�| �d } ~ ww )Nr< z
new exception�rk rs �r} r r �second� rv z@TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.secondc
s s, � zdV W d S t y } z| �d } ~ ww r� r rs r r r �first s � ��z?TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) rk r] rF r$ r r? Z exceptionrw rV ry rx r )r r~ r� r� Zerr_ctxZes_ctxrt r r� r � test_dont_reraise_RuntimeError� s&
� � z0TestBaseExitStack.test_dont_reraise_RuntimeError)r r r r r r� r� r r( r; r@ rA rD rG rI rL ra rd ri rj rp rr rs ru ry r| r� r r r r r d s0
)"
0 "
r c @ s e Zd ZeZddgZdS )�
TestExitStack)r zraise exc_details[1])r zif cb(*exc_details):N)r r r � ExitStackr rU r r r r r� s
�r� 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_streamr� rG 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_init0 s
z+TestRedirectStream.test_no_redirect_in_initc C s� t �� }d}tt| j�}| �|�� t|tt| j�d� W d � n1 s&w Y | �tt| j�|� |�� � � }| �
||� d S )Nz<Consider an API like help(), which prints directly to stdout��file) �io�StringIOr� r� r� r� �printr �getvalue�striprG )r rN �msgr� �sr r r �test_redirect_to_string_io5 s � z-TestRedirectStream.test_redirect_to_string_ioc C sD t �� }| �|��}| �||� W d � d S 1 sw Y d S r )r� r� r� r )r rN � 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 s'w Y |� tdtt| j�d� W d � n1 sCw Y | �tt| j�|� |�� }| � |d� d S �N�Hello� )�endr� zWorld!r� z
Hello World!
�
r� r� r� r� r� r� r� r r� rG �r rN Z
write_to_fr� r� r r r �test_cm_is_reusableD s
� �z&TestRedirectStream.test_cm_is_reusablec C s� t �� }| �|�}tt| j�}|�/ tddtt| j�d� |� tdtt| j�d� W d � n1 s4w Y W d � n1 sCw Y | �tt| j�|� |�� }| � |d� d S r� r� r� r r r �test_cm_is_reentrantP s
�� �z'TestRedirectStream.test_cm_is_reentrant)
r r r r� r� r r� r� r� r� r� r� r� r r r r r� $ s
r� c @ � e Zd ZeZdZdS )�TestRedirectStdout�stdoutN)r r r �redirect_stdoutr� r� r r r r r� ] � r� c @ r� )�TestRedirectStderr�stderrN)r r r �redirect_stderrr� r� r r r r r� c r� r� c @ sZ e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dS )� TestSuppressc C r� r )�suppressr� rG r� r r r r� k r� z TestSuppress.test_instance_docsc C s8 t t��}| �|� W d � d S 1 sw Y d S r )r� rT ry )r r� r r r �test_no_result_from_enterr s
"�z&TestSuppress.test_no_result_from_enterc C s@ t t�� | �tdd�d� W d � d S 1 sw Y d S )Nr r� � )r� rT rG �powr) r r r �test_no_exceptionv s
"�z TestSuppress.test_no_exceptionc C s6 t t��
td� W d � d S 1 sw Y d S )Nr� )r� r% rZ r) r r r �test_exact_exceptionz �
"�z!TestSuppress.test_exact_exceptionc C s6 t t��
dd W d � d S 1 sw Y d S )Nr� �2 )r� � LookupErrorr) r r r �test_exception_hierarchy~ r� z%TestSuppress.test_exception_hierarchyc C sp | � t��) tt�� dd W d � n1 sw Y W d � d S W d � d S 1 s1w Y d S r� )r$ rK r� r% r) r r r �test_other_exception� s
��"�z!TestSuppress.test_other_exceptionc C sn | � t��( t� � dd W d � n1 sw Y W d � d S W d � d S 1 s0w Y d S r� )r$ rK r� r) r r r � test_no_args� s
��"�zTestSuppress.test_no_argsc C sj t tt�� dd W d � n1 sw Y t tt��
td� W d � d S 1 s.w Y d S )Nr< r r� )r� rK r% rZ r) r r r � test_multiple_exception_args� s
�
"�z)TestSuppress.test_multiple_exception_argsc C s� t t�}|� W d � n1 sw Y |� td� W d � n1 s'w Y |�$ |� td� W d � n1 s@w Y d}dd W d � n1 sUw Y | �|� d S )Nr� Tr< r )r� rk rZ r6 )r Zignore_exceptionsZouter_continuedr r r r� � s �
�
�
�z!TestSuppress.test_cm_is_reentrantN)r r r r r� r� r� r� r� r� r� r� r� r� r r r r r� i s
r� �__main__) r� r� r� r� r� r. Zunittest�
contextlibr� r Z test.supportr r� ZTestCaser r: r� r� r� r� r r� r� r r� r� r� r� r� r �mainr r r r �<module> sD + " 0 ! ;96 �