3 files changed, 35 insertions, 11 deletions
diff --git a/spec/factories/projects/import_export/export_relation.rb b/spec/factories/projects/import_export/export_relation.rb
deleted file mode 100644
index 2b6419dcecb..00000000000
--- a/spec/factories/projects/import_export/export_relation.rb
+++ /dev/null
@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-FactoryBot.define do
-  factory :project_relation_export, class: 'Projects::ImportExport::RelationExport' do
-    project_export_job factory: :project_export_job
-
-    relation { 'labels' }
-    status { 0 }
-    sequence(:jid) { |n| "project_relation_export_#{n}" }
-  end
-end
diff --git a/spec/factories/projects/import_export/relation_export.rb b/spec/factories/projects/import_export/relation_export.rb
new file mode 100644
index 00000000000..7fab5808d2f
--- /dev/null
+++ b/spec/factories/projects/import_export/relation_export.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :project_relation_export, class: 'Projects::ImportExport::RelationExport' do
+    project_export_job factory: :project_export_job
+
+    relation { 'labels' }
+    status { Projects::ImportExport::RelationExport::STATUS[:queued] }
+    sequence(:jid) { |n| "project_relation_export_#{n}" }
+
+    trait :queued do
+      status { Projects::ImportExport::RelationExport::STATUS[:queued] }
+    end
+
+    trait :started do
+      status { Projects::ImportExport::RelationExport::STATUS[:started] }
+    end
+
+    trait :finished do
+      status { Projects::ImportExport::RelationExport::STATUS[:finished] }
+    end
+
+    trait :failed do
+      status { Projects::ImportExport::RelationExport::STATUS[:failed] }
+    end
+  end
+end
diff --git a/spec/factories/projects/import_export/relation_export_upload.rb b/spec/factories/projects/import_export/relation_export_upload.rb
new file mode 100644
index 00000000000..eaa57d6ee59
--- /dev/null
+++ b/spec/factories/projects/import_export/relation_export_upload.rb
@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :project_relation_export_upload, class: 'Projects::ImportExport::RelationExportUpload' do
+    relation_export factory: :project_relation_export
+    export_file { fixture_file_upload("spec/fixtures/gitlab/import_export/labels.tar.gz") }
+  end
+end