def test_frames_with_optional_newlines_succeeds(self):
parser = StompParser()
frame = '\n%s\n' % stomper.disconnect()
parser.add(2 * frame)
for _ in xrange(2):
self.assertEqual(parser.getMessage(), {'cmd': 'DISCONNECT', 'headers': {}, 'body': ''})
self.assertEqual(parser.getMessage(), None)
def test_frameParse_succeeds(self):
message = {
'cmd': 'SEND',
'headers': {'foo': 'bar', 'hello ': 'there-world with space ', 'empty-value':'', '':'empty-header', 'destination': '/queue/blah'},
'body': 'some stuff\nand more'
}
frame = createFrame(message)
parser = StompParser()
parser.add(frame.pack())
self.assertEqual(parser.getMessage(), {'cmd': frame.cmd, 'headers': frame.headers, 'body': frame.body})
self.assertEqual(parser.getMessage(), None)
#This time remove optional trailing newline which stomper adds to frames
parser.add(frame.pack()[:-1])
self.assertEqual(parser.getMessage(), {'cmd': frame.cmd, 'headers': frame.headers, 'body': frame.body})
self.assertEqual(parser.getMessage(), None)
def test_processLine_throws_FrameError_on_header_line_missing_separator(self):
parser = StompParser()
parser.add('SEND\n')
self.assertRaises(StompFrameError, lambda: parser.add('no separator\n'))
def test_processLine_throws_FrameError_on_invalid_command(self):
parser = StompParser()
self.assertRaises(StompFrameError, lambda: parser.add('HELLO\n'))
def test_getMessage_returns_None_if_not_done(self):
parser = StompParser()
self.assertEqual(None, parser.getMessage())
parser.add('CONNECT')
self.assertEqual(None, parser.getMessage())
def test_frame_without_header_or_body_succeeds(self):
parser = StompParser()
parser.add(stomper.disconnect())
msg = parser.getMessage()
self.assertEqual(msg, {'cmd': 'DISCONNECT', 'headers': {}, 'body': ''})