def test_ack_writes_correct_frame(self): id = '12345' stomp = Stomp('localhost', 61613) stomp._checkConnected = Mock() stomp._write = Mock() stomp.ack({'cmd': 'MESSAGE', 'headers': {'message-id': id}, 'body': 'blah'}) args,kargs = stomp._write.call_args sentFrame = self.parseFrame(args[0]) self.assertEquals({'cmd': 'ACK', 'headers': {'message-id': id, }, 'body': ''}, sentFrame)
def test_send_writes_correct_frame(self): dest = '/queue/foo' msg = 'test message' headers = {'foo': 'bar', 'fuzz': 'ball'} stomp = Stomp('localhost', 61613) stomp._checkConnected = Mock() stomp._write = Mock() stomp.send(dest, msg, headers) args,kargs = stomp._write.call_args sentFrame = self.parseFrame(args[0]) self.assertEquals({'cmd': 'SEND', 'headers': {'destination': dest, 'foo': 'bar', 'fuzz': 'ball', }, 'body': msg}, sentFrame)
def test_subscribe_writes_correct_frame(self): dest = '/queue/foo' headers = {'foo': 'bar', 'fuzz': 'ball'} stomp = Stomp('localhost', 61613) stomp._checkConnected = Mock() stomp._write = Mock() stomp.subscribe(dest, headers) args,kargs = stomp._write.call_args sentFrame = self.parseFrame(args[0]) self.assertEquals({'cmd': 'SUBSCRIBE', 'headers': {'destination': dest, 'ack': 'auto', 'activemq.prefetchSize': '1', 'foo': 'bar', 'fuzz': 'ball', }, 'body': ''}, sentFrame)