1 files changed, 134 insertions, 0 deletions
diff --git a/spec/services/packages/rubygems/process_gem_service_spec.rb b/spec/services/packages/rubygems/process_gem_service_spec.rb
new file mode 100644
index 00000000000..83e868d9579
--- /dev/null
+++ b/spec/services/packages/rubygems/process_gem_service_spec.rb
@@ -0,0 +1,134 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Packages::Rubygems::ProcessGemService do
+  include ExclusiveLeaseHelpers
+  include RubygemsHelpers
+
+  let_it_be_with_reload(:package) { create(:rubygems_package, :processing, name: 'temp_name', version: '0.0.0') }
+
+  let(:package_file) { create(:package_file, :unprocessed_gem, package: package) }
+  let(:gem) { gem_from_file(package_file.file) }
+  let(:gemspec) { gem.spec }
+  let(:service) { described_class.new(package_file) }
+
+  describe '#execute' do
+    subject { service.execute }
+
+    context 'no gem file', :aggregate_failures do
+      let(:package_file) { nil }
+
+      it 'returns an error' do
+        expect(subject.error?).to be(true)
+        expect(subject.message).to eq('Gem was not processed')
+      end
+    end
+
+    context 'success' do
+      let(:sub_service) { double }
+
+      before do
+        expect(Packages::Rubygems::MetadataExtractionService).to receive(:new).with(package, gemspec).and_return(sub_service)
+        expect(Packages::Rubygems::CreateGemspecService).to receive(:new).with(package, gemspec).and_return(sub_service)
+        expect(Packages::Rubygems::CreateDependenciesService).to receive(:new).with(package, gemspec).and_return(sub_service)
+
+        expect(sub_service).to receive(:execute).exactly(3).times.and_return(true)
+      end
+
+      it 'returns successfully', :aggregate_failures do
+        result = subject
+
+        expect(result.success?).to be true
+        expect(result.payload[:package]).to eq(package)
+      end
+
+      it 'updates the package name and version', :aggregate_failures do
+        expect(package.name).to eq('temp_name')
+        expect(package.version).to eq('0.0.0')
+        expect(package).to be_processing
+
+        subject
+
+        expect(package.reload.name).to eq('package')
+        expect(package.version).to eq('0.0.1')
+        expect(package).to be_default
+      end
+
+      it 'updates the package file name', :aggregate_failures do
+        expect(package_file.file_name).to eq('package.gem')
+
+        subject
+
+        expect(package_file.reload.file_name).to eq('package-0.0.1.gem')
+      end
+    end
+
+    context 'when the package already exists' do
+      let_it_be(:existing_package) { create(:rubygems_package, name: 'package', version: '0.0.1', project: package.project) }
+
+      let(:sub_service) { double }
+
+      before do
+        expect(Packages::Rubygems::MetadataExtractionService).to receive(:new).with(existing_package, gemspec).and_return(sub_service)
+        expect(Packages::Rubygems::CreateGemspecService).to receive(:new).with(existing_package, gemspec).and_return(sub_service)
+        expect(Packages::Rubygems::CreateDependenciesService).to receive(:new).with(existing_package, gemspec).and_return(sub_service)
+
+        expect(sub_service).to receive(:execute).exactly(3).times.and_return(true)
+      end
+
+      it 'assigns the package_file to the existing package and deletes the temporary package', :aggregate_failures do
+        expect(package).to receive(:destroy)
+
+        expect { subject }.to change { existing_package.package_files.count }.by(1)
+
+        expect(package_file.reload.package).to eq(existing_package)
+      end
+    end
+
+    context 'sub-service failure' do
+      before do
+        expect(Packages::Rubygems::MetadataExtractionService).to receive(:new).with(package, gemspec).and_raise(::Packages::Rubygems::ProcessGemService::ExtractionError.new('failure'))
+      end
+
+      it 'returns an error' do
+        expect { subject }.to raise_error(::Packages::Rubygems::ProcessGemService::ExtractionError, 'failure')
+      end
+    end
+
+    context 'bad gem file' do
+      before do
+        expect(Gem::Package).to receive(:new).and_raise(ArgumentError)
+      end
+
+      it 'returns an error' do
+        expect { subject }.to raise_error(::Packages::Rubygems::ProcessGemService::ExtractionError, 'Unable to read gem file')
+      end
+    end
+
+    context 'without obtaining an exclusive lease' do
+      let(:lease_key) { "packages:rubygems:process_gem_service:package:#{package.id}" }
+
+      before do
+        stub_exclusive_lease_taken(lease_key, timeout: 1.hour)
+      end
+
+      it 'does not perform the services', :aggregate_failures do
+        # The #use_file call triggers a separate lease on the package file being opened
+        # for use with the gem. We don't want to test that here, so we allow the call to proceed
+        expect(Gitlab::ExclusiveLease).to receive(:new).with("object_storage_migrate:Packages::PackageFile:#{package_file.id}", anything).and_call_original
+
+        expect(Packages::Rubygems::MetadataExtractionService).not_to receive(:new)
+        expect(Packages::Rubygems::CreateGemspecService).not_to receive(:new)
+        expect(Packages::Rubygems::CreateDependenciesService).not_to receive(:new)
+
+        subject
+
+        expect(package.reload.name).to eq('temp_name')
+        expect(package.version).to eq('0.0.0')
+        expect(package).to be_processing
+        expect(package_file.reload.file_name).to eq('package.gem')
+      end
+    end
+  end
+end