File: //lib64/python3.8/unittest/__pycache__/async_case.cpython-38.pyc
U
e5d² ã @ s0 d dl Z d dlZddlmZ G dd„ deƒZdS )é Né )ÚTestCasec sŒ e Zd Zd ‡ fdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Zdd„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zd ‡ fd d „ Z‡ ZS )!Ú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 )ZsetUpÚ
_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 ZtearDownr r r r Ú
_callTearDownC s z%IsolatedAsyncioTestCase._callTearDownc O s | j |f|ž|Ž d S r r )r Zfunctionr r r r r Ú _callCleanupG s z$IsolatedAsyncioTestCase._callCleanupc O sL | j d k st‚|||Ž}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 st‚|||Ž}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 kr:d S |\}}z |I d H }| ¡ s`| |¡ W q ttfk
r| ‚ Y q t t j
fk
r´ } z| ¡ s¤| |¡ W 5 d }~X Y qX qd S r ) ÚasyncioZQueuer
Z
set_resultÚgetZ task_doneÚ cancelledÚ
SystemExitÚKeyboardInterruptÚ
BaseExceptionZCancelledErrorZ
set_exception)r r' ZqueueZqueryZ awaitabler& Zexr r r Ú_asyncioLoopRunner\ s
z*IsolatedAsyncioTestCase._asyncioLoopRunnerc C sX | j d kst‚t ¡ }t |¡ | d¡ || _ | ¡ }| | |¡¡| _ |
|¡ d S )NT) r r r( Znew_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 st‚| j }d | _ | j d ¡ | | j ¡ ¡ zˆt |¡}|sNW ¢vd S |D ] }|
¡ qR| tj ||ddœŽ¡ |D ]0}| ¡ rŠq||
¡ d k r|| d|
¡ |dœ¡ q|| | ¡ ¡ W 5 t d ¡ | ¡ X d S )NT)r0 Zreturn_exceptionsz(unhandled exception during test shutdown)ÚmessageÚ exceptionÚtask)r r r
r# r$ Újoinr( r/ ÚcloseZ all_tasksZcancelZgatherr* r3 Zcall_exception_handlerZshutdown_asyncgens)r r0 Z to_cancelr4 r r r Ú_tearDownAsyncioLoopy s4
ÿ ý
z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s( | ¡ ztƒ |¡W ¢S | ¡ X d S r )r1 r7 r Úrun)r Úresultr r r r8 › s zIsolatedAsyncioTestCase.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! Zcaser r r r r r Ú<module> s