Merge branch 'sidekiq-query-ee' into 'master'

[EE] Disable query limit counter in sidekiq

Closes #7087

See merge request gitlab-org/gitlab-ee!7204

spec/support/sidekiq.rb

 require 'sidekiq/testing/inline'
 
+# If Sidekiq::Testing.inline! is used, SQL transactions done inside
+# Sidekiq worker are included in the SQL query limit (in a real
+# deployment sidekiq worker is executed separately). To avoid
+# increasing SQL limit counter, the request is marked as whitelisted
+# during Sidekiq block
+class DisableQueryLimit
+  def call(worker_instance, msg, queue)
+    transaction = Gitlab::QueryLimiting::Transaction.current
+
+    if !transaction.respond_to?(:whitelisted) || transaction.whitelisted
+      yield
+    else
+      transaction.whitelisted = true
+      yield
+      transaction.whitelisted = false
+    end
+  end
+end
+
 Sidekiq::Testing.server_middleware do |chain|
   chain.add Gitlab::SidekiqStatus::ServerMiddleware
+  chain.add DisableQueryLimit
 end
 
 RSpec.configure do |config|