f9ce6832f2
This commit provides the following uuid management: (1) Remove uuid record from database after the uuid owned by a Ranger service (cms, fms, ims) resource is deleted. (2) Remove stale uuid record from database when such record is inadvertently created during unsuccessful Ranger service resource creation. (3) Add authentication for uuidgen service delete uuid API. Change-Id: Ifebd68851ff6c0996e7b19d331f4dd99682bd7d2
202 lines
9.4 KiB
Python
Executable File
202 lines
9.4 KiB
Python
Executable File
import pprint
|
|
|
|
from orm.common.orm_common.utils import utils
|
|
from unittest import mock, TestCase
|
|
|
|
|
|
class TestUtil(TestCase):
|
|
@mock.patch('pecan.conf')
|
|
def setUp(self, mock_conf):
|
|
self.mock_response = mock.Mock()
|
|
utils.conf = mock_conf
|
|
|
|
auth_headers = {'X-RANGER-Client': 'cliet',
|
|
'X-RANGER-Requester': 'requestor',
|
|
'X-Auth-Region': 'region',
|
|
'X-Auth-Token': 'token'}
|
|
utils.request = mock.Mock()
|
|
utils.request.headers = auth_headers
|
|
|
|
def respond(self, value, code):
|
|
self.mock_response.json.return_value = value
|
|
self.mock_response.status_code = code
|
|
self.mock_response.text = 'xxxxx'
|
|
return self.mock_response
|
|
|
|
@mock.patch('requests.post')
|
|
def test_make_uuid(self, mock_post):
|
|
mock_post.return_value = self.respond({'uuid': '987654321'}, 200)
|
|
uuid = utils.create_or_validate_uuid('', 'uuidtype')
|
|
self.assertEqual(uuid, '987654321')
|
|
|
|
@mock.patch('orm.common.orm_common.utils.utils.LOG')
|
|
@mock.patch('requests.post')
|
|
def test_make_uuid_offline(self, mock_post, log):
|
|
mock_post.side_effect = Exception('boom')
|
|
uuid = utils.create_or_validate_uuid('', 'uuidtype')
|
|
self.assertEqual(uuid, None)
|
|
log.check(('orm.common.orm_common.utils.utils',
|
|
'INFO', 'Failed in make_uuid:boom'))
|
|
|
|
@mock.patch('requests.post')
|
|
def test_make_transid(self, mock_post):
|
|
mock_post.return_value = self.respond({'uuid': '987654321'}, 200)
|
|
uuid = utils.make_transid()
|
|
self.assertEqual(uuid, '987654321')
|
|
|
|
@mock.patch('orm.common.orm_common.utils.utils.LOG')
|
|
@mock.patch('requests.post')
|
|
def test_make_transid_offline(self, mock_post, l):
|
|
mock_post.side_effect = Exception('boom')
|
|
uuid = utils.make_transid()
|
|
self.assertEqual(uuid, None)
|
|
l.check(('orm_common.utils.utils',
|
|
'INFO', 'Failed in make_transid:boom'))
|
|
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.init')
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.audit')
|
|
def test_audit_trail(self, mock_init, mock_audit):
|
|
resp = utils.audit_trail('create customer', '1234',
|
|
{'X-RANGER-Client': 'Fred'}, '5678')
|
|
self.assertEqual(resp, 200)
|
|
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.audit')
|
|
def test_audit_trail_offline(self, mock_audit):
|
|
mock_audit.side_effect = Exception('boom')
|
|
resp = utils.audit_trail('create customer', '1234',
|
|
{'X-RANGER-Client': 'Fred'}, '5678')
|
|
self.assertEqual(resp, None)
|
|
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.init')
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.audit')
|
|
def test_audit_service_args_least(self, mock_audit, mock_init):
|
|
resp = utils.audit_trail('create customer', '1234',
|
|
{'X-RANGER-Client': 'Fred'}, '5678')
|
|
self.assertEqual(mock_audit.call_args[0][1], 'Fred') # application_id
|
|
self.assertEqual(mock_audit.call_args[0][2], '1234') # tracking_id
|
|
self.assertEqual(mock_audit.call_args[0][3], '1234') # transaction_id
|
|
self.assertEqual(mock_audit.call_args[0][4],
|
|
'create customer') # transaction_type
|
|
self.assertEqual(mock_audit.call_args[0][5], '5678') # resource_id
|
|
# self.assertEqual(mock_audit.call_args[0][6], 'cms') # service
|
|
self.assertEqual(mock_audit.call_args[0][7], '') # user_id
|
|
self.assertEqual(mock_audit.call_args[0][8], 'NA') # external_id
|
|
self.assertEqual(mock_audit.call_args[0][9], '') # event_details
|
|
# self.assertEqual(mock_audit.call_args[0][10], 'Saved to DB') # status
|
|
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.init')
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.audit')
|
|
def test_audit_service_with_tracking(self, mock_audit, mock_init):
|
|
utils.audit_trail('create customer', '1234',
|
|
{'X-RANGER-Client': 'Fred',
|
|
'X-RANGER-Tracking-Id': 'Track12'}, '5678')
|
|
self.assertEqual(mock_audit.call_args[0][1], 'Fred') # application_id
|
|
self.assertEqual(mock_audit.call_args[0][2], 'Track12') # tracking_id
|
|
self.assertEqual(mock_audit.call_args[0][3], '1234') # transaction_id
|
|
self.assertEqual(mock_audit.call_args[0][4],
|
|
'create customer') # transaction_type
|
|
self.assertEqual(mock_audit.call_args[0][5], '5678') # resource_id
|
|
# self.assertEqual(mock_audit.call_args[0][6], 'cms') # service
|
|
self.assertEqual(mock_audit.call_args[0][7], '') # user_id
|
|
self.assertEqual(mock_audit.call_args[0][8], 'NA') # external_id
|
|
self.assertEqual(mock_audit.call_args[0][9], '') # event_details
|
|
# self.assertEqual(mock_audit.call_args[0][10], 'Saved to DB') # status
|
|
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.init')
|
|
@mock.patch('orm.common.client.audit.audit_client.api.audit.audit')
|
|
def test_audit_service_with_requester(self, mock_audit, mock_init):
|
|
resp = utils.audit_trail('create customer', '1234',
|
|
{'X-RANGER-Client': 'Fred',
|
|
'X-RANGER-Requester': 'Req04'}, '5678')
|
|
self.assertEqual(mock_audit.call_args[0][1], 'Fred') # application_id
|
|
self.assertEqual(mock_audit.call_args[0][2], '1234') # tracking_id
|
|
self.assertEqual(mock_audit.call_args[0][3], '1234') # transaction_id
|
|
self.assertEqual(mock_audit.call_args[0][4], 'create customer') # transaction_type
|
|
self.assertEqual(mock_audit.call_args[0][5], '5678') # resource_id
|
|
# self.assertEqual(mock_audit.call_args[0][6], 'cms') # service
|
|
self.assertEqual(mock_audit.call_args[0][7], 'Req04') # user_id
|
|
self.assertEqual(mock_audit.call_args[0][8], 'NA') # external_id
|
|
self.assertEqual(mock_audit.call_args[0][9], '') # event_details
|
|
# self.assertEqual(mock_audit.call_args[0][10], 'Saved to DB') # status
|
|
|
|
def test_set_utils_conf(self):
|
|
utils.set_utils_conf('test')
|
|
self.assertEqual(utils.conf, 'test')
|
|
|
|
def test_check_conf_initialization(self):
|
|
utils.set_utils_conf(None)
|
|
self.assertRaises(AssertionError, utils._check_conf_initialization)
|
|
|
|
@mock.patch('requests.post')
|
|
def test_create_existing_uuid(self, mock_post):
|
|
uuid = '987654321'
|
|
uuid_type = 'testtype'
|
|
mock_post.return_value = self.respond(
|
|
{'uuid': uuid, 'uuid_type': uuid_type}, 200)
|
|
returned_uuid = utils.create_or_validate_uuid(uuid, uuid_type)
|
|
self.assertEqual(returned_uuid, uuid)
|
|
|
|
@mock.patch('requests.post')
|
|
def test_create_existing_uuid_with_exception(self, mock_post):
|
|
mock_post.side_effect = Exception('boom')
|
|
uuid = '987654321'
|
|
uuid_type = 'testtype'
|
|
returned_uuid = utils.create_or_validate_uuid(uuid, uuid_type)
|
|
self.assertEqual(returned_uuid, None)
|
|
|
|
@mock.patch('requests.post')
|
|
def test_create_existing_uuid_with_uuid_fail(self, mock_post):
|
|
uuid = '987654321'
|
|
uuid_type = 'testId'
|
|
mock_post.return_value = self.respond({'uuid': uuid, 'uuid_type': uuid_type}, 409)
|
|
self.assertRaises(TypeError, utils.create_or_validate_uuid, uuid, uuid_type)
|
|
|
|
@mock.patch('pecan.conf')
|
|
def test_report_config(self, mock_conf):
|
|
expected_value = pprint.pformat(mock_conf.to_dict(), indent=4)
|
|
returned_value = utils.report_config(mock_conf)
|
|
self.assertEqual(expected_value, returned_value)
|
|
|
|
@mock.patch('pecan.conf')
|
|
def test_report_config_with_log_write(self, mock_conf):
|
|
expected_value = pprint.pformat(mock_conf.to_dict(), indent=4)
|
|
returned_value = utils.report_config(mock_conf, True)
|
|
self.assertEqual(expected_value, returned_value)
|
|
|
|
@mock.patch('requests.get')
|
|
def test_get_resource_status_sanity(self, mock_get):
|
|
my_response = mock.MagicMock()
|
|
my_response.status_code = 200
|
|
my_response.json.return_value = 'test'
|
|
mock_get.return_value = my_response
|
|
result = utils.get_resource_status('A')
|
|
self.assertEqual(result, 'test')
|
|
|
|
@mock.patch('requests.get', side_effect=ValueError())
|
|
def test_get_resource_status_get_failed(self, mock_get):
|
|
self.assertIsNone(utils.get_resource_status('A'))
|
|
|
|
@mock.patch('requests.get')
|
|
def test_get_resource_status_invalid_response(self, mock_get):
|
|
my_response = mock.MagicMock()
|
|
my_response.status_code = 404
|
|
mock_get.return_value = my_response
|
|
self.assertIsNone(utils.get_resource_status('A'))
|
|
|
|
@mock.patch('requests.delete')
|
|
def test_delete_uuid_success(self, mock_delete):
|
|
mock_delete.return_value = self.respond(None, 200)
|
|
result = utils.delete_uuid('987654321')
|
|
self.assertEqual(result, True)
|
|
|
|
@mock.patch('requests.delete')
|
|
def test_delete_uuid_failed(self, mock_delete):
|
|
mock_delete.return_value = self.respond("Failed", 500)
|
|
result = utils.delete_uuid('987654321')
|
|
self.assertEqual(result, False)
|
|
|
|
@mock.patch('requests.delete', side_effect=SystemError())
|
|
def test_delete_uuid_exception(self, mock_delete):
|
|
result = utils.delete_uuid('987654321')
|
|
self.assertEqual(result, False)
|