diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-12-11 00:11:20 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-12-11 00:11:20 +0300 |
commit | 9fee94931194cf43bbf804356a75226b282227f7 (patch) | |
tree | df3303f3103683fa442dcebf86b1d79996365754 /spec/support/database | |
parent | ca5de528358c23e9cfcb0ff0f42c4106310c3811 (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'spec/support/database')
-rw-r--r-- | spec/support/database/query_analyzer.rb | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/spec/support/database/query_analyzer.rb b/spec/support/database/query_analyzer.rb index 85fa55f81ef..6d6627d54b9 100644 --- a/spec/support/database/query_analyzer.rb +++ b/spec/support/database/query_analyzer.rb @@ -4,11 +4,15 @@ # can be disabled selectively RSpec.configure do |config| - config.around do |example| + config.before do |example| if example.metadata.fetch(:query_analyzers, true) - ::Gitlab::Database::QueryAnalyzer.instance.within { example.run } - else - example.run + ::Gitlab::Database::QueryAnalyzer.instance.begin! + end + end + + config.after do |example| + if example.metadata.fetch(:query_analyzers, true) + ::Gitlab::Database::QueryAnalyzer.instance.end! end end end |