def test_span_delete(self): # ensure the helper removes properly a propagated Span @self.app.task def fn_task(): return 42 # propagate a Span task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" span = self.tracer.trace("celery.run") attach_span(fn_task, task_id, span) # delete the Span weak_dict = getattr(fn_task, "__dd_task_span") detach_span(fn_task, task_id) assert weak_dict.get((task_id, False)) is None
def test_span_delete(self): # ensure the helper removes properly a propagated Span @self.app.task def fn_task(): return 42 # propagate a Span task_id = '7c6731af-9533-40c3-83a9-25b58f0d837f' span = self.tracer.trace('celery.run') attach_span(fn_task, task_id, span) # delete the Span weak_dict = getattr(fn_task, '__dd_task_span') detach_span(fn_task, task_id) ok_(weak_dict.get(task_id) is None)
def test_span_delete_empty(self): # ensure the helper works even if the Task doesn't have # a propagation @self.app.task def fn_task(): return 42 # delete the Span exception = None task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" try: detach_span(fn_task, task_id) except Exception as e: exception = e assert exception is None