File: //usr/local/lib/python3.7/test/__pycache__/test_contextlib_async.cpython-37.pyc
B
��g�: � @ s� d dl Z d dlmZmZmZ d dlZd dlmZ d dlZd dl m
Z
dd� Z G dd� dej �Z
G d d
� d
ej �ZG d d � d e
ej �Zed
kr�e�� dS )� N)�asynccontextmanager�AbstractAsyncContextManager�AsyncExitStack)�support)�TestBaseExitStackc s t �� �� fdd��}|S )z5Decorator to turn an async function into a test case.c s@ � | |�}t �� }t �|� z
|�|�S |�� t �d � X d S )N)�asyncio�new_event_loop�set_event_loopZrun_until_complete�close)�args�kwargs�coro�loop)�func� �6/usr/local/lib/python3.7/test/test_contextlib_async.py�wrapper s
z _async_test.<locals>.wrapper)� functools�wraps)r r r )r r � _async_test
s
r c @ s4 e Zd Zedd� �Zedd� �Zdd� Zdd� Zd S )
� TestAbstractAsyncContextManagerc
� sZ G dd� dt �}|� }| �|�� I d H |� |4 I d H �}| �||� W d Q I d H R X d S )Nc s e Zd Z� fdd�Z� ZS )z@TestAbstractAsyncContextManager.test_enter.<locals>.DefaultEnterc � s t � j|� I d H d S )N)�super� __aexit__)�selfr )� __class__r r r s zJTestAbstractAsyncContextManager.test_enter.<locals>.DefaultEnter.__aexit__)�__name__�
__module__� __qualname__r �
__classcell__r r )r r � DefaultEnter s r )r �assertIs�
__aenter__)r r Zmanager�contextr r r �
test_enter s
z*TestAbstractAsyncContextManager.test_enterc � s� t dd� �� � fdd�}g }td�}| �t��b � � 4 I d H �B x:|� 2 y 3 d H }W n
tk
spX |�|� |�qH Y W W d Q I d H R X W d Q R X | �|dg� d S )Nc S s
d V d S )Nr r r r r �ctx+ s zUTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exit.<locals>.ctxc
s* � � 4 I d H � dV W d Q I d H R X d S )N� r r )r$ r r �gen/ s zUTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exit.<locals>.gen� r% )r �
ValueError� assertRaises�StopAsyncIteration�append� assertEqual)r r& �ret�exc�valr )r$ r �(test_async_gen_propagates_generator_exit' s
,zHTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exitc C s0 G dd� dt �}| �t�� |� W d Q R X d S )Nc @ s e Zd ZdS )zKTestAbstractAsyncContextManager.test_exit_is_abstract.<locals>.MissingAexitN)r r r r r r r � MissingAexit> s r1 )r r) � TypeError)r r1 r r r �test_exit_is_abstract= s z5TestAbstractAsyncContextManager.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 )zWTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.ManagerFromScratchc � s | S )Nr )r r r r r! F s zbTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__aenter__c � s d S )Nr )r �exc_type� exc_value� tracebackr r r r H s zaTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__aexit__N)r r r r! r r r r r �ManagerFromScratchE s r7 c s e Zd Z� fdd�Z� ZS )zQTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.DefaultEnterc � s t � j|� I d H d S )N)r r )r r )r r r r N s z[TestAbstractAsyncContextManager.test_structural_subclassing.<locals>.DefaultEnter.__aexit__)r r r r r r r )r r r M s r c @ s e Zd ZdZdS )zOTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.NoneAenterN)r r r r! r r r r �
NoneAenterS s r8 c @ s e Zd ZdZdS )zNTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.NoneAexitN)r r r r r r r r � NoneAexitX s r9 )Z
assertTrue�
issubclassr � assertFalse)r r7 r r8 r9 r r r �test_structural_subclassingD s z;TestAbstractAsyncContextManager.test_structural_subclassingN)r r r r r# r0 r3 r< r r r r r s r c @ s� e Zd Zedd� �Zedd� �Zedd� �Zedd� �Zed d
� �Zed d � �Z ed
d� �Z
edd� �Z edd� �Z edd� �Z
dd� Zdd� Zejdd� �Zejedd � ��Zed d � �Zd S ) �AsyncContextManagerTestCasec
� sn g � t � fdd��}|� 4 I d H �*}| �� dg� | �|d� � �|� W d Q I d H R X | �� dddg� d S )Nc s � � d� dV � � d� d S )N� �* i� )r+ r )�stater r �woohooc s
zEAsyncContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor> r? i� )r r, r+ )r rA �xr )r@ r �test_contextmanager_plain` s z5AsyncContextManagerTestCase.test_contextmanager_plainc � s� g � t � fdd��}| �t��P |� 4 I d H �0}| �� dg� | �|d� � �|� t� �W d Q I d H R X W d Q R X | �� dddg� d S )Nc s& � � d� z
dV W d � � d� X d S )Nr> r? i� )r+ r )r@ r r rA q s
zGAsyncContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor> r? i� )r r) �ZeroDivisionErrorr, r+ )r rA rB r )r@ r �test_contextmanager_finallyn s
z7AsyncContextManagerTestCase.test_contextmanager_finallyc � sB t dd� �}|� }|�� I d H | �|�ttd�d �I d H � d S )Nc S s
d V d S )Nr r r r r �whee� s zHAsyncContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r r! r; r r2 )r rF r$ r r r � test_contextmanager_no_reraise� s z:AsyncContextManagerTestCase.test_contextmanager_no_reraisec � sR t dd� �}|� }|�� I d H | �t�� |�ttd�d �I d H W d Q R X d S )Nc S s" y
d V W n d V Y nX d S )Nr r r r r �whoo� s
zTAsyncContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoorG )r r! r) � RuntimeErrorr r2 )r rI r$ r r r �*test_contextmanager_trap_yield_after_throw� s
zFAsyncContextManagerTestCase.test_contextmanager_trap_yield_after_throwc � s: t dd� �}|� }| �t�� |�� I d H W d Q R X d S )Nc S s d S )Nr r r r r rI � s zKAsyncContextManagerTestCase.test_contextmanager_trap_no_yield.<locals>.whoo)r r) rJ r! )r rI r$ r r r �!test_contextmanager_trap_no_yield� s z=AsyncContextManagerTestCase.test_contextmanager_trap_no_yieldc � sN t dd� �}|� }|�� I d H | �t�� |�d d d �I d H W d Q R X d S )Nc S s d V d V d S )Nr r r r r rI � s zOAsyncContextManagerTestCase.test_contextmanager_trap_second_yield.<locals>.whoo)r r! r) rJ r )r rI r$ r r r �%test_contextmanager_trap_second_yield� s
zAAsyncContextManagerTestCase.test_contextmanager_trap_second_yieldc � sN t dd� �}|� }|�� I d H | �t�� |�td d �I d H W d Q R X d S )Nc S s( y
d V W n t k
r" t�Y nX d S )N)rJ � SyntaxErrorr r r r rI � s
zLAsyncContextManagerTestCase.test_contextmanager_non_normalised.<locals>.whoo)r r! r) rN r rJ )r rI r$ r r r �"test_contextmanager_non_normalised� s
z>AsyncContextManagerTestCase.test_contextmanager_non_normalisedc
� sx g �t � �fdd��}|� 4 I d H �2}� ��dg� � �|d� ��|� td��W d Q I d H R X � ��dddg� d S )Nc
s^ �� d� y
dV W nD tk
rX } z&�� | jd � � ��dddg� W d d } ~ X Y nX d S )Nr> r? r i� )r+ rD r r, )�e)r r@ r r rA � s
zFAsyncContextManagerTestCase.test_contextmanager_except.<locals>.woohoor> r? i� )r r, r+ rD )r rA rB r )r r@ r �test_contextmanager_except� s
z6AsyncContextManagerTestCase.test_contextmanager_exceptc � s� t dd� �}x�td�td�fD ]�}| jt|�d��n y(|� 4 I d H �
|�W d Q I d H R X W n. tk
r� } z| �||� W d d }~X Y nX | �|� d�� W d Q R X q W d S )Nc S s
d V d S )Nr r r r r rA � s zOAsyncContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohoo�spamZham)�typez was suppressed)r �
StopIterationr* ZsubTestrS � Exceptionr �fail)r rA Zstop_excZexr r r �#test_contextmanager_except_stopiter� s z?AsyncContextManagerTestCase.test_contextmanager_except_stopiterc � s� t dd� �}| �t��. |� 4 I d H � dd W d Q I d H R X W d Q R X | �t��* |� 4 I d H �
t�W d Q I d H R X W d Q R X d S )Nc
S sB y
d V W n2 t k
r< } ztd| � ��| �W d d } ~ X Y nX d S )Nzcaught )rU rJ )r. r r r rA � s
zQAsyncContextManagerTestCase.test_contextmanager_wrap_runtimeerror.<locals>.woohoor> r )r r) rJ r* )r rA r r r �%test_contextmanager_wrap_runtimeerror� s " zAAsyncContextManagerTestCase.test_contextmanager_wrap_runtimeerrorc C s" dd� }t |dd�dd� ��}|S )Nc s � fdd�}|S )Nc s& x � � � D ]\}}t| ||� q
W | S )N)�items�setattr)r �k�v)�kwr r �decorate� s z]AsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )r] r^ r )r] r �attribs� s zKAsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)rG c S s
dV dS )zWhee!Nr )rR r r r �baz� s zGAsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.baz)r )r r_ ra r r r � _create_contextmanager_attribs� s z:AsyncContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nra r` )rb r, r rG )r ra r r r �test_contextmanager_attribs� s z7AsyncContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S )NzWhee!)rb r, �__doc__)r ra r r r � test_contextmanager_doc_attrib s z:AsyncContextManagerTestCase.test_contextmanager_doc_attribc
� s< | � � d �}| �|jd� |4 I d H � W d Q I d H R X d S )NzWhee!)rb r, rd )r ra r r r �*test_instance_docstring_given_cm_docstring s zFAsyncContextManagerTestCase.test_instance_docstring_given_cm_docstringc
� sF t dd� �}|ddddd�4 I d H �}| �|d� W d Q I d H R X d S ) Nc S s | |||fV d S )Nr )r r r �kwdsr r r rA s z9AsyncContextManagerTestCase.test_keywords.<locals>.woohoor% r' �! �, )r r r rg )r% r' rh ri )r r, )r rA �targetr r r �
test_keywords s z)AsyncContextManagerTestCase.test_keywordsN)r r r r rC rE rH rK rL rM rO rQ rW rX rb rc r Zrequires_docstringsre rf rk r r r r r= ^ s
r= c @ sX e Zd ZG dd� de�ZeZdd� Zedd� �Zedd� �Z ed d
� �Z
ed d � �Z d
S )�TestAsyncExitStackc @ s0 e Zd Zedd� �Zdd� Zdd� Zdd� Zd S )
z%TestAsyncExitStack.SyncAsyncExitStackc sj t �� � t �| �}|�� fdd�� � �� |�� }|s@|�� S |j}y|�W n ||_|�Y nX d S )Nc s � � � S )N)�stop)�f)r r r �<lambda>! � zETestAsyncExitStack.SyncAsyncExitStack.run_coroutine.<locals>.<lambda>)r Zget_event_loopZ
ensure_futureZadd_done_callbackZ run_foreverZ exception�result� __context__)r
rn r. r" r )r r �
run_coroutine s
z3TestAsyncExitStack.SyncAsyncExitStack.run_coroutinec C s | � | �� �S )N)rs �aclose)r r r r r
1 s z+TestAsyncExitStack.SyncAsyncExitStack.closec C s | � | �� �S )N)rs r! )r r r r � __enter__4 s z/TestAsyncExitStack.SyncAsyncExitStack.__enter__c G s | � | j|� �S )N)rs r )r � exc_detailsr r r �__exit__7 s z.TestAsyncExitStack.SyncAsyncExitStack.__exit__N)r r r � staticmethodrs r
ru rw r r r r �SyncAsyncExitStack s ry c C s( t �� | _t �| j� | �| jj� d S )N)r r r r Z
addCleanupr
)r r r r �setUp< s
zTestAsyncExitStack.setUpc
� s� di fdi fdi fdt dd�fdt dd�fdt dd�fg}g � � fdd�}t� 4 I d H ��}xpt|�D ]d\}}|r�|r�|j|f|�|�}n4|r�|j|f|�� }n |r�|j|f|�}n
|�|�}| �||� qdW xH|jD ]>}| �|d j|� | �|d j|j� | � |d j
|j
� q�W W d Q I d H R X | � � |� g � t� 4 I d H �Z}| � t
�� |jdd� W d Q R X | � t
�� | jjd d� W d Q R X |j|d
d � W d Q I d H R X | � � dd d
ifg� d S )
Nr )r> )r> � r> )Zexamplec � s � � | |f� dS )zTest metadata propagationN)r+ )r rg )rq r r �_exitL s z5TestAsyncExitStack.test_async_callback.<locals>._exit)�argr{ � )�callbackr} r} )�dictr �reversed�push_async_callbackr �_exit_callbacks� __wrapped__ZassertNotEqualr � assertIsNonerd r, r) r2 �
exit_stack)r Zexpectedr| �stackr rg rn r r )rq r �test_async_callbackA s>
( z&TestAsyncExitStack.test_async_callbackc
� s8 t � � �fdd�}dd� }�fdd�}G dd� dt�}��� 4 I d H ��}|�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� |�|� ��|jd d
|� d
d W d Q I d H R X d S ) Nc � s �� | � � d S )N)r )r4 r. �exc_tb)�
exc_raisedr r r � _expect_excn s z7TestAsyncExitStack.test_async_push.<locals>._expect_excc � s dS )NTr )rv r r r �
_suppress_excp s z9TestAsyncExitStack.test_async_push.<locals>._suppress_excc � s" � � | � � � |� � � |� d S )N)r� )r4 r. r� )r r r �
_expect_okr s
z6TestAsyncExitStack.test_async_push.<locals>._expect_okc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z2TestAsyncExitStack.test_async_push.<locals>.ExitCMc S s
|| _ d S )N)� check_exc)r r� r r r �__init__w s z;TestAsyncExitStack.test_async_push.<locals>.ExitCM.__init__c � s | � d� d S )NzShould not be called!)rV )r r r r r! y s z=TestAsyncExitStack.test_async_push.<locals>.ExitCM.__aenter__c � s | j |� I d H d S )N)r� )r rv r r r r { s z<TestAsyncExitStack.test_async_push.<locals>.ExitCM.__aexit__N)r r r r� r! r r r r r �ExitCMv s r� ���r> r )rD �objectr� �push_async_exitr r� �__self__)r r� r� r� r� r� �cmr )r� r r �test_async_pushk s*
z"TestAsyncExitStack.test_async_pushc
� s� G � fdd�dt �}g � |� }t� 4 I d H �T}|j� fdd��}| �|� |�|�I d H | �|jd d j|� � �d� W d Q I d H R X | � � dddd g� d S )
Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z;TestAsyncExitStack.test_async_enter_context.<locals>.TestCMc � s � � d� d S )Nr> )r+ )r )rq r r r! � s zFTestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aenter__c � s � � d� d S )Nr~ )r+ )r rv )rq r r r � s zETestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aexit__N)r r r r! r r )rq r r �TestCM� s r� c � s � � d� d S )N� )r+ r )rq r r r| � s z:TestAsyncExitStack.test_async_enter_context.<locals>._exitr� r> r{ r~ r� )
r� r r� ZassertIsNotNone�enter_async_contextr r� r� r+ r, )r r� r� r� r| r )rq r �test_async_enter_context� s
z+TestAsyncExitStack.test_async_enter_contextc
� s dd� }d � � fdd�}yh| � � 4 I d H �H}|�|t� |�|t� |�|t� |�|� |�|t� dd W d Q I d H R X W nP tk
r� } z2| �|jt� | �|jjt� | � |jjj� W d d }~X Y n X | �
d� � d }| �|t� | �|jt � d S )Nc � s | �d S )Nr )r. r r r � raise_exc� s zHTestAsyncExitStack.test_async_exit_exception_chaining.<locals>.raise_excc � s | � dS )NTr )rv )�
saved_detailsr r � suppress_exc� s zKTestAsyncExitStack.test_async_exit_exception_chaining.<locals>.suppress_excr> r z0Expected IndexError, but no exception was raised) r� r� �
IndexError�KeyError�AttributeErrorr� r( ZassertIsInstancerr r� rV rD )r r� r� r� r. Z inner_excr )r� r �"test_async_exit_exception_chaining� s&
"
z5TestAsyncExitStack.test_async_exit_exception_chainingN) r r r r ry r� rz r r� r� r� r� r r r r rl s * $ rl �__main__)r �
contextlibr r r r Ztestr ZunittestZtest.test_contextlibr r ZTestCaser r= rl r �mainr r r r �<module> s E = -