def test_dynamic_decorator(self):
middleware = SilkyMiddleware()
SilkyConfig().SILKY_DYNAMIC_PROFILING = [
{
'module': 'silk.tests.data.dynamic',
'function': 'foo'
}
]
middleware._apply_dynamic_mappings()
from silk.tests.data.dynamic import foo
mock = Mock()
mock.queries = []
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': 'silk.tests.data.dynamic',
'function': 'foo',
'start_line': 1,
'end_line': 2,
}
]
middleware._apply_dynamic_mappings()
from silk.tests.data.dynamic import foo
mock = Mock()
mock.queries = []
with patch('silk.profiling.profiler.DataCollector', return_value=mock) as mock_DataCollector:
foo()
self.assertTrue(mock_DataCollector.return_value.register_profile.call_count)
