U ڲg @sBddlZddlZddlZddlmZddlmZGdddZdS)N)guarantee_single_callable)timeoutc@sPeZdZdZddZdddZddd Zd d Zd d ZdddZ dddZ dS)ApplicationCommunicatorz} Runs an ASGI application in a test mode, allowing sending of messages to it and retrieval of messages it sends. cCsNt||_||_t|_t|_t tj |||jj |jj |_ dSN)r applicationscopeasyncioQueue input_queue output_queue contextvarsContextrunZ create_taskgetputfuture)selfrrr3/tmp/pip-unpacked-wheel-hay1n6u8/asgiref/testing.py__init__s   z ApplicationCommunicator.__init__rcszXt|4IdH8z|jIdH|jWntjk rDYnXW5QIdHRXW5|js|jz|jIdHWntjk rYnXXdS)zV Waits for the application to stop itself and returns any exceptions. N)rdonecancelr CancelledError async_timeoutresult)rrrrrwaits   zApplicationCommunicator.waitTcCs(|js|jn|r$|jdSr)rrrr)r exceptionsrrrstop0s  zApplicationCommunicator.stopcCs*z|jddWntk r$YnXdS)NF)r)r RuntimeError)rrrr__del__7szApplicationCommunicator.__del__cs|j|IdHdS)z; Sends a single message to the application N)r r)rmessagerrr send_input?sz"ApplicationCommunicator.send_inputc s|jr|jzFt|4IdH&|jIdHW5QIdHRWSQIdHRXWnptjk r}zP|jr|jn2|jz|jIdHWntj k rYnX|W5d}~XYnXdS)zX Receives a single message from the application, with optional timeout. N) rrrrr rr TimeoutErrorrr)rrerrrreceive_outputFs  4   z&ApplicationCommunicator.receive_output皙?{Gz?csBt}t||kr8|js&dSt|IdHq|jS)zO Checks that there is no message to receive in the given time. FN)time monotonicr emptyr sleep)rrintervalstartrrrreceive_nothing]s  z'ApplicationCommunicator.receive_nothingN)r)T)r)r&r') __name__ __module__ __qualname____doc__rrrr r"r%r.rrrrr s   r)r r r(Z compatibilityrrrrrrrrs