1 files changed, 29 insertions, 0 deletions
diff --git a/spec/factories/snippet_repository_storage_moves.rb b/spec/factories/snippet_repository_storage_moves.rb
new file mode 100644
index 00000000000..ed65dc5374f
--- /dev/null
+++ b/spec/factories/snippet_repository_storage_moves.rb
@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :snippet_repository_storage_move, class: 'SnippetRepositoryStorageMove' do
+    container { association(:snippet) }
+
+    source_storage_name { 'default' }
+
+    trait :scheduled do
+      state { SnippetRepositoryStorageMove.state_machines[:state].states[:scheduled].value }
+    end
+
+    trait :started do
+      state { SnippetRepositoryStorageMove.state_machines[:state].states[:started].value }
+    end
+
+    trait :replicated do
+      state { SnippetRepositoryStorageMove.state_machines[:state].states[:replicated].value }
+    end
+
+    trait :finished do
+      state { SnippetRepositoryStorageMove.state_machines[:state].states[:finished].value }
+    end
+
+    trait :failed do
+      state { SnippetRepositoryStorageMove.state_machines[:state].states[:failed].value }
+    end
+  end
+end