diff options
| -rw-r--r-- | websocket/tests/data/header03.txt | 6 | ||||
| -rw-r--r-- | websocket/tests/test_websocket.py | 4 |
2 files changed, 10 insertions, 0 deletions
diff --git a/websocket/tests/data/header03.txt b/websocket/tests/data/header03.txt new file mode 100644 index 0000000..85c619f --- /dev/null +++ b/websocket/tests/data/header03.txt @@ -0,0 +1,6 @@ +HTTP/1.1 101 WebSocket Protocol Handshake
+Connection: Upgrade, Keep-Alive
+Upgrade: WebSocket
+Sec-WebSocket-Accept: Kxep+hNu9n51529fGidYu7a3wO0=
+some_header: something
+
diff --git a/websocket/tests/test_websocket.py b/websocket/tests/test_websocket.py index 5111c0d..3b02314 100644 --- a/websocket/tests/test_websocket.py +++ b/websocket/tests/test_websocket.py @@ -170,6 +170,10 @@ class WebSocketTest(unittest.TestCase): self.assertEqual(status, 101) self.assertEqual(header["connection"], "Upgrade") + status, header, status_message = read_headers(HeaderSockMock("data/header03.txt")) + self.assertEqual(status, 101) + self.assertEqual(header["connection"], "Upgrade, Keep-Alive") + HeaderSockMock("data/header02.txt") self.assertRaises(ws.WebSocketException, read_headers, HeaderSockMock("data/header02.txt")) |