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)