File: //usr/local/lib/python3.10/test/__pycache__/test_contextlib.cpython-310.pyc
o
�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__� �1/usr/local/lib/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 �managerr 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! � r1 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/ � r4 c @ r3 )zFTestAbstractContextManager.test_structural_subclassing.<locals>.NoExitN)r r r r r r r r �NoExit4 r5 r6 )�
assertTrue�
issubclassr � assertFalse)r r1 r r4 r6 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 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� assertEqualrA �r rD �xr rB 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 rB r r rD K s
�
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor= r> rF )rG r% �ZeroDivisionErrorrH rA rI r rB 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$ rS r> z raise RuntimeErrorSubclass(42)c @ r! )zSContextManagerTestCase.test_contextmanager_traceback.<locals>.StopIterationSubclassNr" r r r r �StopIterationSubclassw r$ rT �spam��type� was suppressedzraise stop_exc)rG rL r/ �
extract_tb�
__traceback__rH �len�name�line� RuntimeError�
StopIteration�subTestrW r �fail)r rO �eZframesrS rT �stop_excr r r rQ Z s^
��������� �� ���z4ContextManagerTestCase.test_contextmanager_tracebackc C s6 t dd� �}|� }|�� | �|�ttd�d �� d S )Nc s rN r r r r r r �whee� rP zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)rG r r9 r r& )r rd �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>.whoore )rG r r% r^ r r& )r rh rf 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 rF )rA rL r rH )rb �r rC r r rD � s �
��zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor= r> r? rF )rG rH rA rL rI r rj 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 rN r r r r r r rD � rP zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohooc @ r! )zYContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.StopIterationSubclassNr" r r r r rT � r$ rT rU rV rX )rG r_ r` rW � Exceptionr ra )r rD rT rc �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
rD rU z StopIteration was suppressed)�execr_ rl r ra )r �code�localsrD rc rm 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)rl r^ ��excr r r �test_issue29692� � �
��zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r rs zissue29692:Unchained) rG rL rl r rW r^ rH r �assertIsInstance� __cause__r_ � assertIsNone)r rv rm 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)re c S � dS )�Whee!Nr )rU r r r �baz� � zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz�rG )r r� r� r r r � _create_contextmanager_attribs� s
z5ContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nr� r� )r� rH r re �r r� r r r �test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nr� �r� rH �__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 rD s �z4ContextManagerTestCase.test_keywords.<locals>.woohoo� � �! �, r� )r� r� r� r� �rG rH )r rD �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 �
gc_collectrz ��a�br* r r rD s �
z5ContextManagerTestCase.test_nokeepref.<locals>.woohoo�r� r� )r r� rD 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 rN r r r� r r r rD ' rP z8ContextManagerTestCase.test_param_errors.<locals>.woohoo� � r� )rG r% r& �r rD 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= )rH )�before)�depthr r r rD 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 rK rM rQ rg ri rk rn rr r{ 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� rH �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* rB r r �closeR � z-ClosingTestCase.test_closing.<locals>.C.closeN�r r r r� r rB r r �CQ � r� r= )rH r� �r r� rJ �yr rB 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* rB r r r� ] r� z3ClosingTestCase.test_closing_error.<locals>.C.closeNr� r rB r r r� \ r� r� r= r )rH r% rL r� r� r rB 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 )
�tempfile�mktemp�openr9 �closed�writer7 r% rL rH �readr �unlink)r ZtfnrO 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 )r9 r7 r% rL )r �lock�lockedr r r � boilerPlate� s �
�� �z LockContextTestCase.boilerPlatec C � t �� }| �||j� d S r )� threading�Lockr� 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� � 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� � 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� �BoundedSemaphorer� 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 )ru �catch�r ru r r r r � s zmycontext.__exit__) r r r r� r� ru 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� rH 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 r7 r� rH ru )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 )Nre r T)r� �assertRaisesRegex� NameError�assertIsNotNoneru 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 �rz ru r7 r� r �r� r r r �test� � z1TestContextDecorator.test_decorator.<locals>.testr� )r� rH ru �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��)Nre )rz ru r7 r� r� r r r r r � s z@TestContextDecorator.test_decorator_with_exception.<locals>.testre r )r� r� r� r ru 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
rH 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� r2 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� , r2 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 rt r� r r r r @ �
zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r r r r� ru 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 rH ru )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� rB r r rD P rE zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor= c s � � �dg� ��| � d S r� )rH rA )rJ rj r r r W s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.test� something)r= r r? �something else)r= r r? r� )r rD r r rj 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� rH 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 rz r� rH r% r& )r �expectedr' �stackr r� rO �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- ru �exc_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 )rz r1 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 r: r r r �__init__� r z4TestBaseExitStack.test_push.<locals>.ExitCM.__init__c S s | � d� d S )NzShould not be called!)ra r* r r r r � r� z5TestBaseExitStack.test_push.<locals>.ExitCM.__enter__c W s | j |� d S r r9 �r r6 r r r r � r� z4TestBaseExitStack.test_push.<locals>.ExitCM.__exit__N�r r r r; r r r r r r �ExitCM� � r> ���r= r )rL r r" �pushr r+ �__self__)r r4 r7 r8 r> r. �cmr r3 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@ r< r& r r r � r� z=TestBaseExitStack.test_enter_context.<locals>.TestCM.__exit__Nr0 r r& r r �TestCM� s rF c r� )Nr$ r@ r r& r r r' � � z3TestBaseExitStack.test_enter_context.<locals>._exitr@ r= r
)r= r
r� r$ )
r r" r% r �
enter_contextr r+ rB rA rH )r rF rC 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' � rG z+TestBaseExitStack.test_close.<locals>._exitr
r= )r" r% r r� rA rH )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� rE r@ r r& r r r' � rG z-TestBaseExitStack.test_pop_all.<locals>._exitr= r
)r= r
r� )r" r% r �pop_allrA r� rH )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 rt r r r �<lambda>� r� z3TestBaseExitStack.test_exit_raise.<locals>.<lambda>r= r )r% rL r" rA �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 rt r r r rN � r� z6TestBaseExitStack.test_exit_suppress.<locals>.<lambda>r= r �r" rA rO 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 rt 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))rT z raise excc S � g | ]}|j |jf�qS r �r\ r] ��.0rO r r r �
<listcomp> � zCTestBaseExitStack.test_exit_exception_traceback.<locals>.<listcomp>c S rW r rX rY r r r r[ r\ )rU rR ) r" r% �
ValueErrorrx r/ rY rZ � callback_error_internal_framesrH � __context__rL )r rT r. rb ru Z ve_framesr- Z
zde_framesr r r rU � 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 rt r� r r r r; 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 rt r< r r r r s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__Nr= r r r r �RaiseExc r? r` c @ r )zUTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s || _ || _d S r )�outer�inner)r ra rb r r r r; 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 )rb ra r< r r r r $ s z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__Nr= r r r r �RaiseExcWithContext s rc 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� )rW �
saved_detailsr< r r r r - s
zVTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__Nr0 r r r r � SuppressExc* r2 re r= r �0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr r] rx r_ rz ra rd rL )r r` rc re ru � 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 rS r r rt r r r rT E r+ zATestBaseExitStack.test_exit_exception_chaining.<locals>.raise_excc s | � dS r� r r5 �rd r r � suppress_excI s zDTestBaseExitStack.test_exit_exception_chaining.<locals>.suppress_excr= r rf ) r" r% rg rh r rA r] rx r_ rz ra rL )r rT rl r. ru ri r rk 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$ rn c 3 s2 � zd V W d S t y � � } z| �d | _w w r )�
BaseExceptionr_ rt )rn 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" rH )r. )rp r r r �my_cm_with_exit_stackt s
�
"�zZTestBaseExitStack.test_exit_exception_explicit_none_context.<locals>.my_cm_with_exit_stackrf )rl rG r` rg rz r_ ra )r rq rC ru r )rn rp 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 rS r r rt r r r rT � r+ zHTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W r� r� r r5 r r r rl � r+ zKTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S r, r r r r r r rN � r� zGTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>rf z.Expected KeyError, but no exception was raised)r" r% rg rl rx ra rh rA )r rT rl r. ru 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 rt r r r �gets_the_context_right� s �zZTestBaseExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr= r
r� r$ )rG rl r" rH r r_ rz )r rt �exc1�exc2�exc3�exc4r. ru 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 )ri Z outer_excr r r � raise_nested� s zQTestBaseExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr= r
r� r$ r� )rl r" r% r r_ rz ) r rz ru rv rw rx Zexc5r. ru 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 r5 r r r rl � r+ zDTestBaseExitStack.test_body_exception_suppress.<locals>.suppress_excr= r z%Expected no exception, got IndexError)r" rA rg ra )r rl r. ru 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 rt r r r rN � r� zITestBaseExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s dd S r� r rt r r r rN � � c W s i d S r� r rt r r r rN � r} rQ rO 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� � r� r@ r= ) r r r" r% r rH rA r r+ )r r� rC 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�rl rt �r� r r �second� rw z@TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.secondc
s s, � zdV W d S t y } z| �d } ~ ww r� r� rt r r r �first s � ��z?TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) rl r^ rG r% r" rH � exceptionrx r_ rz ry r )r r� r� r� Zerr_ctxZes_ctxru r r� r � test_dont_reraise_RuntimeError� s&
� � z0TestBaseExitStack.test_dont_reraise_RuntimeError)r r r r" r r� r� r# r0 rD rI rJ rM rP rR rU rj rm rr rs ry r{ r| r~ r� 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" r^ 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� rH 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�striprH )r rO �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 rO � 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� rH �r rO 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� rH 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� r] rz )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� r] rH �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& r[ 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% rL 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% rL 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� rL r& r[ 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� rl r[ r7 )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 �