1 files changed, 45 insertions, 0 deletions
diff --git a/spec/services/repositories/replicate_service_spec.rb b/spec/services/repositories/replicate_service_spec.rb
new file mode 100644
index 00000000000..b4fbc478d2f
--- /dev/null
+++ b/spec/services/repositories/replicate_service_spec.rb
@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Repositories::ReplicateService, feature_category: :source_code_management do
+  let(:new_checksum) { 'match' }
+  let(:repository) { instance_double('Gitlab::Git::Repository', checksum: 'match') }
+  let(:new_repository) { instance_double('Gitlab::Git::Repository', checksum: new_checksum) }
+
+  subject { described_class.new(repository) }
+
+  it 'replicates repository' do
+    expect(new_repository).to receive(:replicate).with(repository)
+    expect(new_repository).not_to receive(:remove)
+
+    expect { subject.execute(new_repository, :project) }.not_to raise_error
+  end
+
+  context 'when checksum does not match' do
+    let(:new_checksum) { 'does not match' }
+
+    it 'raises an error and removes new repository' do
+      expect(new_repository).to receive(:replicate).with(repository)
+      expect(new_repository).to receive(:remove)
+
+      expect do
+        subject.execute(new_repository, :project)
+      end.to raise_error(described_class::Error, /Failed to verify project repository/)
+    end
+  end
+
+  context 'when an error is raised during checksum calculation' do
+    it 'raises the error and removes new repository' do
+      error = StandardError.new
+
+      expect(new_repository).to receive(:replicate).with(repository)
+      expect(new_repository).to receive(:checksum).and_raise(error)
+      expect(new_repository).to receive(:remove)
+
+      expect do
+        subject.execute(new_repository, :project)
+      end.to raise_error(error)
+    end
+  end
+end