diff options
| author | engn33r <engn33r@users.noreply.github.com> | 2021-03-29 12:21:27 -0700 |
|---|---|---|
| committer | engn33r <engn33r@users.noreply.github.com> | 2021-03-29 12:21:27 -0700 |
| commit | aa0ddd291beb116d7625cd96f2b3971fa196c7f6 (patch) | |
| tree | 969b5378cf7c4348a191e6a1001d87a26f770419 | |
| parent | 331d601eb599a5a7be7588c98fa2da9d23fa89c5 (diff) | |
| download | websocket-client-aa0ddd291beb116d7625cd96f2b3971fa196c7f6.tar.gz | |
Add unit test for PR #627
| -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")) |