berkeley_library-av-core 0.4.0

data/spec/lib/berkeley_library/av/metadata/source_spec.rb

@@ -0,0 +1,261 @@
+require 'spec_helper'
+
+# TODO: break some of this out into a Metadata::Reader spec
+module BerkeleyLibrary
+  module AV
+    class Metadata
+      describe Source do
+        before do
+          allow(BerkeleyLibrary::Logging).to receive(:logger).and_return(Logger.new(File::NULL))
+        end
+
+        describe :for_record_id do
+          it 'returns ALMA for a Millennium bib number' do
+            expect(Source.for_record_id('b12345678')).to be(Source::ALMA)
+          end
+
+          it 'returns ALMA for a Millennium bib number with check digit' do
+            expect(Source.for_record_id('b12345678a')).to be(Source::ALMA)
+          end
+
+          it "returns TIND for something that looks a little like a Millennium bib but isn't" do
+            expect(Source.for_record_id('b1234567')).to be(Source::TIND)
+            expect(Source.for_record_id('b123456789abcdef')).to be(Source::TIND)
+          end
+
+          it 'returns TIND for a TIND record number' do
+            expect(Source.for_record_id('(coll)12345')).to be(Source::TIND)
+          end
+
+          it 'returns TIND for an OCLC number' do
+            expect(Source.for_record_id('o12345678')).to be(Source::TIND)
+          end
+
+          it 'returns ALMA for an Alma MMS ID' do
+            expect(Source.for_record_id('991054360089706532')).to be(Source::ALMA)
+          end
+        end
+
+        describe :catalog_link_text do
+          it 'returns TIND text for a TIND record' do
+            expect(Source::TIND.catalog_link_text).to eq(Source::LINK_TEXT_TIND)
+          end
+
+          it 'returns Alma text for an Alma record' do
+            expect(Source::ALMA.catalog_link_text).to eq(Source::LINK_TEXT_ALMA)
+          end
+        end
+
+        describe 'ALMA' do
+
+          before do
+            Config.avplayer_base_uri = 'https://avplayer.lib.berkeley.edu'
+            Config.tind_base_uri = 'https://digicoll.lib.berkeley.edu'
+            Config.alma_sru_host = 'berkeley.alma.exlibrisgroup.com'
+            Config.alma_institution_code = '01UCS_BER'
+            Config.alma_primo_host = 'search.library.berkeley.edu'
+            Config.alma_permalink_key = 'iqob43'
+          end
+
+          describe :marc_uri_for do
+            it 'returns the search URI for an Alma MMS ID' do
+              mms_id = '991054360089706532'
+              url_expected = "#{sru_url_base}alma.mms_id%3D#{mms_id}"
+              uri_expected = URI.parse(url_expected)
+              uri_actual = Source::ALMA.marc_uri_for(mms_id)
+              expect(uri_actual).to eq(uri_expected)
+            end
+
+            it 'returns the search URI for a Millennium bib number with or without check digit' do
+              full_bib = 'b257169738'
+              url_expected = "#{sru_url_base}alma.other_system_number%3DUCB-#{full_bib}-01ucs_ber"
+              uri_expected = URI.parse(url_expected)
+              aggregate_failures do
+                %w[b25716973 b257169738 b25716973a].each do |bib|
+                  uri_actual = Source::ALMA.marc_uri_for(bib)
+                  expect(uri_actual).to eq(uri_expected)
+                end
+              end
+            end
+          end
+
+          describe :display_uri_for do
+            it 'returns the Primo permalink' do
+              mms_id = '991054360089706532'
+              marc_record = MARC::XMLReader.new("spec/data/alma/#{mms_id}-sru.xml").first
+              uri_expected = URI.parse("#{permalink_base}#{mms_id}")
+
+              metadata = Metadata.new(
+                record_id: mms_id,
+                source: Source::ALMA,
+                marc_record:
+              )
+
+              uri_actual = Source::ALMA.display_uri_for(metadata)
+              expect(uri_actual).to eq(uri_expected)
+            end
+
+            it 'raises ArgumentError for a TIND identifier' do
+              marc_xml = File.read('spec/data/record-(pacradio)00107.xml')
+
+              metadata = Metadata.new(
+                record_id: '(pacradio)00107',
+                source: Source::TIND,
+                marc_record: AV::Marc.from_xml(marc_xml)
+              )
+
+              expect { Source::ALMA.display_uri_for(metadata) }.to raise_error(ArgumentError)
+            end
+          end
+
+          describe :bib_number do
+            it 'finds the bib number' do
+              mms_id = '991054360089706532'
+              marc_record = MARC::XMLReader.new("spec/data/alma/#{mms_id}-sru.xml").first
+
+              metadata = Metadata.new(
+                record_id: mms_id,
+                source: Source::ALMA,
+                marc_record:
+              )
+              expect(metadata.bib_number).to eq('b25716973')
+            end
+          end
+
+          describe :record_for do
+            it 'loads a record from an MMS ID' do
+              mms_id = '991054360089706532'
+              marc_xml_path = "spec/data/alma/#{mms_id}-sru.xml"
+              sru_url = "#{sru_url_base}alma.mms_id%3D#{mms_id}"
+              marc_xml = File.read(marc_xml_path)
+              stub_request(:get, sru_url).to_return(status: 200, body: marc_xml)
+
+              marc_record = Source::ALMA.record_for(mms_id)
+              expect(marc_record).to eq(MARC::XMLReader.new(marc_xml_path).first)
+            end
+
+            it 'loads a record from a bib number' do
+              short_bib = 'b25716973'
+              full_bib = RecordId.ensure_check_digit(short_bib)
+              marc_xml_path = "spec/data/alma/#{short_bib}-sru.xml"
+              marc_xml = File.read(marc_xml_path)
+              sru_url = "#{sru_url_base}alma.other_system_number%3DUCB-#{full_bib}-01ucs_ber"
+              stub_request(:get, sru_url).to_return(status: 200, body: marc_xml)
+
+              aggregate_failures do
+                %w[b25716973 b257169738 b25716973a].each do |bib|
+                  marc_record = Source::ALMA.record_for(bib)
+                  expect(marc_record).to eq(MARC::XMLReader.new(marc_xml_path).first)
+                end
+              end
+            end
+
+            it 'raises ArgumentError for the wrong type of ID' do
+              expect { Source::ALMA.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+          end
+        end
+
+        describe 'TIND' do
+          attr_reader :record_url
+
+          before do
+            AV::Config.tind_base_uri = 'https://digicoll.lib.berkeley.edu'
+            @record_url = 'https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22%28pacradio%2900107%22&of=xm'
+          end
+
+          after do
+            Config.send(:clear!)
+          end
+
+          describe :marc_uri_for do
+            it 'returns the search URI' do
+              uri_expected = URI.parse(record_url)
+              uri_actual = Source::TIND.marc_uri_for('(pacradio)00107')
+              expect(uri_actual).to eq(uri_expected)
+            end
+          end
+
+          describe :display_uri_for do
+            it 'returns the record display page URI' do
+              marc_xml = File.read('spec/data/record-(pacradio)00107.xml')
+              stub_request(:get, record_url).to_return(status: 200, body: marc_xml)
+
+              metadata = Metadata.new(
+                record_id: '(pacradio)00107',
+                source: Source::TIND,
+                marc_record: Source::TIND.record_for('(pacradio)00107')
+              )
+
+              uri_expected = URI.parse('https://digicoll.lib.berkeley.edu/record/19816')
+              uri_actual = Source::TIND.display_uri_for(metadata)
+              expect(uri_actual).to eq(uri_expected)
+            end
+
+            it 'raises ArgumentError for the wrong type of record' do
+              marc_xml = File.read('spec/data/alma/b20786580-sru.xml')
+              marc_record = AV::Marc.from_xml(marc_xml)
+
+              metadata = Metadata.new(
+                record_id: '(pacradio)00107',
+                source: Source::ALMA,
+                marc_record:
+              )
+
+              expect { Source::TIND.display_uri_for(metadata) }.to raise_error(ArgumentError)
+            end
+          end
+
+          describe :record_for do
+            it 'finds a MARC record' do
+              marc_xml = File.read('spec/data/record-(pacradio)00107.xml')
+              stub_request(:get, record_url).to_return(status: 200, body: marc_xml)
+
+              marc_record = Source::TIND.record_for('(pacradio)00107')
+              expect(marc_record).not_to be_nil
+              expect(marc_record).to be_a(MARC::Record)
+
+              fields_001 = marc_record.fields('001')
+              expect(fields_001.size).to eq(1)
+              expect(fields_001[0].value).to eq('19816')
+            end
+
+            it "raises #{AV::RecordNotFound} if the record cannot be found" do
+              stub_request(:get, record_url).to_return(status: 404, body: '')
+
+              expect { Source::TIND.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+
+            it "raises #{AV::RecordNotFound} if the record cannot be parsed" do
+              stub_request(:get, record_url).to_return(status: 200, body: '<? this is not valid MARCXML>')
+
+              expect { Source::TIND.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+
+            it "raises #{AV::RecordNotFound} if no records are returned" do
+              empty_result = File.read('spec/data/record-empty-result.xml')
+              stub_request(:get, record_url).to_return(status: 200, body: empty_result)
+
+              expect { Source::TIND.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+
+            it "raises #{AV::RecordNotFound} on a redirect to login" do
+              redirect_to_login = File.read('spec/data/record-redirect-to-login.html')
+              stub_request(:get, record_url).to_return(status: 200, body: redirect_to_login)
+
+              expect { Source::TIND.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+
+            it "raises #{AV::RecordNotFound} if TIND returns a weird HTTP status" do
+              marc_xml = File.read('spec/data/record-(pacradio)00107.xml')
+              stub_request(:get, record_url).to_return(status: 207, body: marc_xml)
+
+              expect { Source::TIND.record_for('(pacradio)00107') }.to raise_error(AV::RecordNotFound)
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/spec/lib/berkeley_library/av/metadata/value_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+module BerkeleyLibrary
+  module AV
+    class Metadata
+      describe Value do
+        describe :<=> do
+          it 'can compare text with links' do
+            v1 = Value.new(tag: '956', label: 'test', entries: ['test'], order: 1)
+            v2 = Value.new(tag: '956', label: 'test', entries: [Link.new(body: 'test', url: 'http://example.org')], order: 1)
+            actual = v1 <=> v2
+            expect(actual).not_to eq(0)
+            expect(actual).to eq(v1.entries.to_s <=> v2.entries.to_s)
+          end
+
+          it 'returns nil for nil' do
+            v = Value.new(tag: '956', label: 'test', entries: ['test'], order: 1)
+            expect(v <=> nil).to be_nil
+          end
+
+          it 'returns nil for things that are not Values' do
+            v = Value.new(tag: '956', label: 'test', entries: ['test'], order: 1)
+            expect(v <=> v.to_s).to be_nil
+          end
+        end
+      end
+    end
+  end
+end

data/spec/lib/berkeley_library/av/record_id_spec.rb

@@ -0,0 +1,72 @@
+require 'spec_helper'
+
+module BerkeleyLibrary
+  module AV
+    describe RecordId do
+
+      describe :hash do
+        it 'returns the same hash for identical objects' do
+          id1 = RecordId.new('991054360089706532')
+          id2 = RecordId.new('991054360089706532')
+          expect(id1.hash).to eq(id2.hash)
+        end
+      end
+
+      describe :inspect do
+        it 'includes the ID and type' do
+          mms_id = '991054360089706532'
+          id = RecordId.new(mms_id)
+          expect(id.inspect).to include(mms_id)
+          expect(id.inspect).to include(id.type.to_s)
+        end
+      end
+
+      describe :ensure_check_digit do
+        let(:expected_check_digits_by_bib) do
+          infile = 'spec/data/record_id/bibs_with_check_digits.txt'
+          File.readlines(infile, chomp: true).each_with_object({}) do |bib, x|
+            x[bib[0, 9]] = bib[9]
+          end
+        end
+
+        it 'produces the expected check digit' do
+          aggregate_failures 'check digit' do
+            expected_check_digits_by_bib.each do |bib_expected, cd_expected|
+              bib_with_cd = RecordId.ensure_check_digit(bib_expected)
+              expect(bib_with_cd[0, 9]).to eq(bib_expected) # just to be sure
+
+              cd_actual = bib_with_cd[9]
+              expect(cd_actual).to eq(cd_expected), "Wrong check digit for #{bib_expected}; should be #{cd_expected}, was #{cd_actual}"
+            end
+          end
+        end
+
+        it 'raises an error if passed an invalid check digit' do
+          aggregate_failures 'bad check digit' do
+            expected_check_digits_by_bib.each do |bib, cd|
+              cd_i = cd == 'x' ? 10 : cd.to_i
+              bad_cd = cd_i == '0' ? 'x' : (cd_i - 1).to_s
+              bad_bib = "#{bib}#{bad_cd}"
+              expect { RecordId.ensure_check_digit(bad_bib) }.to raise_error(ArgumentError), "#{bib}: No error raised for #{bad_cd} (should be #{cd})"
+            end
+          end
+        end
+
+        it 'ignores a wildcard "a" check digit, but returns the correct digit' do
+          aggregate_failures 'check digit "a"' do
+            expected_check_digits_by_bib.each do |bib, cd|
+              wildcard = "#{bib}a"
+              expected = "#{bib}#{cd}"
+              begin
+                actual = RecordId.ensure_check_digit(wildcard)
+                expect(actual).to eq(expected)
+              rescue ArgumentError => e
+                raise("Expected #{wildcard} not to raise error, got #{e.class}: #{e}")
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/spec/lib/berkeley_library/av/record_spec.rb

@@ -0,0 +1,284 @@
+require 'spec_helper'
+
+module BerkeleyLibrary
+  module AV
+    describe Record do
+      before do
+        Config.avplayer_base_uri = 'https://avplayer.lib.berkeley.edu'
+        Config.tind_base_uri = 'https://digicoll.lib.berkeley.edu'
+        Config.alma_sru_host = 'berkeley.alma.exlibrisgroup.com'
+        Config.alma_institution_code = '01UCS_BER'
+        Config.alma_primo_host = 'search.library.berkeley.edu'
+        Config.alma_permalink_key = 'iqob43'
+      end
+
+      after do
+        Config.send(:clear!)
+      end
+
+      describe :new do
+        it 'sorts the tracks' do
+          t1 = Track.new(sort_order: 1, title: 'Part 1', path: 'MRCAudio/frost-read1.mp3')
+          t2 = Track.new(sort_order: 2, title: 'Part 2', path: 'MRCAudio/frost-read2.mp3')
+          record = Record.new(
+            collection: 'MRCAudio',
+            tracks: [t2, t1],
+            metadata: instance_double(Metadata)
+          )
+          tracks = record.tracks
+          expect(tracks[0]).to eq(t1)
+          expect(tracks[1]).to eq(t2)
+        end
+      end
+
+      describe :player_uri do
+        it 'returns a player URI based on the bib number for Millennium records' do
+          bib_number = 'b22139658'
+          stub_sru_request(bib_number)
+
+          collection = 'MRCVideo'
+          record = Record.from_metadata(collection:, record_id: bib_number)
+
+          expected_uri = URI.parse("https://avplayer.lib.berkeley.edu/#{collection}/#{bib_number}")
+          expect(record.player_uri).to eq(expected_uri)
+        end
+
+        it 'returns a player URI based on the record ID for TIND records' do
+          tind_035 = '(pacradio)01469'
+          marc_xml = File.read("spec/data/record-#{tind_035}.xml")
+          search_url = "https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22#{CGI.escape(tind_035)}%22&of=xm"
+          stub_request(:get, search_url).to_return(status: 200, body: marc_xml)
+
+          collection = 'Pacifica'
+          record = Record.from_metadata(collection:, record_id: tind_035)
+
+          expected_uri = URI.parse("https://avplayer.lib.berkeley.edu/#{collection}/#{tind_035}")
+          expect(record.player_uri).to eq(expected_uri)
+        end
+
+        it 'generates the player URI' do
+          collection = 'MRCAudio'
+          bib_number = 'b11082434'
+
+          metadata = instance_double(Metadata)
+          expect(metadata).to receive(:record_id).and_return(bib_number)
+
+          record = Record.new(collection:, tracks: [], metadata:)
+
+          expected_uri = URI.parse("https://avplayer.lib.berkeley.edu/#{collection}/#{bib_number}")
+          expect(record.player_uri).to eq(expected_uri)
+        end
+      end
+
+      describe :display_uri do
+        it 'returns the display URI' do
+          collection = 'MRCAudio'
+          bib_number = 'b11082434'
+          expected_uri = URI.parse("http://oskicat.example.edu/record=#{bib_number}")
+
+          metadata = instance_double(Metadata)
+          expect(metadata).to receive(:display_uri).and_return(expected_uri)
+
+          record = Record.new(collection:, tracks: [], metadata:)
+          expect(record.display_uri).to eq(expected_uri)
+        end
+      end
+
+      describe :type_label do
+        it 'handles audio' do
+          t1 = Track.new(sort_order: 1, title: 'Part 1', path: 'MRCAudio/frost-read1.mp3')
+          t2 = Track.new(sort_order: 2, title: 'Part 2', path: 'MRCAudio/frost-read2.mp3')
+          record = Record.new(
+            collection: 'MRCAudio',
+            tracks: [t2, t1],
+            metadata: instance_double(Metadata)
+          )
+
+          expect(record.type_label).to eq('Audio')
+        end
+
+        it 'handles video' do
+          t1 = Track.new(sort_order: 1, title: 'Part 1', path: 'MRCAudio/frost-read1.mp4')
+          t2 = Track.new(sort_order: 2, title: 'Part 2', path: 'MRCAudio/frost-read2.mp4')
+          record = Record.new(
+            collection: 'MRCVideo',
+            tracks: [t2, t1],
+            metadata: instance_double(Metadata)
+          )
+
+          expect(record.type_label).to eq('Video')
+        end
+
+        it 'handles mixed records' do
+          t1 = Track.new(sort_order: 1, title: 'Part 1', path: 'MRCAudio/frost-read1.mp3')
+          t2 = Track.new(sort_order: 2, title: 'Part 2', path: 'MRCAudio/frost-read2.mp4')
+          record = Record.new(
+            collection: 'MRCAudioVideo',
+            tracks: [t2, t1],
+            metadata: instance_double(Metadata)
+          )
+
+          expect(record.type_label).to eq('Audio / Video')
+        end
+      end
+
+      describe :description do
+        it 'gets the description from the 520 tag' do
+          bib_number = 'b22139658'
+          stub_sru_request(bib_number)
+
+          marc_record = alma_marc_record_for(bib_number)
+          expected_desc = marc_record['520']['a']
+
+          record = Record.from_metadata(collection: 'MRCVideo', record_id: bib_number)
+          expect(record.description).to eq(expected_desc)
+        end
+      end
+
+      describe :tind_id do
+        it 'returns nil for Millennium records' do
+          bib_number = 'b22139658'
+          stub_sru_request(bib_number)
+          record = Record.from_metadata(collection: 'MRCVideo', record_id: bib_number)
+          expect(record.tind_id).to be_nil
+        end
+
+        it 'returns the TIND ID for TIND records' do
+          marc_xml = File.read('spec/data/record-(pacradio)01469.xml')
+          search_url = 'https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22%28pacradio%2901469%22&of=xm'
+          stub_request(:get, search_url).to_return(status: 200, body: marc_xml)
+
+          record = Record.from_metadata(
+            collection: 'Pacifica',
+            record_id: '(pacradio)01469'
+          )
+          expected_id = RecordId.new('21178')
+          expect(record.tind_id).to eq(expected_id)
+        end
+      end
+
+      describe :bib_number do
+        it 'returns nil for Alma records with no bib number' do
+          mms_id = '991034756419706532'
+          stub_sru_request(mms_id)
+          record = Record.from_metadata(collection: 'Video-Public-Bancroft', record_id: mms_id)
+          expect(record.bib_number).to be_nil
+        end
+
+        it 'returns nil for TIND records with no bib number' do
+          tind_id = '(clir)00020'
+          marc_xml = File.read('spec/data/record-(clir)00020.xml')
+          search_url = 'https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22%28clir%2900020%22&of=xm'
+          stub_request(:get, search_url).to_return(status: 200, body: marc_xml)
+
+          record = Record.from_metadata(collection: 'Video-Public-Bancroft', record_id: tind_id)
+          expect(record.bib_number).to be_nil
+        end
+      end
+
+      describe :from_metadata do
+        it 'loads the metadata' do
+          marc_xml = File.read('spec/data/record-(pacradio)01469.xml')
+          search_url = 'https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22%28pacradio%2901469%22&of=xm'
+          stub_request(:get, search_url).to_return(status: 200, body: marc_xml)
+
+          record = Record.from_metadata(collection: 'Pacifica', record_id: '(pacradio)01469')
+
+          tracks = record.tracks
+          expect(tracks.size).to eq(1)
+          track = tracks[0]
+          expect(track.sort_order).to eq(0)
+          expect(track.title).to be_nil
+          expect(track.path).to eq('Pacifica/PRA_NHPRC1_AZ1084_00_000_00.mp3')
+          expect(track.duration).to eq(AV::Types::Duration.from_string('00:54:03'))
+
+          record.metadata.tap do |metadata|
+            expect(metadata.bib_number).to eq('b23305522')
+            values = metadata.values_by_field.values
+
+            expected = [
+              'Title (245): Wanda Coleman',
+              'Description (520): Poet Opal Palmer Adisa interviews writer/poet Wanda Coleman, author of Mad Dog, Black Lady, African Sleeping Sickness and Hand Dance, among other books. Coleman discusses when she found her poetic voice, talks about the function of poetry, her personal encounters with anti-Black discrimination, and about the reluctance of white liberals to discuss issues that affect the Black community. She also talks about the plight of the African American community in South Central Los Angeles. The poems Coleman reads are A civilized plague, David Polion, Notes of a cultural terrorist and Jazz wazz.',
+              'Creator (700): Coleman, Wanda. interviewee. Adisa, Opal Palmer. interviewer.',
+              'Creator (710): Pacifica Radio Archive. KPFA (Radio station : Berkeley, Calif.).',
+              'Published (260): Los Angeles, Pacifica Radio Archives, 1993.',
+              'Full Collection Name (982): Pacifica Radio Archives Social Activism Sound Recording Project',
+              'Type (336): Audio',
+              'Extent (300): 1 online resource.',
+              'Archive (852): The Library',
+              "Grant Information (536): Sponsored by the National Historical Publications and Records Commission at the National Archives and Records Administration as part of Pacifica's American Women Making History and Culture: 1963-1982 grant preservation project.",
+              'Usage Statement (540): RESTRICTED.  Permissions, licensing requests, and all other inquiries should be directed in writing to: Director of the Archives, Pacifica Radio Archives, 3729 Cahuenga Blvd. West, North Hollywood, CA 91604, 800-735-0230 x 263, fax 818-506-1084, info@pacificaradioarchives.org, http://www.pacificaradioarchives.org',
+              'Collection (982): Pacifica Radio Archives',
+              'Tracks (998): PRA_NHPRC1_AZ1084_00_000_00.mp3 00:54:03',
+              'Linked Resources (856): [View record in Digital Collections.](https://digicoll.lib.berkeley.edu/record/21178)'
+            ]
+            expect(values.size).to eq(expected.size)
+            aggregate_failures 'fields' do
+              values.each_with_index { |f, i| expect(f.to_s).to eq(expected[i]) }
+            end
+
+            expect(record.title).to eq(metadata.title)
+            expect(record.bib_number).to eq(metadata.bib_number)
+          end
+        end
+
+        it "raises #{AV::RecordNotFound} if the record cannot be found" do
+          search_url = 'https://digicoll.lib.berkeley.edu/search?p=035__a%3A%22%28pacradio%2901469%22&of=xm'
+          stub_request(:get, search_url).to_return(status: 404)
+          expect do
+            Record.from_metadata(
+              collection: 'Pacifica',
+              record_id: '(pacradio)01469'
+            )
+          end.to raise_error(AV::RecordNotFound)
+        end
+      end
+
+      describe :calnet_or_ip? do
+        it 'returns true for restricted, false for unrestricted' do
+          restricted = %w[b18538031 b24071548 (cityarts)00002 (cityarts)00773]
+          unrestricted = %w[b22139658 b23161018 (pacradio)00107 (pacradio)01469]
+          (restricted + unrestricted).each do |record_id|
+            source = Metadata::Source.for_record_id(record_id)
+            if source == Metadata::Source::TIND
+              test_data = "spec/data/record-#{record_id}.xml"
+              stub_request(:get, source.marc_uri_for(record_id)).to_return(status: 200, body: File.read(test_data))
+            else
+              stub_sru_request(record_id)
+            end
+          end
+
+          aggregate_failures 'restrictions' do
+            restricted.each do |record_id|
+              record = Record.from_metadata(collection: 'test', record_id:)
+              expect(record.calnet_or_ip?).to eq(true), "Expected #{record_id} to be restricted, was not"
+            end
+
+            unrestricted.each do |record_id|
+              record = Record.from_metadata(collection: 'test', record_id:)
+              expect(record.calnet_or_ip?).to eq(false), "Expected #{record_id} not to be restricted, was"
+            end
+          end
+        end
+      end
+
+      describe :calnet_only? do
+        it 'returns true for CalNet-only records' do
+          mms_id = '991047179369706532'
+          stub_sru_request(mms_id)
+          record = Record.from_metadata(collection: 'test', record_id: mms_id)
+          expect(record.calnet_or_ip?).to eq(true)
+          expect(record.calnet_only?).to eq(true)
+        end
+
+        it 'returns false for records open to UCB IP addresses' do
+          mms_id = '991054360089706532'
+          stub_sru_request(mms_id)
+          record = Record.from_metadata(collection: 'test', record_id: mms_id)
+          expect(record.calnet_or_ip?).to eq(true)
+          expect(record.calnet_only?).to eq(false)
+        end
+      end
+    end
+  end
+end