def test_dynamic_decorator(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [{
'module': 'tests.data.dynamic',
'function': 'foo'
}]
middleware._apply_dynamic_mappings()
from .data.dynamic import foo
mock = mock_data_collector()
with patch('silk.profiling.profiler.DataCollector',
return_value=mock) as mock_DataCollector:
foo() # Should be wrapped in a decorator
self.assertTrue(
mock_DataCollector.return_value.register_profile.call_count)
def test_dynamic_decorator(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [
{
'module': 'tests.data.dynamic',
'function': 'foo'
}
]
middleware._apply_dynamic_mappings()
from .data.dynamic import foo
mock = mock_data_collector()
with patch('silk.profiling.profiler.DataCollector', return_value=mock) as mock_DataCollector:
foo() # Should be wrapped in a decorator
self.assertTrue(mock_DataCollector.return_value.register_profile.call_count)
def test_dynamic_context_manager(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [{
'module': 'tests.data.dynamic',
'function': 'foo',
'start_line': 1,
'end_line': 2,
}]
middleware._apply_dynamic_mappings()
from .data.dynamic import foo
mock = mock_data_collector()
with patch('silk.profiling.profiler.DataCollector',
return_value=mock) as mock_DataCollector:
foo()
self.assertTrue(
mock_DataCollector.return_value.register_profile.call_count)
def test_dynamic_context_manager(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [
{
'module': 'tests.data.dynamic',
'function': 'foo',
'start_line': 1,
'end_line': 2,
}
]
middleware._apply_dynamic_mappings()
from .data.dynamic import foo
mock = mock_data_collector()
with patch('silk.profiling.profiler.DataCollector', return_value=mock) as mock_DataCollector:
foo()
self.assertTrue(mock_DataCollector.return_value.register_profile.call_count)
def test_no_mappings(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = []
middleware._apply_dynamic_mappings() # Just checking no crash
def test_no_mappings(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [
]
middleware._apply_dynamic_mappings() # Just checking no crash
