def zaqar_to_proton(message):
"""Convert a message retrieved from storage to a Proton message"""
msg = Message()
msg.ttl = message.get('ttl')
msg.body = message.get('body')
# NOTE(vkmc) This won't work for now - there is no 'amqp10' field yet
if message.get('amqp10'):
msg.priority = message.get('amqp10').get('priority')
msg.first_acquirer = message.get('amqp10').get('first_acquirer')
msg.delivery_count = message.get('amqp10').get('delivery_count')
msg.id = message.get('amqp10').get('id'),
msg.user_id = message.get('amqp10').get('user_id')
msg.address = message.get('amqp10').get('address')
msg.subject = message.get('amqp10').get('subject')
msg.reply_to = message.get('amqp10').get('reply_to')
msg.correlation_id = message.get('amqp10').get('correlation_id')
msg.content_type = message.get('amqp10').get('content_type')
msg.content_encoding = message.get('amqp10').get('content_encoding')
msg.expiry_time = message.get('amqp10').get('expiry_time')
msg.creation_time = message.get('amqp10').get('creation_time'),
msg.group_id = message.get('amqp10').get('group_id')
msg.group_sequence = message.get('amqp10').get('group_sequence')
msg.reply_to_group_id = message.get('amqp10').get('reply_to_group_id')
msg.format = message.get('amqp10').get('format')
return msg
def _send_request(self):
"""Send a message containing the RPC method call
"""
msg = Message()
msg.subject = "An RPC call!"
msg.address = self._to
msg.reply_to = self._reply_to
msg.body = self._method
msg.correlation_id = 5 # whatever...
print("sending RPC call request: %s" % str(self._method))
# @todo send timeout self._sender.send(msg, self, None, time.time() +
# 10)
self._sender.send(msg, self)
def _send_request(self):
"""Send a message containing the RPC method call
"""
msg = Message()
msg.subject = "An RPC call!"
msg.address = self._to
msg.reply_to = self._reply_to
msg.body = self._method
msg.correlation_id = 5 # whatever...
print("sending RPC call request: %s" % str(self._method))
# @todo send timeout self._sender.send(msg, self, None, time.time() +
# 10)
self._sender.send(msg, self)
def message_received(self, receiver_link, message, handle):
LOG.debug("message received callback")
global reply_senders
# extract to reply-to, correlation id
reply_to = message.reply_to
if not reply_to or reply_to not in reply_senders:
LOG.error("sender for reply-to not found, reply-to=%s",
str(reply_to))
info = Condition("not-found",
"Bad reply-to address: %s" % str(reply_to))
self._link.message_rejected(handle, info)
else:
my_sender = reply_senders[reply_to]
correlation_id = message.correlation_id
method_map = message.body
if (not isinstance(method_map, dict) or
'method' not in method_map):
LOG.error("no method given, map=%s", str(method_map))
info = Condition("invalid-field",
"no method given, map=%s" % str(method_map))
self._link.message_rejected(handle, info)
else:
response = Message()
response.address = reply_to
response.subject = message.subject
response.correlation_id = correlation_id
response.body = {"response": method_map}
print("RPC request received, msg=%s" % str(method_map))
print(" to address=%s" % str(message.address))
print(" replying to=%s" % str(reply_to))
link = my_sender.sender_link
# @todo send timeouts
# link.send( response, my_sender,
# message, time.time() + 5.0)
link.send(response, my_sender, message)
self._link.message_accepted(handle)
if self._link.capacity == 0:
LOG.debug("increasing credit...")
self._link.add_capacity(5)
def message_received(self, receiver_link, message, handle):
LOG.debug("message received callback")
global reply_senders
# extract to reply-to, correlation id
reply_to = message.reply_to
if not reply_to or reply_to not in reply_senders:
LOG.error("sender for reply-to not found, reply-to=%s",
str(reply_to))
info = Condition("not-found",
"Bad reply-to address: %s" % str(reply_to))
self._link.message_rejected(handle, info)
else:
my_sender = reply_senders[reply_to]
correlation_id = message.correlation_id
method_map = message.body
if (not isinstance(method_map, dict)
or 'method' not in method_map):
LOG.error("no method given, map=%s", str(method_map))
info = Condition("invalid-field",
"no method given, map=%s" % str(method_map))
self._link.message_rejected(handle, info)
else:
response = Message()
response.address = reply_to
response.subject = message.subject
response.correlation_id = correlation_id
response.body = {"response": method_map}
print("RPC request received, msg=%s" % str(method_map))
print(" to address=%s" % str(message.address))
print(" replying to=%s" % str(reply_to))
link = my_sender.sender_link
# @todo send timeouts
# link.send( response, my_sender,
# message, time.time() + 5.0)
link.send(response, my_sender, message)
self._link.message_accepted(handle)
if self._link.capacity == 0:
LOG.debug("increasing credit...")
self._link.add_capacity(5)
def on_sendable(self, event):
if not self.sent:
msg = Message()
msg.address = self.address
msg.id = '123455'
msg.user_id = BINARY('testuser')
msg.subject = 'test-subject'
msg.content_type = 'text/html; charset=utf-8'
msg.correlation_id = 89
msg.creation_time = 1487772623.883
msg.group_id = "group1"
msg.reply_to = 'hello_world'
msg.content_encoding = 'gzip, deflate'
msg.reply_to_group_id = "group0"
application_properties = dict()
application_properties['app-property'] = [10, 20, 30]
application_properties['some-other'] = symbol("O_one")
msg.properties = application_properties
msg.body = u"Hello World!"
event.sender.send(msg)
self.sent = True
def on_sendable(self, event):
if not self.sent:
msg = Message()
msg.address = self.address
msg.id = '123455'
msg.user_id = 'testuser'
msg.subject = 'test-subject'
msg.content_type = 'text/html; charset=utf-8'
msg.correlation_id = 89
msg.creation_time = 1487772623.883
msg.group_id = "group1"
msg.reply_to = 'hello_world'
msg.content_encoding = 'gzip, deflate'
msg.reply_to_group_id = "group0"
application_properties = dict()
application_properties['app-property'] = [10, 20, 30]
application_properties['some-other'] = symbol("O_one")
msg.properties = application_properties
msg.body = u"Hello World!"
event.sender.send(msg)
self.sent = True
def test_02_bad_request_message(self):
"""
Test various improperly constructed request messages
"""
server = TestServer(server_port=self.http_server_port,
client_port=self.http_listener_port,
tests={})
body_filler = "?" * 1024 * 300 # Q2
msg = Message(body="NOMSGID " + body_filler)
ts = AsyncTestSender(address=self.INT_A.listener,
target="testServer",
message=msg)
ts.wait()
self.assertEqual(1, ts.rejected)
msg = Message(body="NO REPLY TO " + body_filler)
msg.id = 1
ts = AsyncTestSender(address=self.INT_A.listener,
target="testServer",
message=msg)
ts.wait()
self.assertEqual(1, ts.rejected)
msg = Message(body="NO SUBJECT " + body_filler)
msg.id = 1
msg.reply_to = "amqp://fake/reply_to"
ts = AsyncTestSender(address=self.INT_A.listener,
target="testServer",
message=msg)
ts.wait()
self.assertEqual(1, ts.rejected)
msg = Message(body="NO APP PROPERTIES " + body_filler)
msg.id = 1
msg.reply_to = "amqp://fake/reply_to"
msg.subject = "GET"
ts = AsyncTestSender(address=self.INT_A.listener,
target="testServer",
message=msg)
ts.wait()
self.assertEqual(1, ts.rejected)
# TODO: fix body parsing (returns NEED_MORE)
# msg = Message(body="INVALID BODY " + body_filler)
# msg.id = 1
# msg.reply_to = "amqp://fake/reply_to"
# msg.subject = "GET"
# msg.properties = {"http:target": "/Some/target"}
# ts = AsyncTestSender(address=self.INT_A.listener,
# target="testServer",
# message=msg)
# ts.wait()
# self.assertEqual(1, ts.rejected);
server.wait()
# verify router is still sane:
count, error = http1_ping(self.http_server_port,
self.http_listener_port)
self.assertIsNone(error)
self.assertEqual(1, count)
