1 files changed, 16 insertions, 4 deletions
diff --git a/spec/factories/gitlab/database/background_migration/batched_migrations.rb b/spec/factories/gitlab/database/background_migration/batched_migrations.rb
index 79b4447b76e..5ff90ff44b9 100644
--- a/spec/factories/gitlab/database/background_migration/batched_migrations.rb
+++ b/spec/factories/gitlab/database/background_migration/batched_migrations.rb
@@ -13,12 +13,24 @@ FactoryBot.define do
     total_tuple_count { 10_000 }
     pause_ms { 100 }
 
-    trait :finished do
-      status { :finished }
+    trait(:paused) do
+      status { 0 }
     end
 
-    trait :failed do
-      status { :failed }
+    trait(:active) do
+      status { 1 }
+    end
+
+    trait(:finished) do
+      status { 3 }
+    end
+
+    trait(:failed) do
+      status { 4 }
+    end
+
+    trait(:finalizing) do
+      status { 5 }
     end
   end
 end