1 files changed, 25 insertions, 0 deletions
diff --git a/spec/factories/snippet_statistics.rb b/spec/factories/snippet_statistics.rb
new file mode 100644
index 00000000000..ab2d9525466
--- /dev/null
+++ b/spec/factories/snippet_statistics.rb
@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :snippet_statistics do
+    snippet
+
+    initialize_with do
+      # statistics are automatically created when a snippet is created
+      snippet&.statistics || new
+    end
+
+    transient do
+      with_data { false }
+      size_multiplier { 1 }
+    end
+
+    after(:build) do |snippet_statistics, evaluator|
+      if evaluator.with_data
+        snippet_statistics.repository_size = evaluator.size_multiplier
+        snippet_statistics.commit_count = evaluator.size_multiplier * 2
+        snippet_statistics.file_count = evaluator.size_multiplier * 3
+      end
+    end
+  end
+end