1 files changed, 14 insertions, 2 deletions
diff --git a/spec/factories/packages/packages.rb b/spec/factories/packages/packages.rb
index caec7580e46..7896cfd8bb8 100644
--- a/spec/factories/packages/packages.rb
+++ b/spec/factories/packages/packages.rb
@@ -199,8 +199,14 @@ FactoryBot.define do
       sequence(:version) { |n| "1.0.#{n}" }
       package_type { :nuget }
 
-      after :create do |package|
-        create :package_file, :nuget, package: package, file_name: "#{package.name}.#{package.version}.nupkg"
+      transient do
+        without_package_files { false }
+      end
+
+      after :create do |package, evaluator|
+        unless evaluator.without_package_files
+          create :package_file, :nuget, package: package, file_name: "#{package.name}.#{package.version}.nupkg"
+        end
       end
 
       trait(:with_metadatum) do
@@ -214,6 +220,12 @@ FactoryBot.define do
           create :package_file, :snupkg, package: package, file_name: "#{package.name}.#{package.version}.snupkg"
         end
       end
+
+      trait :with_build do
+        after :create do |package|
+          create(:package_build_info, package: package)
+        end
+      end
     end
 
     factory :pypi_package do