o
    nYi                     @   s:   d dl Z d dlmZ d dlmZ dd ZG dd dZdS )    N)	MagicMock)Messagec                     s$   t | i |  fdd} |_|S )Nc                     s    | i |S N )argskwargsmr   ]/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/signalwire/tests/__init__.py	mock_coro   s   zAsyncMock.<locals>.mock_coro)r   mock)r   r   r   r   r   r
   	AsyncMock   s   r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )MockedConnectionc                 C   s0   || _ t | _t | _g | _t | _d| _d S )NT)	clientr   connectasyncioQueuequeue	responsesclose	connected)selfr   r   r   r
   __init__   s   

zMockedConnection.__init__c                    s   | j |jI d H  d S r   )r   putid)r   messager   r   r
   send   s   zMockedConnection.sendc                    s:   | j D ]}t|}| j I d H |_| j| qd S r   )r   r   	from_jsonr   getr   r   message_handler)r   responsemsgr   r   r
   read   s   

zMockedConnection.readN)__name__
__module____qualname__r   r   r"   r   r   r   r
   r      s    r   )r   unittest.mockr   !signalwire.blade.messages.messager   r   r   r   r   r   r
   <module>   s
    	