71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
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_encode_jsonp_payload(self):
|
|
pkt = packet.Packet(packet.MESSAGE, data=six.text_type('abc'))
|
|
p = payload.Payload([pkt])
|
|
self.assertEqual(p.packets, [pkt])
|
|
self.assertEqual(p.encode(jsonp_index=233),
|
|
b'___eio[233]("\x00\x04\xff4abc");')
|
|
self.assertEqual(p.encode(jsonp_index=233, b64=True),
|
|
b'___eio[233]("4:4abc");')
|
|
|
|
def test_decode_jsonp_payload(self):
|
|
p = payload.Payload(encoded_payload=b'd=4:4abc')
|
|
self.assertEqual(p.encode(), b'\x00\x04\xff4abc')
|
|
|
|
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')
|