
    af`	                         d dl mZ 	 d dlmZmZmZ dZn# e$ r dZY nw xY wd dlm	Z	 d dl
mZ  G d dej                  Zd	S )
    )unittest)create_memory_agentMemoryReactorClockResolvercreate_pumperTF)inlineCallbacks)WebSocketServerProtocolc                   6    e Zd Ze Zd Zd Zed             ZdS )	TestAgentc                     t                      | _        t                      | _        | j                                        S N)r   pumperr   reactorstartselfs    _/var/www/html/env/lib/python3.11/site-packages/autobahn/twisted/test/test_tx_websocket_agent.pysetUpzTestAgent.setUp   s.    #oo133{  """    c                 4    | j                                         S r   )r   stopr   s    r   tearDownzTestAgent.tearDown   s    {!!!r   c              #   ~  K    G d dt                     }t          | j        | j        |          }|                    dt                                V }g fd}|                    d|           |                    d           	 |                                 |j        V  |                     dg           d S )Nc                       e Zd Zd ZdS ).TestAgent.test_echo_server.<locals>.EchoServerc                 0    |                      |           d S r   )sendMessage)r   msg	is_binarys      r   	onMessagez8TestAgent.test_echo_server.<locals>.EchoServer.onMessage   s      %%%%%r   N)__name__
__module____qualname__r    r   r   
EchoServerr      s#        & & & & &r   r$   zws://localhost:1234/wsc                 2                         |            d S r   )append)r   r   messagess     r   gotz'TestAgent.test_echo_server.<locals>.got%   s    OOC     r   messages   hello)r   r   r   r   opendictonr   	sendClose	transportloseConnection	is_closedassertEqual)r   r$   agentprotor(   r'   s        @r   test_echo_serverzTestAgent.test_echo_server   s      	& 	& 	& 	& 	&0 	& 	& 	& $DL$+zJJjj!9466BBBB	! 	! 	! 	! 	!C   (###	-OO o(X.....r   N)	r    r!   r"   HAVE_TESTINGskipr   r   r   r4   r#   r   r   r
   r
      sU        D# # #
" " " / / _/ / /r   r
   N)twisted.trialr   autobahn.twisted.testingr   r   r   r5   ImportErrortwisted.internet.deferr   autobahn.twisted.websocketr   TestCaser
   r#   r   r   <module>r=      s    " " " " " "ggggggggggLL   LLL 3 2 2 2 2 2 > > > > > >%/ %/ %/ %/ %/! %/ %/ %/ %/ %/s    