
    h`	                         d dl mZ 	 d dlmZmZmZ dZd dlm	Z	 d dl
mZ  G d dej                        Zy	# e$ r dZY +w xY w)
    )unittest)create_memory_agentMemoryReactorClockResolvercreate_pumperTF)inlineCallbacks)WebSocketServerProtocolc                   .    e Zd Ze Zd Zd Zed        Zy)	TestAgentc                 r    t               | _        t               | _        | j                  j	                         S N)r   pumperr   reactorstartselfs    i/var/www/Befach/backend/env/lib/python3.12/site-packages/autobahn/twisted/test/test_tx_websocket_agent.pysetUpzTestAgent.setUp   s(    #o13{{  ""    c                 6    | j                   j                         S r   )r   stopr   s    r   tearDownzTestAgent.tearDown   s    {{!!r   c              #   ^  K    G d dt               }t        | j                  | j                  |      }|j	                  dt                     }g fd}|j                  d|       |j                  d       	 |j                          |j                   | j                  dg       y w)Nc                       e Zd Zd Zy).TestAgent.test_echo_server.<locals>.EchoServerc                 &    | j                  |       y 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                 (    j                  |        y 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 	& $DLL$++zJjj!946BB	!C (#OO oo(X.s   B*B-N)	r    r!   r"   HAVE_TESTINGskipr   r   r   r4   r#   r   r   r
   r
      s(    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=      sH    "ggL 3 >%/!! %/  Ls   6 A A 