BigW Consortium Gitlab

query_matcher.rb 1.02 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
RSpec::Matchers.define :make_queries_matching do |matcher, expected_count = nil|
  supports_block_expectations

  match do |block|
    @counter = query_count(matcher, &block)
    if expected_count
      @counter.count == expected_count
    else
      @counter.count > 0
    end
  end

  failure_message_when_negated do |_|
    if expected_count
      "expected #{matcher} not to match #{expected_count} queries, got #{@counter.count} matches:\n\n#{@counter.inspect}"
    else
      "expected #{matcher} not to match any query, got #{@counter.count} matches:\n\n#{@counter.inspect}"
    end
  end

  failure_message do |_|
    if expected_count
      "expected #{matcher} to match #{expected_count} queries, got #{@counter.count} matches:\n\n#{@counter.inspect}"
    else
      "expected #{matcher} to match at least one query, got #{@counter.count} matches:\n\n#{@counter.inspect}"
    end
  end

  def query_count(regex, &block)
    @recorder = ActiveRecord::QueryRecorder.new(&block).log
31
    @recorder.select { |q| q.match(regex) }
32 33
  end
end