1 files changed, 21 insertions, 0 deletions
diff --git a/spec/factories/gitlab/jwt_token.rb b/spec/factories/gitlab/jwt_token.rb
new file mode 100644
index 00000000000..bc00c6a5ff4
--- /dev/null
+++ b/spec/factories/gitlab/jwt_token.rb
@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :jwt_token, class: 'Gitlab::JWTToken' do
+    skip_create
+
+    initialize_with { new }
+
+    trait :with_custom_payload do
+      transient do
+        custom_payload { {} }
+      end
+
+      after(:build) do |jwt, evaluator|
+        evaluator.custom_payload.each do |key, value|
+          jwt[key] = value
+        end
+      end
+    end
+  end
+end