diff options
author | James Edwards-Jones <jedwardsjones@gitlab.com> | 2016-12-20 14:48:04 +0000 |
---|---|---|
committer | James Edwards-Jones <jedwardsjones@gitlab.com> | 2016-12-20 16:59:48 +0000 |
commit | 9c623e3e5d7434f2e30f7c389d13e5af4ede770a (patch) | |
tree | 03e905658ce5edeb8fb7d3b97be7e4ead5da835c /spec/support | |
parent | a2f57f23616bc1ab7547e5a7c517cdd49b172bd8 (diff) | |
download | gitlab-ce-9c623e3e5d7434f2e30f7c389d13e5af4ede770a.tar.gz |
Added QueryRecorder to test N+1 fix on Milestone#showjej-fix-n+1-queries-milestones-show
Diffstat (limited to 'spec/support')
-rw-r--r-- | spec/support/query_recorder.rb | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/spec/support/query_recorder.rb b/spec/support/query_recorder.rb new file mode 100644 index 00000000000..e40d5ebd9a8 --- /dev/null +++ b/spec/support/query_recorder.rb @@ -0,0 +1,40 @@ +module ActiveRecord + class QueryRecorder + attr_reader :log + + def initialize(&block) + @log = [] + ActiveSupport::Notifications.subscribed(method(:callback), 'sql.active_record', &block) + end + + def callback(name, start, finish, message_id, values) + return if %w(CACHE SCHEMA).include?(values[:name]) + @log << values[:sql] + end + + def count + @log.count + end + + def log_message + @log.join("\n\n") + end + end +end + +RSpec::Matchers.define :exceed_query_limit do |expected| + supports_block_expectations + + match do |block| + query_count(&block) > expected + end + + failure_message_when_negated do |actual| + "Expected a maximum of #{expected} queries, got #{@recorder.count}:\n\n#{@recorder.log_message}" + end + + def query_count(&block) + @recorder = ActiveRecord::QueryRecorder.new(&block) + @recorder.count + end +end |