def test_extract_message_multi(self):
""" Test the handling of multiple concatenated messages by the buffer. """
m1 = 'CONNECT\nsession:207567f3-cce7-4a0a-930b-46fc394dd53d\n\n0123456789\x00\n'
m2 = 'SUBSCRIBE\nack:auto\ndestination:/queue/test\n\n\x00SEND\ndestination:/queue/test\n\n\x00'
sb = StompFrameBuffer()
sb.append(m1)
f1 = sb.extract_message()
assert f1.cmd == 'CONNECT'
assert f1.body == '0123456789'
assert sb.extract_message() is None
sb.append(m2)
f2 = sb.extract_message()
f3 = sb.extract_message()
assert f2.cmd == 'SUBSCRIBE'
assert f2.body == ''
assert f3.cmd == 'SEND'
assert f3.destination == '/queue/test'
assert f3.body == ''
assert sb.extract_message() is None
def test_extract_message_binary(self):
""" Test extracting a binary frame. """
sb = StompFrameBuffer()
binmsg = "\x00\x00HELLO\x00\x00DONKEY\x00\x00"
m1 = self.createMessage('send', {'content-length': len(binmsg)}, binmsg)
sb.append(m1)
msg = sb.extract_message()
assert isinstance(msg, stomper.Frame)
assert msg.pack() == m1
m2 = self.createMessage('send', {'content-length': len(binmsg), 'x-other-header': 'value'}, binmsg)
sb.append(m2)
msg = sb.extract_message()
assert isinstance(msg, stomper.Frame)
assert msg.pack() == m2
def test_extract_message(self):
""" Test extracting a single frame. """
sb = StompFrameBuffer()
m1 = self.createMessage('connect', {'session': uuid.uuid4()}, 'This is the body')
sb.append(m1)
msg = sb.extract_message()
assert isinstance(msg, stomper.Frame)
assert m1 == msg.pack()
def test_extract_message_fragmented(self):
""" Test the handling of fragmented frame data. """
m1_1 = 'CONNECT\nsession:207567f3-cce7-4a0a-930b-'
m1_2 = '46fc394dd53d\n\n0123456789\x00\nSUBSCRIBE\nack:a'
m1_3 = 'uto\ndestination:/queue/test\n\n\x00SE'
m1_4 = 'ND\ndestination:/queue/test\n\n0123456789\x00'
sb = StompFrameBuffer()
sb.append(m1_1)
assert sb.extract_message() is None
sb.append(m1_2)
f1 = sb.extract_message()
assert f1.cmd == 'CONNECT'
assert f1.body == '0123456789'
assert sb.extract_message() is None
sb.append(m1_3)
f2 = sb.extract_message()
assert f2.cmd == 'SUBSCRIBE'
assert sb.extract_message() is None
sb.append(m1_4)
f3 = sb.extract_message()
assert f3.cmd == 'SEND'
assert f3.destination == '/queue/test'
assert f3.body == '0123456789'