def test_set_hook(self): with mock.patch.object(osprofiler.sqlalchemy, 'add_tracing') as profiler: cfg.CONF.set_override('enabled', True, group='profiler') cfg.CONF.set_override('trace_sqlalchemy', True, group='profiler') engine_mock = mock.Mock() db_api.set_hook(engine_mock) profiler.assert_called_with(sqlalchemy, engine_mock, 'neutron.db')
def test_set_hook(self): with mock.patch.object(osprofiler.opts, 'is_trace_enabled', return_value=True),\ mock.patch.object(osprofiler.opts, 'is_db_trace_enabled', return_value=True): with mock.patch.object(osprofiler.sqlalchemy, 'add_tracing') as add_tracing: engine_mock = mock.Mock() db_api.set_hook(engine_mock) add_tracing.assert_called_with(sqlalchemy, engine_mock, 'neutron.db')