def test_getNewItemLinks_pd(self, mockCreate, mockReq):
testItem = {
'rightsCode': 'pd',
'itemURL': 'hathitrust.org/302-redirect'
}
mockResp = MagicMock()
mockResp.headers = {
'Location': 'https://hathitrust.org/test?id=test.123465'
}
mockReq.head.return_value = mockResp
mockInstance = MagicMock()
testInst = HoldingParser('mock856', mockInstance)
testInst.source = 'hathitrust'
testInst.identifier = '123465'
outItem = testInst.getNewItemLinks(testItem)
self.assertEqual(outItem['source'], 'hathitrust')
self.assertEqual(len(outItem['links']), 2)
mockCreate.assert_has_calls([
call('hathitrust.org/test?id=test.123465',
'text/html',
local=False,
download=False,
images=True,
ebook=False),
call('babel.hathitrust.org/cgi/imgsrv/download/pdf?id=test.123465',
'application/pdf',
local=False,
download=True,
images=True,
ebook=False)
])
def test_getNewItemLinks_not_pd(self):
testItem = {'rightsCode': 'ic-world'}
testInst = HoldingParser('mock856', 'mockInstance')
self.assertEqual(testInst.getNewItemLinks(testItem), None)
