Your IP : 216.73.216.85
B
� f�: � @ 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� �?/opt/alt/python37/lib64/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& Zret�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 r0 )r r) � TypeError)r r0 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 r6 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 r7 c @ s e Zd ZdZdS )zNTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.NoneAexitN)r r r r r r r r � NoneAexitX s r8 )Z
assertTrue�
issubclassr � assertFalse)r r6 r r7 r8 r r r �test_structural_subclassingD s z;TestAbstractAsyncContextManager.test_structural_subclassingN)r r r r r# r/ r2 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 r@ �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 r@ q s
zGAsyncContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor= r> i� )r r) �ZeroDivisionErrorr, r+ )r r@ rA 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 r1 )r rE 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>.whoorF )r r! r) � RuntimeErrorr r1 )r rH 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 rH � s zKAsyncContextManagerTestCase.test_contextmanager_trap_no_yield.<locals>.whoo)r r) rI r! )r rH 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 rH � s zOAsyncContextManagerTestCase.test_contextmanager_trap_second_yield.<locals>.whoo)r r! r) rI r )r rH 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)rI � SyntaxErrorr r r r rH � s
zLAsyncContextManagerTestCase.test_contextmanager_non_normalised.<locals>.whoo)r r! r) rM r rI )r rH 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+ rC r r, )�e)r r? r r r@ � s
zFAsyncContextManagerTestCase.test_contextmanager_except.<locals>.woohoor= r> i� )r r, r+ rC )r r@ rA 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 r@ � s zOAsyncContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohoo�spamZham)�typez was suppressed)r �
StopIterationr* ZsubTestrR � Exceptionr �fail)r r@ 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 )rT rI )r- r r r r@ � s
zQAsyncContextManagerTestCase.test_contextmanager_wrap_runtimeerror.<locals>.woohoor= r )r r) rI r* )r r@ 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)rF c S s
dV dS )zWhee!Nr )rQ r r r �baz� s zGAsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.baz)r )r r^ r` r r r � _create_contextmanager_attribs� s z:AsyncContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nr` r_ )ra r, r rF )r r` r r r �test_contextmanager_attribs� s z7AsyncContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S )NzWhee!)ra r, �__doc__)r r` 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!)ra r, rc )r r` 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 r@ s z9AsyncContextManagerTestCase.test_keywords.<locals>.woohoor% r' �! �, )r r r rf )r% r' rg rh )r r, )r r@ �targetr r r �
test_keywords s z)AsyncContextManagerTestCase.test_keywordsN)r r r r rB rD rG rJ rK rL rN rP rV rW ra rb r Zrequires_docstringsrd re rj 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
rm r- r" r )r r �
run_coroutine s
z3TestAsyncExitStack.SyncAsyncExitStack.run_coroutinec C s | � | �� �S )N)rr �aclose)r r r r r
1 s z+TestAsyncExitStack.SyncAsyncExitStack.closec C s | � | �� �S )N)rr r! )r r r r � __enter__4 s z/TestAsyncExitStack.SyncAsyncExitStack.__enter__c G s | � | j|� �S )N)rr r )r � exc_detailsr r r �__exit__7 s z.TestAsyncExitStack.SyncAsyncExitStack.__exit__N)r r r � staticmethodrr r
rt rv r r r r �SyncAsyncExitStack s rx 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 rf )rp r r �_exitL s z5TestAsyncExitStack.test_async_callback.<locals>._exit)�argrz � )�callbackr| r| )�dictr �reversed�push_async_callbackr �_exit_callbacks� __wrapped__ZassertNotEqualr � assertIsNonerc r, r) r1 �
exit_stack)r Zexpectedr{ �stackr rf rm r r )rp 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 )r3 r- �exc_tb)�
exc_raisedr r r � _expect_excn s z7TestAsyncExitStack.test_async_push.<locals>._expect_excc � s dS )NTr )ru r r r �
_suppress_excp s z9TestAsyncExitStack.test_async_push.<locals>._suppress_excc � s" � � | � � � |� � � |� d S )N)r� )r3 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!)rU )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 ru 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 )rC �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 )rp r r r! � s zFTestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aenter__c � s � � d� d S )Nr} )r+ )r ru )rp r r r � s zETestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aexit__N)r r r r! r r )rp r r �TestCM� s r� c � s � � d� d S )N� )r+ r )rp r r r{ � s z:TestAsyncExitStack.test_async_enter_context.<locals>._exitr� r= rz r} r� )
r� r r� ZassertIsNotNone�enter_async_contextr r� r� r+ r, )r r� r� r� r{ r )rp 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 )ru )�
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( ZassertIsInstancerq r� rU rC )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 rx r� ry r r� r� r� r� r r r r rk s * $ rk �__main__)r �
contextlibr r r r Ztestr ZunittestZtest.test_contextlibr r ZTestCaser r< rk r �mainr r r r �<module> s E = -