a <+bd@sdZddlmZddlZddlZddlZddlmZm Z m Z m Z ddl m Z daaddZd d Zd d Zejd gdddZdS)zTest async/await integration) LooseVersionN)executeflush_channelsstart_new_kernelTIMEOUT)validate_messagecCst\aattdS)zCstart the global kernel (if it isn't running) and return its clientN)rKMKCrr r 9lib/python3.9/site-packages/ipykernel/tests/test_async.pysetup_functions r cCsttjdddS)NT)Znow)r Z stop_channelsr Zshutdown_kernelr r r r teardown_functionsrcCs.tttdt\}}|ddks*J|dS)Nz(import asyncio; await asyncio.sleep(0.1)statusok)rr r)msg_idcontentr r r test_async_awaitsrasynclib)ZasyncioZtrioZcuriocCs>z t|Wn ty,td|Yn0|ddtttd|t\}}|ddksjJ|tttd|d|d }tjt d }t |d||d d d ksJtjt d }t |dtjt d }t |d|d ddksJt t d }|ddks J||ddvs2JttdS)Nz Requires %scSs tdtS)Nz%autoawait asyncio)rr r r r r *z&test_async_interrupt..z %autoawait rrzprint('begin'); import z; await z .sleep(5))ZtimeoutrZexecution_statebusyZ execute_inputstreamtextzbegin errorZename>ZCancelledErrorKeyboardInterrupt) __import__ ImportErrorpytestskipZ addfinalizerrr rZ get_iopub_msgrrr Zinterrupt_kernelZ get_shell_msg)rZrequestrrrZechorZreplyr r r test_async_interrupt$s2         r )__doc__Zdistutils.versionrVsysrZIPythonZutilsrrrrZtest_message_specrr r r rrZmarkZ parametrizer r r r r s