def disable(*args, **kwargs):
sys.stdout, sys.stderr = StringIO(), StringIO()
try:
return fun(*args, **kwargs)
finally:
sys.stdout = sys.__stdout__
sys.stderr = sys.__stderr__
def wrap_logger(logger, loglevel=logging.ERROR):
old_handlers = get_handlers(logger)
sio = StringIO()
siohandler = logging.StreamHandler(sio)
set_handlers(logger, [siohandler])
yield sio
set_handlers(logger, old_handlers)
def test_get_logger(self):
T1 = self.createTaskCls("T1", "c.unittest.t.t1")
t1 = T1()
logfh = StringIO()
logger = t1.get_logger(logfile=logfh, loglevel=0)
self.assertTrue(logger)
T1.request.loglevel = 3
logger = t1.get_logger(logfile=logfh, loglevel=None)
self.assertTrue(logger)
def _test_on_failure(self, exception):
app = app_or_default()
tid = gen_unique_id()
tw = TaskRequest(mytask.name, tid, [4], {"f": "x"})
try:
raise exception
except Exception:
exc_info = ExceptionInfo(sys.exc_info())
logfh = StringIO()
tw.logger.handlers = []
tw.logger = setup_logger(logfile=logfh, loglevel=logging.INFO,
root=False)
app.conf.CELERY_SEND_TASK_ERROR_EMAILS = True
tw.on_failure(exc_info)
logvalue = logfh.getvalue()
self.assertIn(mytask.name, logvalue)
self.assertIn(tid, logvalue)
self.assertIn("ERROR", logvalue)
app.conf.CELERY_SEND_TASK_ERROR_EMAILS = False
def _test_on_failure(self, exception):
app = app_or_default()
tid = gen_unique_id()
tw = TaskRequest(mytask.name, tid, [4], {"f": "x"})
try:
raise exception
except Exception:
exc_info = ExceptionInfo(sys.exc_info())
logfh = StringIO()
tw.logger.handlers = []
tw.logger = setup_logger(logfile=logfh, loglevel=logging.INFO,
root=False)
app.conf.CELERY_SEND_TASK_ERROR_EMAILS = True
tw.on_failure(exc_info)
logvalue = logfh.getvalue()
self.assertIn(mytask.name, logvalue)
self.assertIn(tid, logvalue)
self.assertIn("ERROR", logvalue)
app.conf.CELERY_SEND_TASK_ERROR_EMAILS = False
def _mocked(url, *args, **kwargs):
response_data, headers = response_method(url)
return addinfourl(StringIO(response_data), headers, url)
