def __init__(self, vumi_api, message):
self.vumi_api = vumi_api
self.message = message
super(MessageMetadataHelper,
self).__init__(message.get('helper_metadata', {}))
# A place to store objects we don't want serialised.
if not hasattr(message, '_store_objects'):
message._store_objects = {}
self._store_objects = message._store_objects
# If we don't have a tag, we want to blow up early in some places.
self.tag = TaggingMiddleware.map_msg_to_tag(message)
def __init__(self, vumi_api, message):
self.vumi_api = vumi_api
self.message = message
super(MessageMetadataHelper, self).__init__(
message.get('helper_metadata', {}))
# A place to store objects we don't want serialised.
if not hasattr(message, '_store_objects'):
message._store_objects = {}
self._store_objects = message._store_objects
# If we don't have a tag, we want to blow up early in some places.
self.tag = TaggingMiddleware.map_msg_to_tag(message)
def get_tag(self, message):
tag = TaggingMiddleware.map_msg_to_tag(message)
if tag is None:
return None
else:
return ":".join(tag)
def get_tag(self, to_addr):
msg = self.mk_msg(to_addr)
msg = self.mw.handle_inbound(msg, "dummy_connector")
return TaggingMiddleware.map_msg_to_tag(msg)
def test_map_msg_to_tag(self):
msg = self.mk_msg("123456")
self.assertEqual(TaggingMiddleware.map_msg_to_tag(msg), None)
msg['helper_metadata']['tag'] = {'tag': ['pool', 'mytag']}
self.assertEqual(TaggingMiddleware.map_msg_to_tag(msg),
("pool", "mytag"))
def handle_inbound(self, message, connector_name):
tag = TaggingMiddleware.map_msg_to_tag(message)
yield self.store.add_inbound_message(message, tag=tag)
returnValue(message)
def handle_outbound(self, message, endpoint):
tag = TaggingMiddleware.map_msg_to_tag(message)
self.store.add_outbound_message(message, tag=tag)
return message
def handle_outbound(self, message, connector_name):
tag = TaggingMiddleware.map_msg_to_tag(message)
yield self.store.add_outbound_message(message, tag=tag)
returnValue(message)
def set_tag(self, tag):
TaggingMiddleware.add_tag_to_msg(self.message, tag)
self.tag = TaggingMiddleware.map_msg_to_tag(self.message)
def handle_outbound(self, message, endpoint):
tag = TaggingMiddleware.map_msg_to_tag(message)
self.store.add_outbound_message(message, tag=tag)
return message
def test_map_msg_to_tag(self):
msg = self.mk_msg("123456")
self.assertEqual(TaggingMiddleware.map_msg_to_tag(msg), None)
msg["helper_metadata"]["tag"] = {"tag": ["pool", "mytag"]}
self.assertEqual(TaggingMiddleware.map_msg_to_tag(msg), ("pool", "mytag"))
def get_tag(self, to_addr):
msg = self.mk_msg(to_addr)
msg = self.mw.handle_inbound(msg, "dummy_endpoint")
return TaggingMiddleware.map_msg_to_tag(msg)
def set_tag(self, tag):
TaggingMiddleware.add_tag_to_msg(self.message, tag)
self.tag = TaggingMiddleware.map_msg_to_tag(self.message)
def get_tag(self, message):
tag = TaggingMiddleware.map_msg_to_tag(message)
if tag is None:
return None
else:
return ":".join(tag)
