Sindbad~EG File Manager
a
��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 sX � | i |��}t �� }t �|� z |�|�W |�� t �d � S |�� t �d � 0 d S �N)�asyncio�new_event_loop�set_event_loopZrun_until_complete�close�set_event_loop_policy)�args�kwargs�coro�loop��func� �6/usr/local/lib/python3.9/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 � st G dd� dt �}|� }| �|�� I d H |� |4 I d H �"}| �||� W d �I d H qp1 I d H sf0 Y d S )Nc s e Zd Z� fdd�Z� ZS )z@TestAbstractAsyncContextManager.test_enter.<locals>.DefaultEnterc � s t � j|� I d H d S r ��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& �manager�contextr r r �
test_enter s
z*TestAbstractAsyncContextManager.test_enterc
� s� t dd� �� � fdd�}g }td�}| �t��n � � 4 I d H �: |� 2 z3 d H W }|�|� |�qF6 W d �I d H q�1 I d H s�0 Y W d � n1 s�0 Y | �|dg� d S )Nc S s
d V d S r r r r r r �ctx+ s zUTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exit.<locals>.ctxc sD � � 4 I d H � dV W d �I d H q@1 I d H s60 Y d S )N� r r �r, r r �gen/ s zUTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exit.<locals>.gen� r- )r �
ValueError� assertRaises�append� assertEqual)r r/ �ret�exc�valr r. r �(test_async_gen_propagates_generator_exit' s
PzHTestAbstractAsyncContextManager.test_async_gen_propagates_generator_exitc C sD G dd� dt �}| �t�� |� W d � n1 s60 Y d S )Nc @ s e Zd ZdS )zKTestAbstractAsyncContextManager.test_exit_is_abstract.<locals>.MissingAexitN�r" r# r$ r r r r � MissingAexit> s r: )r r2 � TypeError)r r: 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 r r �r r r r r( F s zbTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__aenter__c � s d S r r )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 rB c s e Zd Z� fdd�Z� ZS )zQTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.DefaultEnterc � s t � j|� I d H d S r r r r r r r N s z[TestAbstractAsyncContextManager.test_structural_subclassing.<locals>.DefaultEnter.__aexit__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 rC c @ s e Zd ZdZdS )zNTestAbstractAsyncContextManager.test_structural_subclassing.<locals>.NoneAexitN)r" r# r$ r r r r r � NoneAexitX s rD )Z
assertTrue�
issubclassr � assertFalse)r rB r& rC rD r r r �test_structural_subclassingD s z;TestAbstractAsyncContextManager.test_structural_subclassingN)r" r# r$ r r+ r8 r< rG 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 � s� g � t � fdd��}|� 4 I d H �:}| �� dg� | �|d� � �|� W d �I d H qr1 I d H sh0 Y | �� g d�� d S )Nc s � � d� dV � � d� d S �N� �* �� �r3 r ��stater r �woohooc s
zEAsyncContextManagerTestCase.test_contextmanager_plain.<locals>.woohoorJ rK �rJ rK rL )r r4 r3 �r rP �xr rN r �test_contextmanager_plain` s 4z5AsyncContextManagerTestCase.test_contextmanager_plainc
� s� g � t � fdd��}| �t��t |� 4 I d H �@}| �� dg� | �|d� � �|� t� �W d �I d H q�1 I d H sz0 Y W d � n1 s�0 Y | �� g d�� d S )Nc s0 � � d� zdV W � � d� n � � d� 0 d S rI rM r rN r r rP q s
zGAsyncContextManagerTestCase.test_contextmanager_finally.<locals>.woohoorJ rK rQ )r r2 �ZeroDivisionErrorr4 r3 rR r rN r �test_contextmanager_finallyn s
Nz7AsyncContextManagerTestCase.test_contextmanager_finallyc � sB t dd� �}|� }|�� I d H | �|�ttd�d �I d H � d S )Nc S s
d V d S r r r r r r �whee� s zHAsyncContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r r( rF r r; )r rW r, r r r � test_contextmanager_no_reraise� s
z:AsyncContextManagerTestCase.test_contextmanager_no_reraisec � sf t dd� �}|� }|�� I d H | �t��( |�ttd�d �I d H W d � n1 sX0 Y d S )Nc S s" z
d V W n d V Y n0 d S r r r r r r �whoo� s
zTAsyncContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoorX )r r( r2 � RuntimeErrorr r; �r rZ r, r r r �*test_contextmanager_trap_yield_after_throw� s
zFAsyncContextManagerTestCase.test_contextmanager_trap_yield_after_throwc � sN t dd� �}|� }| �t�� |�� I d H W d � n1 s@0 Y d S )Nc S s d S r r r r r r rZ � s zKAsyncContextManagerTestCase.test_contextmanager_trap_no_yield.<locals>.whoo)r r2 r[ r( r\ r r r �!test_contextmanager_trap_no_yield� s
z=AsyncContextManagerTestCase.test_contextmanager_trap_no_yieldc � sb t dd� �}|� }|�� I d H | �t��$ |�d d d �I d H W d � n1 sT0 Y d S )Nc S s d V d V d S r r r r r r rZ � s zOAsyncContextManagerTestCase.test_contextmanager_trap_second_yield.<locals>.whoo)r r( r2 r[ r r\ r r r �%test_contextmanager_trap_second_yield� s
zAAsyncContextManagerTestCase.test_contextmanager_trap_second_yieldc � sb t dd� �}|� }|�� I d H | �t��$ |�td d �I d H W d � n1 sT0 Y d S )Nc S s& z
d V W n t y t�Y n0 d S r )r[ � SyntaxErrorr r r r rZ � s
zLAsyncContextManagerTestCase.test_contextmanager_non_normalised.<locals>.whoo)r r( r2 r` r r[ r\ r r r �"test_contextmanager_non_normalised� s
z>AsyncContextManagerTestCase.test_contextmanager_non_normalisedc � s� g �t � �fdd��}|� 4 I d H �B}� ��dg� � �|d� ��|� td��W d �I d H q|1 I d H sr0 Y � ��g d�� d S )Nc
s^ �� d� z
dV W nD tyX } z,�� | jd � � ��g d�� W Y d } ~ n
d } ~ 0 0 d S )NrJ rK r rQ )r3 rU r
r4 )�e�r rO r r rP � s
zFAsyncContextManagerTestCase.test_contextmanager_except.<locals>.woohoorJ rK rL rQ )r r4 r3 rU rR r rc r �test_contextmanager_except� s
2z6AsyncContextManagerTestCase.test_contextmanager_exceptc � s t dd� �}G dd� dt�}G dd� dt�}td�td�|d�|d�fD ]�}| jt|�d ��� zB|� 4 I d H � |�W d �I d H q�1 I d H s�0 Y W n0 ty� } z| �||� W Y d }~nd }~0 0 | �|� d
�� W d � qH1 s�0 Y qHd S ) Nc S s
d V d S r r r r r r rP � s zOAsyncContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohooc @ s e Zd ZdS )z^AsyncContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.StopIterationSubclassNr9 r r r r �StopIterationSubclass� s re c @ s e Zd ZdS )zcAsyncContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.StopAsyncIterationSubclassNr9 r r r r �StopAsyncIterationSubclass� s rf �spamZham)�typez was suppressed)r �
StopIteration�StopAsyncIteration�subTestrh � Exceptionr' �fail)r rP re rf Zstop_exc�exr r r �#test_contextmanager_except_stopiter� s
�2"z?AsyncContextManagerTestCase.test_contextmanager_except_stopiterc
� s� t dd� �}| �t��R |� 4 I d H � dd W d �I d H qZ1 I d H sP0 Y W d � n1 sn0 Y | �t��N |� 4 I d H � t�W d �I d H q�1 I d H s�0 Y W d � n1 s�0 Y d S )Nc
S sD z
d V W n4 t y> } z td| � ��| �W Y d } ~ n
d } ~ 0 0 d S )Nzcaught )rl r[ �r6 r r r rP � s
zQAsyncContextManagerTestCase.test_contextmanager_wrap_runtimeerror.<locals>.woohoorJ r )r r2 r[ rj )r rP r r r �%test_contextmanager_wrap_runtimeerror� s
P zAAsyncContextManagerTestCase.test_contextmanager_wrap_runtimeerrorc C s" dd� }t |dd�dd� ��}|S )Nc s � fdd�}|S )Nc s" � � � D ]\}}t| ||� q| S r )�items�setattr)r �k�v��kwr r �decorate� s z]AsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )rw rx r rv r �attribs� s zKAsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)rX c S s
dV dS )�Whee!Nr )rg r r r �baz s zGAsyncContextManagerTestCase._create_contextmanager_attribs.<locals>.baz)r )r ry r| r r r � _create_contextmanager_attribs� s
z:AsyncContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nr| rz )r} r4 r" rX �r r| r r r �test_contextmanager_attribs s z7AsyncContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nr{ �r} r4 �__doc__r~ r r r � test_contextmanager_doc_attrib s z:AsyncContextManagerTestCase.test_contextmanager_doc_attribc � sV | � � d �}| �|jd� |4 I d H � W d �I d H qR1 I d H sH0 Y d S r� r� r~ r r r �*test_instance_docstring_given_cm_docstring s zFAsyncContextManagerTestCase.test_instance_docstring_given_cm_docstringc � s` t dd� �}|ddddd�4 I d H �"}| �|d� W d �I d H q\1 I d H sR0 Y d S ) Nc S s | |||fV d S r r �r r r
�kwdsr r r rP s z9AsyncContextManagerTestCase.test_keywords.<locals>.woohoor- r0 �! �, r� )r- r0 r� r� )r r4 )r rP �targetr r r �
test_keywords s
z)AsyncContextManagerTestCase.test_keywordsN)r" r# r$ r rT rV rY r] r^ r_ ra rd ro rq r} r r Zrequires_docstringsr� r� r� r r r r rH ^ s:
rH c @ sd e Zd ZG dd� de�ZeZdd� Zedd� �Zedd� �Z ed d
� �Z
ed d � �Z ed
d� �Z dS )�TestAsyncExitStackc @ s0 e Zd Zedd� �Zdd� Zdd� Zdd� Zd S )
z%TestAsyncExitStack.SyncAsyncExitStackc sj t �� � t �| �}|�� fdd�� � �� |�� }|s@|�� S |j}z|�W n ||_|�Y n0 d S )Nc s � � � S r )�stop)�f�r r r �<lambda>, � zETestAsyncExitStack.SyncAsyncExitStack.run_coroutine.<locals>.<lambda>)r Zget_event_loopZ
ensure_future�add_done_callbackZ run_forever� exception�result� __context__)r r� r6 r* r r� r �
run_coroutine' s
z3TestAsyncExitStack.SyncAsyncExitStack.run_coroutinec C s | � | �� �S r )r� �acloser= r r r r < s z+TestAsyncExitStack.SyncAsyncExitStack.closec C s | � | �� �S r )r� r( r= r r r � __enter__? s z/TestAsyncExitStack.SyncAsyncExitStack.__enter__c G s | � | j|� �S r )r� r �r � exc_detailsr r r �__exit__B s z.TestAsyncExitStack.SyncAsyncExitStack.__exit__N)r" r# r$ � staticmethodr� r r� r� r r r r �SyncAsyncExitStack& s
r� c C s6 t �� | _t �| j� | �| jj� | �t jd � d S r )r r r r
Z
addCleanupr r r= r r r �setUpG s
zTestAsyncExitStack.setUpc � sR di fdi fdi fdt dd�fdt dd�fdt dd�fg}g � � fdd�}t� 4 I d H ��}t|�D ]p\}}|r�|r�|j|g|�R i |��}n:|r�|j|g|�R � }n"|r�|j|fi |��}n
|�|�}| �||� qb|jD ]>}| �|d j|� | �|d j|j� | � |d j
|j
� q�W d �I d H �qH1 I d H �s>0 Y | � � |� g � t� 4 I d H ��}| � t
�� |jdd� W d � n1 �s�0 Y | � t
�� | jjd d� W d � n1 �s�0 Y | � t
�� |j|d
d � W d � n1 �s
0 Y W d �I d H �qB1 I d H �s80 Y | � � g � d S ) Nr )rJ )rJ � rJ )Zexamplec � s � � | |f� dS )zTest metadata propagationNrM )r
r� �r� r r �_exitX s z5TestAsyncExitStack.test_async_callback.<locals>._exit)�argr� � )�callbackr� )�dictr �reversed�push_async_callbackr' �_exit_callbacks� __wrapped__ZassertNotEqualr" � assertIsNoner� r4 r2 r; �
exit_stack)r Zexpectedr� �stackr
r� r� r r r� r �test_async_callbackM sB �
D , . \z&TestAsyncExitStack.test_async_callbackc � sV 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 �I d H �qR1 I d H �sH0 Y d S ) Nc � s �� | � � d S r )r' �r> r6 �exc_tb�Z
exc_raisedr r r � _expect_exc{ s z7TestAsyncExitStack.test_async_push.<locals>._expect_excc � s dS �NTr �r� r r r �
_suppress_exc} s z9TestAsyncExitStack.test_async_push.<locals>._suppress_excc � s" � � | � � � |� � � |� d S r )r� r� r= r r �
_expect_ok 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 r �� check_exc)r r� r r r �__init__� s z;TestAsyncExitStack.test_async_push.<locals>.ExitCM.__init__c � s | � d� d S )NzShould not be called!)rm r= r r r r( � s z=TestAsyncExitStack.test_async_push.<locals>.ExitCM.__aenter__c � s | j |� I d H d S r r� r� r r r r � s z<TestAsyncExitStack.test_async_push.<locals>.ExitCM.__aexit__N)r" r# r$ r� r( r r r r r �ExitCM� s r� ���rJ r )rU �objectr� �push_async_exitr' r� �__self__)r r� r� r� r� r� �cmr r� r �test_async_pushx s*
z"TestAsyncExitStack.test_async_pushc � s� G � fdd�dt �}g � |� }t� 4 I d H �d}|j� fdd��}| �|� |�|�I d H | �|jd d j|� � �d� W d �I d H q�1 I d H s�0 Y | � � g d�� d S ) Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z;TestAsyncExitStack.test_async_enter_context.<locals>.TestCMc � s � � d� d S )NrJ rM r= r� r r r( � s zFTestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aenter__c � s � � d� d S )Nr� rM r� r� r r r � s zETestAsyncExitStack.test_async_enter_context.<locals>.TestCM.__aexit__NrA r r� r r �TestCM� s r� c � s � � d� d S )N� rM r r� r r r� � s z:TestAsyncExitStack.test_async_enter_context.<locals>._exitr� rJ r� )rJ r� r� r� )
r� r r� ZassertIsNotNone�enter_async_contextr' r� r� r3 r4 )r r� r� r� r� r r� r �test_async_enter_context� s
4z+TestAsyncExitStack.test_async_enter_contextc
� s dd� }d � � fdd�}z�| � � 4 I d H �X}|�|t� |�|t� |�|t� |�|� |�|t� dd W d �I d H q�1 I d H s�0 Y W nR ty� } z:| �|jt� | �|jjt� | � |jjj� W Y d }~nd }~0 0 | �
d� � d }| �|t� | �|jt � d S )Nc � s | �d S r r rp r r r � raise_exc� s zHTestAsyncExitStack.test_async_exit_exception_chaining.<locals>.raise_excc � s | � dS r� r r� �Z
saved_detailsr r � suppress_exc� s zKTestAsyncExitStack.test_async_exit_exception_chaining.<locals>.suppress_excrJ r �0Expected IndexError, but no exception was raised) r� r� �
IndexError�KeyError�AttributeErrorr� r1 ZassertIsInstancer� r� rm rU )r r� r� r� r6 Z inner_excr r� r �"test_async_exit_exception_chaining� s&
6&
z5TestAsyncExitStack.test_async_exit_exception_chainingc � s� G dd� dt �� t� fdd���t��fdd��}�|fD ]�}��� �� zD|� 4 I d H � t� �W d �I d H q�1 I d H s�0 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 )zWTestAsyncExitStack.test_async_exit_exception_explicit_none_context.<locals>.MyExceptionNr9 r r r r � MyException� s r� c s@ z
d V W n0 t y: � � } z| �W d | _nd | _0 Y n0 d S r )�
BaseExceptionr� rp )r� r r �my_cm� s
zQTestAsyncExitStack.test_async_exit_exception_explicit_none_context.<locals>.my_cmc sX �� � 4 I d H �.} | �� � �I d H | V W d �I d H qT1 I d H sJ0 Y d S r )r� r� )r� )r� r r r �my_cm_with_exit_stack� s zaTestAsyncExitStack.test_async_exit_exception_explicit_none_context.<locals>.my_cm_with_exit_stackr� )rl r rk r� r� r� rm )r r� r� r6 r )r� r� r r �/test_async_exit_exception_explicit_none_context� s
4"zBTestAsyncExitStack.test_async_exit_exception_explicit_none_contextN)
r" r# r$ r r� r� r� r r� r� r� r� r� r r r r r� % s
*
#
r� �__main__)r �
contextlibr r r r �testr ZunittestZtest.test_contextlibr r ZTestCaser rH r� r" �mainr r r r �<module> s E H R
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists