diff options
author | liris <liris.pp@gmail.com> | 2011-03-15 15:08:54 +0900 |
---|---|---|
committer | liris <liris.pp@gmail.com> | 2011-03-15 15:08:54 +0900 |
commit | ad4b0aa550cfe8491410f1d1649c16f2df874576 (patch) | |
tree | 734e0d39a71db05005f6ee01c70b71cc33e7dbe2 /test_websocket.py | |
parent | 8c6452006f0dfc4a812ae451ccc6bd883faa1c98 (diff) | |
download | websocket-client-ad4b0aa550cfe8491410f1d1649c16f2df874576.tar.gz |
- more test code.
Diffstat (limited to 'test_websocket.py')
-rw-r--r-- | test_websocket.py | 26 |
1 files changed, 26 insertions, 0 deletions
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() |