1 files changed, 80 insertions, 0 deletions
diff --git a/spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb b/spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb
new file mode 100644
index 00000000000..a3cc866a406
--- /dev/null
+++ b/spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb
@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe BulkImports::Common::Pipelines::UploadsPipeline do
+  let_it_be(:tmpdir) { Dir.mktmpdir }
+  let_it_be(:project) { create(:project) }
+  let_it_be(:entity) { create(:bulk_import_entity, :project_entity, project: project, source_full_path: 'test') }
+  let_it_be(:tracker) { create(:bulk_import_tracker, entity: entity) }
+  let_it_be(:context) { BulkImports::Pipeline::Context.new(tracker) }
+  let_it_be(:uploads_dir_path) { File.join(tmpdir, '72a497a02fe3ee09edae2ed06d390038') }
+  let_it_be(:upload_file_path) { File.join(uploads_dir_path, 'upload.txt')}
+
+  subject(:pipeline) { described_class.new(context) }
+
+  before do
+    stub_uploads_object_storage(FileUploader)
+
+    FileUtils.mkdir_p(uploads_dir_path)
+    FileUtils.touch(upload_file_path)
+  end
+
+  after do
+    FileUtils.remove_entry(tmpdir) if Dir.exist?(tmpdir)
+  end
+
+  describe '#run' do
+    it 'imports uploads into destination portable and removes tmpdir' do
+      allow(Dir).to receive(:mktmpdir).with('bulk_imports').and_return(tmpdir)
+      allow(pipeline).to receive(:extract).and_return(BulkImports::Pipeline::ExtractedData.new(data: [upload_file_path]))
+
+      pipeline.run
+
+      expect(project.uploads.map { |u| u.retrieve_uploader.filename }).to include('upload.txt')
+
+      expect(Dir.exist?(tmpdir)).to eq(false)
+    end
+  end
+
+  describe '#extract' do
+    it 'downloads & extracts upload paths' do
+      allow(Dir).to receive(:mktmpdir).and_return(tmpdir)
+      expect(pipeline).to receive(:untar_zxf)
+      file_download_service = instance_double("BulkImports::FileDownloadService")
+
+      expect(BulkImports::FileDownloadService)
+        .to receive(:new)
+        .with(
+          configuration: context.configuration,
+          relative_url: "/projects/test/export_relations/download?relation=uploads",
+          dir: tmpdir,
+          filename: 'uploads.tar.gz')
+        .and_return(file_download_service)
+
+      expect(file_download_service).to receive(:execute)
+
+      extracted_data = pipeline.extract(context)
+
+      expect(extracted_data.data).to contain_exactly(uploads_dir_path, upload_file_path)
+    end
+  end
+
+  describe '#load' do
+    it 'creates a file upload' do
+      expect { pipeline.load(context, upload_file_path) }.to change { project.uploads.count }.by(1)
+    end
+
+    context 'when dynamic path is nil' do
+      it 'returns' do
+        expect { pipeline.load(context, File.join(tmpdir, 'test')) }.not_to change { project.uploads.count }
+      end
+    end
+
+    context 'when path is a directory' do
+      it 'returns' do
+        expect { pipeline.load(context, uploads_dir_path) }.not_to change { project.uploads.count }
+      end
+    end
+  end
+end