spec/workers/bulk_imports/entity_worker_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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe BulkImports::EntityWorker do
  describe '#execute' do
    let(:bulk_import) { create(:bulk_import) }

    context 'when started entity exists' do
      let(:entity) { create(:bulk_import_entity, :started, bulk_import: bulk_import) }

      it 'executes BulkImports::Importers::GroupImporter' do
        expect(BulkImports::Importers::GroupImporter).to receive(:new).with(entity).and_call_original

        subject.perform(entity.id)
      end

      it 'sets jid' do
        jid = 'jid'

        allow(subject).to receive(:jid).and_return(jid)

        subject.perform(entity.id)

        expect(entity.reload.jid).to eq(jid)
      end

      context 'when exception occurs' do
        it 'tracks the exception & marks entity as failed' do
          allow(BulkImports::Importers::GroupImporter).to receive(:new) { raise StandardError }

          expect(Gitlab::ErrorTracking)
            .to receive(:track_exception)
            .with(kind_of(StandardError), bulk_import_id: bulk_import.id, entity_id: entity.id)

          subject.perform(entity.id)

          expect(entity.reload.failed?).to eq(true)
        end
      end
    end

    context 'when started entity does not exist' do
      it 'does not execute BulkImports::Importers::GroupImporter' do
        entity = create(:bulk_import_entity, bulk_import: bulk_import)

        expect(BulkImports::Importers::GroupImporter).not_to receive(:new)

        subject.perform(entity.id)
      end
    end
  end
end