Пример #1
0
 def teardown(self):
     """
     Removes current failure handler
     """
     task_prerun.disconnect(self.setup_context)
     task_postrun.disconnect(self.reset_context)
     if self.report_exceptions:
         task_failure.disconnect(self._failure_handler)
     logger.info('Honeybadger Celery support uninstalled')
Пример #2
0
 def disconnect_traced_handlers(self):
     if self._propagate:
         before_task_publish.disconnect(self._prepublish)
         after_task_publish.disconnect(self._postpublish)
     task_prerun.disconnect(self._start_span)
     task_failure.disconnect(self._tag_error)
     task_retry.disconnect(self._tag_retry)
     task_postrun.disconnect(self._finish_span)
     log.debug('Disconnected CeleryTracing signal handlers.')
Пример #3
0
def test_celery_run_without_parent_span(span_in_context_mock, celery_eager,
                                        tracer, task_error):
    def task_prerun_hook(task, **kwargs):
        task.request.delivery_info['is_eager'] = False

    task_prerun.connect(task_prerun_hook)
    task_prerun.receivers = list(reversed(task_prerun.receivers))
    try:
        result = _test_foo_task(celery_eager, task_error)
    finally:
        task_prerun.disconnect(task_prerun_hook)

    span_server = tracer.recorder.get_spans()[0]
    assert span_server.parent_id is None
    assert_span(span_server, result, 'run', tags.SPAN_KIND_RPC_SERVER)
Пример #4
0
 def uninstall(self):
     task_prerun.disconnect(self.handle_task_prerun)
     task_postrun.disconnect(self.handle_task_postrun)
     task_failure.disconnect(self.process_failure_signal)
Пример #5
0
def uninstall():
    before_task_publish.disconnect(before_task_publish_callback)
    task_prerun.disconnect(task_prerun_callback)
    task_postrun.disconnect(task_postrun_callback)
Пример #6
0
 def _reset_patches(self):
     Task.apply_async = _task_apply_async
     before_task_publish.disconnect(before_task_publish_handler)
     task_prerun.disconnect(task_prerun_handler)
     task_success.disconnect(task_success_handler)
     task_failure.disconnect(task_failure_handler)
Пример #7
0
 def tearDown(self):
     super(CSVImportTestCase, self).tearDown()
     task_prerun.disconnect(self.task_sent_handler)
     from haystack import backend
     backend.SearchBackend.inmemory_db = None
Пример #8
0
def uninstall():
    task_prerun.disconnect(prerun_callback)
    task_postrun.disconnect(postrun_callback)
Пример #9
0
 def tearDown(self):
     super(CSVImportTestCase, self).tearDown()
     task_prerun.disconnect(self.task_sent_handler)
     from haystack import backend
     backend.SearchBackend.inmemory_db = None