SocketIOTestClient can handle disconnect event from server (#967)

flask_socketio/test_client.py

@@ -45,6 +45,8 @@ def _mock_send_packet(sid, pkt):
                     pkt.packet_type == packet.BINARY_ACK:
                 self.acks[sid] = {'args': pkt.data,
                                   'namespace': pkt.namespace or '/'}
+            elif pkt.packet_type == packet.DISCONNECT:
+                self.connected[pkt.namespace or '/'] = False
 
         self.app = app
         self.flask_test_client = flask_test_client

test_socketio.py

@@ -7,7 +7,7 @@
 
 from flask import Flask, session, request, json as flask_json
 from flask_socketio import SocketIO, send, emit, join_room, leave_room, \
-    Namespace
+    Namespace, disconnect
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret'
@@ -216,6 +216,9 @@ def on_json(self, data):
         if not data.get('noackargs'):
             return data
 
+    def on_exit(self, data):
+        disconnect()
+
     def on_my_custom_event(self, data):
         emit('my custom response', data)
         if not data.get('noackargs'):
@@ -608,6 +611,14 @@ def test_send_json_class_based(self):
         self.assertEqual(len(received), 1)
         self.assertEqual(received[0]['args']['a'], 'b')
 
+    def test_server_disconnected(self):
+        client = socketio.test_client(app, namespace='/ns')
+        client.get_received('/ns')
+        client.emit('exit', {}, namespace='/ns')
+        self.assertFalse(client.is_connected('/ns'))
+        with self.assertRaises(RuntimeError):
+            client.emit('hello', {}, namespace='/ns')
+
     def test_emit_class_based(self):
         client = socketio.test_client(app, namespace='/ns')
         client.get_received('/ns')