summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/discussions_diff/file_collection_spec.rb
blob: 6ef1e41450fb9fd300a33229296fc3662b7f89a8 (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
71
72
73
74
75
76
77
78
# frozen_string_literal: true

require 'spec_helper'

describe Gitlab::DiscussionsDiff::FileCollection do
  let(:merge_request) { create(:merge_request) }
  let!(:diff_note_a) { create(:diff_note_on_merge_request, project: merge_request.project, noteable: merge_request) }
  let!(:diff_note_b) { create(:diff_note_on_merge_request, project: merge_request.project, noteable: merge_request) }
  let(:note_diff_file_a) { diff_note_a.note_diff_file }
  let(:note_diff_file_b) { diff_note_b.note_diff_file }

  subject { described_class.new([note_diff_file_a, note_diff_file_b]) }

  describe '#load_highlight', :clean_gitlab_redis_shared_state do
    it 'writes uncached diffs highlight' do
      file_a_caching_content = diff_note_a.diff_file.highlighted_diff_lines.map(&:to_hash)
      file_b_caching_content = diff_note_b.diff_file.highlighted_diff_lines.map(&:to_hash)

      expect(Gitlab::DiscussionsDiff::HighlightCache)
        .to receive(:write_multiple)
        .with({ note_diff_file_a.id => file_a_caching_content,
                note_diff_file_b.id => file_b_caching_content })
        .and_call_original

      subject.load_highlight
    end

    it 'does not write cache for already cached file' do
      file_a_caching_content = diff_note_a.diff_file.highlighted_diff_lines.map(&:to_hash)
      Gitlab::DiscussionsDiff::HighlightCache
        .write_multiple({ note_diff_file_a.id => file_a_caching_content })

      file_b_caching_content = diff_note_b.diff_file.highlighted_diff_lines.map(&:to_hash)

      expect(Gitlab::DiscussionsDiff::HighlightCache)
        .to receive(:write_multiple)
        .with({ note_diff_file_b.id => file_b_caching_content })
        .and_call_original

      subject.load_highlight
    end

    it 'does not write cache for resolved notes' do
      diff_note_a.update_column(:resolved_at, Time.now)

      file_b_caching_content = diff_note_b.diff_file.highlighted_diff_lines.map(&:to_hash)

      expect(Gitlab::DiscussionsDiff::HighlightCache)
        .to receive(:write_multiple)
        .with({ note_diff_file_b.id => file_b_caching_content })
        .and_call_original

      subject.load_highlight
    end

    it 'loaded diff files have highlighted lines loaded' do
      subject.load_highlight

      diff_file_a = subject.find_by_id(note_diff_file_a.id)
      diff_file_b = subject.find_by_id(note_diff_file_b.id)

      expect(diff_file_a).to be_highlight_loaded
      expect(diff_file_b).to be_highlight_loaded
    end

    it 'not loaded diff files does not have highlighted lines loaded' do
      diff_note_a.update_column(:resolved_at, Time.now)

      subject.load_highlight

      diff_file_a = subject.find_by_id(note_diff_file_a.id)
      diff_file_b = subject.find_by_id(note_diff_file_b.id)

      expect(diff_file_a).not_to be_highlight_loaded
      expect(diff_file_b).to be_highlight_loaded
    end
  end
end