£Á°èZ¨Ä…–K§‚«“ô4“ÒÙ´dîfUÙÃÅ WKbyÊ¦•êŽ…È®FÒ¿ÊÎóCozá¬S@6{Í:›œêZÌ:Š•_%:¢¾¾~;‘Ã~èŠ©ÊÇí`ÔÑ©úë™µ'5I¿fš×WO%ø9¾«¾DK|€ùÍD”Ýs]nHÕ¶ê×Ó¼ãžªéUWŸÈË%DÒÕ¬ï‘]/Åcx  ‰ï2ß]ä6G[]S£ÔÏ¯rs{úëóµmÒï#UQxo·õÞCe]"±/aÙ&Eã4ú9Jé_ÞåëdãöKë)AÞ                  ¯¹ægƒÛowÐø^d™ý½ßB7áyMä9ÜÖUã
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
<html>
U
    e5d                     @   s0   d dl Z d dlZddlmZ G dd deZdS )    N   )TestCasec                       s   e Zd Zd fdd	Zdd Zdd Zdd	 Zd
d Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zd  fdd	Z  ZS )!IsolatedAsyncioTestCaserunTestc                    s   t  | d | _d | _d S N)super__init___asyncioTestLoop_asyncioCallsQueue)selfZ
methodName	__class__ +/usr/lib64/python3.8/unittest/async_case.pyr   "   s    z IsolatedAsyncioTestCase.__init__c                    s   d S r   r   r   r   r   r   
asyncSetUp'   s    z"IsolatedAsyncioTestCase.asyncSetUpc                    s   d S r   r   r   r   r   r   asyncTearDown*   s    z%IsolatedAsyncioTestCase.asyncTearDownc                O   s   | j |f|| d S r   )Z
addCleanup)r   funcargskwargsr   r   r   addAsyncCleanup-   s    z'IsolatedAsyncioTestCase.addAsyncCleanupc                 C   s   |    | | j d S r   )ZsetUp
_callAsyncr   r   r   r   r   
_callSetUp<   s    z"IsolatedAsyncioTestCase._callSetUpc                 C   s   |  | d S r   _callMaybeAsync)r   methodr   r   r   _callTestMethod@   s    z'IsolatedAsyncioTestCase._callTestMethodc                 C   s   |  | j |   d S r   )r   r   ZtearDownr   r   r   r   _callTearDownC   s    z%IsolatedAsyncioTestCase._callTearDownc                 O   s   | j |f|| d S r   r   )r   Zfunctionr   r   r   r   r   _callCleanupG   s    z$IsolatedAsyncioTestCase._callCleanupc                O   sL   | j d k	st|||}t|s&t| j  }| j||f | j |S r   r	   AssertionErrorinspectZisawaitablecreate_futurer
   
put_nowaitrun_until_completer   r   r   r   retfutr   r   r   r   J   s    

z"IsolatedAsyncioTestCase._callAsyncc                O   sP   | j d k	st|||}t|rH| j  }| j||f | j |S |S d S r   r   r%   r   r   r   r   R   s    


z'IsolatedAsyncioTestCase._callMaybeAsyncc              
      s   t   | _}|d  | I d H }|  |d kr:d S |\}}z |I d H }| s`|| W q ttfk
r|    Y q t	t j
fk
r } z| s|| W 5 d }~X Y qX qd S r   )asyncioZQueuer
   Z
set_resultgetZ	task_done	cancelled
SystemExitKeyboardInterruptBaseExceptionZCancelledErrorZset_exception)r   r'   ZqueueZqueryZ	awaitabler&   Zexr   r   r   _asyncioLoopRunner\   s     

z*IsolatedAsyncioTestCase._asyncioLoopRunnerc                 C   sX   | j d kstt }t| |d || _ | }|| || _	|
| d S )NT)r	   r    r(   Znew_event_loopset_event_loopZ	set_debugr"   Zcreate_taskr.   Z_asyncioCallsTaskr$   )r   loopr'   r   r   r   _setupAsyncioLoopo   s    

z)IsolatedAsyncioTestCase._setupAsyncioLoopc              	   C   s   | j d k	st| j }d | _ | jd  || j  zt	|}|sNW vd S |D ]}|
  qR|tj||dd |D ]0}| rq|| d k	r||d| |d q|||  W 5 td  |  X d S )NT)r0   Zreturn_exceptionsz(unhandled exception during test shutdown)message	exceptiontask)r	   r    r
   r#   r$   joinr(   r/   closeZ	all_tasksZcancelZgatherr*   r3   Zcall_exception_handlerZshutdown_asyncgens)r   r0   Z	to_cancelr4   r   r   r   _tearDownAsyncioLoopy   s4    



z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc                    s(   |    zt |W S |   X d S r   )r1   r7   r   run)r   resultr   r   r   r8      s    zIsolatedAsyncioTestCase.run)r   )N)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r.   r1   r7   r8   __classcell__r   r   r   r   r      s   

"r   )r(   r!   Zcaser   r   r   r   r   r   <module>   s   