summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/diff/position_collection_spec.rb
blob: dd8434ab10b6d8b5da12d5dc0fbd92c498593ebc (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
# frozen_string_literal: true

require 'spec_helper'

describe Gitlab::Diff::PositionCollection do
  let(:merge_request) { build(:merge_request) }

  let(:text_position) do
    build(:text_diff_position, :added, diff_refs: diff_refs)
  end
  let(:folded_text_position) do
    build(:text_diff_position, diff_refs: diff_refs, old_line: 1, new_line: 1)
  end
  let(:image_position) do
    build(:image_diff_position, diff_refs: diff_refs)
  end

  let(:diff_refs) { merge_request.diff_refs }
  let(:invalid_position) { 'a position' }
  let(:head_sha) { merge_request.diff_head_sha }

  let(:collection) do
    described_class.new([text_position, folded_text_position, image_position, invalid_position], head_sha)
  end

  describe '#to_a' do
    it 'returns all positions that are Gitlab::Diff::Position' do
      expect(collection.to_a).to eq([text_position, folded_text_position, image_position])
    end
  end

  describe '#unfoldable' do
    it 'returns unfoldable diff positions' do
      expect(collection.unfoldable).to eq([folded_text_position])
    end

    context 'when given head_sha does not match with positions head_sha' do
      let(:head_sha) { 'unknown' }

      it 'returns no position' do
        expect(collection.unfoldable).to be_empty
      end
    end

    context 'when given head_sha is nil' do
      let(:head_sha) { nil }

      it 'returns unfoldable diff positions unfiltered by head_sha' do
        expect(collection.unfoldable).to eq([folded_text_position])
      end
    end
  end

  describe '#concat' do
    let(:new_text_position) do
      build(:text_diff_position, diff_refs: diff_refs, old_line: 1, new_line: 1)
    end

    it 'returns a Gitlab::Diff::Position' do
      expect(collection.concat([new_text_position])).to be_a(described_class)
    end

    it 'concatenates the new position to the collection' do
      collection.concat([new_text_position])

      expect(collection.to_a).to eq([text_position, folded_text_position, image_position, new_text_position])
    end
  end
end