ÿØÿà JFIF      ÿÛ C      

!"$"$ÿÛ C  ÿÂ p " ÿÄ              ÿÄ             ÿÚ    ÕÔË®
(%	aA*‚XYD¡(J„¡E¢RE,P€XYae )(E¤²€B¤R¥	BQ¤¢ X«)X…€¤   @  

  ..............................................................................................................................................................................
.............................................................................                                                  
                                                                                                                                                                                     ÿØÿà JFIF      ÿÛ C      

!"$"$ÿÛ C  ÿÂ p " ÿÄ              ÿÄ             ÿÚ    ÕÔË®
(%	aA*‚XYD¡(J„¡E¢RE,P€XYae )(E¤²€B¤R¥	BQ¤¢ X«)X…€¤   @  

  ..............................................................................................................................................................................
.............................................................................                                                  
                                                                                                                                                                                     o
    p̦i                     @   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 fdd Zd!d" Z  ZS )%IsolatedAsyncioTestCaserunTestc                    s   t  | d | _d | _d S N)super__init___asyncioTestLoop_asyncioCallsQueue)self
methodName	__class__ */usr/lib/python3.10/unittest/async_case.pyr   !   s   
z IsolatedAsyncioTestCase.__init__c                       d S r   r   r   r   r   r   
asyncSetUp&      z"IsolatedAsyncioTestCase.asyncSetUpc                    r   r   r   r   r   r   r   asyncTearDown)   r   z%IsolatedAsyncioTestCase.asyncTearDownc                O      | j |g|R i | d S r   )
addCleanup)r   funcargskwargsr   r   r   addAsyncCleanup,   s   z'IsolatedAsyncioTestCase.addAsyncCleanupc                 C   s   |    | | j d S r   )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   tearDownr   r   r   r   _callTearDownB   s   z%IsolatedAsyncioTestCase._callTearDownc                 O   r   r   r   )r   functionr   r   r   r   r   _callCleanupF   s   z$IsolatedAsyncioTestCase._callCleanupc                O   s^   | j d us	J d||i |}t|sJ |d| j  }| j||f | j |S )N$asyncio test loop is not initializedz returned non-awaitabler	   inspectisawaitablecreate_futurer
   
put_nowaitrun_until_completer   r   r   r   retfutr   r   r   r   I   s   
z"IsolatedAsyncioTestCase._callAsyncc                O   sT   | j d us	J d||i |}t|r(| j  }| j||f | j |S |S )Nr'   r(   r.   r   r   r   r    Q   s   

z'IsolatedAsyncioTestCase._callMaybeAsyncc              
      s   t   | _}|d  	 | I d H }|  |d u rd S |\}}z|I d H }| s2|| W n' ttfy=     t	t j
fyZ } z| sP|| W Y d }~nd }~ww qr   )asyncioQueuer
   
set_resultget	task_done	cancelled
SystemExitKeyboardInterruptBaseExceptionCancelledErrorset_exception)r   r0   queuequery	awaitabler/   exr   r   r   _asyncioLoopRunner[   s,   



z*IsolatedAsyncioTestCase._asyncioLoopRunnerc                 C   s\   | j d u s	J dt }t| |d || _ | }|| || _|	| d S )Nz%asyncio test loop already initializedT)
r	   r1   new_event_loopset_event_loop	set_debugr+   create_taskr@   _asyncioCallsTaskr-   )r   loopr0   r   r   r   _setupAsyncioLoopn   s   

z)IsolatedAsyncioTestCase._setupAsyncioLoopc              
   C   s$  | j d us	J d| j }d | _ | jd  || j  zct|}|s8W ||  td  |	  d S |D ]}|
  q:|tj|ddi |D ]}| rUqN| d urf|d| |d qN||  W ||  td  |	  d S ||  td  |	  w )Nr'   return_exceptionsTz(unhandled exception during test shutdown)message	exceptiontask)r	   r
   r,   r-   joinr1   	all_tasksshutdown_default_executorrB   closecancelgatherr6   rJ   call_exception_handlershutdown_asyncgens)r   rF   	to_cancelrK   r   r   r   _tearDownAsyncioLoopx   sD   





z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc                    s*   |    zt |W |   S |   w r   )rG   r   runrU   )r   resultr   r   r   rV      s   zIsolatedAsyncioTestCase.runc                    s   |    t   |   d S r   )rG   r   debugrU   r   r   r   r   rX      s   
zIsolatedAsyncioTestCase.debugc                 C   s   | j d ur|   d S d S r   )r	   rU   r   r   r   r   __del__   s   
zIsolatedAsyncioTestCase.__del__)r   r   )__name__
__module____qualname__r   r   r   r   r   r"   r$   r&   r   r    r@   rG   rU   rV   rX   rY   __classcell__r   r   r   r   r      s"    

$r   )r1   r)   caser   r   r   r   r   r   <module>   s    