diff --git a/lib/iiif_print/data/work_files.rb b/lib/iiif_print/data/work_files.rb index c372a592..fb79cb42 100644 --- a/lib/iiif_print/data/work_files.rb +++ b/lib/iiif_print/data/work_files.rb @@ -153,7 +153,7 @@ def original_name(fileset) def filesets # file sets with non-nil original file contained: - work.members.select { |m| m.is_a? FileSet && m.original_file } + work.members.select { |m| m.is_a?(FileSet) && m.original_file } end def user diff --git a/spec/iiif_print/data/work_files_spec.rb b/spec/iiif_print/data/work_files_spec.rb index fc098df8..9b796e0e 100644 --- a/spec/iiif_print/data/work_files_spec.rb +++ b/spec/iiif_print/data/work_files_spec.rb @@ -168,7 +168,7 @@ adapter.unassign(adapter.keys[0]) adapter.commit! expect(adapter.keys.size).to eq 0 - expect(work.members.count { |m| m.is_a? FileSet }).to eq 0 + expect(work.members.to_a.count { |m| m.is_a? FileSet }).to eq 0 end context "when it is a new work" do @@ -203,7 +203,7 @@ # should refresh the work.members, and by consequence adapter.keys work.reload expect(adapter.keys.size).to eq 1 - expect(work.members.count { |m| m.is_a? FileSet }).to eq 1 + expect(work.members.to_a.count { |m| m.is_a? FileSet }).to eq 1 expect(adapter.names).to include 'ocr_gray.tiff' end diff --git a/spec/services/iiif_print/text_formats_from_alto_service_spec.rb b/spec/services/iiif_print/text_formats_from_alto_service_spec.rb index 86cc2e30..e820c6c0 100644 --- a/spec/services/iiif_print/text_formats_from_alto_service_spec.rb +++ b/spec/services/iiif_print/text_formats_from_alto_service_spec.rb @@ -112,7 +112,7 @@ def attach_alto(work) attach_primary_file(work) attach_alto(work) work.reload - file_set = work.ordered_members.to_a.find { |m| m.is_a? FileSet) } + file_set = work.ordered_members.to_a.find { |m| m.is_a? FileSet } service = described_class.new(file_set) service.create_derivatives('/a/path/here/needed/but/will/not/matter') coords = JSON.parse(derivatives_of(work).data('json'))