wg+DddlZddlmZddlmZddlmZGddZy)N)packet) PubSubManager)EnvironBuildercReZdZdZiZ d dZd dZ d dZd dZdZ d dZ d d Z y)SocketIOTestClienta This class is useful for testing a Flask-SocketIO server. It works in a similar way to the Flask Test Client, but adapted to the Socket.IO server. :param app: The Flask application instance. :param socketio: The application's ``SocketIO`` instance. :param namespace: The namespace for the client. If not provided, the client connects to the server on the global namespace. :param query_string: A string with custom query string arguments. :param headers: A dictionary with custom HTTP headers. :param auth: Optional authentication data, given as a dictionary. :param flask_test_client: The instance of the Flask test client currently in use. Passing the Flask test client is optional, but is necessary if you want the Flask user session and any other cookies set in HTTP routes accessible from Socket.IO events. Nc fd d fd}|_|_tjj_j j <d_|_i_ g_ d_ |j_ ||j_i|jjj <d|j_d|jj"_t%|jj&t(r t+d|jj&j-j/||||y)NcB|j}t|tstj|}n4tj|d}|ddD]}|j |j j|}|sy|jtjk(s|jtjk(r|jddk(s|jddk(rH|jj|jd|jd|jxsddy|jj|jd|jdd|jxsddy|jtjk(s|jtj k(r#|j|jxsdd|_y|jtj$tj&fvrd |j(|jxsd<yy) Nencoded_packetrmessagejson/)nameargs namespace)rrF)encode isinstancelistrPacketadd_attachmentclientsget packet_typeEVENT BINARY_EVENTdataqueueappendrACK BINARY_ACKacks DISCONNECT CONNECT_ERROR connected)eio_sidpktepktattclientselfs _/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/flask_socketio/test_client.py_mock_send_packetz6SocketIOTestClient.__init__.._mock_send_packets::  !5#6NcJj|jjtjk(r |dyyt j |j}|jdk(r ||y|y)Nr r)rrattachment_countlen attachmentsrr)r&eio_pktr'_current_packetr-s r,_mock_send_eio_packetz:SocketIOTestClient.__init__.._mock_send_eio_packet@s|*..w||<"33O7789%g?&*O9 mm7<<@''1,%gs3&)Or.rFz^Test client cannot be used with a message queue. Disable the queue on your test configuration.)r query_stringheadersauth)appflask_test_clientuuiduuid4hexr&rcallback_countersocketior%rr"server _send_packet_send_eio_packetenvironasync_handlerseiormanagerr RuntimeError initializeconnect) r+r9r?rr6r7r8r:r5r4r-s ` @@r,__init__zSocketIOTestClient.__init__s ?> *!2zz|'' %) T\\" !    '8$+@(02 -).&-2* hoo--} = 01 1 **, y|$4  1r.cB|jj|xsddS)zCheck if a namespace is connected. :param namespace: The namespace to check. The global namespace is assumed if this argument is not provided. rF)r%r)r+rs r, is_connectedzSocketIOTestClient.is_connectedes ~~!!)"2sE::r.cd}|xsd}|r|ddk7rd|z}||z }t||j}|j|d<|jrWt |jdr|jj |n%|jj j||jjj|j|tjtj||}|jjj|j|j!|jjj"j%|j|}|rd |j&|<y y ) aConnect the client. :param namespace: The namespace for the client. If not provided, the client connects to the server on the global namespace. :param query_string: A string with custom query string arguments. :param headers: A dictionary with custom HTTP headers. :param auth: Optional authentication data, given as a dictionary. Note that it is usually not necessary to explicitly call this method, since a connection is automatically established when an instance of this class is created. An example where it this method would be useful is when the application accepts multiple namespace connections. z /socket.iorr?)r7z flask.app_add_cookies_to_wsgirTN)r get_environr9r:hasattrrO cookie_jar inject_wsgir?r@_handle_eio_connectr&rrCONNECT_handle_eio_messagerrFsid_from_eio_sidr%) r+rr6r7r8urlrCr'sids r,rIzSocketIOTestClient.connectms; $ A#%"\1 < C g6BBD#xx   ! !t--/EF&&;;GD&&11==gF 00wGmmFNNDIF 00szz|Lmm""**;;DLLrrrrrrr?r@rWr&r"r1) r+eventrkwargsrr_r`r' encoded_pktr(acks r,emitzSocketIOTestClient.emits6JJ{D1   +/ /::j%0    ! !Q & !&&BmmFLLwd/C&/B8jjl k4 (# M $$88tL M MM 4 4T\\; O 99 ))CDI%(V%5%:3v;q> ![ ! !r.c:|rd}nd}|j||||S)axSend a text or JSON message to the server. :param data: A string, dictionary or list to send to the server. :param json: ``True`` to send a JSON message, ``False`` to send a text message. :param callback: ``True`` if the client requests a callback, ``False`` if not. Note that client-side callbacks are not implemented, a callback request will just tell the server to provide the arguments to invoke the callback, but no callback is invoked. Instead, the arguments that the server provided for the callback are returned by this function. :param namespace: The namespace of the event. The global namespace is assumed if this argument is not provided. rr )r_r)rf)r+rrr_rmsgs r,sendzSocketIOTestClient.sends( CCyydXyKKr.c|j|s td|xsd}|jDcgc] }|d|k(s |}}|jDcgc] }||vs| c}|_|Scc}wcc}w)aReturn the list of messages received from the server. Since this is not a real client, any time the server emits an event, the event is simply stored. The test code can invoke this method to obtain the list of events that were received since the last call. :param namespace: The namespace to get events from. The global namespace is assumed if this argument is not provided. r\rr)rLrGr)r+rr'rs r, get_receivedzSocketIOTestClient.get_receivedst  +/ /$  JJ HS#k*:i*GS H H%)ZZ@c3arts 1(ddr.