a
    ތxd/                     @   s:   d dl Z d dlZddlmZ ddlmZ G dd dZdS )    N   )guarantee_single_callable)timeoutc                   @   sP   e Zd ZdZdd ZdddZddd	Zd
d Zdd ZdddZ	dddZ
dS )ApplicationCommunicatorz}
    Runs an ASGI application in a test mode, allowing sending of
    messages to it and retrieval of messages it sends.
    c                 C   sF   t || _|| _t | _t | _t| || jj| jj	| _
d S N)r   applicationscopeasyncioQueueinput_queueoutput_queueensure_futuregetputfuture)selfr   r    r   G/var/www/html/Ranjet/env/lib/python3.9/site-packages/asgiref/testing.py__init__   s    


z ApplicationCommunicator.__init__r   c                    s   zt |4 I dH F z| jI dH  | j  W n tjyB   Y n0 W d  I dH  qn1 I dH sd0    Y  W | j s| j  z| jI dH  W q tjy   Y q0 n<| j s| j  z| jI dH  W n tjy   Y n0 0 dS )zV
        Waits for the application to stop itself and returns any exceptions.
        N)async_timeoutr   resultr	   CancelledErrordonecancel)r   r   r   r   r   wait   s&    2



zApplicationCommunicator.waitTc                 C   s(   | j  s| j   n|r$| j   d S r   )r   r   r   r   )r   
exceptionsr   r   r   stop*   s    
zApplicationCommunicator.stopc                 C   s(   z| j dd W n ty"   Y n0 d S )NF)r   )r   RuntimeError)r   r   r   r   __del__1   s    zApplicationCommunicator.__del__c                    s   | j |I dH  dS )z;
        Sends a single message to the application
        N)r   r   )r   messager   r   r   
send_input9   s    z"ApplicationCommunicator.send_inputc                    s   | j  r| j   zRt|4 I dH ( | j I dH W  d  I dH  W S 1 I dH sZ0    Y  W np tjy } zV| j  r| j   n0| j   z| j I dH  W n tj	y   Y n0 |W Y d}~n
d}~0 0 dS )zX
        Receives a single message from the application, with optional timeout.
        N)
r   r   r   r   r   r   r	   TimeoutErrorr   r   )r   r   er   r   r   receive_output@   s    

@

z&ApplicationCommunicator.receive_output皙?{Gz?c                    sB   t  }t  | |k r8| j s&dS t|I dH  q| j S )zO
        Checks that there is no message to receive in the given time.
        FN)time	monotonicr   emptyr	   sleep)r   r   intervalstartr   r   r   receive_nothingW   s    
z'ApplicationCommunicator.receive_nothingN)r   )T)r   )r$   r%   )__name__
__module____qualname____doc__r   r   r   r   r    r#   r,   r   r   r   r   r      s   	


r   )r	   r&   compatibilityr   r   r   r   r   r   r   r   <module>   s   