diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2022-01-10 23:42:50 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2022-01-10 23:43:16 +0300 |
commit | b55e13ec164336d1e5d5bbdbca939edcc31d557f (patch) | |
tree | b3283adcd5aa0cd6ac55afc75793189e7507e4ae /spec/lib | |
parent | ceaba594d07563402ac901c958b9a6e27c4964dc (diff) |
Add latest changes from gitlab-org/security/gitlab@14-6-stable-ee
Diffstat (limited to 'spec/lib')
-rw-r--r-- | spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb | 24 | ||||
-rw-r--r-- | spec/lib/gitlab/import_export/command_line_util_spec.rb | 35 |
2 files changed, 55 insertions, 4 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 index 0f6238e10dc..3b5ea131d0d 100644 --- a/spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb +++ b/spec/lib/bulk_imports/common/pipelines/uploads_pipeline_spec.rb @@ -70,8 +70,10 @@ RSpec.describe BulkImports::Common::Pipelines::UploadsPipeline do 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") + + download_service = instance_double("BulkImports::FileDownloadService") + decompression_service = instance_double("BulkImports::FileDecompressionService") + extraction_service = instance_double("BulkImports::ArchiveExtractionService") expect(BulkImports::FileDownloadService) .to receive(:new) @@ -80,9 +82,13 @@ RSpec.describe BulkImports::Common::Pipelines::UploadsPipeline do relative_url: "/#{entity.pluralized_name}/test/export_relations/download?relation=uploads", dir: tmpdir, filename: 'uploads.tar.gz') - .and_return(file_download_service) + .and_return(download_service) + expect(BulkImports::FileDecompressionService).to receive(:new).with(dir: tmpdir, filename: 'uploads.tar.gz').and_return(decompression_service) + expect(BulkImports::ArchiveExtractionService).to receive(:new).with(tmpdir: tmpdir, filename: 'uploads.tar').and_return(extraction_service) - expect(file_download_service).to receive(:execute) + expect(download_service).to receive(:execute) + expect(decompression_service).to receive(:execute) + expect(extraction_service).to receive(:execute) extracted_data = pipeline.extract(context) @@ -106,6 +112,16 @@ RSpec.describe BulkImports::Common::Pipelines::UploadsPipeline do expect { pipeline.load(context, uploads_dir_path) }.not_to change { portable.uploads.count } end end + + context 'when path is a symlink' do + it 'does not upload the file' do + symlink = File.join(tmpdir, 'symlink') + + FileUtils.ln_s(File.join(tmpdir, upload_file_path), symlink) + + expect { pipeline.load(context, symlink) }.not_to change { portable.uploads.count } + end + end end end diff --git a/spec/lib/gitlab/import_export/command_line_util_spec.rb b/spec/lib/gitlab/import_export/command_line_util_spec.rb index 59c4e1083ae..31512259bb1 100644 --- a/spec/lib/gitlab/import_export/command_line_util_spec.rb +++ b/spec/lib/gitlab/import_export/command_line_util_spec.rb @@ -101,4 +101,39 @@ RSpec.describe Gitlab::ImportExport::CommandLineUtil do end end end + + describe '#untar_xf' do + let(:archive_dir) { Dir.mktmpdir } + + after do + FileUtils.remove_entry(archive_dir) + end + + it 'extracts archive without decompression' do + filename = 'archive.tar.gz' + archive_file = File.join(archive_dir, 'archive.tar') + + FileUtils.copy_file(archive, File.join(archive_dir, filename)) + subject.gunzip(dir: archive_dir, filename: filename) + + result = subject.untar_xf(archive: archive_file, dir: archive_dir) + + expect(result).to eq(true) + expect(File.exist?(archive_file)).to eq(true) + expect(File.exist?(File.join(archive_dir, 'project.json'))).to eq(true) + expect(Dir.exist?(File.join(archive_dir, 'uploads'))).to eq(true) + end + + context 'when something goes wrong' do + it 'raises an error' do + expect(Gitlab::Popen).to receive(:popen).and_return(['Error', 1]) + + klass = Class.new do + include Gitlab::ImportExport::CommandLineUtil + end.new + + expect { klass.untar_xf(archive: 'test', dir: 'test') }.to raise_error(Gitlab::ImportExport::Error, 'System call failed') + end + end + end end |