def test_with_origin(self):
with settings(origin='unit_test'):
go()
for metric in self._query_backend(go.fqn)['hits']['hits']:
self.assertEqual(metric['_source']['origin'], 'unit_test')
def test_duration_field(self):
with settings(duration_field='my_duration'):
go()
for metric in self._query_backend(go.fqn):
self.assertTrue('my_duration' in metric)
def test_with_origin(self):
with settings(origin='unit_test'):
go()
for metric in self._query_backend():
assert metric.value['origin'] == 'unit_test'
def test_with_origin(self):
with settings(origin='unit_test'):
go()
for metric in self._query_backend(go.fqn)['hits']['hits']:
self.assertEqual(metric['_source']['origin'], 'unit_test')
def test_old_hooks(self):
def old_hook(response, exception, metric, func_args, func_kwargs):
return dict(old_hook='old_hook')
def new_hook(response, exception, metric, func, func_args,
func_kwargs):
return dict(new_hook='new_hook',
attribute_from_class=func.__self__.attr)
class Class:
attr = 1
@time_execution
def method(self):
return True
with settings(backends=[CollectorBackend()],
hooks=[old_hook, new_hook]):
collector = settings.backends[0]
Class().method()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][get_fqn(Class().method)]
assert "old_hook" in metadata
assert "new_hook" in metadata
assert metadata["attribute_from_class"] == Class.attr
def test_duration_field(self):
with settings(duration_field="my_duration"):
go()
for metric in self._query_backend(go.fqn):
self.assertTrue("my_duration" in metric)
def test_with_origin(self):
with settings(origin="unit_test"):
go()
for metric in self._query_backend(go.get_fqn())["hits"]["hits"]:
self.assertEqual(metric["_source"]["origin"], "unit_test")
def test_with_origin(self):
with settings(origin="unit_test"):
go()
for metric in self._query_backend(go.fqn):
self.assertEqual(metric["origin"], "unit_test")
def test_hook_func_kwargs(self):
param = 'foo'
@time_execution
def go(param1):
return '200 OK'
def hook(response, exception, metric, func_args, func_kwargs):
self.assertEqual(func_kwargs['param1'], param)
with settings(hooks=[hook]):
go(param1=param)
def test_hook_func_kwargs(self):
param = 'foo'
@time_execution
def go(param1):
return '200 OK'
def hook(response, exception, metric, func_args, func_kwargs):
assert func_kwargs['param1'] == param
with settings(hooks=[hook]):
go(param1=param)
def test_pipeline_present(self, mocked_index):
backend = ElasticsearchBackend(ELASTICSEARCH_HOST,
index="pipelinetest",
pipeline="custom-pipeline")
with settings(backends=[backend]):
go()
assert "pipeline" in mocked_index.call_args.kwargs
assert mocked_index.call_args.kwargs[
"pipeline"] == "custom-pipeline"
ElasticTestMixin._clear(backend)
def test_hook_func_kwargs(self):
param = "foo"
@time_execution
def go(param1):
return "200 OK"
def hook(response, exception, metric, func, func_args, func_kwargs):
assert func_kwargs["param1"] == param
with settings(hooks=[hook]):
go(param1=param)
def test_hook_func_args(self):
param = 'foo'
@time_execution
def go(param1):
return '200 OK'
def hook(response, exception, metric, func_args, func_kwargs):
self.assertEqual(func_args[0], param)
with settings(hooks=[hook]):
go(param)
def test_hook(self):
def test_args(**kwargs):
self.assertIn('response', kwargs)
self.assertIn('exception', kwargs)
self.assertIn('metric', kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key='test value')
with settings(hooks=[test_args, test_metadata]):
go()
for metric in self._query_backend(go.fqn)['hits']['hits']:
self.assertEqual(metric['_source']['test_key'], 'test value')
def test_hook(self):
def test_args(**kwargs):
self.assertIn("response", kwargs)
self.assertIn("exception", kwargs)
self.assertIn("metric", kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key="test value")
with settings(hooks=[test_args, test_metadata]):
go()
for metric in self._query_backend(go.get_fqn())["hits"]["hits"]:
self.assertEqual(metric["_source"]["test_key"], "test value")
def test_hook(self):
def test_args(**kwargs):
self.assertIn('response', kwargs)
self.assertIn('exception', kwargs)
self.assertIn('metric', kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key='test value')
with settings(hooks=[test_args, test_metadata]):
go()
for metric in self._query_backend():
assert metric.value['test_key'] == 'test value'
def test_hook(self):
def test_args(**kwargs):
self.assertIn('response', kwargs)
self.assertIn('exception', kwargs)
self.assertIn('metric', kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key='test value')
with settings(hooks=[test_args, test_metadata]):
go()
for metric in self._query_backend(go.fqn)['hits']['hits']:
self.assertEqual(metric['_source']['test_key'], 'test value')
def test_hook(self):
def test_args(**kwargs):
self.assertIn("response", kwargs)
self.assertIn("exception", kwargs)
self.assertIn("metric", kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key="test value")
with settings(hooks=[test_args, test_metadata]):
go()
for metric in self._query_backend(go.fqn):
self.assertEqual(metric["test_key"], "test value")
def test_hook(self):
def test_args(**kwargs):
self.assertIn('response', kwargs)
self.assertIn('exception', kwargs)
self.assertIn('metric', kwargs)
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key='test value')
def asserts(name, **data):
self.assertEqual(data['test_key'], 'test value')
with settings(backends=[AssertBackend(asserts)],
hooks=[test_args, test_metadata]):
go()
def test_hook(self):
def test_args(**kwargs):
assert 'response' in kwargs
assert 'exception' in kwargs
assert 'metric' in kwargs
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key='test value')
def asserts(name, **data):
assert data['test_key'] == 'test value'
with settings(backends=[AssertBackend(asserts)],
hooks=[test_args, test_metadata]):
go()
def test_hook(self):
def test_args(**kwargs):
assert "response" in kwargs
assert "exception" in kwargs
assert "metric" in kwargs
return dict()
def test_metadata(*args, **kwargs):
return dict(test_key="test value")
def asserts(name, **data):
assert data["test_key"] == "test value"
with settings(backends=[AssertBackend(asserts)],
hooks=[test_args, test_metadata]):
go()
def test_hook_exception(self):
message = 'exception message'
class TimeExecutionException(Exception):
pass
def exception_hook(exception, **kwargs):
self.assertIsInstance(exception, TimeExecutionException)
return dict(exception_message=str(exception))
def asserts(name, **data):
self.assertEqual(data['exception_message'], message)
@time_execution
def go():
raise TimeExecutionException(message)
with settings(backends=[AssertBackend(asserts)],
hooks=[exception_hook]):
with self.assertRaises(TimeExecutionException):
go()
def test_hook_exception(self):
message = 'exception message'
class TimeExecutionException(Exception):
pass
def exception_hook(exception, **kwargs):
assert isinstance(exception, (TimeExecutionException,))
return dict(exception_message=str(exception))
def asserts(name, **data):
assert data['exception_message'] == message
@time_execution
def go():
raise TimeExecutionException(message)
with settings(backends=[AssertBackend(asserts)],
hooks=[exception_hook]):
with pytest.raises(TimeExecutionException):
go()
def test_hook_exception_args(self):
message = 'exception message'
class TimeExecutionException(Exception):
def __init__(self, msg, required):
super(TimeExecutionException, self).__init__(msg)
def exception_hook(exception, **kwargs):
assert isinstance(exception, (TimeExecutionException,))
return dict(exception_message=str(exception))
def asserts(name, **data):
assert data['exception_message'] == message
@time_execution
def go():
raise TimeExecutionException(message, True)
with settings(backends=[AssertBackend(asserts)],
hooks=[exception_hook]):
with pytest.raises(TimeExecutionException):
go()
def test_duration_field(self):
with settings(duration_field="my_duration"):
go()
for metric in self._query_backend(go.get_fqn())["hits"]["hits"]:
self.assertTrue("my_duration" in metric["_source"])
def test_duration_field(self):
with settings(duration_field='my_duration'):
go()
for metric in self._query_backend(go.fqn)['hits']['hits']:
self.assertTrue('my_duration' in metric['_source'])
def test_duration_field(self):
with settings(duration_field='my_duration'):
go()
for metric in self._query_backend():
assert 'my_duration' in metric.value
def test_custom_hook(self):
with settings(backends=[CollectorBackend()], hooks=[global_hook]):
collector = settings.backends[0]
@time_execution(extra_hooks=[local_hook])
def func_local_hook(*args, **kwargs):
return True
func_local_hook()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][func_local_hook.get_fqn()]
assert metadata["local_hook_key"] == "local hook value"
assert metadata["global_hook_key"] == "global hook value"
collector.clean()
@time_execution(extra_hooks=[local_hook], disable_default_hooks=True)
def func_local_hook_disable_default_hooks(*args, **kwargs):
return True
func_local_hook_disable_default_hooks()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][func_local_hook_disable_default_hooks.get_fqn()]
assert metadata["local_hook_key"] == "local hook value"
assert "global_hook_key" not in metadata
collector.clean()
@time_execution
def func_global_hook(*args, **kwargs):
return True
func_global_hook()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][func_global_hook.get_fqn()]
assert metadata["global_hook_key"] == "global hook value"
assert "local_hook_key" not in metadata
collector.clean()
class ClassNoHooks:
@time_execution(extra_hooks=[local_hook])
def method_local_hook(self):
return True
@time_execution
def method_global_hook(self):
return True
ClassNoHooks().method_local_hook()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][ClassNoHooks().method_local_hook.get_fqn()]
assert metadata["global_hook_key"] == "global hook value"
assert metadata["local_hook_key"] == "local hook value"
collector.clean()
ClassNoHooks().method_global_hook()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][ClassNoHooks().method_global_hook.get_fqn()]
assert metadata["global_hook_key"] == "global hook value"
assert "local_hook_key" not in metadata
collector.clean()
@time_execution(extra_hooks=[local_hook])
class ClassLocalHook:
def method(self):
return True
@time_execution
def method_global_hook(self):
return True
ClassLocalHook().method()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][get_fqn(ClassLocalHook)]
assert metadata["global_hook_key"] == "global hook value"
assert metadata["local_hook_key"] == "local hook value"
collector.clean()
ClassLocalHook().method_global_hook()
assert len(collector.metrics) == 2
metadata_class = collector.metrics[0][get_fqn(ClassLocalHook)]
assert metadata_class["local_hook_key"] == "local hook value"
assert metadata_class["global_hook_key"] == "global hook value"
metadata = collector.metrics[1][ClassLocalHook().method_global_hook.get_fqn()]
assert metadata["global_hook_key"] == "global hook value"
assert "local_hook_key" not in metadata
collector.clean()
@time_execution
class ClassGlobalHook:
def method(self):
return True
@time_execution(extra_hooks=[local_hook])
def method_local_hook(self):
return True
ClassGlobalHook().method()
assert len(collector.metrics) == 1
metadata = collector.metrics[0][get_fqn(ClassGlobalHook)]
assert metadata["global_hook_key"] == "global hook value"
assert "local_hook_key" not in metadata
collector.clean()
ClassGlobalHook().method_local_hook()
assert len(collector.metrics) == 2
metadata_class = collector.metrics[0][get_fqn(ClassGlobalHook)]
assert metadata_class["global_hook_key"] == "global hook value"
assert "local_hook_key" not in metadata_class
metadata = collector.metrics[1][ClassGlobalHook().method_local_hook.get_fqn()]
assert metadata["global_hook_key"] == "global hook value"
assert metadata["local_hook_key"] == "local hook value"
collector.clean()