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'
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(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 setup(self):
self.debug = False
self.log = logging.getLogger(
'%s.%s' % (self.__module__, self.__class__.__name__))
self.buffer = StompFrameBuffer()
self.engine = StompEngine(connection=self,
authenticator=self.server.authenticator,
queue_manager=self.server.queue_manager,
topic_manager=self.server.topic_manager)
def __init__(self, queue_manager, topic_manager):
self.log = logging.getLogger(
'%s.%s' % (self.__class__.__module__, self.__class__.__name__))
self.log.debug("Initializing StompProtocol.")
self.buffer = StompFrameBuffer()
self.engine = StompEngine(
connection=self,
authenticator=None, # FIXME: Add the authenticator
queue_manager=queue_manager,
topic_manager=topic_manager)
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 __init__(self, addr, connect=True):
"""
@param addr: The (host,port) tuple for connection.
@type addr: C{tuple}
@param connect: Whether to connect socket to specified addr.
@type connect: C{bool}
"""
self.log = logging.getLogger('%s.%s' % (self.__module__, self.__class__.__name__))
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.addr = addr
self.received_frames = Queue()
self.read_stopped = threading.Event()
self.buffer = StompFrameBuffer()
if connect:
self._connect()
def test_iteration(self):
""" Test the iteration feature of our buffer."""
sb = StompFrameBuffer()
m1 = self.createMessage('connect', {'session': uuid.uuid4()}, 'This is the body')
m2 = self.createMessage('send', {'destination': '/queue/sample'}, 'This is the body-2')
print '%r' % m1
print '%r' % m2
sb.append(m1)
sb.append(m2)
assert sb is iter(sb)
idx = 0
expected = (m1, m2)
for frame in sb:
assert isinstance(frame, stomper.Frame)
assert expected[idx] == frame.pack()
idx += 1
assert idx == 2
