def test_http_config(self): config = Config() config._add("django", dict()) assert config.django.trace_query_string is None config.http.trace_query_string = True assert config.http.trace_query_string is True assert config.django.trace_query_string is True # Integration usage config = Config() config._add("django", dict()) config.django.http.trace_query_string = True assert config.http.trace_query_string is None assert config.django.trace_query_string is True assert config.django.http.trace_query_string is True
class GlobalConfigTestCase(TestCase): """Test the `Configuration` class that stores integration settings""" def setUp(self): self.config = Config() def test_registration(self): # ensure an integration can register a new list of settings settings = { 'distributed_tracing': True, } self.config._add('requests', settings) ok_(self.config.requests['distributed_tracing'] is True) def test_settings_copy(self): # ensure that once an integration is registered, a copy # of the settings is stored to avoid side-effects experimental = { 'request_enqueuing': True, } settings = { 'distributed_tracing': True, 'experimental': experimental, } self.config._add('requests', settings) settings['distributed_tracing'] = False experimental['request_enqueuing'] = False ok_(self.config.requests['distributed_tracing'] is True) ok_(self.config.requests['experimental']['request_enqueuing'] is True) def test_missing_integration(self): # ensure a meaningful exception is raised when an integration # that is not available is retrieved in the configuration # object with assert_raises(ConfigException) as e: self.config.new_integration['some_key'] ok_(isinstance(e.exception, ConfigException)) def test_global_configuration(self): # ensure a global configuration is available in the `ddtrace` module ok_(isinstance(global_config, Config))
def config(): c = Config() c._add("myint", dict()) return c
class GlobalConfigTestCase(TestCase): """Test the `Configuration` class that stores integration settings""" def setUp(self): self.config = Config() self.tracer = get_dummy_tracer() def test_registration(self): # ensure an integration can register a new list of settings settings = { "distributed_tracing": True, } self.config._add("requests", settings) assert self.config.requests["distributed_tracing"] is True def test_settings_copy(self): # ensure that once an integration is registered, a copy # of the settings is stored to avoid side-effects experimental = { "request_enqueuing": True, } settings = { "distributed_tracing": True, "experimental": experimental, } self.config._add("requests", settings) settings["distributed_tracing"] = False experimental["request_enqueuing"] = False assert self.config.requests["distributed_tracing"] is True assert self.config.requests["experimental"]["request_enqueuing"] is True def test_missing_integration_key(self): # ensure a meaningful exception is raised when an integration # that is not available is retrieved in the configuration # object with pytest.raises(KeyError) as e: self.config.new_integration["some_key"] assert isinstance(e.value, KeyError) def test_global_configuration(self): # ensure a global configuration is available in the `ddtrace` module assert isinstance(global_config, Config) def test_settings_merge(self): """ When calling `config._add()` when existing settings exist we do not overwrite the existing settings """ self.config.requests["split_by_domain"] = True self.config._add("requests", dict(split_by_domain=False)) assert self.config.requests["split_by_domain"] is True def test_settings_overwrite(self): """ When calling `config._add(..., merge=False)` when existing settings exist we overwrite the existing settings """ self.config.requests["split_by_domain"] = True self.config._add("requests", dict(split_by_domain=False), merge=False) assert self.config.requests["split_by_domain"] is False def test_settings_merge_deep(self): """ When calling `config._add()` when existing "deep" settings exist we do not overwrite the existing settings """ self.config.requests["a"] = dict( b=dict( c=True, ), ) self.config._add( "requests", dict( a=dict( b=dict( c=False, d=True, ), ), ), ) assert self.config.requests["a"]["b"]["c"] is True assert self.config.requests["a"]["b"]["d"] is True def test_settings_hook(self): """ When calling `Hooks.emit()` When there is a hook registered we call the hook as expected """ # Setup our hook @self.config.web.hooks.on("request") def on_web_request(span): span.set_tag("web.request", "/") # Create our span span = self.tracer.start_span("web.request") assert "web.request" not in span.meta # Emit the span self.config.web.hooks.emit("request", span) # Assert we updated the span as expected assert span.get_tag("web.request") == "/" def test_settings_hook_args(self): """ When calling `Hooks.emit()` with arguments When there is a hook registered we call the hook as expected """ # Setup our hook @self.config.web.hooks.on("request") def on_web_request(span, request, response): span.set_tag("web.request", request) span.set_tag("web.response", response) # Create our span span = self.tracer.start_span("web.request") assert "web.request" not in span.meta # Emit the span # DEV: The actual values don't matter, we just want to test args + kwargs usage self.config.web.hooks.emit("request", span, "request", response="response") # Assert we updated the span as expected assert span.get_tag("web.request") == "request" assert span.get_tag("web.response") == "response" def test_settings_hook_args_failure(self): """ When calling `Hooks.emit()` with arguments When there is a hook registered that is missing parameters we do not raise an exception """ # Setup our hook # DEV: We are missing the required "response" argument @self.config.web.hooks.on("request") def on_web_request(span, request): span.set_tag("web.request", request) # Create our span span = self.tracer.start_span("web.request") assert "web.request" not in span.meta # Emit the span # DEV: This also asserts that no exception was raised self.config.web.hooks.emit("request", span, "request", response="response") # Assert we did not update the span assert "web.request" not in span.meta def test_settings_multiple_hooks(self): """ When calling `Hooks.emit()` When there are multiple hooks registered we do not raise an exception """ # Setup our hooks @self.config.web.hooks.on("request") def on_web_request(span): span.set_tag("web.request", "/") @self.config.web.hooks.on("request") def on_web_request2(span): span.set_tag("web.status", 200) @self.config.web.hooks.on("request") def on_web_request3(span): span.set_tag("web.method", "GET") # Create our span span = self.tracer.start_span("web.request") assert "web.request" not in span.meta assert "web.status" not in span.metrics assert "web.method" not in span.meta # Emit the span self.config.web.hooks.emit("request", span) # Assert we updated the span as expected assert span.get_tag("web.request") == "/" assert span.get_metric("web.status") == 200 assert span.get_tag("web.method") == "GET" def test_settings_hook_failure(self): """ When calling `Hooks.emit()` When the hook raises an exception we do not raise an exception """ # Setup our failing hook on_web_request = mock.Mock(side_effect=Exception) self.config.web.hooks.register("request")(on_web_request) # Create our span span = self.tracer.start_span("web.request") # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks.emit("request", span) on_web_request.assert_called() def test_settings_no_hook(self): """ When calling `Hooks.emit()` When no hook is registered we do not raise an exception """ # Create our span span = self.tracer.start_span("web.request") # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks.emit("request", span) def test_settings_no_span(self): """ When calling `Hooks.emit()` When no span is provided we do not raise an exception """ # Setup our hooks @self.config.web.hooks.on("request") def on_web_request(span): span.set_tag("web.request", "/") # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks.emit("request", None) def test_dd_version(self): c = Config() assert c.version is None with override_env(dict(DD_VERSION="1.2.3")): c = Config() assert c.version == "1.2.3" c.version = "4.5.6" assert c.version == "4.5.6" def test_dd_env(self): c = Config() assert c.env is None with override_env(dict(DD_ENV="prod")): c = Config() assert c.env == "prod" # manual override still possible c.env = "prod-staging" assert c.env == "prod-staging" # between DD_ENV and DATADOG_ENV, the former takes priority with override_env(dict(DATADOG_ENV="prod", DD_ENV="prod-staging")): c = Config() assert c.env == "prod-staging"
class GlobalConfigTestCase(TestCase): """Test the `Configuration` class that stores integration settings""" def setUp(self): self.config = Config() self.tracer = get_dummy_tracer() def test_registration(self): # ensure an integration can register a new list of settings settings = { 'distributed_tracing': True, } self.config._add('requests', settings) assert self.config.requests['distributed_tracing'] is True def test_settings_copy(self): # ensure that once an integration is registered, a copy # of the settings is stored to avoid side-effects experimental = { 'request_enqueuing': True, } settings = { 'distributed_tracing': True, 'experimental': experimental, } self.config._add('requests', settings) settings['distributed_tracing'] = False experimental['request_enqueuing'] = False assert self.config.requests['distributed_tracing'] is True assert self.config.requests['experimental']['request_enqueuing'] is True def test_missing_integration_key(self): # ensure a meaningful exception is raised when an integration # that is not available is retrieved in the configuration # object with pytest.raises(KeyError) as e: self.config.new_integration['some_key'] assert isinstance(e.value, KeyError) def test_global_configuration(self): # ensure a global configuration is available in the `ddtrace` module assert isinstance(global_config, Config) def test_settings_merge(self): """ When calling `config._add()` when existing settings exist we do not overwrite the existing settings """ self.config.requests['split_by_domain'] = True self.config._add('requests', dict(split_by_domain=False)) assert self.config.requests['split_by_domain'] is True def test_settings_overwrite(self): """ When calling `config._add(..., merge=False)` when existing settings exist we overwrite the existing settings """ self.config.requests['split_by_domain'] = True self.config._add('requests', dict(split_by_domain=False), merge=False) assert self.config.requests['split_by_domain'] is False def test_settings_merge_deep(self): """ When calling `config._add()` when existing "deep" settings exist we do not overwrite the existing settings """ self.config.requests['a'] = dict( b=dict( c=True, ), ) self.config._add('requests', dict( a=dict( b=dict( c=False, d=True, ), ), )) assert self.config.requests['a']['b']['c'] is True assert self.config.requests['a']['b']['d'] is True def test_settings_hook(self): """ When calling `Hooks._emit()` When there is a hook registered we call the hook as expected """ # Setup our hook @self.config.web.hooks.on('request') def on_web_request(span): span.set_tag('web.request', '/') # Create our span span = self.tracer.start_span('web.request') assert 'web.request' not in span.meta # Emit the span self.config.web.hooks._emit('request', span) # Assert we updated the span as expected assert span.get_tag('web.request') == '/' def test_settings_hook_args(self): """ When calling `Hooks._emit()` with arguments When there is a hook registered we call the hook as expected """ # Setup our hook @self.config.web.hooks.on('request') def on_web_request(span, request, response): span.set_tag('web.request', request) span.set_tag('web.response', response) # Create our span span = self.tracer.start_span('web.request') assert 'web.request' not in span.meta # Emit the span # DEV: The actual values don't matter, we just want to test args + kwargs usage self.config.web.hooks._emit('request', span, 'request', response='response') # Assert we updated the span as expected assert span.get_tag('web.request') == 'request' assert span.get_tag('web.response') == 'response' def test_settings_hook_args_failure(self): """ When calling `Hooks._emit()` with arguments When there is a hook registered that is missing parameters we do not raise an exception """ # Setup our hook # DEV: We are missing the required "response" argument @self.config.web.hooks.on('request') def on_web_request(span, request): span.set_tag('web.request', request) # Create our span span = self.tracer.start_span('web.request') assert 'web.request' not in span.meta # Emit the span # DEV: This also asserts that no exception was raised self.config.web.hooks._emit('request', span, 'request', response='response') # Assert we did not update the span assert 'web.request' not in span.meta def test_settings_multiple_hooks(self): """ When calling `Hooks._emit()` When there are multiple hooks registered we do not raise an exception """ # Setup our hooks @self.config.web.hooks.on('request') def on_web_request(span): span.set_tag('web.request', '/') @self.config.web.hooks.on('request') def on_web_request2(span): span.set_tag('web.status', 200) @self.config.web.hooks.on('request') def on_web_request3(span): span.set_tag('web.method', 'GET') # Create our span span = self.tracer.start_span('web.request') assert 'web.request' not in span.meta assert 'web.status' not in span.metrics assert 'web.method' not in span.meta # Emit the span self.config.web.hooks._emit('request', span) # Assert we updated the span as expected assert span.get_tag('web.request') == '/' assert span.get_metric('web.status') == 200 assert span.get_tag('web.method') == 'GET' def test_settings_hook_failure(self): """ When calling `Hooks._emit()` When the hook raises an exception we do not raise an exception """ # Setup our failing hook on_web_request = mock.Mock(side_effect=Exception) self.config.web.hooks.register('request')(on_web_request) # Create our span span = self.tracer.start_span('web.request') # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks._emit('request', span) on_web_request.assert_called() def test_settings_no_hook(self): """ When calling `Hooks._emit()` When no hook is registered we do not raise an exception """ # Create our span span = self.tracer.start_span('web.request') # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks._emit('request', span) def test_settings_no_span(self): """ When calling `Hooks._emit()` When no span is provided we do not raise an exception """ # Setup our hooks @self.config.web.hooks.on('request') def on_web_request(span): span.set_tag('web.request', '/') # Emit the span # DEV: This is the test, to ensure no exceptions are raised self.config.web.hooks._emit('request', None)
class GlobalConfigTestCase(TestCase): """Test the `Configuration` class that stores integration settings""" def setUp(self): self.config = Config() def test_registration(self): # ensure an integration can register a new list of settings settings = { 'distributed_tracing': True, } self.config._add('requests', settings) ok_(self.config.requests['distributed_tracing'] is True) def test_settings_copy(self): # ensure that once an integration is registered, a copy # of the settings is stored to avoid side-effects experimental = { 'request_enqueuing': True, } settings = { 'distributed_tracing': True, 'experimental': experimental, } self.config._add('requests', settings) settings['distributed_tracing'] = False experimental['request_enqueuing'] = False ok_(self.config.requests['distributed_tracing'] is True) ok_(self.config.requests['experimental']['request_enqueuing'] is True) def test_missing_integration(self): # ensure a meaningful exception is raised when an integration # that is not available is retrieved in the configuration # object with assert_raises(ConfigException) as e: self.config.new_integration['some_key'] ok_(isinstance(e.exception, ConfigException)) def test_global_configuration(self): # ensure a global configuration is available in the `ddtrace` module ok_(isinstance(global_config, Config)) def test_settings_merge(self): """ When calling `config._add()` when existing settings exist we do not overwrite the existing settings """ self.config.requests['split_by_domain'] = True self.config._add('requests', dict(split_by_domain=False)) eq_(self.config.requests['split_by_domain'], True) def test_settings_overwrite(self): """ When calling `config._add(..., merge=False)` when existing settings exist we overwrite the existing settings """ self.config.requests['split_by_domain'] = True self.config._add('requests', dict(split_by_domain=False), merge=False) eq_(self.config.requests['split_by_domain'], False) def test_settings_merge_deep(self): """ When calling `config._add()` when existing "deep" settings exist we do not overwrite the existing settings """ self.config.requests['a'] = dict(b=dict(c=True, ), ) self.config._add('requests', dict(a=dict(b=dict( c=False, d=True, ), ), )) eq_(self.config.requests['a']['b']['c'], True) eq_(self.config.requests['a']['b']['d'], True)