
ŠÃ8W¿  ã               @   sG   d  d l  Z  d d l Td d l Td d l m Z Gd d „  d ƒ Z d S)é    Né   )Ú*)Úhandlerc               @   s4   e  Z d  Z d d „  Z d d „  Z d d „  Z d S)ÚClientServerContextManagerc             C   s#   t  j ƒ  |  _ t  j |  j ƒ d  S)N)ÚasyncioÚnew_event_loopÚloopÚset_event_loop)Úself© r   úD/Users/myk/Documents/dev/websockets/websockets/py35/client_server.pyÚsetUp   s    z ClientServerContextManager.setUpc             C   s   |  j  j ƒ  d  S)N)r   Úclose)r
   r   r   r   ÚtearDown   s    z#ClientServerContextManager.tearDownc                sv   t  t d d ƒ } ˆ  j j | ƒ ˆ  _ ‡  f d d †  } ˆ  j j | ƒ  ƒ ˆ  j j ƒ  ˆ  j j ˆ  j j ƒ  ƒ d  S)NÚ	localhostiÂ!  c           
   ³   sX   t  d ƒ 4Id  Hš: }  |  j d ƒ Id  H|  j ƒ  Id  H} ˆ  j | d ƒ Wd  QId  HRXd  S)Nzws://localhost:8642/zHello!)ÚconnectÚsendÚrecvÚassertEqual)ÚclientÚreply)r
   r   r   Úbasic   s    z4ClientServerContextManager.test_basic.<locals>.basic)Úserver   r   Úrun_until_completeÚserverr   Úwait_closed)r
   r   r   r   )r
   r   Ú
test_basic   s    z%ClientServerContextManager.test_basicN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r      s   r   )r   r   r   Ztest_client_serverr   r   r   r   r   r   Ú<module>   s   

