def setUp(self):
"""
测试case开始
:resurn:
"""
Gremlin().gremlin_post('graph.truncateBackend();',
auth=auth) # gremlin语句进行clear操作
# 创建group
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
# 创建 target
body = {
"target_url":
'%s:%d' % (_cfg.graph_host, _cfg.server_port),
"target_name":
"gremlin",
"target_graph":
_cfg.graph_name,
"target_resources": [{
"type": "GREMLIN",
"properties": None,
"label": "*"
}]
}
code, res = Auth().post_targets(body, auth=auth)
print(code, res)
def test_target_update(self):
"""
获取 target
"""
# premise
body = {
"target_url":
'%s:%d' % (_cfg.graph_host, _cfg.server_port),
"target_name":
"gremlin",
"target_graph":
_cfg.graph_name,
"target_resources": [{
"type": "GREMLIN",
"properties": None,
"label": "*"
}]
}
code, res = Auth().post_targets(body, auth=auth)
print(code, res)
# test
body = {
"target_name": "gremlin",
"target_url": '%s:%d' % (_cfg.graph_host, _cfg.server_port),
"target_graph": _cfg.graph_name,
"target_resources": [{
"type": "VERTEX"
}]
}
code, res = Auth().update_targets(body, res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['target_resources'][0]['type'], 'VERTEX',
'res check fail')
def __init__(self):
"""初始化"""
self.auth = Auth()
self.target_graph = _cfg.graph_name
self.target_url = _cfg.graph_host + ":" + str(_cfg.server_port)
self.auth_user = "******"
self.auth_password = _cfg.test_password['tester']
def test_user_delete(self):
"""
删除 user
"""
# premise
body = {"user_name": "tester", "user_password": "******"}
code, res = Auth().post_users(body, auth=auth)
print(code, res)
# test
code, res = Auth().delete_users(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 204, msg='code check fail')
def test_user_one(self):
"""
获取user
"""
# premise
body = {"user_name": "tester", "user_password": "******"}
code, res = Auth().post_users(body, auth=auth)
print(code, res)
# test
code, res = Auth().get_user(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['id'], '-63:tester', 'res check fail')
def test_access_delete(self):
"""
删除 access
"""
# premise
body = {
'group': '-69:gremlin',
'target': '-77:gremlin',
'access_permission': 'EXECUTE'
}
code, res = Auth().post_accesses(body, auth=auth)
# test
code, res = Auth().delete_accesses(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 204, msg='code check fail')
def test_groups_delete(self):
"""
删除 groups
"""
# premise
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
# test
code, res = Auth().delete_groups(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 204, msg='code check fail')
def test_groups_one(self):
"""
获取 group
"""
# premise
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
# test
code, res = Auth().get_group(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['id'], '-69:gremlin', 'res check fail')
def test_user_one_role(self):
"""
获取user角色
"""
# premise
permission_list = [{
'target_list': [
{
'type': 'GRANT'
},
# {'type': 'STATUS'}
],
'permission': 'WRITE',
'name': 'grant_write'
}]
user_id = set_auth.post_auth(permission_list)
# test
code, res = Auth().get_users_role(user_id, auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(
res['roles'], {
'hugegraph': {
'WRITE': [{
'type': 'GRANT',
'label': '*',
'properties': None
}]
}
}, 'res check fail')
def test_belong_delete(self):
"""
删除 belong
"""
# premise
body = {
'group': '-69:gremlin',
'user': '******',
'belong_description': 'belong gremlin'
}
code, res = Auth().post_belongs(body, auth=auth)
print(code, res)
# test
code, res = Auth().delete_belongs(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 204, msg='code check fail')
def test_access_list(self):
"""
获取 access
"""
# premise
body = {
'group': '-69:gremlin',
'target': '-77:gremlin',
'access_permission': 'EXECUTE'
}
code, res = Auth().post_accesses(body, auth=auth)
# test
code, res = Auth().get_accesses(auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['accesses'][0]['id'],
'S-69:gremlin>-88>18>S-77:gremlin', 'res check fail')
def test_access_update(self):
"""
更新 access
"""
# premise
body = {
'group': '-69:gremlin',
'target': '-77:gremlin',
'access_permission': 'EXECUTE'
}
code, res = Auth().post_accesses(body, auth=auth)
# test
body = {"access_description": "access description rename"}
code, res = Auth().update_accesses(body, res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['id'], 'S-69:gremlin>-88>18>S-77:gremlin',
'res check fail')
def test_belong_one(self):
"""
获取 belong
"""
# premise
body = {
'group': '-69:gremlin',
'user': '******',
'belong_description': 'belong gremlin'
}
code, res = Auth().post_belongs(body, auth=auth)
print(code, res)
# test
code, res = Auth().get_belong(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['id'], 'S-63:tester>-82>>S-69:gremlin',
'res check fail')
def setUp(self):
"""
测试case开始
:resurn:
"""
Gremlin().gremlin_post('graph.truncateBackend();',
auth=auth) # gremlin语句进行clear操作
# 创建group
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
# 创建 user
body = {"user_name": "tester", "user_password": "******"}
code, res = Auth().post_users(body, auth=auth)
print(code, res)
def test_user_create(self):
"""
创建 user
"""
body = {"user_name": "tester", "user_password": "******"}
code, res = Auth().post_users(body, auth=auth)
print(code, res)
self.assertEqual(code, 201, msg='code check fail')
self.assertEqual(res['id'], '-63:tester', 'res check fail')
def test_belong_update(self):
"""
更新 belong
"""
# premise
body = {
'group': '-69:gremlin',
'user': '******',
'belong_description': 'belong gremlin'
}
code, res = Auth().post_belongs(body, auth=auth)
print(code, res)
# test
body = {"belong_description": "belong description rename"}
code, res = Auth().update_belongs(body, res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['id'], 'S-63:tester>-82>>S-69:gremlin',
'res check fail')
def test_user_update(self):
"""
获取 user
"""
# premise
body = {"user_name": "tester", "user_password": "******"}
code, res = Auth().post_users(body, auth=auth)
print(code, res)
# test
body = {
"user_name": "tester",
"user_password": "******",
"user_phone": "138",
"user_avatar": "tester.png",
"user_email": "*****@*****.**"
}
code, res = Auth().update_users(body, res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['user_avatar'], 'tester.png', 'res check fail')
def test_groups_update(self):
"""
更新groups
"""
# premise
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
# test
body = {
"group_name": "gremlin",
"group_description": "group_update description"
}
code, res = Auth().update_groups(body, res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 200, msg='code check fail')
self.assertEqual(res['group_description'], 'group_update description',
'res check fail')
def test_groups_create(self):
"""
创建 groups
"""
body = {
"group_name": "gremlin",
"group_description": "group can execute gremlin"
}
code, res = Auth().post_groups(body, auth=auth)
print(code, res)
self.assertEqual(code, 201, 'code check fail')
self.assertEqual(res['group_name'], 'gremlin', 'code check fail')
def test_access_create(self):
"""
创建 access
"""
body = {
'group': '-69:gremlin',
'target': '-77:gremlin',
'access_permission': 'EXECUTE'
}
code, res = Auth().post_accesses(body, auth=auth)
print(code, res)
self.assertEqual(code, 201, msg='code check fail')
self.assertEqual(res['id'], 'S-69:gremlin>-88>18>S-77:gremlin',
'res check fail')
def test_belong_create(self):
"""
创建 belong
"""
body = {
'group': '-69:gremlin',
'user': '******',
'belong_description': 'belong gremlin'
}
code, res = Auth().post_belongs(body, auth=auth)
print(code, res)
self.assertEqual(code, 201, msg='code check fail')
self.assertEqual(res['id'], 'S-63:tester>-82>>S-69:gremlin',
'res check fail')
def test_target_delete(self):
"""
删除 target
"""
# premise
body = {
"target_url":
'%s:%d' % (_cfg.graph_host, _cfg.server_port),
"target_name":
"gremlin",
"target_graph":
_cfg.graph_name,
"target_resources": [{
"type": "GREMLIN",
"properties": None,
"label": "*"
}]
}
code, res = Auth().post_targets(body, auth=auth)
print(code, res)
# test
code, res = Auth().delete_targets(res['id'], auth=auth)
print(code, res)
self.assertEqual(code, 204, msg='code check fail')
def test_target_create(self):
"""
创建 target
"""
body = {
"target_url":
'%s:%d' % (_cfg.graph_host, _cfg.server_port),
"target_name":
"gremlin",
"target_graph":
_cfg.graph_name,
"target_resources": [{
"type": "GREMLIN",
"properties": None,
"label": "*"
}]
}
code, res = Auth().post_targets(body, auth=auth)
print(code, res)
self.assertEqual(code, 201, msg='code check fail')
self.assertEqual(res['id'], '-77:gremlin', 'res check fail')
class AuthBody(object):
"""
权限公共方法
"""
def __init__(self):
"""初始化"""
self.auth = Auth()
self.target_graph = _cfg.graph_name
self.target_url = _cfg.graph_host + ":" + str(_cfg.server_port)
self.auth_user = "******"
self.auth_password = _cfg.test_password['tester']
def create_target(self, target_list, name):
"""
创建资源
"""
body = {
"target_name": name + "_target",
"target_graph": self.target_graph,
"target_url": self.target_url,
"target_resources": target_list
}
status, res = self.auth.post_targets(body=body, auth=auth)
# print(status, res)
return res["id"]
def create_group(self, name):
"""创建组"""
body = {
"group_name": name + "_group",
"group_description": "%s graph_test" % (name + "_group")
}
status, res = self.auth.post_groups(body, auth=auth)
# print(status, res)
return res["id"]
def create_access(self, group, target, premission):
"""创建group到target的连接"""
body = {
"group": group,
"target": target,
"access_permission": premission
}
status, res = self.auth.post_accesses(body, auth=auth)
# print(status, res)
return res["id"]
def create_user(self):
"""创建用户user"""
user_id = ''
status, res = self.auth.get_users(auth=auth)
for user in res['users']:
if user['user_name'] == self.auth_user:
user_id = user['id']
else:
pass
if user_id == '':
body = {
"user_name": self.auth_user,
"user_password": self.auth_password,
}
status, res = self.auth.post_users(body, auth=auth)
user_id = res['id']
return user_id
def create_belong(self, user, group):
"""创建用户的授权"""
body = {
"user": user,
"group": group,
}
status, res = self.auth.post_belongs(body, auth=auth)
# print(status, res)
def create_auth(self,
type_list,
permission,
target_name="target_name",
groupname="group_test",
name="test001",
pw="123456"):
"""创建一个用户，并为其指定权限"""
target_id = self.create_target(type_list, target_name)
group_id = self.create_group(groupname)
access_id = self.create_access(group_id, target_id, permission)
user_id = self.create_user()
self.create_belong(user_id, group_id)
return user_id
