def test_message_not_saved_data(): data = test_message_data(saved=False) e = Message(actor=_ACTOR_NAME, phase=_PHASE_NAME, context=_CONTEXT_NAME, hostname=_HOSTNAME, topic=_TOPIC_NAME, msg_type=_MESSAGE_TYPE, data=data) e.store() assert e.message_id assert e.host_id assert e.data_source_id
def _process_message(self, message): message['event'] = 'new-message' message_keys = ('stamp', 'topic', 'actor', 'phase', 'hostname', 'context', 'msg_type') audit_keys = ('event', 'stamp', 'data', 'actor', 'phase', 'hostname', 'context') message['msg_type'] = message.pop('type') payload = message.pop('message') msg = Message(**dict(((k, message[k]) for k in message_keys if k in message))) audit = Audit(**dict(((k, message[k]) for k in audit_keys if k in message))) audit.message = msg audit.message.data = MessageData(data=payload['data'], hash_id=payload['hash']) audit.store() return message