spec/tasks/gitlab/external_diffs_rake_spec.rb


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

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe 'gitlab:external_diffs rake tasks', :silence_stdout do
  before do
    Rake.application.rake_require 'tasks/gitlab/external_diffs'
  end

  describe 'force_object_storage task' do
    it 'forces externally stored merge request diffs to object storage' do
      db = create(:merge_request).merge_request_diff
      file = create(:merge_request).merge_request_diff.tap { |o| o.update_columns(stored_externally: true, external_diff_store: 1) }
      object = create(:merge_request).merge_request_diff.tap { |o| o.update_columns(stored_externally: true, external_diff_store: 2) }

      run_rake_task('gitlab:external_diffs:force_object_storage')

      expect(db.reload).not_to be_stored_externally
      expect(file.reload).to be_stored_externally
      expect(object.reload).to be_stored_externally

      expect(file.external_diff_store).to eq(2)
      expect(object.external_diff_store).to eq(2)
    end

    it 'limits batches according to BATCH_SIZE, START_ID, and END_ID' do
      stub_env('START_ID' => 'foo', 'END_ID' => 'bar', 'BATCH_SIZE' => 'baz')

      expect(MergeRequestDiff).to receive(:in_batches).with(start: 'foo', finish: 'bar', of: 'baz')

      run_rake_task('gitlab:external_diffs:force_object_storage')
    end
  end
end