OwlCyberSecurity - MANAGER
Edit File: testing.cpython-39.pyc
a gH�f� � @ sB d dl Z d dlZd 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 sN t |�| _|| _t�� | _t�� | _t�� � tj | �|| jj | jj ��| _ d S �N)r � application�scope�asyncio�Queue� input_queue� output_queue� contextvars�Context�run� create_task�get�put�future)�selfr r � r �Q/home/gouroczh/virtualenv/test/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 �stop0 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__7 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_input? 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_outputF 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_nothing] 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 r) Z compatibilityr r r r r r r r �<module> s