Get tests passing

curationexperts · Oct 5, 2021 · 3961d4a · 3961d4a
1 parent b0efdcf
commit 3961d4a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/zizia/record_importer.rb b/lib/zizia/record_importer.rb
@@ -24,6 +24,14 @@ def import(record:)
       raise e
     end
 
+    def import_all_records(records)
+      @collection_records = records.select { |record| record.object_type == "collection" }
+      @collection_records.each { |record| create_collection(record) }
+      @work_records = records.select { |record| record.object_type == "work" }
+      @file_records = records.select { |record| record.object_type == "file" }
+      @work_records.each { |record| import(record: record) }
+    end
+
     def import_type(record)
       raise 'No curation_concern found for import' unless
         defined?(Hyrax) && Hyrax&.config&.curation_concerns&.any?

diff --git a/spec/zizia/importer_spec.rb b/spec/zizia/importer_spec.rb
@@ -15,6 +15,11 @@ def import(record:)
         records << record.attributes
       end
 
+      def import_all_records(records)
+        @work_records = records.select { |record| record.object_type == "work" }
+        @work_records.each { |record| import(record: record) }
+      end
+
       def records
         @records ||= []
       end