added a ping(m) method

Lawouach · Lawouach · commit 9e06d71569e4 · 2015-07-23T23:03:11.000+02:00
diff --git a/test/test_websocket.py b/test/test_websocket.py
@@ -169,6 +169,14 @@ def test_closing_message_received(self):
             ws.process(b'unused for this test')
             c.assert_called_once_with(1000, b'test closing')
             
+    def test_sending_ping(self):
+        tm = PingControlMessage("hello").single(mask=False)
+        
+        m = MagicMock()
+        ws = WebSocket(sock=m)
+        ws.ping("hello")
+        m.sendall.assert_called_once_with(tm)
+
 
 if __name__ == '__main__':
     suite = unittest.TestSuite()
diff --git a/ws4py/websocket.py b/ws4py/websocket.py
@@ -8,7 +8,8 @@
 from ws4py import WS_KEY, WS_VERSION
 from ws4py.exc import HandshakeError, StreamClosed
 from ws4py.streaming import Stream
-from ws4py.messaging import Message, PongControlMessage
+from ws4py.messaging import Message, PingControlMessage,\
+    PongControlMessage
 from ws4py.compat import basestring, unicode
 
 DEFAULT_READING_SIZE = 2
@@ -210,6 +211,13 @@ def close_connection(self):
             finally:
                 self.sock = None
 
+    def ping(self, message):
+        """
+        Send a ping message to the remote peer.
+        The given `message` must be a unicode string.
+        """
+        self.send(PingControlMessage(message))
+
     def ponged(self, pong):
         """
         Pong message, as a :class:`messaging.PongControlMessage` instance,
