1 files changed, 33 insertions, 0 deletions
diff --git a/spec/support/helpers/dependency_proxy_helpers.rb b/spec/support/helpers/dependency_proxy_helpers.rb
new file mode 100644
index 00000000000..545b9d1f4d0
--- /dev/null
+++ b/spec/support/helpers/dependency_proxy_helpers.rb
@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module DependencyProxyHelpers
+  include StubRequests
+
+  def stub_registry_auth(image, token, status = 200, body = nil)
+    auth_body = { 'token' => token }.to_json
+    auth_link = registry.auth_url(image)
+
+    stub_full_request(auth_link)
+      .to_return(status: status, body: body || auth_body)
+  end
+
+  def stub_manifest_download(image, tag, status = 200, body = nil)
+    manifest_url = registry.manifest_url(image, tag)
+
+    stub_full_request(manifest_url)
+      .to_return(status: status, body: body || manifest)
+  end
+
+  def stub_blob_download(image, blob_sha, status = 200, body = '123456')
+    download_link = registry.blob_url(image, blob_sha)
+
+    stub_full_request(download_link)
+      .to_return(status: status, body: body)
+  end
+
+  private
+
+  def registry
+    @registry ||= DependencyProxy::Registry
+  end
+end