def test_parse_hybi00_incomplete(self):
frame = six.b("\x00Test")
frames, buf = parse_hybi00_frames(frame)
self.assertEqual(len(frames), 0)
self.assertEqual(buf, six.b("\x00Test"))
def test_parse_hybi00_garbage(self):
frame = six.b("trash\x00Test\xff")
frames, buf = parse_hybi00_frames(frame)
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0], (NORMAL, six.b("Test")))
self.assertEqual(buf, six.b(""))
def test_parse_hybi00_single(self):
frame = "\x00Test\xff"
frames, buf = parse_hybi00_frames(frame)
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0], (NORMAL, "Test"))
self.assertEqual(buf, "")
def test_parse_hybi00_single(self):
frame = "\x00Test\xff"
frames, buf = parse_hybi00_frames(frame)
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0], (NORMAL, "Test"))
self.assertEqual(buf, "")
def test_parse_hybi00_multiple(self):
frame = six.b("\x00Test\xff\x00Again\xff")
frames, buf = parse_hybi00_frames(frame)
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0], (NORMAL, six.b("Test")))
self.assertEqual(frames[1], (NORMAL, six.b("Again")))
self.assertEqual(buf, six.b(""))
def test_socketio_crashers(self):
"""
A series of snippets which crash other WebSockets implementations
(specifically, Socket.IO) are harmless to this implementation.
"""
frames = [
"""[{"length":1}]""",
"""{"messages":[{"length":1}]}""",
"hello",
"hello<script>alert(/xss/)</script>",
"hello<img src=x:x onerror=alert(/xss.2/)>",
"{",
"~m~EVJLFDJP~",
]
for frame in frames:
prepared = make_hybi00_frame(frame)
frames, buf = parse_hybi00_frames(prepared)
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0], (NORMAL, frame.encode('utf-8')))
self.assertEqual(buf, six.b(""))
