add :with_member_works_and_file_sets trait for hyrax_work factory

samvera · Aug 23, 2023 · db4d46b · db4d46b
1 parent f18cdb3
commit db4d46b
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/spec/factories/hyrax_work.rb b/spec/factories/hyrax_work.rb
@@ -74,7 +74,7 @@
     trait :with_member_works do
       transient do
         members do
-          # If you set a depositor on the containing work, propogate that into these members
+          # If eyou set a depositor on the containing work, propogate that into these members
           additional_attributes = {}
           additional_attributes[:depositor] = depositor if depositor
           [valkyrie_create(:hyrax_work, additional_attributes), valkyrie_create(:hyrax_work, additional_attributes)]
@@ -83,6 +83,20 @@
     end
 
     trait :with_member_file_sets do
+      transient do
+        members do
+          # If you set a depositor on the containing work, propogate that into these members
+          additional_attributes = {}
+          additional_attributes[:depositor] = depositor if depositor
+          [valkyrie_create(:hyrax_work, additional_attributes),
+           valkyrie_create(:hyrax_work, additional_attributes),
+           valkyrie_create(:hyrax_file_set, additional_attributes),
+           valkyrie_create(:hyrax_file_set, additional_attributes)]
+        end
+      end
+    end
+
+    trait :with_member_works_and_file_sets do
       transient do
         members do
           # If you set a depositor on the containing work, propogate that into these members