def test_processLine_throws_FrameError_on_invalid_command(self):
parser = StompParser()
self.assertRaises(StompFrameError, lambda: parser.add('HELLO\n'))
self.assertFalse(parser.canRead())
parser.add('DISCONNECT\n\n\x00')
self.assertEquals(StompFrame('DISCONNECT'), parser.get())
self.assertFalse(parser.canRead())
def test_frames_with_heart_beats_succeeds(self):
parser = StompParser(version=StompSpec.VERSION_1_1)
disconnect = commands.disconnect()
frame = '\n%s\n' % disconnect
parser.add(2 * frame)
frames = []
while parser.canRead():
frames.append(parser.get())
self.assertEquals(frames, [StompHeartBeat(), disconnect, StompHeartBeat(), StompHeartBeat(), disconnect, StompHeartBeat()])
#self.assert frames
#StompFrame(command='DISCONNECT', headers={}, body=''), StompFrame(command='DISCONNECT', headers={}, body='')]
#self.assertEqual(parser.get(), commands.disconnect())
self.assertEqual(parser.get(), None)
