def setUp(self):
response = requests.get('https://isso.nypl.org')
self.assertTrue(response.ok)
user_data = shelve.open(setup_dirs.USER_DATA)
api_params = user_data['PlatformAPIs']['NYPL TEST']
self.client_id = base64.b64decode(api_params['client_id'])
self.oauth_server = api_params['oauth_server']
self.base_url = api_params['host']
user_data.close()
# get password
self.client_secret = credentials.get_from_vault(
self.oauth_server, self.client_id)
self.auth = AuthorizeAccess(self.client_id, self.client_secret,
self.oauth_server)
self.token = self.auth.get_token()
self.sess = PlatformSession(self.base_url, self.token)
self.sess.headers.update({'user-agent': 'overload/TESTS'})
def test_query_bibId(self, mock_query):
# way to verify correct methods are hijacked
assert mock_query is PlatformSession.query_bibUpdatedDate
expires_on = datetime.now() + timedelta(seconds=5)
token = {"expires_on": expires_on, "id": "abc1234"}
sess = PlatformSession(self.base_url, token)
keywords = ["12345"]
limit = 1
mock_query.return_value = "bar"
res = sess.query_bibUpdatedDate(keywords, limit=limit)
self.assertEqual(res, "bar")
mock_query.assert_called_with(["12345"], limit=limit)
# test exceptions are raised
mock_query.side_effect = [Timeout, ConnectionError]
with self.assertRaises(Timeout):
sess.query_bibUpdatedDate(keywords)
sess.query_bibUpdatedDate(keywords)
def test_open_session_method(self):
expires_on = datetime.now() + timedelta(seconds=5)
token = {"expires_on": expires_on, "id": "abc1234"}
sess = PlatformSession(self.base_url, token)
sess.headers.update({"User-Agent": "Overload/TESTS"})
self.assertIsInstance(sess, requests.Session)
self.assertEqual(
sess.headers,
{
"Authorization": "Bearer abc1234",
"Connection": "keep-alive",
"Accept-Encoding": "gzip, deflate",
"Accept": "*/*",
"User-Agent": "Overload/TESTS",
"Authorization": "Bearer " + token.get("id"),
},
)
def test_validate_token_exception(self):
expires_on = datetime.now() - timedelta(seconds=1)
token = {"expires_on": expires_on, "id": "abc1234"}
with self.assertRaises(APITokenExpiredError):
PlatformSession(self.base_url, token)
def test_platform_session_invalid_attrib(self):
with self.assertRaises(ValueError):
PlatformSession()
def test_methods_in_platform_session(self):
expires_on = datetime.now() + timedelta(seconds=5)
token = {"expires_on": expires_on, "id": "abc1234"}
sess = PlatformSession(self.base_url, token)
self.assertEquals(
dir(sess),
[
"__attrs__",
"__class__",
"__delattr__",
"__dict__",
"__doc__",
"__enter__",
"__exit__",
"__format__",
"__getattribute__",
"__getstate__",
"__hash__",
"__init__",
"__module__",
"__new__",
"__reduce__",
"__reduce_ex__",
"__repr__",
"__setattr__",
"__setstate__",
"__sizeof__",
"__str__",
"__subclasshook__",
"__weakref__",
"_validate_token",
"adapters",
"auth",
"base_url",
"cert",
"close",
"cookies",
"delete",
"get",
"get_adapter",
"get_bibItems",
"get_item",
"get_redirect_target",
"head",
"headers",
"hooks",
"max_redirects",
"merge_environment_settings",
"mount",
"options",
"params",
"patch",
"post",
"prepare_request",
"proxies",
"put",
"query_bibControlNo",
"query_bibCreatedDate",
"query_bibId",
"query_bibStandardNo",
"query_bibUpdatedDate",
"query_itemBarcode",
"query_itemBibId",
"query_itemCreatedDate",
"query_itemId",
"query_itemUpdateddDate",
"rebuild_auth",
"rebuild_method",
"rebuild_proxies",
"request",
"resolve_redirects",
"send",
"stream",
"timeout",
"token",
"trust_env",
"verify",
],
)
class TestPlatformMethodsOnDev(unittest.TestCase):
def setUp(self):
response = requests.get('https://isso.nypl.org')
self.assertTrue(response.ok)
user_data = shelve.open(setup_dirs.USER_DATA)
api_params = user_data['PlatformAPIs']['NYPL TEST']
self.client_id = base64.b64decode(api_params['client_id'])
self.oauth_server = api_params['oauth_server']
self.base_url = api_params['host']
user_data.close()
# get password
self.client_secret = credentials.get_from_vault(
self.oauth_server, self.client_id)
self.auth = AuthorizeAccess(self.client_id, self.client_secret,
self.oauth_server)
self.token = self.auth.get_token()
self.sess = PlatformSession(self.base_url, self.token)
self.sess.headers.update({'user-agent': 'overload/TESTS'})
def tearDown(self):
self.sess.close()
def test_bib_data_structure(self):
endpoint = self.base_url + '/bibs'
payload = dict(nyplSource='sierra-nypl', limit=1, id='21541362')
res = self.sess.get(endpoint, params=payload)
self.assertEqual(
sorted(res.json().keys()),
sorted(['count', 'totalCount', 'data', 'debugInfo', 'statusCode']))
self.assertEqual(
sorted(res.json()['data'][0].keys()),
sorted([
u'varFields', u'materialType', u'locations',
u'standardNumbers', 'controlNumber', u'id', u'author',
u'normAuthor', u'deletedDate', u'normTitle', u'nyplSource',
u'deleted', u'createdDate', u'suppressed', u'publishYear',
u'lang', u'catalogDate', u'fixedFields', u'country',
u'nyplType', u'updatedDate', u'title', u'bibLevel'
]))
self.assertEqual(sorted(res.json()['data'][0]['locations'][0].keys()),
sorted(['code', 'name']))
self.assertEqual(
sorted(res.json()['data'][0]['varFields'][0].keys()),
sorted([
'marcTag', 'ind1', 'ind2', 'content', 'fieldTag', 'subfields'
]))
self.assertEqual(
sorted(
res.json()['data'][0]['varFields'][0]['subfields'][0].keys()),
sorted(['content', 'tag']))
def test_bibItems_data_structure(self):
endpoint = self.base_url + '/bibs/sierra-nypl/17746307/items'
res = self.sess.get(endpoint)
self.assertEqual(
sorted(res.json().keys()),
sorted(['count', 'debugInfo', 'data', 'statusCode', 'totalCount']))
self.assertEqual(
sorted(res.json()['data'][0].keys()),
sorted([
u'deletedDate', u'status', u'nyplSource', u'varFields',
u'itemType', u'fixedFields', u'nyplType', u'deleted',
u'barcode', u'bibIds', u'callNumber', u'updatedDate',
u'location', u'createdDate', u'id'
]))
self.assertEqual(sorted(res.json()['data'][0]['location'].keys()),
sorted([u'code', u'name']))
self.assertEqual(res.json()['data'][0]['varFields'][0].keys(), [
u'marcTag', u'ind1', u'ind2', u'content', u'fieldTag', u'subfields'
])
self.assertEqual(res.json()['data'][0]['status'].keys(),
[u'code', u'display', u'duedate'])
def test_query_bibStandardNo(self):
res = self.sess.query_bibStandardNo(keywords=['9781302905620'])
self.assertTrue(res.status_code in (200, 404))
def test_query_bibControlNo(self):
res = self.sess.query_bibControlNo(keywords=['58948871'])
self.assertTrue(res.status_code in (200, 404))
def test_query_bibId(self):
res = self.sess.query_bibId(keywords=['21310805'])
self.assertTrue(res.status_code in (200, 404))
def test_query_bibCreatedDate(self):
res = self.sess.query_bibCreatedDate('2013-09-03T13:17:45Z',
'2014-09-03T13:17:45Z',
limit=1)
self.assertTrue(res.status_code in (200, 404))
def test_bibUpdatedDate(self):
res = self.sess.query_bibUpdatedDate('2013-09-03T13:17:45Z',
'2014-09-03T13:17:45Z',
limit=1)
self.assertTrue(res.status_code in (200, 404))
def test_get_bibItems(self):
res = self.sess.get_bibItems('21310805')
self.assertTrue(res.status_code in (200, 404))
def query_itemId(self):
res = self.sess.query_itemId(keywords=['35366967, 35366972'])
self.assertTrue(res.status_code in (200, 404))
def query_itemBarcode(self):
res = self.sess.query_itemBarcode(keyword='33333844889317')
self.assertTrue(res.status_code in (200, 404))
def query_itemBibId(self):
res = self.sess.queryBibId(keyword='21310805', limit=5)
self.assertTrue(res.status_code in (200, 404))
def query_itemCreatedDate(self):
res = self.sess.query_itemCreatedDate('2013-09-03T13:17:45Z',
'2014-09-03T13:17:45Z',
limit=5)
self.assertTrue(res.status_code in (200, 404))
def query_itemUpdatedDate(self):
res = self.sess.query_itemUpdatedDate('2013-09-03T13:17:45Z',
'2014-09-03T13:17:45Z',
limit=5)
self.assertTrue(res.status_code in (200, 404))
def get_item(self):
res = self.sess.get_item(keyword='35362061')
self.assertTrue(res.status_code in (200, 404))