1 files changed, 27 insertions, 0 deletions
diff --git a/spec/factories/projects/build_artifacts_size_refreshes.rb b/spec/factories/projects/build_artifacts_size_refreshes.rb
new file mode 100644
index 00000000000..b05f5dfab1c
--- /dev/null
+++ b/spec/factories/projects/build_artifacts_size_refreshes.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :project_build_artifacts_size_refresh, class: 'Projects::BuildArtifactsSizeRefresh' do
+    project factory: :project
+
+    trait :created do
+      state { Projects::BuildArtifactsSizeRefresh::STATES[:created] }
+    end
+
+    trait :pending do
+      state { Projects::BuildArtifactsSizeRefresh::STATES[:pending] }
+      refresh_started_at { Time.zone.now }
+    end
+
+    trait :running do
+      state { Projects::BuildArtifactsSizeRefresh::STATES[:running] }
+      refresh_started_at { Time.zone.now }
+    end
+
+    trait :stale do
+      running
+      refresh_started_at { 30.days.ago }
+      updated_at { 30.days.ago }
+    end
+  end
+end