Implement disconnect method for external processes (Fixes miguelgrinberg#684)

miguelgrinberg · miguelgrinberg · commit a830c9f7887d · 2021-08-20T17:16:42.000+01:00
diff --git a/src/socketio/pubsub_manager.py b/src/socketio/pubsub_manager.py
@@ -75,6 +75,10 @@ def can_disconnect(self, sid, namespace):
             self._publish({'method': 'disconnect', 'sid': sid,
                            'namespace': namespace or '/'})
 
+    def disconnect(self, sid, namespace=None):
+        self._publish({'method': 'disconnect', 'sid': sid,
+                       'namespace': namespace or '/'})
+
     def close_room(self, room, namespace=None):
         self._publish({'method': 'close_room', 'room': room,
                        'namespace': namespace or '/'})
diff --git a/tests/common/test_pubsub_manager.py b/tests/common/test_pubsub_manager.py
@@ -169,6 +169,12 @@ def test_can_disconnect(self):
             {'method': 'disconnect', 'sid': sid, 'namespace': '/foo'}
         )
 
+    def test_disconnect(self):
+        self.pm.disconnect('foo')
+        self.pm._publish.assert_called_once_with(
+            {'method': 'disconnect', 'sid': 'foo', 'namespace': '/'}
+        )
+
     def test_close_room(self):
         self.pm.close_room('foo')
         self.pm._publish.assert_called_once_with(
