class TestRecvFrame(TestCase):
def setUp(self):
self.protocol = StompProtocol()
def test_decode(self):
self.assertEqual(self.protocol._decode(u'éĂ'), u'éĂ')
def test_on_decode_error_show_string(self):
data = MagicMock(spec=six.binary_type)
data.decode.side_effect = UnicodeDecodeError(
'hitchhiker', b"", 42, 43, 'the universe and everything else')
with self.assertRaises(UnicodeDecodeError):
self.protocol._decode(data)
def test_can_reset(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00')
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 1)
self.protocol.reset()
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 0)
def test_single_packet(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00')
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_no_body_command_packet(self):
self.protocol.feed_data(
b'CONNECT\n'
b'accept-version:1.0\n\n'
b'Hey dude\x00', )
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_partial_packet(self):
stream_data = (
b'CONNECT\n',
b'accept-version:1.0\n\n\x00',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
def test_multi_partial_packet1(self):
stream_data = (b'CONNECT\n', b'accept-version:1.0\n\n\x00\n',
b'CONNECTED\n', b'version:1.0\n\n\x00\n')
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(frames[2].command, u'CONNECTED')
self.assertEqual(frames[2].headers, {u'version': u'1.0'})
self.assertEqual(frames[2].body, None)
self.assertEqual(frames[3].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length(self):
stream_data = (
b'ERROR\n',
b'header:1.0\n',
b'content-length:3\n\n'
b'Hey dude\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'ERROR')
self.assertEqual(frames[0].headers, {
u'header': u'1.0',
u'content-length': u'3'
})
self.assertEqual(frames[0].body.decode(), u'Hey')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet2(self):
stream_data = (
b'CONNECTED\n'
b'version:1.0\n\n',
b'\x00\nERROR\n',
b'header:1.0\n\n',
b'Hey dude\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u'CONNECTED')
self.assertEqual(frames[0].headers, {u'version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(frames[2].command, u'ERROR')
self.assertEqual(frames[2].headers, {u'header': u'1.0'})
self.assertEqual(frames[2].body.decode(), u'Hey dude')
self.assertEqual(frames[3].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet_with_utf8(self):
stream_data = (
b'CONNECTED\n'
b'accept-version:1.0\n\n',
b'\x00\nERROR\n',
b'header:1.0\n\n\xc3',
b'\xa7\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
self.assertEqual(len(self.protocol._frames_ready), 4)
self.assertEqual(self.protocol._pending_parts, [])
self.assertEqual(self.protocol._frames_ready[0].body, None)
self.assertEqual(self.protocol._frames_ready[1].command, u'HEARTBEAT')
self.assertEqual(str(self.protocol._frames_ready[1]),
'<Frame: HEARTBEAT headers: >')
self.assertEqual(self.protocol._frames_ready[2].body.decode(), u'ç')
self.assertEqual(self.protocol._frames_ready[3].command, u'HEARTBEAT')
def test_heart_beat_packet1(self):
self.protocol.feed_data(b'\n')
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet2(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00\n')
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet3(self):
self.protocol.feed_data(b'\nCONNECT\n' b'accept-version:1.0\n\n\x00')
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'HEARTBEAT')
self.assertEqual(frames[1].command, u'CONNECT')
self.assertEqual(frames[1].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[1].body, None)
self.assertEqual(self.protocol._pending_parts, [])
class TestRecvFrame(TestCase):
def setUp(self):
self.protocol = StompProtocol()
def test_decode(self):
self.assertEqual(self.protocol._decode(u'éĂ'), u'éĂ')
def test_on_decode_error_show_string(self):
data = MagicMock(spec=bytes)
data.decode.side_effect = UnicodeDecodeError(
'hitchhiker', b"", 42, 43, 'the universe and everything else')
with self.assertRaises(UnicodeDecodeError):
self.protocol._decode(data)
def test_can_reset(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00')
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 1)
self.protocol.reset()
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 0)
def test_single_packet(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00')
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_no_body_command_packet(self):
self.protocol.feed_data(
b'CONNECT\n'
b'accept-version:1.0\n\n'
b'Hey dude\x00', )
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, b'Hey dude')
self.assertEqual(self.protocol._pending_parts, [])
def test_partial_packet(self):
stream_data = (
b'CONNECT\n',
b'accept-version:1.0\n\n\x00',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
def test_long_packet(self):
data = b'MESSAGE\n' \
b'content-length:14\nexpires:0\ndestination:/topic/' \
b'xxxxxxxxxxxxxxxxxxxxxxxxxl' \
b'\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id' \
b':ID\\cxxxxxx-35207-1543430467768-204' \
b'\\c363\\c-1\\c1\\c463859\npersistent:true\ntimestamp' \
b':1548945234003\n\n222.222.22.222' \
b'\x00\nMESSAGE\ncontent-length:12\nexpires:0\ndestination:' \
b'/topic/xxxxxxxxxxxxxxxxxxxxxxxxxx' \
b'\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id' \
b':ID\\cxxxxxx-35207-1543430467768-204' \
b'\\c363\\c-1\\c1\\c463860\npersistent:true\ntimestamp' \
b':1548945234005\n\n88.88.888.88' \
b'\x00\nMESSAGE\ncontent-length:11\nexpires:0\ndestination:' \
b'/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx' \
b'\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id' \
b':ID\\cxxxxxx-35207-1543430467768-204'\
b'\\c362\\c-1\\c1\\c290793\npersistent:true\ntimestamp' \
b':1548945234005\n\n111.11.1.11' \
b'\x00\nMESSAGE\ncontent-length:14\nexpires:0\ndestination:' \
b'/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx' \
b'\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id' \
b':ID\\cxxxxxx-35207-1543430467768-204' \
b'\\c362\\c-1\\c1\\c290794\npersistent:true\ntimestamp:' \
b'1548945234005\n\n222.222.22.222' \
b'\x00\nMESSAGE\ncontent-length:12\nexpires:0\ndestination:' \
b'/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx' \
b'\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id' \
b':ID\\cxxxxxx-35207-1543430467768-204' \
b'\\c362\\c-1\\c1\\c290795\npersistent:true\ntimestamp:' \
b'1548945234005\n\n88.88.888.88\x00\nMESS'
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 10)
self.assertEqual(frames[0].command, u'MESSAGE')
self.assertEqual(
frames[0].headers, {
'ActiveMQ.MQTT.QoS': '1',
'content-length': '14',
'destination': '/topic/xxxxxxxxxxxxxxxxxxxxxxxxxl',
'expires': '0',
'message-id':
'ID:xxxxxx-35207-1543430467768-204:363:-1:1:463859',
'persistent': 'true',
'priority': '4',
'subscription': '1',
'timestamp': '1548945234003'
})
self.assertEqual(frames[0].body, b'222.222.22.222')
self.assertEqual(frames[1].command, u'HEARTBEAT')
def test_multi_partial_packet1(self):
stream_data = (b'CONNECT\n', b'accept-version:1.0\n\n\x00\n',
b'CONNECTED\n', b'version:1.0\n\n\x00\n')
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u'CONNECT')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(frames[2].command, u'CONNECTED')
self.assertEqual(frames[2].headers, {u'version': u'1.0'})
self.assertEqual(frames[2].body, None)
self.assertEqual(frames[3].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length(self):
stream_data = (
b'ERROR\n',
b'header:1.0\n',
b'content-length:8\n\n'
b'Hey dude\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'ERROR')
self.assertEqual(frames[0].headers, {
u'header': u'1.0',
u'content-length': u'8'
})
self.assertEqual(frames[0].body.decode(), u'Hey dude')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length_EOF(self):
stream_data = (
b'ERROR\n',
b'header:1.0\n',
b'content-length:3\n\n'
b'\x00\x00\x00\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'ERROR')
self.assertEqual(frames[0].headers, {
u'header': u'1.0',
u'content-length': u'3'
})
self.assertEqual(frames[0].body, b'\x00\x00\x00')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_read_partial_content_by_length_EOF(self):
stream_data = (
b'ERROR\n',
b'header:1.0\n',
b'content-length:3\n\n'
b'\x00\x00',
b'\x00\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'ERROR')
self.assertEqual(frames[0].headers, {
u'header': u'1.0',
u'content-length': u'3'
})
self.assertEqual(frames[0].body, b'\x00\x00\x00')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length_EOF_multipacket(self):
stream_data = (b'ERROR\n', b'header:1.0\n', b'content-length:3\n\n'
b'\x00\x00', b'\x00', b'\x00\n')
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'ERROR')
self.assertEqual(frames[0].headers, {
u'header': u'1.0',
u'content-length': u'3'
})
self.assertEqual(frames[0].body, b'\x00\x00\x00')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet2(self):
stream_data = (
b'CONNECTED\n'
b'version:1.0\n\n',
b'\x00\nERROR\n',
b'header:1.0\n\n',
b'Hey dude\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u'CONNECTED')
self.assertEqual(frames[0].headers, {u'version': u'1.0'})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(frames[2].command, u'ERROR')
self.assertEqual(frames[2].headers, {u'header': u'1.0'})
self.assertEqual(frames[2].body.decode(), u'Hey dude')
self.assertEqual(frames[3].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet_with_utf8(self):
stream_data = (
b'CONNECTED\n'
b'accept-version:1.0\n\n',
b'\x00\nERROR\n',
b'header:1.0\n\n\xc3',
b'\xa7\x00\n',
)
for data in stream_data:
self.protocol.feed_data(data)
self.assertEqual(len(self.protocol._frames_ready), 4)
self.assertEqual(self.protocol._pending_parts, [])
self.assertEqual(self.protocol._frames_ready[0].body, None)
self.assertEqual(self.protocol._frames_ready[1].command, u'HEARTBEAT')
self.assertEqual(str(self.protocol._frames_ready[1]),
'<Frame: HEARTBEAT headers: >')
self.assertEqual(self.protocol._frames_ready[2].body.decode(), u'ç')
self.assertEqual(self.protocol._frames_ready[3].command, u'HEARTBEAT')
def test_heart_beat_packet1(self):
self.protocol.feed_data(b'\n')
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet2(self):
self.protocol.feed_data(b'CONNECT\n' b'accept-version:1.0\n\n\x00\n')
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet3(self):
self.protocol.feed_data(b'\nCONNECT\n' b'accept-version:1.0\n\n\x00')
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'HEARTBEAT')
self.assertEqual(frames[1].command, u'CONNECT')
self.assertEqual(frames[1].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[1].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet_with_pending_data(self):
self.protocol.feed_data(b'MESSAGE\n' b'accept-version:1.0')
self.protocol.feed_data(b'\n\nsome_data\x00\n')
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u'MESSAGE')
self.assertEqual(frames[0].headers, {u'accept-version': u'1.0'})
self.assertEqual(frames[0].body, b'some_data')
self.assertEqual(frames[1].command, u'HEARTBEAT')
self.assertEqual(self.protocol._pending_parts, [])
class TestRecvFrame(TestCase):
def setUp(self):
self.protocol = StompProtocol()
def test_decode(self):
self.assertEqual(self.protocol._decode(u"éĂ"), u"éĂ")
def test_on_decode_error_show_string(self):
data = MagicMock(spec=bytes)
data.decode.side_effect = UnicodeDecodeError(
"hitchhiker", b"", 42, 43, "the universe and everything else"
)
with self.assertRaises(UnicodeDecodeError):
self.protocol._decode(data)
def test_can_reset(self):
self.protocol.feed_data(b"CONNECT\n" b"accept-version:1.0\n\n\x00")
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 1)
self.protocol.reset()
self.assertEqual(len(self.protocol._pending_parts), 0)
self.assertEqual(len(self.protocol._frames_ready), 0)
def test_single_packet(self):
self.protocol.feed_data(b"CONNECT\n" b"accept-version:1.0\n\n\x00")
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u"CONNECT")
self.assertEqual(frames[0].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[0].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_no_body_command_packet(self):
self.protocol.feed_data(b"CONNECT\n" b"accept-version:1.0\n\n" b"Hey dude\x00")
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u"CONNECT")
self.assertEqual(frames[0].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[0].body, b"Hey dude")
self.assertEqual(self.protocol._pending_parts, [])
def test_partial_packet(self):
stream_data = (b"CONNECT\n", b"accept-version:1.0\n\n\x00")
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].command, u"CONNECT")
self.assertEqual(frames[0].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[0].body, None)
def test_long_packet(self):
data = (
b"MESSAGE\n"
b"content-length:14\nexpires:0\ndestination:/topic/"
b"xxxxxxxxxxxxxxxxxxxxxxxxxl"
b"\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id"
b":ID\\cxxxxxx-35207-1543430467768-204"
b"\\c363\\c-1\\c1\\c463859\npersistent:true\ntimestamp"
b":1548945234003\n\n222.222.22.222"
b"\x00\nMESSAGE\ncontent-length:12\nexpires:0\ndestination:"
b"/topic/xxxxxxxxxxxxxxxxxxxxxxxxxx"
b"\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id"
b":ID\\cxxxxxx-35207-1543430467768-204"
b"\\c363\\c-1\\c1\\c463860\npersistent:true\ntimestamp"
b":1548945234005\n\n88.88.888.88"
b"\x00\nMESSAGE\ncontent-length:11\nexpires:0\ndestination:"
b"/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
b"\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id"
b":ID\\cxxxxxx-35207-1543430467768-204"
b"\\c362\\c-1\\c1\\c290793\npersistent:true\ntimestamp"
b":1548945234005\n\n111.11.1.11"
b"\x00\nMESSAGE\ncontent-length:14\nexpires:0\ndestination:"
b"/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
b"\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id"
b":ID\\cxxxxxx-35207-1543430467768-204"
b"\\c362\\c-1\\c1\\c290794\npersistent:true\ntimestamp:"
b"1548945234005\n\n222.222.22.222"
b"\x00\nMESSAGE\ncontent-length:12\nexpires:0\ndestination:"
b"/topic/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
b"\nsubscription:1\npriority:4\nActiveMQ.MQTT.QoS:1\nmessage-id"
b":ID\\cxxxxxx-35207-1543430467768-204"
b"\\c362\\c-1\\c1\\c290795\npersistent:true\ntimestamp:"
b"1548945234005\n\n88.88.888.88\x00\nMESS"
)
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 10)
self.assertEqual(frames[0].command, u"MESSAGE")
self.assertEqual(
frames[0].headers,
{
"ActiveMQ.MQTT.QoS": "1",
"content-length": "14",
"destination": "/topic/xxxxxxxxxxxxxxxxxxxxxxxxxl",
"expires": "0",
"message-id": "ID:xxxxxx-35207-1543430467768-204:363:-1:1:463859",
"persistent": "true",
"priority": "4",
"subscription": "1",
"timestamp": "1548945234003",
},
)
self.assertEqual(frames[0].body, b"222.222.22.222")
self.assertEqual(frames[1].command, u"HEARTBEAT")
def test_multi_partial_packet1(self):
stream_data = (
b"CONNECT\n",
b"accept-version:1.0\n\n\x00\n",
b"CONNECTED\n",
b"version:1.0\n\n\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u"CONNECT")
self.assertEqual(frames[0].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(frames[2].command, u"CONNECTED")
self.assertEqual(frames[2].headers, {u"version": u"1.0"})
self.assertEqual(frames[2].body, None)
self.assertEqual(frames[3].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length(self):
stream_data = (
b"ERROR\n",
b"header:1.0\n",
b"content-length:8\n\n" b"Hey dude\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"ERROR")
self.assertEqual(
frames[0].headers, {u"header": u"1.0", u"content-length": u"8"}
)
self.assertEqual(frames[0].body.decode(), u"Hey dude")
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length_EOF(self):
stream_data = (
b"ERROR\n",
b"header:1.0\n",
b"content-length:3\n\n" b"\x00\x00\x00\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"ERROR")
self.assertEqual(
frames[0].headers, {u"header": u"1.0", u"content-length": u"3"}
)
self.assertEqual(frames[0].body, b"\x00\x00\x00")
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_read_partial_content_by_length_EOF(self):
stream_data = (
b"ERROR\n",
b"header:1.0\n",
b"content-length:3\n\n" b"\x00\x00",
b"\x00\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"ERROR")
self.assertEqual(
frames[0].headers, {u"header": u"1.0", u"content-length": u"3"}
)
self.assertEqual(frames[0].body, b"\x00\x00\x00")
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_read_content_by_length_EOF_multipacket(self):
stream_data = (
b"ERROR\n",
b"header:1.0\n",
b"content-length:3\n\n" b"\x00\x00",
b"\x00",
b"\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"ERROR")
self.assertEqual(
frames[0].headers, {u"header": u"1.0", u"content-length": u"3"}
)
self.assertEqual(frames[0].body, b"\x00\x00\x00")
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet2(self):
stream_data = (
b"CONNECTED\n" b"version:1.0\n\n",
b"\x00\nERROR\n",
b"header:1.0\n\n",
b"Hey dude\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 4)
self.assertEqual(frames[0].command, u"CONNECTED")
self.assertEqual(frames[0].headers, {u"version": u"1.0"})
self.assertEqual(frames[0].body, None)
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(frames[2].command, u"ERROR")
self.assertEqual(frames[2].headers, {u"header": u"1.0"})
self.assertEqual(frames[2].body.decode(), u"Hey dude")
self.assertEqual(frames[3].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
def test_multi_partial_packet_with_utf8(self):
stream_data = (
b"CONNECTED\n" b"accept-version:1.0\n\n",
b"\x00\nERROR\n",
b"header:1.0\n\n\xc3",
b"\xa7\x00\n",
)
for data in stream_data:
self.protocol.feed_data(data)
self.assertEqual(len(self.protocol._frames_ready), 4)
self.assertEqual(self.protocol._pending_parts, [])
self.assertEqual(self.protocol._frames_ready[0].body, None)
self.assertEqual(self.protocol._frames_ready[1].command, u"HEARTBEAT")
self.assertEqual(
str(self.protocol._frames_ready[1]), "<Frame: HEARTBEAT headers: >"
)
self.assertEqual(self.protocol._frames_ready[2].body.decode(), u"ç")
self.assertEqual(self.protocol._frames_ready[3].command, u"HEARTBEAT")
def test_heart_beat_packet1(self):
self.protocol.feed_data(b"\n")
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet2(self):
self.protocol.feed_data(b"CONNECT\n" b"accept-version:1.0\n\n\x00\n")
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet3(self):
self.protocol.feed_data(b"\nCONNECT\n" b"accept-version:1.0\n\n\x00")
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"HEARTBEAT")
self.assertEqual(frames[1].command, u"CONNECT")
self.assertEqual(frames[1].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[1].body, None)
self.assertEqual(self.protocol._pending_parts, [])
def test_heart_beat_packet_with_pending_data(self):
self.protocol.feed_data(b"MESSAGE\n" b"accept-version:1.0")
self.protocol.feed_data(b"\n\nsome_data\x00\n")
frames = self.protocol.pop_frames()
self.assertEqual(len(frames), 2)
self.assertEqual(frames[0].command, u"MESSAGE")
self.assertEqual(frames[0].headers, {u"accept-version": u"1.0"})
self.assertEqual(frames[0].body, b"some_data")
self.assertEqual(frames[1].command, u"HEARTBEAT")
self.assertEqual(self.protocol._pending_parts, [])
