Add latest changes from gitlab-org/gitlab@master

1 files changed, 34 insertions, 0 deletions

diff --git a/spec/support/helpers/packages_manager_api_spec_helpers.rb b/spec/support/helpers/packages_manager_api_spec_helpers.rb
new file mode 100644
index 00000000000..1c9fce183e9
--- /dev/null
+++ b/spec/support/helpers/packages_manager_api_spec_helpers.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module PackagesManagerApiSpecHelpers
+  def build_jwt(personal_access_token, secret: jwt_secret, user_id: nil)
+    JSONWebToken::HMACToken.new(secret).tap do |jwt|
+      jwt['access_token'] = personal_access_token.token
+      jwt['user_id'] = user_id || personal_access_token.user_id
+    end
+  end
+
+  def build_jwt_from_job(job, secret: jwt_secret)
+    JSONWebToken::HMACToken.new(secret).tap do |jwt|
+      jwt['access_token'] = job.token
+      jwt['user_id'] = job.user.id
+    end
+  end
+
+  def build_jwt_from_deploy_token(deploy_token, secret: jwt_secret)
+    JSONWebToken::HMACToken.new(secret).tap do |jwt|
+      jwt['access_token'] = deploy_token.token
+      jwt['user_id'] = deploy_token.username
+    end
+  end
+
+  def temp_file(package_tmp)
+    upload_path = ::Packages::PackageFileUploader.workhorse_local_upload_path
+    file_path = "#{upload_path}/#{package_tmp}"
+
+    FileUtils.mkdir_p(upload_path)
+    File.write(file_path, 'test')
+
+    UploadedFile.new(file_path, filename: File.basename(file_path))
+  end
+end