diff options
Diffstat (limited to 'spec/support/helpers/dependency_proxy_helpers.rb')
-rw-r--r-- | spec/support/helpers/dependency_proxy_helpers.rb | 33 |
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 |