Example #1
0
    def __init__(self,
                 api_key=None,
                 disable_trace=False,
                 disable_metric=True,
                 disable_log=True,
                 opts=None):
        super(TornadoWrapper, self).__init__(api_key, disable_trace,
                                             disable_metric, disable_log, opts)
        self.application_info_provider = GlobalApplicationInfoProvider()
        ExecutionContextManager.set_provider(TracingExecutionContextProvider())
        self.plugin_context = PluginContext(
            application_info=self.application_info_provider.
            get_application_info(),
            request_count=0,
            executor=tornado_executor,
            api_key=self.api_key)

        self.plugins = wrapper_utils.initialize_plugins(self.plugin_context,
                                                        disable_trace,
                                                        disable_metric,
                                                        disable_log,
                                                        config=self.config)

        web_wrapper_utils.update_application_info(
            self.application_info_provider,
            self.plugin_context.application_info,
            constants.ClassNames['TORNADO'])
Example #2
0
 def __init__(self,
              *,
              plugin_executor=None,
              api_key=None,
              disable_trace=False,
              disable_metric=True,
              disable_log=True,
              opts=None,
              **ignored):
     super(TestWrapper, self).__init__(api_key, disable_trace,
                                       disable_metric, disable_log, opts)
     ExecutionContextManager.set_provider(
         TracingExecutionContextProvider())  #TODO
     self.application_info_provider = GlobalApplicationInfoProvider()
     self._set_application_info("Foresight", "TestSuite", "TestSuite")
     self.plugin_context = PluginContext(
         application_info=self.application_info_provider.
         get_application_info(),
         request_count=0,
         executor=plugin_executor,
         api_key=self.api_key)
     max_test_log_count = ConfigProvider.get(
         config_names.THUNDRA_TEST_LOG_COUNT_MAX)
     self.config.log_config = LogConfig(
         sampler=MaxCountAwareSampler(max_test_log_count))
     self.plugins = wrapper_utils.initialize_plugins(self.plugin_context,
                                                     disable_trace,
                                                     disable_metric,
                                                     disable_log,
                                                     config=self.config)
     TestWrapper.__instance = self
Example #3
0
    def __init__(self, api_key=None, disable_trace=False, disable_metric=True, disable_log=True, opts=None):
        try:
            from django.conf import settings
            django_settings = getattr(settings, 'THUNDRA', None)
            if django_settings is not None:
                configure({'config': django_settings})
        except:
            pass

        super(DjangoWrapper, self).__init__(api_key, disable_trace, disable_metric, disable_log, opts)
        self.application_info_provider = GlobalApplicationInfoProvider()
        ExecutionContextManager.set_provider(TracingExecutionContextProvider())
        self.plugin_context = PluginContext(application_info=self.application_info_provider.get_application_info(),
                                            request_count=0,
                                            executor=django_executor,
                                            api_key=self.api_key)

        self.plugins = wrapper_utils.initialize_plugins(self.plugin_context, disable_trace, disable_metric, disable_log,
                                                        config=self.config)

        web_wrapper_utils.update_application_info(self.application_info_provider, self.plugin_context.application_info,
                                                  constants.ClassNames['DJANGO'])
Example #4
0
    def __init__(self,
                 api_key=None,
                 disable_trace=False,
                 disable_metric=True,
                 disable_log=True,
                 opts=None):
        super(LambdaWrapper, self).__init__(api_key, disable_trace,
                                            disable_metric, disable_log, opts)
        self.application_info_provider = GlobalApplicationInfoProvider(
            LambdaApplicationInfoProvider())
        self.plugin_context = PluginContext(
            application_info=self.application_info_provider.
            get_application_info(),
            request_count=0,
            executor=lambda_executor,
            api_key=self.api_key)

        ExecutionContextManager.set_provider(GlobalExecutionContextProvider())
        self.plugins = wrapper_utils.initialize_plugins(
            self.plugin_context, disable_trace, disable_metric, disable_log,
            self.config)

        self.timeout_margin = ConfigProvider.get(
            config_names.THUNDRA_LAMBDA_TIMEOUT_MARGIN,
            constants.DEFAULT_LAMBDA_TIMEOUT_MARGIN)

        if not ConfigProvider.get(
                config_names.THUNDRA_TRACE_INSTRUMENT_DISABLE):
            # Pass thundra instance to integration for wrapping handler wrappers
            handler_wrappers.patch_modules(self)

        self.ptvsd_imported = False
        if ConfigProvider.get(
                config_names.THUNDRA_LAMBDA_DEBUGGER_ENABLE,
                ConfigProvider.get(
                    config_names.THUNDRA_LAMBDA_DEBUGGER_AUTH_TOKEN)):
            self.initialize_debugger()