1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import unittest
import six
from engineio import packet
from engineio import payload
class TestPayload(unittest.TestCase):
def test_encode_empty_payload(self):
p = payload.Payload()
self.assertEqual(p.packets, [])
self.assertEqual(p.encode(), b'')
def test_decode_empty_payload(self):
p = payload.Payload(encoded_payload=b'')
self.assertEqual(p.encode(), b'')
def test_encode_payload_xhr2(self):
pkt = packet.Packet(packet.MESSAGE, data=six.text_type('abc'))
p = payload.Payload([pkt])
self.assertEqual(p.packets, [pkt])
self.assertEqual(p.encode(), b'\x00\x04\xff4abc')
def test_decode_payload_xhr2(self):
p = payload.Payload(encoded_payload=b'\x00\x04\xff4abc')
self.assertEqual(p.encode(), b'\x00\x04\xff4abc')
def test_encode_payload_xhr_text(self):
pkt = packet.Packet(packet.MESSAGE, data=six.text_type('abc'))
p = payload.Payload([pkt])
self.assertEqual(p.packets, [pkt])
self.assertEqual(p.encode(b64=True), b'4:4abc')
def test_decode_payload_xhr_text(self):
p = payload.Payload(encoded_payload=b'4:4abc')
self.assertEqual(p.encode(), b'\x00\x04\xff4abc')
def test_encode_payload_xhr_binary(self):
pkt = packet.Packet(packet.MESSAGE, data=b'\x00\x01\x02', binary=True)
p = payload.Payload([pkt])
self.assertEqual(p.packets, [pkt])
self.assertEqual(p.encode(b64=True), b'6:b4AAEC')
def test_decode_payload_xhr_binary(self):
p = payload.Payload(encoded_payload=b'6:b4AAEC')
self.assertEqual(p.encode(), b'\x01\x04\xff\x04\x00\x01\x02')
def test_decode_invalid_payload(self):
self.assertRaises(ValueError, payload.Payload,
encoded_payload=b'bad payload')
def test_decode_multi_payload(self):
p = payload.Payload(encoded_payload=b'4:4abc\x00\x04\xff4def')
self.assertEqual(len(p.packets), 2)
self.assertEqual(p.packets[0].data, 'abc')
self.assertEqual(p.packets[1].data, 'def')
|
