def test_format_raises(self, safe_str): x = ColorFormatter('HELLO') def on_safe_str(s): try: raise ValueError('foo') finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str class Record(object): levelname = 'ERROR' msg = 'HELLO' exc_text = 'error text' stack_info = None def __str__(self): return on_safe_str('') def getMessage(self): return self.msg record = Record() safe_str.return_value = record x.format(record) self.assertIn('<Unrepresentable', record.msg) self.assertEqual(safe_str.call_count, 2)
def test_format_raises(self, safe_str): x = ColorFormatter("HELLO") def on_safe_str(s): try: raise ValueError("foo") finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str class Record(object): levelname = "ERROR" msg = "HELLO" exc_text = "error text" stack_info = None def __str__(self): return on_safe_str("") def getMessage(self): return self.msg record = Record() safe_str.return_value = record x.format(record) self.assertIn("<Unrepresentable", record.msg) self.assertEqual(safe_str.call_count, 2)
def test_format_raises_no_color(self, safe_str): x = ColorFormatter("HELLO", False) record = Mock() record.levelname = "ERROR" record.msg = "HELLO" record.exc_text = "error text" x.format(record) self.assertEqual(safe_str.call_count, 1)
def test_format_raises_no_color(self, safe_str): x = ColorFormatter(use_color=False) record = Mock() record.levelname = 'ERROR' record.msg = 'HELLO' record.exc_text = 'error text' x.format(record) self.assertEqual(safe_str.call_count, 1)
def test_format_raises_no_color(self, safe_str): if sys.version_info[0] == 3: raise SkipTest("py3k") x = ColorFormatter("HELLO", False) record = Mock() record.levelname = "ERROR" record.msg = "HELLO" record.exc_text = "error text" x.format(record) self.assertEqual(safe_str.call_count, 1)
def test_format_raises_no_color(self, safe_str): if sys.version_info[0] == 3: raise SkipTest('py3k') x = ColorFormatter('HELLO', False) record = Mock() record.levelname = 'ERROR' record.msg = 'HELLO' record.exc_text = 'error text' x.format(record) self.assertEqual(safe_str.call_count, 1)
def test_format_raises(self, safe_str): x = ColorFormatter() def on_safe_str(s): try: raise ValueError('foo') finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str class Record: levelname = 'ERROR' msg = 'HELLO' exc_info = 1 exc_text = 'error text' stack_info = None def __str__(self): return on_safe_str('') def getMessage(self): return self.msg record = Record() safe_str.return_value = record msg = x.format(record) assert '<Unrepresentable' in msg assert safe_str.call_count == 1
def test_format_object(self, _format): x = ColorFormatter() x.use_color = True record = Mock() record.levelname = 'ERROR' record.msg = object() self.assertTrue(x.format(record))
def test_format_raises(self, safe_str): x = ColorFormatter() def on_safe_str(s): try: raise ValueError('foo') finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str @python_2_unicode_compatible class Record(object): levelname = 'ERROR' msg = 'HELLO' exc_info = 1 exc_text = 'error text' stack_info = None def __str__(self): return on_safe_str('') def getMessage(self): return self.msg record = Record() safe_str.return_value = record msg = x.format(record) self.assertIn('<Unrepresentable', msg) self.assertEqual(safe_str.call_count, 1)
def test_format_raises(self, safe_str): x = ColorFormatter() def on_safe_str(s): try: raise ValueError('foo') finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str @python_2_unicode_compatible class Record(object): levelname = 'ERROR' msg = 'HELLO' exc_info = 1 exc_text = 'error text' stack_info = None def __str__(self): return on_safe_str('') def getMessage(self): return self.msg record = Record() safe_str.return_value = record msg = x.format(record) assert '<Unrepresentable' in msg assert safe_str.call_count == 1
def format(self, record): task = get_current_task() if task and task.request: record.__dict__.update(task_id=task.request.id, task_name=task.name) else: record.__dict__.setdefault("task_name", "???") record.__dict__.setdefault("task_id", "???") return ColorFormatter.format(self, record)
def format(self, record): task = get_current_task() if task and task.request: name = task.name.split('.')[-1] record.__dict__.update(task_id=task.request.id, task_name=name) else: record.__dict__.update(task_id='xxx', task_name='xxx') return ColorFormatter.format(self, record)
def format(self, record): task = get_current_task() if task: record.__dict__.update(task_id=task.request.id, task_name=task.name) else: record.__dict__.setdefault("task_name", "???") record.__dict__.setdefault("task_id", "???") return ColorFormatter.format(self, record)
def format(self, record): task = get_current_task() if task and task.request: record.__dict__.update(task_id=task.request.id, task_name=task.name) else: record.__dict__.setdefault('task_name', '???') record.__dict__.setdefault('task_id', '???') return ColorFormatter.format(self, record)
def format(self, record): task = get_current_task() if task: record.__dict__.update(task_id=task.request.id, task_name=task.name) else: record.__dict__.setdefault('task_name', '???') record.__dict__.setdefault('task_id', '???') return ColorFormatter.format(self, record)
def test_format_raises(self, safe_str): x = ColorFormatter("HELLO") def on_safe_str(s): try: raise ValueError("foo") finally: safe_str.side_effect = None safe_str.side_effect = on_safe_str record = Mock() record.levelname = "ERROR" record.msg = "HELLO" record.exc_text = "error text" safe_str.return_value = record x.format(record) self.assertIn("<Unrepresentable", record.msg) self.assertEqual(safe_str.call_count, 2)
def format(self, record): self.use_color = False task = get_current_task() if task and task.request: record.__dict__.update(root_id=task.request.root_id, task_id=task.request.id, task_name=task.name) else: record.__dict__.setdefault('task_name', None) record.__dict__.setdefault('task_id', None) record.__dict__.setdefault('root_id', None) return ColorFormatter.format(self, record)
def format(self, record): task = get_current_task() if task and task.request: # For gevent pool, task_id will be something like `7ab44cb4-aacf-444e-bc20-4cbaa2a7b082`. For logs # is better to get it short task_id = task.request.id[:8] # Task name usually has all the package, better cut the first part for logging task_name = task.name.split('.')[-1] record.__dict__.update(task_id=task_id, task_name=task_name) else: record.__dict__.setdefault('task_name', '???') record.__dict__.setdefault('task_id', '???') return ColorFormatter.format(self, record)
def format(self, record): """Append task-related values to the record upon format.""" task = get_current_task() if task and task.request: record.__dict__.update( task_id=task.request.id, task_name=task.name, task_root_id=task.request.root_id, task_parent_id=task.request.parent_id, ) else: record.__dict__.setdefault("task_name", "None") record.__dict__.setdefault("task_id", "None") record.__dict__.setdefault("task_root_id", "None") record.__dict__.setdefault("task_parent_id", "None") return ColorFormatter.format(self, record)