o
    >h                     @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ dd Zej	j
e jdd dd	ej	jd
d Zej	j
e jdd dd	dd Zej	j
e jdd dd	ej	jdd ZdS )    N)Mock)WebSocketServerFactoryc                    s   t |I d H  | S Nasynciosleep)whatwhen r
   /var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/autobahn/asyncio/test/test_aio_websocket.py
echo_async   s   r   USE_ASYNCIOFztest runs on asyncio only)reasonc                      s   dt ddI d H ksJ d S )NzHello!r   )r   r
   r
   r
   r   test_echo_async   s   r   c                 C   s$   t | d}| }t }|| d S )N)loop)r   r   connection_made)
event_loopfactoryserver	transportr
   r
   r   test_websocket_custom_loop   s   
r   c                    s   dt   g dd  fdd}t }| }||_t }|| dg d|_|   I d H  t	dks@J d	  ksJJ d S )
N*   c                    s   t dI d H  | |  S )N   r   )xr
   r
   r   foo)   s   z)test_async_on_connect_server.<locals>.fooc                    s*   I d H } | t |  d S r   )appendtxaioresolve)reqvdoner   numvaluesr
   r   
on_connect-   s   
z0test_async_on_connect_server.<locals>.on_connect   
)	s   GET /ws HTTP/1.1s   Host: www.example.coms   Sec-WebSocket-Version: 13s,   Origin: http://www.example.com.malicious.coms,   Sec-WebSocket-Extensions: permessage-deflates+   Sec-WebSocket-Key: tXAxWFUqnhi86Ajj7dRY5g==s   Connection: keep-alive, Upgrades   Upgrade: websocketr%   r   r   )
r   create_futurer   	onConnectr   r   joindataprocessHandshakelen)r   r$   r   r   r   r
   r    r   test_async_on_connect_server!   s    

r,   )osr   pytestr   unittest.mockr   autobahn.asyncio.websocketr   r   markskipifenvirongetr   r   r,   r
   r
   r
   r   <module>   s    
