1 files changed, 25 insertions, 0 deletions
diff --git a/spec/factories/packages/debian/distribution.rb b/spec/factories/packages/debian/distribution.rb
new file mode 100644
index 00000000000..2015f2923b8
--- /dev/null
+++ b/spec/factories/packages/debian/distribution.rb
@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :debian_project_distribution, class: 'Packages::Debian::ProjectDistribution' do
+    container { association(:project) }
+
+    sequence(:codename) { |n| "project-dist-#{n}" }
+
+    factory :debian_group_distribution, class: 'Packages::Debian::GroupDistribution' do
+      container { association(:group) }
+
+      sequence(:codename) { |n| "group-dist-#{n}" }
+    end
+
+    trait(:with_file) do
+      after(:build) do |distribution, evaluator|
+        distribution.file = fixture_file_upload('spec/fixtures/packages/debian/README.md')
+      end
+    end
+
+    trait(:object_storage) do
+      file_store { Packages::PackageFileUploader::Store::REMOTE }
+    end
+  end
+end