def handle(self, *args, **options):
if not options["task_code"]:
log.error("Expected a task_code argument, use --help.")
return
numeric_level = getattr(logging, options["log_level"].upper(), None)
if not isinstance(numeric_level, int):
log.error("Invalid log level: %s" % options["log_level"])
numeric_level = 10
broker = BrokerConnection()
connection = broker.connect_to_broker()
removeFileHandlers()
setFileHandler(options["worker_nr"])
setLevelToAllHandlers(numeric_level)
if connection is None:
log.error("Could not connect to broker.")
return
action = ActionTask(options["task_code"], options["worker_nr"])
logging.handlers.AMQPMessageHandler = AMQPMessageHandler
broker_logging_handler = logging.handlers.AMQPMessageHandler(
action, numeric_level)
action.set_broker_logging_handler(broker_logging_handler)
task_worker = Worker(connection, options["task_code"], action,
options["worker_nr"])
task_worker.run_worker()
removeFileHandlers()
def test_status_task(self):
"""
"""
action_task = ActionTask(self.task_code, self.worker_nr)
self.assertEqual(True, action_task.status_task(True))
self.assertEqual(True, action_task.status_task((True,)))
self.assertEqual(False, action_task.status_task(None))
def handle(self, *args, **options):
if not options["task_code"]:
log.error("Expected a task_code argument, use --help.")
return
numeric_level = getattr(logging, options["log_level"].upper(), None)
if not isinstance(numeric_level, int):
log.error("Invalid log level: %s" % options["log_level"])
numeric_level = 10
broker = BrokerConnection()
connection = broker.connect_to_broker()
removeFileHandlers()
setFileHandler(options["worker_nr"])
setLevelToAllHandlers(numeric_level)
if connection is None:
log.error("Could not connect to broker.")
return
action = ActionTask(options["task_code"], options["worker_nr"])
logging.handlers.AMQPMessageHandler = AMQPMessageHandler
broker_logging_handler = logging.handlers.AMQPMessageHandler(
action, numeric_level)
action.set_broker_logging_handler(broker_logging_handler)
task_worker = Worker(connection,
options["task_code"],
action,
options["worker_nr"])
task_worker.run_worker()
removeFileHandlers()
def test_action(self, child, task_code, worker_nr):
#action = ActionTask('120', '10')
from lizard_worker.worker.worker import Worker
from lizard_worker.worker.broker_connection import BrokerConnection
connection = BrokerConnection().connect_to_broker()
action = ActionTask(task_code, worker_nr)
worker = Worker(connection, task_code, action, worker_nr)
worker.run_worker()
def test_requeue_failed_message(self):
"""
Test requeueing of failed tasks.
"""
action_task = ActionTask(self.task_code, self.worker_nr)
action_task.properties = None
action_task.body = self.create_body(0, 0)
action_task.requeue_failed_message(self.mock_channel, self.mock_method)
action_task.body = self.create_body(1, 2)
action_task.requeue_failed_message(self.mock_channel, self.mock_method)
action_task.body = {}
action_task.requeue_failed_message(self.mock_channel, self.mock_method)
