def test_create_event_update_message(self): service_id = 'facebook' service_user_id = 'service_user_id' service_event_id = 'service_event_id' message = messages.create_event_update_message(service_id, service_user_id, service_event_id) self.assertEqual(message['header']['type'], service_id + '.update') self.assertEqual(message['message']['service_author_id'], service_user_id) self.assertEqual(message['message']['service_event_id'], service_event_id)
def _send_update_message_from_event(message_client, exchange, event): current_id = "" if event is not None: message = messages.create_event_update_message( event.service_name, event.service_user_id, event.service_event_id) message_queue.send_messages(message_client, exchange, [message]) current_id = event.hash_id return current_id
def app_main(self, config, options, args): logging.info("Beginning...") db.configure_session(url=db.create_url_from_config(config['db'])) # if services option is empty default to all available configured services if not options.services: service_names = [] # generate the list of service name's from queue list for key in config['queues'].iterkeys(): service_names.append(key) else: service_names = options.services # if authors option is empty default to all authors if not options.authors: author_names = [author_name for author_name in db.Session().query(Author.author_name).all()] else: author_names = options.authors # get the broker and queue config broker_url = config['broker']['url'] # get message broker client and store in instance client = message_queue.create_message_client(broker_url) # for each specified service for service_name in service_names: # create the queue for this service message_queue.create_queues_from_config(client, config['amqp']) # for each specified author for author_name in author_names: # post a update message for service & author for asm, event in db.Session().query(AuthorServiceMap, ServiceEvent). \ join(ServiceEvent, and_(AuthorServiceMap.author_id == ServiceEvent.author_id, AuthorServiceMap.service_id == ServiceEvent.service_id)). \ join(Service, AuthorServiceMap.service_id == Service.id). \ join(Author, AuthorServiceMap.author_id == Author.id). \ filter(and_(Service.service_name == service_name, Author.author_name == author_name)).all(): if event.type_id == 2: continue message_queue.send_messages(client, [create_event_update_message(service_name, asm.service_author_id, event.event_id)]) logging.info("Finished...")