def test_get_user_dataset_access(self, mock_get):
with app.app_context() as ac:
ac.g.trace_id = None
ac.g.requests = requests.Session()
with app.test_request_context():
dataset_access = [{
'name': 'res_cov',
'title': 'Restrictive Covenants',
'licences': {
'res_cov_direct': {
'title': 'Direct Use',
'agreed': True
}
}
}]
mock_get.return_value.json.return_value = dataset_access
mock_get.return_value.status_code = 200
response = UlapdAPI.get_user_dataset_access(self, 1)
mock_get.assert_called_once()
mock_get.assert_called_with(
'{0}/users/dataset-access/{1}'.format(self.url, 1),
headers=self.headers,
timeout=self.timeout)
self.assertEqual(response, dataset_access)
def test_get_user_dataset_access_connection_error(self, mock_get):
with app.app_context() as ac:
ac.g.trace_id = None
ac.g.requests = requests.Session()
with app.test_request_context():
mock_get.side_effect = ConnectionError(self.error_msg)
with self.assertRaises(ApplicationError) as context:
UlapdAPI.get_user_dataset_access(self, 1)
expected_err = ('verification_api', 'ULAPD_API_CONN_ERROR')
expected_err_message = errors.get_message(
*expected_err, filler=self.error_msg)
expected_err_code = errors.get_code(*expected_err)
mock_get.assert_called_once()
self.assertEqual(context.exception.message,
expected_err_message)
self.assertEqual(context.exception.code, expected_err_code)