Fixed RESQL test cases for Roles module.

This commit is contained in:
Aditya Toshniwal
2020-08-28 18:22:25 +05:30
committed by Akshay Joshi
parent 3984544bdb
commit 11d5613866
11 changed files with 77 additions and 22 deletions
+8 -4
View File
@@ -281,7 +281,7 @@ class ReverseEngineeredSQLTestCases(BaseTestGenerator):
# Store the object id based on endpoints
if 'store_object_id' in scenario:
self.store_object_ids(object_id,
scenario['data']['name'],
scenario['data'],
scenario['endpoint'])
# Compare the reverse engineering SQL
@@ -324,8 +324,9 @@ class ReverseEngineeredSQLTestCases(BaseTestGenerator):
elif 'type' in scenario and scenario['type'] == 'delete':
# Get the delete url and delete the object created above.
delete_url = self.get_url(scenario['endpoint'], object_id)
delete_response = self.tester.delete(delete_url,
follow_redirects=True)
delete_response = self.tester.delete(
delete_url, data=json.dumps(scenario.get('data', {})),
follow_redirects=True)
try:
self.assertEquals(delete_response.status_code, 200)
except Exception as e:
@@ -628,7 +629,7 @@ class ReverseEngineeredSQLTestCases(BaseTestGenerator):
return sql
def store_object_ids(self, object_id, object_name, endpoint):
def store_object_ids(self, object_id, object_data, endpoint):
"""
This functions will store the object id based on endpoints
:param object_id: Object id of the created node
@@ -636,12 +637,15 @@ class ReverseEngineeredSQLTestCases(BaseTestGenerator):
:param endpoint:
:return:
"""
object_name = object_data.get('name', '')
if endpoint.__contains__("NODE-table"):
self.parent_ids['tid'] = object_id
elif endpoint.__contains__("NODE-foreign_data_wrapper"):
self.parent_ids['fid'] = object_id
elif endpoint.__contains__("NODE-foreign_server"):
self.parent_ids['fsid'] = object_id
elif endpoint.__contains__("NODE-role.obj"):
object_name = object_data['rolname']
# Store object id with object name
self.all_object_ids[object_name] = object_id