def test_missing_key(self): ws = websocket.WebSocket() sock = FakeSocket() self.assertRaises(Exception, ws.accept, sock, { 'upgrade': 'websocket', 'Sec-WebSocket-Version': '13' })
def test_decode_hybi_binary(self): buf = b'\x82\x04\x01\x02\x03\x04' ws = websocket.WebSocket() res = ws._decode_hybi(buf) self.assertEqual(res['fin'], 1) self.assertEqual(res['opcode'], 0x2) self.assertEqual(res['length'], len(buf)) self.assertEqual(res['payload'], b'\x01\x02\x03\x04')
def test_decode_hybi_extended_64bit_binary(self): data = (b'\x01\x02\x03\x04' * 65) # len > 126 -- len == 260 buf = b'\x82\x7f\x00\x00\x00\x00\x00\x00\x01\x04' + data ws = websocket.WebSocket() res = ws._decode_hybi(buf) self.assertEqual(res['fin'], 1) self.assertEqual(res['opcode'], 0x2) self.assertEqual(res['length'], len(buf)) self.assertEqual(res['payload'], data)
def test_decode_hybi_text(self): buf = b'\x81\x85\x37\xfa\x21\x3d\x7f\x9f\x4d\x51\x58' ws = websocket.WebSocket() res = ws._decode_hybi(buf) self.assertEqual(res['fin'], 1) self.assertEqual(res['opcode'], 0x1) self.assertEqual(res['masked'], True) self.assertEqual(res['length'], len(buf)) self.assertEqual(res['payload'], b'Hello')
def test_missing_protocol(self): ws = websocket.WebSocket() sock = FakeSocket() self.assertRaises( Exception, ws.accept, sock, { 'upgrade': 'websocket', 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==', 'Sec-WebSocket-Protocol': 'foobar gazonk' })
def setUp(self): self.ws = websocket.WebSocket() self.sock = FakeSocket() self.ws.accept( self.sock, { 'upgrade': 'websocket', 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==' }) self.assertEqual(self.sock.data[:13], b'HTTP/1.1 101 ') self.sock.data = b''
def test_no_protocol(self): ws = websocket.WebSocket() sock = FakeSocket() ws.accept( sock, { 'upgrade': 'websocket', 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==' }) self.assertEqual(sock.data[:13], b'HTTP/1.1 101 ') self.assertFalse(b'\r\nSec-WebSocket-Protocol:' in sock.data)
def test_bad_upgrade(self): ws = websocket.WebSocket() sock = FakeSocket() self.assertRaises( Exception, ws.accept, sock, { 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==' }) self.assertRaises( Exception, ws.accept, sock, { 'upgrade': 'websocket2', 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==' })
def test_success(self): ws = websocket.WebSocket() sock = FakeSocket() ws.accept( sock, { 'upgrade': 'websocket', 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==' }) self.assertEqual(sock.data[:13], b'HTTP/1.1 101 ') self.assertTrue(b'\r\nUpgrade: websocket\r\n' in sock.data) self.assertTrue(b'\r\nConnection: Upgrade\r\n' in sock.data) self.assertTrue( b'\r\nSec-WebSocket-Accept: pczpYSQsvE1vBpTQYjFQPcuoj6M=\r\n' in sock.data)
def test_decode_hybi_multi(self): buf1 = b'\x01\x03\x48\x65\x6c' buf2 = b'\x80\x02\x6c\x6f' ws = websocket.WebSocket() res1 = ws._decode_hybi(buf1) self.assertEqual(res1['fin'], 0) self.assertEqual(res1['opcode'], 0x1) self.assertEqual(res1['length'], len(buf1)) self.assertEqual(res1['payload'], b'Hel') res2 = ws._decode_hybi(buf2) self.assertEqual(res2['fin'], 1) self.assertEqual(res2['opcode'], 0x0) self.assertEqual(res2['length'], len(buf2)) self.assertEqual(res2['payload'], b'lo')
def test_encode_hybi_basic(self): ws = websocket.WebSocket() res = ws._encode_hybi(0x1, b'Hello') expected = b'\x81\x05\x48\x65\x6c\x6c\x6f' self.assertEqual(res, expected)