File: merge_request_diff_preloader_spec.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (29 lines) | stat: -rw-r--r-- 966 bytes parent folder | download
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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Preloaders::MergeRequestDiffPreloader do
  let_it_be(:merge_request_1) { create(:merge_request) }
  let_it_be(:merge_request_2) { create(:merge_request) }
  let_it_be(:merge_request_3) { create(:merge_request, :skip_diff_creation) }

  let(:merge_requests) { [merge_request_1, merge_request_2, merge_request_3] }

  def trigger(merge_requests)
    Array(merge_requests).each(&:merge_request_diff)
  end

  def merge_requests_with_preloaded_diff
    described_class.new(MergeRequest.where(id: merge_requests.map(&:id)).to_a).preload_all
  end

  it 'does not trigger N+1 queries' do
    # warmup
    trigger(merge_requests_with_preloaded_diff)

    first_merge_request = merge_requests_with_preloaded_diff.first
    clean_merge_requests = merge_requests_with_preloaded_diff

    expect { trigger(clean_merge_requests) }.to issue_same_number_of_queries_as { trigger(first_merge_request) }
  end
end