Add latest changes from gitlab-org/gitlab@master

1 files changed, 8 insertions, 0 deletions

diff --git a/spec/factories/gitlab/database/async_foreign_keys/postgres_async_foreign_key_validation.rb b/spec/factories/gitlab/database/async_foreign_keys/postgres_async_constraint_validation.rb
index 23f4049f1cb..81f67e958c0 100644
--- a/spec/factories/gitlab/database/async_foreign_keys/postgres_async_foreign_key_validation.rb
+++ b/spec/factories/gitlab/database/async_foreign_keys/postgres_async_constraint_validation.rb
@@ -5,5 +5,13 @@ FactoryBot.define do
     class: 'Gitlab::Database::AsyncConstraints::PostgresAsyncConstraintValidation' do
     sequence(:name) { |n| "fk_users_id_#{n}" }
     table_name { "users" }
+
+    trait :foreign_key do
+      constraint_type { :foreign_key }
+    end
+
+    trait :check_constraint do
+      constraint_type { :check_constraint }
+    end
   end
 end