def test_unglueit_summary_fetch_error(self, mock_get):
unglueTest = Unglueit('999999999')
mock_req = MagicMock()
mock_get.return_value = mock_req
mock_req.status_code = 500
with self.assertRaises(UnglueError):
unglueTest.getSummary('1')
def test_uglueit_work_search_error(self, mock_get):
unglueTest = Unglueit('999999999')
mock_req = MagicMock()
mock_get.return_value = mock_req
mock_req.status_code = 500
with self.assertRaises(UnglueError):
unglueTest.getWork()
def test_unglueit_summary_fetch(self, mock_parse, mock_get):
unglueTest = Unglueit('999999999')
mock_req = MagicMock()
mock_get.return_value = mock_req
mock_req.status_code = 200
mock_req.text = 'fakeXMLDocument'
testSummary = unglueTest.getSummary('1')
mock_parse.assert_called_once_with('fakeXMLDocument')
self.assertEqual(testSummary, 'fakeXMLDocument')
def test_unglueit_opds_parse(self):
testXML = """<feed xmlns="http://www.w3.org/2005/Atom">
<title>Test Unglue.it Record</title>
<entry xmlns:ns0="http://www.w3.org/2005/Atom">
<title>Test Unglue.it Record</title>
<content ns0:type="html">A summary of the work</content>
</entry>
</feed>
"""
unglueTest = Unglueit('9999999999')
summaryText = unglueTest.parseOPDS(testXML)
self.assertEqual(summaryText, 'A summary of the work')
def test_summary_fetch(self, mock_resp, mock_summary, mock_work):
unglueTest = Unglueit('9999999999')
output = unglueTest.fetchSummary()
mock_work.assert_called_once()
mock_summary.assert_called_once_with(1)
mock_resp.assert_called_once_with(
200,
{
'match': True,
'isbn': '9999999999',
'summary': 'summary'
}
)
self.assertTrue(output)
def test_uglueit_work_id_error(self, mock_get):
unglueTest = Unglueit('999999999')
mock_req = MagicMock()
mock_get.return_value = mock_req
mock_req.status_code = 200
mock_req.json.return_value = {
'objects': [
{
'work': '/api/test/something/'
}
]
}
with self.assertRaises(UnglueError):
unglueTest.getWork()
def test_uglueit_work_search(self, mock_get):
unglueTest = Unglueit('999999999')
mock_req = MagicMock()
mock_get.return_value = mock_req
mock_req.status_code = 200
mock_req.json.return_value = {
'objects': [
{
'work': '/api/test/1/'
}
]
}
testWorkID = unglueTest.getWork()
self.assertEqual(testWorkID, '1')
def handler(event, context):
"""The central handler response to invocations from event sources. For this
function these come from the API Gateway
Arguments:
event {dict} -- Dictionary containing contents of the event that
invoked the function, primarily the payload of data to be processed.
context {LambdaContext} -- An object containing metadata describing
the event source and client details.
Raises:
InvalidExecutionType -- Raised when GET parameters are missing or are
malformed.
Returns:
[dict] -- An object that is returned to the service that
originated the API call to the API gateway.
"""
logger.info('Starting Lambda Execution')
logger.debug(event)
try:
isbn = event['queryStringParameters']['isbn']
except KeyError:
logger.error('Missing required lookup ISBN parameter')
raise InvalidExecutionType('isbn parameter required')
unglued = Unglueit(isbn)
try:
unglued.validate()
returnObj = unglued.fetchSummary()
except UnglueError as err:
logger.error(err)
returnObj = Unglueit.formatResponse(err.status, {
'match': False,
'message': err.message
})
return returnObj
def test_validate_invalid_character_count(self):
validateTest = Unglueit('999999999')
with self.assertRaises(UnglueError):
validateTest.validate()
def test_validate_invalid_characters(self):
validateTest = Unglueit('fake iSBN 978xxxxx')
with self.assertRaises(UnglueError):
validateTest.validate()
def test_validate_isbn13(self, mock_isbn13):
validateTest = Unglueit('978-99-999-9999-9')
validateTest.validate()
mock_isbn13.assert_called_once_with('9789999999999')
def test_create_search_object(self):
testInstance = Unglueit('9999999999')
self.assertEqual(testInstance.isbn, '9999999999')
self.assertEqual(testInstance.unglueitSearch, 'unglueISBN')