1 files changed, 83 insertions, 0 deletions
diff --git a/spec/services/dependency_proxy/find_or_create_manifest_service_spec.rb b/spec/services/dependency_proxy/find_or_create_manifest_service_spec.rb
new file mode 100644
index 00000000000..c375e5a2fa3
--- /dev/null
+++ b/spec/services/dependency_proxy/find_or_create_manifest_service_spec.rb
@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+require 'spec_helper'
+
+RSpec.describe DependencyProxy::FindOrCreateManifestService do
+  include DependencyProxyHelpers
+
+  let_it_be(:image) { 'alpine' }
+  let_it_be(:tag) { 'latest' }
+  let_it_be(:dependency_proxy_manifest) { create(:dependency_proxy_manifest, file_name: "#{image}:#{tag}.json") }
+  let(:manifest) { dependency_proxy_manifest.file.read }
+  let(:group) { dependency_proxy_manifest.group }
+  let(:token) { Digest::SHA256.hexdigest('123') }
+  let(:headers) { { 'docker-content-digest' => dependency_proxy_manifest.digest } }
+
+  describe '#execute' do
+    subject { described_class.new(group, image, tag, token).execute }
+
+    context 'when no manifest exists' do
+      let_it_be(:image) { 'new-image' }
+
+      before do
+        stub_manifest_head(image, tag, digest: dependency_proxy_manifest.digest)
+        stub_manifest_download(image, tag, headers: headers)
+      end
+
+      it 'downloads manifest from remote registry if there is no cached one', :aggregate_failures do
+        expect { subject }.to change { group.dependency_proxy_manifests.count }.by(1)
+        expect(subject[:status]).to eq(:success)
+        expect(subject[:manifest]).to be_a(DependencyProxy::Manifest)
+        expect(subject[:manifest]).to be_persisted
+      end
+    end
+
+    context 'when manifest exists' do
+      before do
+        stub_manifest_head(image, tag, digest: dependency_proxy_manifest.digest)
+      end
+
+      shared_examples 'using the cached manifest' do
+        it 'uses cached manifest instead of downloading one', :aggregate_failures do
+          expect(subject[:status]).to eq(:success)
+          expect(subject[:manifest]).to be_a(DependencyProxy::Manifest)
+          expect(subject[:manifest]).to eq(dependency_proxy_manifest)
+        end
+      end
+
+      it_behaves_like 'using the cached manifest'
+
+      context 'when digest is stale' do
+        let(:digest) { 'new-digest' }
+
+        before do
+          stub_manifest_head(image, tag, digest: digest)
+          stub_manifest_download(image, tag, headers: { 'docker-content-digest' => digest })
+        end
+
+        it 'downloads the new manifest and updates the existing record', :aggregate_failures do
+          expect(subject[:status]).to eq(:success)
+          expect(subject[:manifest]).to eq(dependency_proxy_manifest)
+          expect(subject[:manifest].digest).to eq(digest)
+        end
+      end
+
+      context 'failed connection' do
+        before do
+          expect(DependencyProxy::HeadManifestService).to receive(:new).and_raise(Net::OpenTimeout)
+        end
+
+        it_behaves_like 'using the cached manifest'
+
+        context 'and no manifest is cached' do
+          let_it_be(:image) { 'new-image' }
+
+          it 'returns an error', :aggregate_failures do
+            expect(subject[:status]).to eq(:error)
+            expect(subject[:http_status]).to eq(503)
+            expect(subject[:message]).to eq('Failed to download the manifest from the external registry')
+          end
+        end
+      end
+    end
+  end
+end