class RestTest(unittest.TestCase):
"""
Test Rest queries
"""
def setUp(self):
inst_id = 111637
wskey = 'TjBKm4f7QdZwxUrvfnukshyAIPkCgt3ZieslDR23Z95rV8rmqU3gIFvKRRDaTwX4UwzoQtQYIbyCqEWe'
self.client_json = Rest(inst_id, wskey, response_format="json")
self.client_xml = Rest(inst_id, wskey)
def tearDown(self):
pass
def testGetProvider(self):
'''Compare XML and JSON get_provider calls'''
npg_xml = self.client_xml.get_provider('NPG')
npg_json = self.client_json.get_provider('NPG')
self.assertEqual(len(npg_xml), 1, 'No XML result.')
self.assertEqual(len(npg_json), 1, 'No JSON result.')
self.assertEqual(npg_xml[0]['id'], npg_json[0]['id'], 'XML and JSON ids are different.')
self.assertEqual(npg_xml, npg_json)
def testListProvidersLength(self):
'''Compare result size of XML and JSON list_providers calls'''
xml = self.client_xml.list_providers()
json = self.client_json.list_providers()
self.assertEqual(len(xml['entries']), len(json['entries']), 'XML and JSON result sizes are different.')
def testSearchProvidersLength(self):
'''Compare result size of XML and JSON search_providers calls'''
xml = self.client_xml.search_providers('Nature')
json = self.client_json.search_providers('Nature')
self.assertEqual(len(xml['entries']), len(json['entries']), 'XML and JSON result sizes are different')
def testGetCollection(self):
'''Compare XML and JSON get_collection calls'''
npg_xml = self.client_xml.get_collection('NPG.journals')
npg_json = self.client_json.get_collection('NPG.journals')
self.assertEqual(len(npg_xml), 1, 'No XML result.')
self.assertEqual(len(npg_json), 1, 'No JSON result.')
self.assertEqual(npg_xml[0]['id'], npg_json[0]['id'], 'XML and JSON ids are different')
self.assertEqual(npg_xml, npg_json)
def testGetEntrybyID(self):
'''Compare results of get_entry with an entry_id'''
entry_xml = self.client_xml.get_entry('036f688a982b7a702aadd05003f9742e')
entry_json = self.client_json.get_entry('036f688a982b7a702aadd05003f9742e')
self.assertEqual(len(entry_xml), 1, 'No XML result.')
self.assertEqual(len(entry_json), 1, 'No JSON result.')
self.assertEqual(entry_xml[0]['id'], entry_json[0]['id'], 'XML and JSON ids are different.')
self.assertEqual(entry_xml, entry_json)
def testGetEntrybyUID(self):
'''Compare results of get_entry with an entry_uid'''
entry_xml = self.client_xml.get_entry('NPG.journals,1987357')
entry_json = self.client_json.get_entry('NPG.journals,1987357')
self.assertEqual(len(entry_xml), 1, 'No XML result.')
self.assertEqual(len(entry_json), 1, 'No JSON result.')
self.assertEqual(entry_xml[0]['id'], entry_json[0]['id'], 'XML and JSON ids are different.')
self.assertEqual(entry_xml, entry_json)
def setUp(self):
inst_id = 111637
wskey = 'TjBKm4f7QdZwxUrvfnukshyAIPkCgt3ZieslDR23Z95rV8rmqU3gIFvKRRDaTwX4UwzoQtQYIbyCqEWe'
self.client_json = Rest(inst_id, wskey, response_format="json")
self.client_xml = Rest(inst_id, wskey)
class RestTest(unittest.TestCase):
"""
Test Rest query client
"""
def setUp(self):
inst_id = 111637
wskey = 'TjBKm4f7QdZwxUrvfnukshyAIPkCgt3ZieslDR23Z95rV8rmqU3gIFvKRRDaTwX4UwzoQtQYIbyCqEWe'
self.client_json = Rest(inst_id, wskey, response_format="json")
self.client_xml = Rest(inst_id, wskey)
def tearDown(self):
pass
def testMapping(self):
'''create_query_params should do some mappings to the right api field'''
self.assertIn('q', self.client_xml.create_query_params(keyword='test'))
self.assertIn('start-index', self.client_xml.create_query_params(start_index=10))
self.assertIn('max-results', self.client_xml.create_query_params(max_results=10))
self.assertIn('order-by', self.client_xml.create_query_params(order_by='title'))
def testAdditionalFields(self):
'''create_query_params should add some additional fields'''
self.assertIn('institution_id', self.client_xml.create_query_params(keyword='test'))
self.assertIn('wskey', self.client_xml.create_query_params(keyword='test'))
def testAltFormat(self):
self.assertIn('alt', self.client_xml.create_query_params(keyword='test'))
self.assertIn('alt', self.client_json.create_query_params(keyword='test'))
self.assertEqual('xml', self.client_xml.create_query_params(keyword='test')['alt'])
self.assertEqual('json', self.client_json.create_query_params(keyword='test')['alt'])
self.assertNotEqual('json', self.client_xml.create_query_params(keyword='test')['alt'])
self.assertNotEqual('xml', self.client_json.create_query_params(keyword='test')['alt'])
