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)