OwlCyberSecurity - MANAGER
Edit File: testing.cpython-39.pyc
a ���e/ � @ s: d dl Z d dlZddlmZ ddlmZ G dd� d�ZdS )� N� )�guarantee_single_callable)�timeoutc @ sP e Zd ZdZdd� Zddd�Zddd �Zd d � Zd d � Zddd�Z ddd�Z dS )�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�| �|| jj| jj ��| _ d S �N) r � application�scope�asyncio�Queue� input_queue� output_queue� ensure_future�get�put�future)�selfr r � r �P/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/asgiref/testing.py�__init__ s �z ApplicationCommunicator.__init__r c � s� z�t |�4 I dH �F z| jI dH | j�� W n tjyB Y n0 W d �I dH qn1 I dH sd0 Y W | j�� s�| j�� z| jI dH W q� tjy� Y q�0 n<| j�� s�| j�� z| jI dH W n tjy� Y n0 0 dS )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 dd� W n ty" Y n0 d S )NF)r )r � RuntimeError)r r r r �__del__1 s z ApplicationCommunicator.__del__c � s | j �|�I dH dS )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 dH �( | j�� I dH W d �I dH W S 1 I dH sZ0 Y W np tjy� } zV| j �� r�| j �� n0| j �� z| j I dH W n tj y� Y n0 |�W Y d}~n d}~0 0 dS )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皙�����?�{�G�z�?c � sB t �� }t �� | |k r8| j�� s&dS t�|�I dH 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