1 files changed, 4 insertions, 0 deletions
diff --git a/spec/factories/gitlab/database/async_indexes/postgres_async_index.rb b/spec/factories/gitlab/database/async_indexes/postgres_async_index.rb
index d6b4b90bbd0..e3f366c17eb 100644
--- a/spec/factories/gitlab/database/async_indexes/postgres_async_index.rb
+++ b/spec/factories/gitlab/database/async_indexes/postgres_async_index.rb
@@ -5,5 +5,9 @@ FactoryBot.define do
     sequence(:name) { |n| "users_id_#{n}" }
     definition { "CREATE INDEX #{name} ON #{table_name} (id)" }
     table_name { "users" }
+
+    trait :with_drop do
+      definition { "DROP INDEX #{name}" }
+    end
   end
 end