summaryrefslogtreecommitdiff
path: root/spec/lib/api/entities/merge_request_basic_spec.rb
blob: fe4c27b70ae806a61fdc501faeb4c02943c097b2 (plain)
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
62
63
64
65
66
67
68
69
70
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe ::API::Entities::MergeRequestBasic do
  let_it_be(:user) { create(:user) }
  let_it_be(:project) { create(:project, :public) }
  let_it_be(:merge_request) { create(:merge_request) }
  let_it_be(:labels) { create_list(:label, 3) }
  let_it_be(:merge_requests) { create_list(:labeled_merge_request, 10, :unique_branches, :with_diffs, labels: labels) }

  # This mimics the behavior of the `Grape::Entity` serializer
  def present(obj)
    described_class.new(obj).presented
  end

  context "with :with_api_entity_associations scope" do
    let(:scope) { MergeRequest.with_api_entity_associations }

    it "avoids N+1 queries" do
      query = scope.find(merge_request.id)

      control = ActiveRecord::QueryRecorder.new do
        present(query).to_json
      end

      # stub the `head_commit_sha` as it will trigger a
      # backward compatibility query that is out-of-scope
      # for this test whenever it is `nil`
      allow_any_instance_of(MergeRequestDiff).to receive(:head_commit_sha).and_return(Gitlab::Git::BLANK_SHA)

      query = scope.all
      batch = ActiveRecord::QueryRecorder.new do
        entities = query.map(&method(:present))

        entities.to_json
      end

      # The current threshold is 3 query per entity maximum.
      expect(batch.count).to be_within(3 * query.count).of(control.count)
    end
  end

  context 'reviewers' do
    context "when merge_request_reviewers FF is enabled" do
      before do
        stub_feature_flags(merge_request_reviewers: true)
        merge_request.reviewers = [user]
      end

      it 'includes assigned reviewers' do
        result = Gitlab::Json.parse(present(merge_request).to_json)

        expect(result['reviewers'][0]['username']).to eq user.username
      end
    end

    context "when merge_request_reviewers FF is disabled" do
      before do
        stub_feature_flags(merge_request_reviewers: false)
      end

      it 'does not include reviewers' do
        result = Gitlab::Json.parse(present(merge_request).to_json)

        expect(result.keys).not_to include('reviewers')
      end
    end
  end
end