From ad4b0aa550cfe8491410f1d1649c16f2df874576 Mon Sep 17 00:00:00 2001 From: liris Date: Tue, 15 Mar 2011 15:08:54 +0900 Subject: - more test code. --- test_websocket.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/test_websocket.py b/test_websocket.py index be0f5e9..8143efc 100644 --- a/test_websocket.py +++ b/test_websocket.py @@ -174,6 +174,32 @@ class WebSocketTest(unittest.TestCase): self.assertEquals(len(data), 255) self.assertEquals(data, "a" * 255) + def testWebSocket(self): + s = ws.create_connection("ws://echo.websocket.org/") + self.assertNotEquals(s, None) + s.send("Hello, World") + result = s.recv() + self.assertEquals(result, "Hello, World") + s.close() + + def testSecureWebsocket(self): + s = ws.create_connection("wss://echo.websocket.org/") + self.assertNotEquals(s, None) + s.send("Hello, World") + result = s.recv() + self.assertEquals(result, "Hello, World") + s.close() + + def testWebSocketWihtCustomHeader(self): + s = ws.create_connection("ws://echo.websocket.org/", + headers={"User-Agent": "PythonWebsocketClient"}) + self.assertNotEquals(s, None) + s.send("Hello, World") + result = s.recv() + self.assertEquals(result, "Hello, World") + s.close() + + if __name__ == "__main__": unittest.main() -- cgit v1.2.1