File: notes_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 (61 lines) | stat: -rw-r--r-- 1,728 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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Preloaders::Projects::NotesPreloader, :request_store, feature_category: :team_planning do
  let_it_be(:project) { create(:project, :repository, :public) }
  let_it_be(:issue) { create(:issue, project: project) }

  it 'preloads author access level and contributor status' do
    developer1 = create(:user, developer_of: project)
    developer2 = create(:user, developer_of: project)
    contributor1 = create(:user)
    contributor2 = create(:user)
    contributor3 = create(:user)

    create_merge_request_for(contributor1)
    create_note_for(developer1)
    create_note_for(contributor1)

    notes = issue.notes.preload(:author, :project).to_a

    control = ActiveRecord::QueryRecorder.new do
      preload_and_fetch_attributes(notes, developer1)
    end

    create_merge_request_for(contributor2)
    create_merge_request_for(contributor3)
    create_note_for(developer2)
    create_note_for(contributor2)
    create_note_for(developer1)
    create_note_for(contributor3)
    issue.reload

    notes = issue.notes.preload(:author, :project).to_a

    expect do
      preload_and_fetch_attributes(notes, developer1)
    end.not_to exceed_query_limit(control)
  end

  def create_note_for(user)
    create(:note, project: project, noteable: issue, author: user)
  end

  def create_merge_request_for(user)
    create(
      :merge_request,
      :merged,
      :simple,
      source_project: project,
      author: user,
      target_branch: project.default_branch.to_s
    )
  end

  def preload_and_fetch_attributes(notes, user)
    described_class.new(project, user).call(notes)

    notes.each { |n| n.contributor? && n.human_max_access }
  end
end